diff options
Diffstat (limited to 'module/web/ajax/views.py')
| -rw-r--r-- | module/web/ajax/views.py | 26 |
1 files changed, 23 insertions, 3 deletions
diff --git a/module/web/ajax/views.py b/module/web/ajax/views.py index 4265e8081..a0e1238aa 100644 --- a/module/web/ajax/views.py +++ b/module/web/ajax/views.py @@ -1,4 +1,3 @@ - # Create your views here. from os.path import join import time @@ -10,6 +9,7 @@ from django.http import HttpResponseForbidden from django.http import HttpResponseServerError from django.utils import simplejson from django.utils.translation import ugettext as _ +import base64 def format_time(seconds): seconds = int(seconds) @@ -89,7 +89,9 @@ def remove_link(request, id): @permission('pyload.can_see_dl') def status(request): try: - return JsonResponse(settings.PYLOAD.status_server()) + status = settings.PYLOAD.status_server() + status['captcha'] = settings.PYLOAD.is_captcha_waiting() + return JsonResponse(status) except: return HttpResponseServerError() @@ -229,4 +231,22 @@ def push_to_queue(request, id): settings.PYLOAD.push_package_2_queue(int(id)) return JsonResponse("sucess") except: - return HttpResponseServerError()
\ No newline at end of file + return HttpResponseServerError() + +@permission('pyload.can_add_dl') +def set_captcha(request): + if request.META['REQUEST_METHOD'] == "POST": + try: + settings.PYLOAD.set_captcha_result(request.POST["cap_id"], request.POST["cap_text"]) + except: + pass + + id, binary, typ = settings.PYLOAD.get_captcha_task() + + if id: + binary = base64.standard_b64encode(str(binary)) + src = "data:image/%s;base64,%s" % (typ, binary) + + return JsonResponse({'captcha': True, 'src': src, 'id': id}) + else: + return JsonResponse({'captcha': False}) |