Update auth function

1 files changed, 22 insertions, 0 deletions

diff --git a/indoteknik_custom/models/res_users.py b/indoteknik_custom/models/res_users.py
index 7f94771f..02433deb 100755
--- a/indoteknik_custom/models/res_users.py
+++ b/indoteknik_custom/models/res_users.py
@@ -1,4 +1,7 @@
 from odoo import models, fields
+from datetime import datetime
+from pytz import UTC
+import random, string
 
 
 class ResUsers(models.Model):
@@ -6,3 +9,22 @@ class ResUsers(models.Model):
 
     reset_password_token = fields.Char(string="Reset Password Token")
     activation_token = fields.Char(string="Activation Token")
+    otp_code = fields.Char(string='OTP Code')
+    otp_create_date = fields.Datetime(string='OTP Create Date')
+
+    def _generate_otp(self):
+        for user in self:
+            user.otp_code = '{:04d}'.format(random.randint(0, 9999))
+            user.otp_create_date = fields.Datetime.now()
+            
+    def _generate_activation_token(self):
+        for user in self:
+            token_source = string.ascii_letters + string.digits
+            user.activation_token = ''.join(random.choice(token_source) for i in range(21))
+            
+    def send_activation_mail(self):
+        template = self.env.ref('indoteknik_custom.mail_template_res_user_activation_request')
+        for user in self:
+            user._generate_otp()
+            user._generate_activation_token()
+            template.send_mail(user.id, force_send=True)
\ No newline at end of file