add srt.load

ox/__init__.py

@@ -7,6 +7,7 @@ import cache
 import js
 import jsonc
 import net
+import srt
 
 from api import *
 from file import *

ox/srt.py

@@ -0,0 +1,82 @@
+# -*- coding: utf-8 -*-
+# vi:si:et:sw=4:sts=4:ts=4
+from __future__ import with_statement, division
+import chardet
+import re
+
+import ox
+
+
+__all__ = []
+
+
+def _detectEncoding(fp):
+    bomDict={ # bytepattern : name
+              (0x00, 0x00, 0xFE, 0xFF): "utf_32_be",
+              (0xFF, 0xFE, 0x00, 0x00): "utf_32_le",
+              (0xFE, 0xFF, None, None): "utf_16_be",
+              (0xFF, 0xFE, None, None): "utf_16_le",
+              (0xEF, 0xBB, 0xBF, None): "utf_8",
+            }
+
+    # go to beginning of file and get the first 4 bytes
+    oldFP = fp.tell()
+    fp.seek(0)
+    (byte1, byte2, byte3, byte4) = tuple(map(ord, fp.read(4)))
+
+    # try bom detection using 4 bytes, 3 bytes, or 2 bytes
+    bomDetection = bomDict.get((byte1, byte2, byte3, byte4))
+    if not bomDetection:
+        bomDetection = bomDict.get((byte1, byte2, byte3, None))
+        if not bomDetection:
+            bomDetection = bomDict.get((byte1, byte2, None, None))
+    ## if BOM detected, we're done :-)
+    fp.seek(oldFP)
+    if bomDetection:
+        return bomDetection
+
+    encoding = 'latin-1'
+    #more character detecting magick using http://chardet.feedparser.org/
+    fp.seek(0)
+    rawdata = fp.read()
+    encoding = chardet.detect(rawdata)['encoding']
+    fp.seek(oldFP)
+    return encoding
+
+
+def load(filename, offset=0):
+    '''
+        filename path to an srt file
+        offset in seconds shift all in/out points by offset
+
+        returns list with objects that have in,out,value and id
+    '''
+    srt = []
+
+    def parse_time(t):
+        return offset + ox.time2ms(t.replace(',', '.')) / 1000
+
+    with open(filename) as f:
+        encoding = _detectEncoding(f)
+        data = f.read()
+    try:
+        data = unicode(data, encoding)
+    except:
+        try:
+            data = unicode(data, 'latin-1')
+        except:
+            print "failed to detect encoding, giving up"
+            return srt
+
+    data = data.replace('\r\n', '\n')
+    srts = re.compile('(\d\d:\d\d:\d\d[,.]\d\d\d)\s*-->\s*(\d\d:\d\d:\d\d[,.]\d\d\d)\s*(.+?)\n\n', re.DOTALL)
+    i = 0
+    for s in srts.findall(data):
+        _s = {'id': str(i),
+              'in': parse_time(s[0]),
+              'out': parse_time(s[1]),
+              'value': s[2].strip()
+        }
+        srt.append(_s)
+        i += 1
+    return srt