完善captchaapi

2017-08-19 06:10:48 +08:00
parent 57b75fd511
commit d1767e775d
3 changed files with 136 additions and 81 deletions
--- a/utils/captcha/views.py
+++ b/utils/captcha/views.py
@@ -1,7 +1,11 @@
-from django.http import HttpResponse
+from base64 import b64encode

-from utils.captcha import Captcha
+from . import Captcha
+from ..api import APIView


-def show_captcha(request):
-    return HttpResponse(Captcha(request).display(), content_type="image/gif")
+class CaptchaAPIView(APIView):
+    def get(self, request):
+        img_prefix = "data:image/png;base64,"
+        img = img_prefix + b64encode(Captcha(request).get()).decode("utf-8")
+        return self.success(img)