Email Verification Update w/ Token Expiration

1 files changed, 39 insertions, 0 deletions

diff --git a/users/accountFunctions.py b/users/accountFunctions.py
new file mode 100644
index 00000000..be60e29f
--- /dev/null
+++ b/users/accountFunctions.py
@@ -0,0 +1,39 @@
+from users.models import TokenStore, UserProfile
+import uuid
+import secrets
+from django.utils import timezone
+
+def generate_token():
+    uid = uuid.uuid4().hex
+    token = secrets.token_urlsafe(32)
+    print(uid, token)
+    return uid, token
+
+def store_token(token_type, user, email=None):
+    previous_tokens = TokenStore.objects.filter(user=user, token_type=token_type)
+    if previous_tokens.exists():
+        previous_tokens.delete()
+    uid, token = generate_token()
+    token_store = TokenStore.objects.create(
+        user=user,
+        email=email if email is not None else user.email,
+        uid=uid,
+        token=token,
+        token_type=token_type,
+        expires=timezone.now() + timezone.timedelta(minutes=30),
+    )
+    token_store.save()
+    return uid, token
+
+def verify_token(token_type, uid, token):
+    try:
+        token_store = TokenStore.objects.get(token_type=token_type, uid=uid, token=token)
+        if token_store.expires > timezone.now() and not token_store.verified and token_store.token_type == token_type and token_store.uid == uid and token_store.token == token:
+            token_store.verified = True
+            UserProfile.objects.filter(user=token_store.user).update(email_verified=True)
+            token_store.save()
+
+        return token_store
+    except TokenStore.DoesNotExist:
+        return None
+    
\ No newline at end of file