mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-05-18 02:19:23 +02:00
Refactor fixtures from JSON to Python
This commit is contained in:
parent
ba7c0b9e8d
commit
df265b10f8
10 changed files with 426 additions and 270 deletions
|
@ -1,9 +1,16 @@
|
|||
from django.db.models.signals import post_save
|
||||
import logging
|
||||
|
||||
from django.conf import settings
|
||||
from django.core.management import call_command
|
||||
from django.db.models.signals import post_save, post_migrate
|
||||
from django.dispatch import receiver
|
||||
|
||||
from .models import User, UserProfile
|
||||
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
@receiver(post_save, sender=User)
|
||||
def handle_profile(sender, instance, **kwargs):
|
||||
|
||||
|
@ -21,3 +28,13 @@ def handle_profile(sender, instance, **kwargs):
|
|||
instance.userprofile.save()
|
||||
else:
|
||||
UserProfile.objects.create(user=instance)
|
||||
|
||||
|
||||
@receiver(post_migrate)
|
||||
def handle_loaddata(**kwargs):
|
||||
"""Attempt to load test fixtures when in DEBUG mode."""
|
||||
if settings.DEBUG:
|
||||
try:
|
||||
call_command("load")
|
||||
except Exception as e:
|
||||
logger.warning(e)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue