swap QualityProfile.profile->quality in init

lardbit · Aug 17, 2024 · 9bcf4dc · 9bcf4dc
1 parent ba83c65
commit 9bcf4dc
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/nefarious/management/commands/nefarious-init.py b/src/nefarious/management/commands/nefarious-init.py
@@ -26,14 +26,16 @@ def handle(self, *args, **options):
         nefarious_settings, _ = NefariousSettings.objects.get_or_create()
 
         # assign default quality profiles
-        nefarious_settings.quality_profile_tv = QualityProfile.objects.get(profile=PROFILE_ANY)
-        nefarious_settings.quality_profile_movies = QualityProfile.objects.get(profile=PROFILE_HD_1080p)
-        nefarious_settings.save()
+        if not nefarious_settings.quality_profile_tv:
+            nefarious_settings.quality_profile_tv = QualityProfile.objects.get(quality=PROFILE_ANY)
+        if not nefarious_settings.quality_profile_movies:
+            nefarious_settings.quality_profile_movies = QualityProfile.objects.get(quality=PROFILE_HD_1080p)
 
         # populate tmdb configuration if necessary
         if not nefarious_settings.tmdb_configuration or not nefarious_settings.tmdb_languages:
             tmdb_client = get_tmdb_client(nefarious_settings)
             configuration = tmdb_client.Configuration()
             nefarious_settings.tmdb_configuration = configuration.info()
             nefarious_settings.tmdb_languages = configuration.languages()
-            nefarious_settings.save()
+
+        nefarious_settings.save()