Check none type scratch codes (#1459)

DMOJ · Jun 17, 2020 · e21d8ad · e21d8ad
1 parent 4188578
commit e21d8ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/judge/forms.py b/judge/forms.py
@@ -243,7 +243,7 @@ def clean(self):
         elif self.profile.is_totp_enabled and totp_or_scratch_code:
             if pyotp.TOTP(self.profile.totp_key).verify(totp_or_scratch_code, valid_window=self.TOLERANCE):
                 return
-            elif totp_or_scratch_code in json.loads(self.profile.scratch_codes):
+            elif self.profile.scratch_codes and totp_or_scratch_code in json.loads(self.profile.scratch_codes):
                 scratch_codes = json.loads(self.profile.scratch_codes)
                 scratch_codes.remove(totp_or_scratch_code)
                 self.profile.scratch_codes = json.dumps(scratch_codes)