1 files changed, 12 insertions, 0 deletions
diff --git a/users/tokens.py b/users/tokens.py
index 7bc5bc90..c127fa45 100644
--- a/users/tokens.py
+++ b/users/tokens.py
@@ -1,4 +1,7 @@
+import cryptocode
+import os
 from django.contrib.auth.tokens import PasswordResetTokenGenerator
+from dotenv import load_dotenv
 from six import text_type
 
 class AccountActivationTokenGenerator(PasswordResetTokenGenerator):
@@ -8,4 +11,13 @@ class AccountActivationTokenGenerator(PasswordResetTokenGenerator):
             text_type(user.is_active)
         )
 
+class EmailChangeTokenGenerator():
+    def encrypt(self, email):
+        auth_string = os.getenv('AUTHORIZATION_STRING')
+        return cryptocode.encrypt(email, auth_string)
+    
+    def decrypt(self, token):
+        auth_string = os.getenv('AUTHORIZATION_STRING')
+        return cryptocode.decrypt(token, auth_string)
+
 account_activation_token = AccountActivationTokenGenerator()