From a49c6381d6b43259a375df846e2f1bbe6048e84f Mon Sep 17 00:00:00 2001 From: Aroy-Art Date: Wed, 2 Apr 2025 12:49:42 +0200 Subject: [PATCH] Add: initial data on migration --- backend/apps/archive/apps.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/backend/apps/archive/apps.py b/backend/apps/archive/apps.py index 7ce2ee2..fa0ab90 100644 --- a/backend/apps/archive/apps.py +++ b/backend/apps/archive/apps.py @@ -1,6 +1,24 @@ from django.apps import AppConfig +from django.db.models.signals import post_migrate +from django.dispatch import receiver + class ArchiveConfig(AppConfig): default_auto_field = "django.db.models.BigAutoField" name = "apps.archive" + + +@receiver(post_migrate) +def insert_defualt_data(sender, **kwargs): + from apps.archive.models import CategoryModel + + categories = { + "furaffinity": "FurAffinity", + "deviantart": "DeviantArt", + "twitter": "Twitter/X", + } + + if sender.name == ArchiveConfig.name: + for key, value in categories.items(): + CategoryModel.objects.get_or_create(slug=key, name=value)