improve sns endpoint

tzangms · tzangms · commit a7b73b4cfad3 · 2014-04-14T20:50:25.000+08:00
diff --git a/dj_elastictranscoder/__init__.py b/dj_elastictranscoder/__init__.py
@@ -1 +1 @@
-__version__ = '0.8'
+__version__ = '0.9'
diff --git a/dj_elastictranscoder/views.py b/dj_elastictranscoder/views.py
@@ -3,6 +3,7 @@
 from django.http import HttpResponse, HttpResponseBadRequest
 from django.core.mail import mail_admins
 
+from .models import EncodeJob
 from .signals import (
     transcode_onprogress,
     transcode_onerror,
@@ -40,10 +41,25 @@ def endpoint(request):
 
     #
     if message['state'] == 'PROGRESSING':
+        job = EncodeJob.objects.get(pk=message['jobId'])
+        job.message = 'Progress'
+        job.state = 1
+        job.save()
+
         transcode_onprogress.send(sender=None, message=message)
     elif message['state'] == 'COMPLETED':
+        job = EncodeJob.objects.get(pk=message['jobId'])
+        job.message = 'Success'
+        job.state = 4
+        job.save()
+
         transcode_oncomplete.send(sender=None, message=message)
     elif message['state'] == 'ERROR':
+        job = EncodeJob.objects.get(pk=message['jobId'])
+        job.message = message['messageDetails']
+        job.state = 2
+        job.save()
+
         transcode_onerror.send(sender=None, message=message)
 
     return HttpResponse('Done')
