first commit

auth.py

@@ -0,0 +1,71 @@
+# coding:utf-8
+# author:ila
+import base64, copy
+from django.http import JsonResponse
+from django.apps import apps
+
+from util.codes import *
+from util import message as mes
+
+
+class Auth(object):
+    def authenticate(self, model, req_dict):
+        """
+        用户登录，登录成功返回token；登录失败返回失败原因
+        :param username:账号
+        :param password:密码
+        :return: json
+        """
+        msg = {'code': normal_code, 'msg': mes.normal_code, 'data': {}}
+        tablename = model.__tablename__
+        encode_dict = {"tablename": tablename, "params": req_dict}
+
+        encode_str = base64.b64encode(str(encode_dict).encode("utf-8"))
+        msg['data']["id"] = req_dict.get("id")
+        msg["id"] = req_dict.get("id")
+        msg['token'] = encode_str.decode('utf-8')
+        return JsonResponse(msg)
+
+    def identify(self, request):
+        """
+        用户鉴权
+        :param request:本次请求对象
+        :return: list
+        """
+
+        msg = {'code': normal_code, 'msg': mes.normal_code, 'data': {}}
+        # django的header被处理过了
+        token = request.META.get('HTTP_TOKEN')
+
+        if token  and token !="null":
+
+            auth_token = copy.deepcopy(token)
+
+            decode_str = base64.b64decode(auth_token).decode("utf8")
+            decode_str=decode_str.replace('"null"','""').replace('null','""')
+            decode_dict = eval(decode_str)
+
+            tablename2 = decode_dict.get("tablename")
+
+            params2 = decode_dict.get("params",{})
+
+            datas=None
+            allModels = apps.get_app_config('main').get_models()
+            for model in allModels:
+                if model.__tablename__ == tablename2:
+                    datas = model.getbyparams(model, model, params2)
+
+            if not datas:
+                msg['code'] = username_error_code
+                msg['msg'] = '找不到该用户信息'
+                result = msg
+            else:
+                request.session['tablename'] = tablename2
+                request.session['params'] = params2
+                msg['msg'] = '身份验证通过。'
+                result = msg
+        else:
+            msg['code'] = 401
+            msg['msg'] = 'headers未包含认证信息。'
+            result = msg
+        return result