tus uploads integrated

Author: qencode-dev

qencode/tus_uploader.py

@@ -0,0 +1,32 @@
+import sys
+from tusclient import client
+from utils import get_tus_from_url
+
+class UploadStatus(object):
+  def __init__(self, error = None, url= None, status= None):
+    self.url = url
+    self.error = error
+    self.status = status
+
+
+def upload(file_path=None, url=None, chunk_size=None, log_func=None):
+    """
+        Returns upload status and url using tus protocol
+
+        :fileUrl:
+
+        Url address where to upload the file
+
+        :Args:
+            see tusclient.uploader.Uploader for required and optional arguments.
+    """
+    try:
+        my_client = client.TusClient(url=url)
+        uploader = my_client.uploader(file_path=file_path, chunk_size=chunk_size, log_func=log_func)
+        uploader.upload()
+        url_storage = uploader.url
+        tus_url = get_tus_from_url(url_storage)
+        return UploadStatus(url=tus_url, status='Ok', error='')
+    except:
+       print('Error uploading file to ' + url)
+       raise

qencode/utils.py

@@ -65,4 +65,17 @@ def log(self, path=None, name=None, log_format=None):
   while 1:
    log.info('{0} | {1} | {2}'.format(self.status, self.percent, self.message))
    if self.task_completed:
-     break
+     break
+
+def get_tus_from_url(url=''):
+  try:
+      if url.find('tus:') == 0:
+        return url
+      else:
+        x = url.split('/')[-1]
+        if x == url:
+          return url
+        else:
+          return 'tus:' + x
+  except:
+      return url

requirements.txt

@@ -1,4 +1,5 @@
 json
 urllib
 urllib2
-urlparse
+urlparse
+tuspy

sample-code/start_custom_tus_upload.py

@@ -0,0 +1,79 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+import sys
+import os.path
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), os.path.pardir)))
+import qencode
+import time
+import json
+from qencode import QencodeClientException, QencodeTaskException, tus_uploader
+
+#replace with your API KEY (can be found in your Project settings on Qencode portal)
+API_KEY = 'your-api-qencode-key'
+
+file_path = '/path/to/file/for/upload'
+
+query = """
+{"query": {
+  "source": "%s",
+  "format": [
+    {
+      "output": "mp4",
+      "size": "320x240",
+      "video_codec": "libx264"
+    }
+  ]
+  }
+}
+"""
+
+
+def start_encode():
+
+  """
+    Create client object
+    :param api_key: string. required
+    :param api_url: string. not required
+    :param api_version: int. not required. default 'v1'
+    :return: task object
+  """
+
+  client = qencode.client(API_KEY)
+  if client.error:
+    raise QencodeClientException(client.message)
+
+  print 'The client created. Expire date: %s' % client.expire
+
+  task = client.create_task()
+
+  if task.error:
+    raise QencodeTaskException(task.message)
+
+  #get upload url from endpoint returned with /v1/create_task and task_token value
+  uploadUrl = task.upload_url + '/' + task.task_token
+
+  #do upload and get uploaded file URI
+  uploadedFile = tus_uploader.upload(file_path=file_path, url=uploadUrl, log_func=log_upload, chunk_size=2000000)
+
+  params = query % uploadedFile.url
+  task.custom_start(params)
+
+  if task.error:
+    raise QencodeTaskException(task.message)
+
+  print 'Start encode. Task: %s' % task.task_token
+
+  while True:
+    status = task.status()
+    # print status
+    print json.dumps(status, indent=2, sort_keys=True)
+    if status['error'] or status['status'] == 'completed':
+      break
+    time.sleep(5)
+
+def log_upload(msg):
+  print(msg)
+
+if __name__ == '__main__':
+  start_encode()