summaryrefslogtreecommitdiff
path: root/indoteknik_custom/models/res_users.py
blob: 09321fc6413f40b75139017b80336edea7cd96d7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
from odoo import models, fields
from datetime import datetime
from pytz import UTC
import random, string


class ResUsers(models.Model):
    _inherit = 'res.users'

    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)
            
    def get_activation_token_url(self):
        base_url = self.env['ir.config_parameter'].get_param('site.base.url')
        return f'{base_url}/register?activation=token&token={self.activation_token}'
    
    def get_voucher_code(self, type):
        if type == 'activation':
            vouchers = self.env['voucher'].get_active_voucher([('show_on_email', '=', 'user_activation')])
            if not vouchers: return None
            return ', '.join(x.code for x in vouchers)
        return None