mirror of
https://github.com/jazzband/django-defender.git
synced 2026-03-16 22:10:32 +00:00
a1d526f318
Run black with Python 2.7 as target version to unify the code styling and make it more linter and style guide compliant
33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
try:
|
|
from django.utils.deprecation import MiddlewareMixin as MIDDLEWARE_BASE_CLASS
|
|
except ImportError:
|
|
MIDDLEWARE_BASE_CLASS = object
|
|
from django.contrib.auth import views as auth_views
|
|
from django.utils.decorators import method_decorator
|
|
|
|
from .decorators import watch_login
|
|
|
|
|
|
class FailedLoginMiddleware(MIDDLEWARE_BASE_CLASS):
|
|
""" Failed login middleware """
|
|
|
|
patched = False
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(FailedLoginMiddleware, self).__init__(*args, **kwargs)
|
|
# Watch the auth login.
|
|
# Monkey-patch only once - otherwise we would be recording
|
|
# failed attempts multiple times!
|
|
if not FailedLoginMiddleware.patched:
|
|
# Django 1.11 turned the `login` function view into the
|
|
# `LoginView` class-based view
|
|
try:
|
|
from django.contrib.auth.views import LoginView
|
|
|
|
our_decorator = watch_login()
|
|
watch_login_method = method_decorator(our_decorator)
|
|
LoginView.dispatch = watch_login_method(LoginView.dispatch)
|
|
except ImportError: # Django < 1.11
|
|
auth_views.login = watch_login()(auth_views.login)
|
|
|
|
FailedLoginMiddleware.patched = True
|