apps.py 2.75 KB
Newer Older
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
# vim: set fileencoding=utf-8 :

###############################################################################
#                                                                             #
# Copyright (c) 2020 Idiap Research Institute, http://www.idiap.ch/           #
# Contact: beat.support@idiap.ch                                              #
#                                                                             #
# This file is part of the beat.web module of the BEAT platform.              #
#                                                                             #
# Commercial License Usage                                                    #
# Licensees holding valid commercial BEAT licenses may use this file in       #
# accordance with the terms contained in a written agreement between you      #
# and Idiap. For further information contact tto@idiap.ch                     #
#                                                                             #
# Alternatively, this file may be used under the terms of the GNU Affero      #
# Public License version 3 as published by the Free Software and appearing    #
# in the file LICENSE.AGPL included in the packaging of this file.            #
# The BEAT platform is distributed in the hope that it will be useful, but    #
# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY  #
# or FITNESS FOR A PARTICULAR PURPOSE.                                        #
#                                                                             #
# You should have received a copy of the GNU Affero Public License along      #
# with the BEAT platform. If not, see http://www.gnu.org/licenses/.           #
#                                                                             #
###############################################################################

from django.apps import AppConfig
from django.utils.translation import ugettext_lazy as _


31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52
def monkey_patch_2fa_authentication_form():
    """
    This methods monkey patches the list of forms used by the 2fa LoginView.

    It replaces the standard form with a custom one that sends an email on successful
    login with an account that is blocked.
    """

    from two_factor.views import LoginView

    from .forms import AuthenticationFormSendingWarning

    form_list = []
    for form_tuple in LoginView.form_list:
        if form_tuple[0] == "auth":
            form_list.append(("auth", AuthenticationFormSendingWarning))
        else:
            form_list.append(form_tuple)

    LoginView.form_list = form_list


53
class UiRegistrationConfig(AppConfig):
54

55 56
    name = "beat.web.ui.registration"
    verbose_name = _("Ui Registration")
57 58 59 60

    def ready(self):
        super().ready()
        monkey_patch_2fa_authentication_form()