From 616973daaeae2fedbfefabe9eaa62f2806ea8ee8 Mon Sep 17 00:00:00 2001
From: j <0x006A@0x2620.org>
Date: Tue, 14 Feb 2012 12:08:01 +0530
Subject: [PATCH] local bin folder, resume, srt import

---
 bin/pandora_client         |  2 +-
 pandora_client/__init__.py | 54 +++++++++++++++++++++++++++++++++++---
 pandora_client/extract.py  | 10 +++++--
 3 files changed, 59 insertions(+), 7 deletions(-)

diff --git a/bin/pandora_client b/bin/pandora_client
index 82433cb..a4e2120 100755
--- a/bin/pandora_client
+++ b/bin/pandora_client
@@ -25,7 +25,7 @@ if __name__ == '__main__':
         parser.print_help()
         sys.exit()
 
-    actions = ('scan', 'sync', 'upload', 'extract', 'clean', 'cmd')
+    actions = ('scan', 'sync', 'upload', 'extract', 'clean', 'cmd', 'import_srt')
     if not args or args[0] not in actions:
         parser.error('you must specify a valid action. \n\t\tknown actions are: %s' % ', '.join(actions))
 
diff --git a/pandora_client/__init__.py b/pandora_client/__init__.py
index 37c1ba5..2809185 100644
--- a/pandora_client/__init__.py
+++ b/pandora_client/__init__.py
@@ -9,6 +9,7 @@ import sqlite3
 import time
 import shutil
 import sys
+import socket
 
 import ox
 
@@ -19,10 +20,11 @@ import utils
 DEBUG = False
 
 __version__ = '0.2'
+
+socket.setdefaulttimeout(300)
 CHUNK_SIZE = 1024*1024
 default_media_cache = os.environ.get('oxMEDIA', os.path.expanduser('~/.ox/media'))
 
-
 def encode(filename, prefix, profile, info=None):
     if not info:
         info = utils.avinfo(filename)
@@ -344,6 +346,11 @@ class Client(object):
                                     post['info'][oshash] = files['info'][oshash]
                             if len(post['info']):
                                 r = self.api.update(post)
+
+        if not 'data' in r:
+            print r
+            return
+
         if r['data']['data']:
             files = []
             for f in r['data']['data']:
@@ -433,6 +440,25 @@ class Client(object):
         if os.path.exists(self.prefix()):
             shutil.rmtree(self.prefix())
 
+    def import_srt(self, args):
+        '''
+            import srt as annotations, usage:
+                pandora_local ITEMID layername /path/to/transcript.srt
+            i.e. 
+                pandora_local ABC transcripts /path/to/transcript.srt
+        '''
+        item = args[0]
+        layer = args[1]
+        filename = args[2]
+        for s in ox.srt.load(filename):
+            self.api.addAnnotation({
+                'item;': item,
+                'layer;': layer,
+                'in': s['in'],
+                'out': s['out'],
+                'value': s['value'],
+            })
+
 class API(ox.API):
     __name__ = 'pandora_client'
     __version__ = __version__
@@ -443,6 +469,7 @@ class API(ox.API):
         self.media_cache = media_cache
         if not self.media_cache:
             self.media_cache = default_media_cache 
+        self._resume_file = '/tmp/pandora_client.%s.json' % os.environ.get('USER')
 
     def uploadVideo(self, filename, data, profile, info=None):
         i = encode(filename, self.media_cache, profile, info)
@@ -491,10 +518,18 @@ class API(ox.API):
 
     def upload_chunks(self, url, filename, data=None):
         form = ox.MultiPartForm()
-        if not data:
+        if os.path.exists(self._resume_file):
+            with open(self._resume_file) as f:
+                resume = json.load(f)
+            if resume.get('url') != url:
+                resume = None
+        if resume:
+            data = resume
+        else:
             for key in data:
                 form.add_field(key, data[key])
-        data = self._json_request(url, form)
+            data = self._json_request(url, form)
+
         print filename
         if 'url' in data:
             print data['url']
@@ -504,7 +539,12 @@ class API(ox.API):
             fsize = os.stat(filename).st_size
             done = 0
             start = time.mktime(time.localtime())
-            chunk = f.read(CHUNK_SIZE)
+            if 'offset' in data and data['offset'] < fsize:
+                while done < data['offset']:
+                    chunk = f.read(CHUNK_SIZE)
+                    done += len(chunk)
+            else:
+                chunk = f.read(CHUNK_SIZE)
             fname = os.path.basename(filename)
             if isinstance(fname, unicode):
                 fname = fname.encode('utf-8')
@@ -548,6 +588,12 @@ class API(ox.API):
                             print data
                         time.sleep(5)
                 if data and data.get('result') == 1:
+                    with open(self._resume_file, 'w') as r:
+                        json.dump({
+                            'uploadUrl': uploadUrl,
+                            'url': url,
+                            'offset': done
+                        }, r, indent=2)
                     done += len(chunk)
                     chunk = f.read(CHUNK_SIZE)
             print ' ' * 80
diff --git a/pandora_client/extract.py b/pandora_client/extract.py
index bdee35e..911410e 100644
--- a/pandora_client/extract.py
+++ b/pandora_client/extract.py
@@ -21,6 +21,12 @@ import ox
 from utils import avinfo, AspectRatio, run_command
 
 
+def command(program):
+    local = os.path.expanduser('~/.ox/bin/%s' % program)
+    if os.path.exists(local):
+        program = local
+    return program
+
 def frame(video, target, position):
     fdir = os.path.dirname(target)
     if fdir and not os.path.exists(fdir):
@@ -54,7 +60,7 @@ def frame(video, target, position):
     return r == 0
     '''
     #ffmpeg
-    cmd = ['ffmpeg', '-y', '-ss', str(position), '-i', video, '-an', '-vframes', '1', target]
+    cmd = [command('ffmpeg'), '-y', '-ss', str(position), '-i', video, '-an', '-vframes', '1', target]
     r = run_command(cmd)
     return r == 0
 
@@ -150,7 +156,7 @@ def video_cmd(video, target, profile, info):
     else:
         audio_settings = ['-an']
 
-    cmd = ['ffmpeg', '-y', '-i', video, '-threads', '4'] \
+    cmd = [command('ffmpeg'), '-y', '-i', video, '-threads', '4'] \
           + audio_settings \
           + video_settings \
           + ['-f','webm', target]