improve sns endpoint

Author: tzangms

dj_elastictranscoder/__init__.py

@@ -1 +1 @@
-__version__ = '0.8'
+__version__ = '0.9'

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')