add support aliyun transcoder

Author: seanhsu0102

dj_elastictranscoder/transcoder.py

@@ -111,3 +111,63 @@ def start_job(self, obj, transcode_kwargs, message=''):
         job.object_id = obj.pk
         job.message = message
         job.save()
+
+
+class AliyunTranscoder(Transcoder):
+
+    def __init__(
+        self,
+        access_key_id=None,
+        access_key_secret=None,
+        pipeline_id=None,
+        region=None,
+        notify_url=None
+    ):
+        if not access_key_id:
+            access_key_id = get_setting_or_raise('ALIYUN_TRANSCODE_ACCESS_KEY_ID')
+        self.access_key_id = access_key_id
+
+        if not access_key_secret:
+            access_key_secret = get_setting_or_raise('ALIYUN_TRANSCODE_ACCESS_KEY_SECRET')
+        self.access_key_secret = access_key_secret
+
+        if not pipeline_id:
+            pipeline_id = get_setting_or_raise('ALIYUN_TRANSCODE_PIPELINE_ID')
+        self.pipeline_id = pipeline_id
+
+        if not region:
+            region = get_setting_or_raise('ALIYUN_TRANSCODE_REGION')
+        self.region = region
+
+        if not notify_url:
+            notify_url = get_setting_or_raise('ALIYUN_TRANSCODE_NOTIFY_URL')
+        self.notify_url = notify_url
+
+        from aliyunsdkcore import client
+
+        self.client = client.AcsClient(self.access_key_id, self.access_key_secret, self.region)
+
+    def start_job(self, obj, transcode_kwargs, message=''):
+        """
+        https://help.aliyun.com/document_detail/57347.html?spm=5176.doc56767.6.724.AJ8z3E
+        """
+
+        import json
+        from aliyunsdkmts.request.v20140618 import SubmitJobsRequest
+
+        request = SubmitJobsRequest.SubmitJobsRequest()
+        request.set_accept_format('json')
+        request.set_Input(json.dumps(transcode_kwargs.get('input_file')))
+        request.set_OutputBucket(transcode_kwargs.get('bucket'))
+        request.set_OutputLocation(transcode_kwargs.get('oss_location'))
+        request.set_Outputs(json.dumps(transcode_kwargs.get('outputs')))
+        request.set_PipelineId(self.pipeline_id)
+        response = json.loads(self.client.do_action_with_exception(request).decode('utf-8'))
+
+        content_type = ContentType.objects.get_for_model(obj)
+        job = EncodeJob()
+        job.id = response['JobResultList']['JobResult'][0]['Job']['JobId']
+        job.content_type = content_type
+        job.object_id = obj.pk
+        job.message = message
+        job.save()

dj_elastictranscoder/urls.py

@@ -8,4 +8,5 @@
     url(r'^endpoint/$', 'aws_endpoint'),
     url(r'^aws_endpoint/$', 'aws_endpoint', name='aws_endpoint'),
     url(r'^qiniu_endpoint/$', 'qiniu_endpoint', name='qiniu_endpoint'),
+    url(r'^aliyun_endpoint', 'aliyun_endpoint', name='aliyun_endpoint'),
 )

dj_elastictranscoder/views.py

@@ -102,3 +102,39 @@ def qiniu_endpoint(request):
         raise RuntimeError('Invalid code')
 
     return HttpResponse('Done')
+
+
+@csrf_exempt
+@require_http_methods(['POST', ])
+def aliyun_endpoint(request):
+    """
+    Receive Aliyun notification
+    """
+
+    try:
+        webhook = request.read().decode('utf-8')
+        data = json.loads(webhook)
+    except ValueError:
+        return HttpResponseBadRequest('Invalid JSON')
+
+    message = json.loads(data['Message'])
+    if message['Type'] == 'Transcode':
+        state = message['state']
+        job_id = message['jobId']
+
+        job = EncodeJob.objects.get(pk=job_id)
+
+        # https://help.aliyun.com/document_detail/57347.html?spm=5176.doc29208.6.724.4zQQQ4
+        if state == 'Success':  # Complate
+            job.message = webhook
+            job.state = 4
+            job.save()
+            transcode_oncomplete.send(sender=None, job=job, job_response=job_id)
+        elif state == 'Fail':  # Error
+            job.message = webhook
+            job.state = 2
+            job.save()
+            transcode_onerror.send(sender=None, job=job, job_response=data)
+        else:
+            raise RuntimeError('Invalid code')
+        return HttpResponse('Done')