oca-ocb-security/odoo-bringout-oca-ocb-auth_totp/auth_totp/models/auth_totp.py
Ernad Husremovic c0efcc53f5 19.0 vanilla
2026-03-09 09:32:28 +01:00

38 lines
1.3 KiB
Python

# -*- coding: utf-8 -*-
from odoo import models
from odoo.addons.auth_totp.controllers.home import TRUSTED_DEVICE_AGE_DAYS
import logging
_logger = logging.getLogger(__name__)
class Auth_TotpDevice(models.Model):
# init is overriden in res.users.apikeys to create a secret column 'key'
# use a different model to benefit from the secured methods while not mixing
# two different concepts
_name = 'auth_totp.device'
_inherit = ["res.users.apikeys"]
_description = "Authentication Device"
_auto = False
def _check_credentials_for_uid(self, *, scope, key, uid):
"""Return True if device key matches given `scope` for user ID `uid`"""
assert uid, "uid is required"
return self._check_credentials(scope=scope, key=key) == uid
def _get_trusted_device_age(self):
ICP = self.env['ir.config_parameter'].sudo()
try:
nbr_days = int(ICP.get_param('auth_totp.trusted_device_age', TRUSTED_DEVICE_AGE_DAYS))
if nbr_days <= 0:
nbr_days = None
except ValueError:
nbr_days = None
if nbr_days is None:
_logger.warning("Invalid value for 'auth_totp.trusted_device_age', using default value.")
nbr_days = TRUSTED_DEVICE_AGE_DAYS
return nbr_days * 86400 # seconds