diff --git a/axes/middleware.py b/axes/middleware.py index b2d5880..c3cf389 100644 --- a/axes/middleware.py +++ b/axes/middleware.py @@ -12,3 +12,17 @@ class FailedLoginMiddleware(object): # and the regular auth login page auth_views.login = watch_login(auth_views.login) + +class FailedAdminLoginMiddleware(object): + def __init__(self, *args, **kwargs): + super(FailedAdminLoginMiddleware, self).__init__(*args, **kwargs) + + # watch the admin login page + admin.site.login = watch_login(admin.site.login) + +class FailedAuthLoginMiddleware(object): + def __init__(self, *args, **kwargs): + super(FailedAuthLoginMiddleware, self).__init__(*args, **kwargs) + + # watch the admin login page + auth_views.login = watch_login(auth_views.login)