From aa1bc1456baf50c7f1f8b6e1fa610ec2a122c5ae Mon Sep 17 00:00:00 2001 From: Mike Blume Date: Mon, 18 Oct 2010 03:31:59 +0000 Subject: [PATCH] provide more fine-grained control over login patching --- axes/middleware.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) 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)