mirror of
https://github.com/jazzband/django-defender.git
synced 2026-03-16 22:10:32 +00:00
db7a9001db
Fixes the `TypeError: object.__init__() takes no parameters` that happens with the new MIDDLEWARE that is new in django 1.10
31 lines
1.2 KiB
Python
31 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
|