68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
#!/usr/bin/python3
|
|
|
|
import ox
|
|
import os
|
|
import json
|
|
import sys
|
|
import re
|
|
|
|
api = ox.api.signin("https://power.0x2620.org/api/")
|
|
|
|
prefix = sys.argv[1]
|
|
|
|
def fix_timecode(match):
|
|
ts, ms = match[0].split(',')
|
|
seconds = int(ms)/1000
|
|
for i, v in enumerate(list(reversed(ts.split(':')))):
|
|
seconds += float(v) * pow(60, i)
|
|
return ox.format_duration(seconds * 1000, years=False).replace('.', ',')
|
|
|
|
def fix_timecodes(srt_data):
|
|
reg = re.compile(r'(\d\d:\d\d:\d\d,\d\d\d\d)')
|
|
old = srt_data
|
|
srt_data = re.sub(reg, fix_timecode, srt_data)
|
|
if srt_data != old:
|
|
rows1 = old.split('\n')
|
|
rows2 = srt_data.split('\n')
|
|
for n, row in enumerate(rows1):
|
|
if rows1[n] != rows2[n]:
|
|
print(rows1[n], '->', rows2[n])
|
|
return srt_data
|
|
|
|
for root, folders, files in os.walk(prefix):
|
|
for file in sorted(files):
|
|
if file.endswith('.mp3'):
|
|
print(file)
|
|
chapter = '%02d' % int(file.split('-')[0].replace('ch', ''))
|
|
file = os.path.join(root, file)
|
|
srt = file.replace('.mp3', '_eng.srt')
|
|
if not os.path.exists(srt):
|
|
name = os.path.basename(srt)
|
|
srt = srt.replace(name, name.replace(' ', '_'))
|
|
with open(srt) as fd:
|
|
srt_data = fd.read()
|
|
srt_data = fix_timecodes(srt_data)
|
|
subs = ox.srt.loads(srt_data)
|
|
oshash = ox.oshash(file)
|
|
url = '%supload/direct/' % api.url
|
|
avinfo = ox.avinfo(file)
|
|
r = api.addMedia({
|
|
'id': oshash,
|
|
'filename': os.path.basename(file),
|
|
'info': avinfo
|
|
})
|
|
id = r['data']['item']
|
|
if api.upload_chunks(url, file, {'id': oshash}):
|
|
info = {
|
|
"id": id,
|
|
"type": 'voice over',
|
|
"chapter": [chapter]
|
|
}
|
|
api.edit(info)
|
|
for sub in subs:
|
|
sub['value'] = sub['value'].replace('\n', '<br>\n')
|
|
api.addAnnotations({
|
|
"item": id,
|
|
"layer": "subtitles",
|
|
"annotations": subs
|
|
})
|