使用Python3和更科学的API写法

utils/shortcuts.py

@@ -1,103 +1,25 @@
 # coding=utf-8
-import json
 import logging
 import random
 
-from django.http import HttpResponse
-from django.views.generic import View
+from django.utils.crypto import get_random_string
+
 
 logger = logging.getLogger(__name__)
 
 
-def JSONResponse(data, content_type="application/json"):
-    resp = HttpResponse(json.dumps(data, indent=4), content_type=content_type)
-    resp.data = data
-    return resp
-
-
-class APIView(View):
-    def _get_request_json(self, request):
-        if request.method != "GET":
-            body = request.body
-            if body:
-                return json.loads(body.decode("utf-8"))
-            return {}
-        return request.GET
-
-    def success(self, data=None):
-        return JSONResponse({"error": None, "data": data})
-
-    def error(self, message, error="error"):
-        return JSONResponse({"error": error, "data": message})
-
-    def invalid_serializer(self, serializer):
-        for k, v in serializer.errors.items():
-            return self.error(k + ": " + v[0], error="invalid-data-format")
-
-    def server_error(self):
-        return self.error("Server Error")
-
-    def dispatch(self, request, *args, **kwargs):
-        try:
-            request.data = self._get_request_json(self.request)
-        except ValueError:
-            return self.error("Invalid JSON")
-        try:
-            return super(APIView, self).dispatch(request, *args, **kwargs)
-        except Exception as e:
-            logging.exception(e)
-            return self.server_error()
-
-
-def paginate_data(request, query_set, object_serializer):
-    """
-    function used to paginate data
-    """
-    need_paginate = request.GET.get("paging", None)
-    # if paging=true not in request.GET, then we return all data
-    if need_paginate != "true":
-        if object_serializer:
-            return object_serializer(query_set, many=True).data
-        else:
-            return query_set
-
-    try:
-        limit = int(request.GET.get("limit", "100"))
-    except ValueError:
-        limit = 100
-    if limit < 0:
-        limit = 100
-
-    try:
-        offset = int(request.GET.get("offset", "0"))
-    except ValueError:
-        offset = 0
-    if offset < 0:
-        offset = 0
-
-    results = query_set[offset:offset + limit]
-    if object_serializer:
-        count = query_set.count()
-        results = object_serializer(results, many=True).data
-    else:
-        count = len(query_set)
-
-    data = {"results": results,
-            "count": count}
-
-    return data
-
-
 def rand_str(length=32, type="lower_hex"):
     """
-    generate types of random string or number with specific length
-    DO NOT USE TO GENERATE SECRET KEY!
+    生成指定长度的随机字符串或者数字, 可以用于密钥等安全场景
+    :param length: 字符串或者数字的长度
+    :param type: str 代表随机字符串，num 代表随机数字
+    :return: 字符串
     """
     if type == "str":
-        return ''.join(random.choice("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789") for i in range(length))
+        return get_random_string(length, allowed_chars="ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789")
     elif type == "lower_str":
-        return ''.join(random.choice("abcdefghijklmnopqrstuvwxyz0123456789") for i in range(length))
+        return get_random_string(length, allowed_chars="abcdefghijklmnopqrstuvwxyz0123456789")
     elif type == "lower_hex":
-        return ''.join(random.choice("0123456789abcdef") for i in range(length))
+        return random.choice("123456789abcdef") + get_random_string(length - 1, allowed_chars="0123456789abcdef")
     else:
-        return random.choice("123456789") + ''.join(random.choice("0123456789") for i in range(length - 1))
+        return random.choice("123456789") + get_random_string(length - 1, allowed_chars="0123456789")