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
|
# -*- coding: utf-8 -*-
import logging
import odoo.addons.auth_totp.controllers.home
from odoo import http
from odoo.exceptions import AccessDenied, UserError
from odoo.http import request
_logger = logging.getLogger(__name__)
class Home(odoo.addons.auth_totp.controllers.home.Home):
@http.route()
def web_totp(self, redirect=None, **kwargs):
response = super().web_totp(redirect=redirect, **kwargs)
if response.status_code != 200 or response.qcontext['user']._mfa_type() != 'totp_mail':
# In case the response from the super is a redirection
# or the user has another TOTP method, we return the response from the call to super.
return response
assert request.session.pre_uid and not request.session.uid, \
"The user must still be in the pre-authentication phase"
# Send the email containing the code to the user inbox
try:
response.qcontext['user']._send_totp_mail_code()
except (AccessDenied, UserError) as e:
response.qcontext['error'] = str(e)
except Exception as e:
_logger.exception('Unable to send TOTP email')
response.qcontext['error'] = str(e)
return response
|
