重构

account/middleware.py

@@ -1,7 +1,6 @@
 # coding=utf-8
 import time
 import json
-import urllib
 
 from django.http import HttpResponse
 from django.utils.translation import ugettext as _
@@ -23,5 +22,18 @@ class SessionSecurityMiddleware(object):
                                             content_type="application/json")
                     else:
                         return redirect_to_login(request)
-            # 更新最后活动日期
+            # update last active time
             request.session["last_activity"] = time.time()
+
+
+class AdminRequiredMiddleware(object):
+    def process_request(self, request):
+        path = request.path_info
+        if path.startswith("/admin/") or path.startswith("/api/admin/"):
+            if not(request.user.is_authenticated() and request.user.is_admin()):
+                if request.is_ajax():
+                    return HttpResponse(json.dumps({"code": 1, "data": _("Please login in first")}),
+                                        content_type="application/json")
+                else:
+                    return HttpResponse(json.dumps({"code": 1, "data": _("Admin required")}),
+                                        content_type="application/json")