93 lines
3.0 KiB
Python
93 lines
3.0 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
# vi:si:et:sw=4:sts=4:ts=4
|
|
# GPL 2010
|
|
|
|
import fractions
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
import time
|
|
|
|
import ox
|
|
|
|
|
|
class AspectRatio(fractions.Fraction):
|
|
def __new__(cls, numerator, denominator=None):
|
|
if not denominator:
|
|
ratio = list(map(int, numerator.split(':')))
|
|
if len(ratio) == 1:
|
|
ratio.append(1)
|
|
numerator = ratio[0]
|
|
denominator = ratio[1]
|
|
# if its close enough to the common aspect ratios rather use that
|
|
if abs(numerator/denominator - 4/3) < 0.03:
|
|
numerator = 4
|
|
denominator = 3
|
|
elif abs(numerator/denominator - 16/9) < 0.02:
|
|
numerator = 16
|
|
denominator = 9
|
|
return super(AspectRatio, cls).__new__(cls, numerator, denominator)
|
|
|
|
@property
|
|
def ratio(self):
|
|
return "%d:%d" % (self.numerator, self.denominator)
|
|
|
|
def avinfo(filename, cached=True):
|
|
if os.path.getsize(filename):
|
|
info = ox.avinfo(filename, cached=cached)
|
|
if 'video' in info and info['video'] and 'width' in info['video'][0]:
|
|
if 'display_aspect_ratio' not in info['video'][0]:
|
|
dar = AspectRatio(info['video'][0]['width'], info['video'][0]['height'])
|
|
info['video'][0]['display_aspect_ratio'] = dar.ratio
|
|
del info['path']
|
|
if os.path.splitext(filename)[-1] in ('.vtt', '.srt', '.sub', '.idx', '.rar') and 'error' in info:
|
|
del info['error']
|
|
if 'code' in info and info['code'] == 'badfile':
|
|
del info['code']
|
|
return info
|
|
return {'path': filename, 'size': 0}
|
|
|
|
def hash_prefix(h):
|
|
return [h[:2], h[2:4], h[4:6], h[6:]]
|
|
|
|
def run_command(cmd, timeout=25):
|
|
# print(cmd)
|
|
p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=subprocess.PIPE)
|
|
while timeout > 0:
|
|
time.sleep(0.2)
|
|
timeout -= 0.2
|
|
if p.poll() is not None:
|
|
return p.returncode
|
|
if p.poll() is None:
|
|
os.kill(p.pid, 9)
|
|
killedpid, stat = os.waitpid(p.pid, os.WNOHANG)
|
|
return p.returncode
|
|
|
|
def video_frame_positions(duration):
|
|
pos = duration / 2
|
|
# return [pos/4, pos/2, pos/2+pos/4, pos, pos+pos/2, pos+pos/2+pos/4]
|
|
return list(map(int, [pos/2, pos, pos+pos/2]))
|
|
|
|
def cleanup_tree(root):
|
|
again = False
|
|
for prefix, folders, files in os.walk(root):
|
|
removed = []
|
|
for f in files:
|
|
if f in ('.DS_Store', 'Thumbs.db'):
|
|
os.unlink(os.path.join(prefix, f))
|
|
removed.append(f)
|
|
files = list(set(files) - set(removed))
|
|
if not folders and not files and len(prefix) > len(root):
|
|
os.rmdir(prefix)
|
|
again = True
|
|
if again:
|
|
cleanup_tree(root)
|
|
|
|
|
|
def basedir():
|
|
base = os.path.join(os.path.expanduser('~'), '.ox')
|
|
if sys.platform.startswith('win'):
|
|
base = os.path.join(sys.environ['APPDATA'], 'pandora_client')
|
|
return base
|