improve signal context

tzangms · tzangms · commit 7a3e74119db8 · 2014-04-14T20:54:22.000+08:00
diff --git a/dj_elastictranscoder/signals.py b/dj_elastictranscoder/signals.py
@@ -1,5 +1,5 @@
 from django.dispatch import Signal
 
-transcode_onprogress = Signal(providing_args=["message"])
-transcode_onerror = Signal(providing_args=["message"])
-transcode_oncomplete = Signal(providing_args=["message"])
+transcode_onprogress = Signal(providing_args=["job", "message"])
+transcode_onerror = Signal(providing_args=["job", "message"])
+transcode_oncomplete = Signal(providing_args=["job", "message"])
diff --git a/dj_elastictranscoder/views.py b/dj_elastictranscoder/views.py
@@ -46,20 +46,20 @@ def endpoint(request):
         job.state = 1
         job.save()
 
-        transcode_onprogress.send(sender=None, message=message)
+        transcode_onprogress.send(sender=None, job=job, 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)
+        transcode_oncomplete.send(sender=None, job=job, 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)
+        transcode_onerror.send(sender=None, job=job, message=message)
 
     return HttpResponse('Done')
