add exception DoesNotExist

scaleoutsystems · Mar 21, 2023 · e9551cc · e9551cc
1 parent 8fb8d10
commit e9551cc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/scripts/admin_token.py b/scripts/admin_token.py
@@ -6,4 +6,7 @@
 
 def run(*args):
     admin = User.objects.get(username="admin")
-    _ = Token.objects.create(user=admin)
+    try:
+        _ = Token.objects.get(user=admin)
+    except Token.DoesNotExist:
+        _ = Token.objects.create(user=admin)