Update fill_db_using_fake_data

weni-ai · Sep 21, 2022 · a98f14e · a98f14e
1 parent d94c49d
commit a98f14e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/bothub/common/management/commands/fill_db_using_fake_data.py b/bothub/common/management/commands/fill_db_using_fake_data.py
@@ -28,12 +28,13 @@ def handle(self, *args, **kwargs):
 
         # Users
 
-        if not User.objects.filter(email="admin@bothub.it").exists:
+        if not User.objects.filter(email="admin@bothub.it").exists():
             User.objects.create_superuser(
                 email="admin@bothub.it", nickname="admin", password="admin", name="Admin"
             )
 
-        if not User.objects.filter(email="user@bothub.it").exists:
+        user = User.objects.filter(email="user@bothub.it").first()
+        if not user:
             user = User.objects.create_user(
                 email="user@bothub.it", nickname="user", password="user", name="User"
             )