From 4e0c4a00e253b139bc6f401be92beca7bf6edab8 Mon Sep 17 00:00:00 2001 From: Fabio Caccamo Date: Wed, 6 Feb 2019 16:21:50 +0100 Subject: [PATCH] Fixed IntegrityError on postgres --- admin_interface/models.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/admin_interface/models.py b/admin_interface/models.py index b55c40d..103eb15 100644 --- a/admin_interface/models.py +++ b/admin_interface/models.py @@ -28,16 +28,17 @@ class Theme(models.Model): @staticmethod def get_active_theme(): - objs_active_qs = Theme.objects.filter(active=True) + objs_manager = Theme.objects + objs_active_qs = objs_manager.filter(active=True) objs_active_ls = list(objs_active_qs) objs_active_count = len(objs_active_ls) if objs_active_count == 0: - default_obj, default_obj_created = Theme.objects.get_or_create( - pk='1', defaults={'active': True}) - if not default_obj_created: - default_obj.set_active() - obj = default_obj + obj = objs_manager.all().first() + if obj: + obj.set_active() + else: + obj = objs_manager.create() elif objs_active_count == 1: obj = objs_active_ls[0]