From fceed0fc70ba70e8076545c04f9d4dcf56ffa19c Mon Sep 17 00:00:00 2001
From: John Jarvis <jarv@edx.org>
Date: Thu, 1 Nov 2012 16:41:22 -0400
Subject: [PATCH] view for the certificate callback URL

---
 lms/djangoapps/certificates/views.py | 53 ++++++++++++++++++----------
 lms/urls.py                          |  3 ++
 2 files changed, 37 insertions(+), 19 deletions(-)

diff --git a/lms/djangoapps/certificates/views.py b/lms/djangoapps/certificates/views.py
index d701794b6d..b71b648aec 100644
--- a/lms/djangoapps/certificates/views.py
+++ b/lms/djangoapps/certificates/views.py
@@ -1,10 +1,12 @@
 import logging
 from certificates.models import GeneratedCertificate
-from pprint import pprint
+from django.views.decorators.csrf import csrf_exempt
+from django.http import HttpResponse
+import json
 
 log = logging.getLogger("mitx.certificates")
 
-
+@csrf_exempt
 def update_certificate(request):
     """
     Will update GeneratedCertificate for a new certificate or
@@ -12,20 +14,33 @@ def update_certificate(request):
     """
 
     if request.method == "POST":
-        pprint(request)
-#        user = request.POST.get('user')
-#        try:
-#            generated_certificate = GeneratedCertificate.objects.get(
-#                   key=key)
-#        except GeneratedCertificate.DoesNotExist:
-#            generated_certificate = GeneratedCertificate(user=user)
-#
-#        enabled = request.POST.get('enabled')
-#        enabled = True if enabled == 'True' else False
-#        generated_certificate.grade = request.POST.get('grade')
-#        generated_certificate.download_url = request.POST.get('download_url')
-#        generated_certificate.graded_download_url = request.POST.get(
-#                'graded_download_url')
-#        generated_certificate.course_id = request.POST.get('course_id')
-#        generated_certificate.enabled = enabled
-#        generated_certificate.save()
+
+        xqueue_body = json.loads(request.POST.get('xqueue_body'))
+        xqueue_header = json.loads(request.POST.get('xqueue_header'))
+
+        try:
+            cert = GeneratedCertificate.objects.get(
+                   user__username=xqueue_body['username'],
+                   course_id=xqueue_body['course_id'],
+                   key=xqueue_header['lms_key'])
+
+        except GeneratedCertificate.DoesNotExist:
+            log.critical('Unable to lookup certificate\n' 
+                         'xqueue_body: {0}\n'
+                         'xqueue_header: {1}'.format(
+                                      xqueue_body, xqueue_header))
+
+            return HttpResponse(json.dumps({
+                            'return_code': 1,
+                            'content': 'unable to lookup key'}),
+                             mimetype='application/json')
+
+        cert.download_uuid = xqueue_body['download_uuid']
+        cert.verify_uuid = xqueue_body['download_uuid']
+        cert.download_url = xqueue_body['url']
+        cert.status = 'downloadable'
+        cert.save()
+        return HttpResponse(json.dumps({'return_code': 0}),
+                             mimetype='application/json')
+
+
diff --git a/lms/urls.py b/lms/urls.py
index 4141a5e6b3..ea663aeaf8 100644
--- a/lms/urls.py
+++ b/lms/urls.py
@@ -4,6 +4,9 @@ from django.conf.urls.static import static
 import django.contrib.auth.views
 
 urlpatterns = ('',
+    # certificate view
+
+    url(r'^update_certificate$', 'certificates.views.update_certificate'),
     url(r'^$', 'branding.views.index', name="root"), # Main marketing page, or redirect to courseware
     url(r'^dashboard$', 'student.views.dashboard', name="dashboard"),