Prevent duplicate character and team tokens

SavageAim · Jan 20, 2022 · e9f97f7 · e9f97f7
1 parent 34107b8
commit e9f97f7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/backend/api/models/character.py b/backend/api/models/character.py
@@ -32,4 +32,8 @@ def generate_token() -> str:
         """
         Generate a token for a Character to use for verification
         """
-        return 'savageaim-' + ''.join(choice(CHARACTERS) for _ in range(RANDOM_CHARS))
+        code = 'savageaim-' + ''.join(choice(CHARACTERS) for _ in range(RANDOM_CHARS))
+        while Character.objects.filter(token=code).exists():
+            code = 'savageaim-' + ''.join(choice(CHARACTERS) for _ in range(RANDOM_CHARS))
+
+        return code
diff --git a/backend/api/models/team.py b/backend/api/models/team.py
@@ -27,4 +27,8 @@ def generate_invite_code() -> str:
         Generate a new invite code for a team
         Can be used to regenerate the code later as needed
         """
-        return ''.join(choice(CHARS) for _ in range(CODE_LEN))
+        code = ''.join(choice(CHARS) for _ in range(CODE_LEN))
+        while Team.objects.filter(invite_code=code).exists():
+            code = ''.join(choice(CHARS) for _ in range(CODE_LEN))
+
+        return code