Files
Sub-Zero.bundle/Contents/Libraries/Shared/submod_test.py
T
2017-12-24 01:57:53 +01:00

48 lines
1.4 KiB
Python

# coding=utf-8
import logging
import sys
import codecs
from ftfy import fix_text
from babelfish import Language
from subliminal_patch import Subtitle
from subliminal_patch.subtitle import ftfy_defaults
logger = logging.getLogger(__name__)
from subzero.modification import SubMod
fn = sys.argv[1]
debug = "--debug" in sys.argv
if debug:
logging.basicConfig(level=logging.DEBUG)
sub = Subtitle(Language.fromietf("heb"), mods=["OCR_fixes", "common", "remove_tags"])
sub.content = open(fn).read()
sub.normalize()
content = sub.get_modified_content(debug=True)
#submod = SubMod(debug=debug)
#submod.load(fn, language=Language.fromietf("pol"), encoding="utf-8")
#submod.modify("OCR_fixes", "common", "remove_tags", "OCR_fixes", "OCR_fixes")
#submod.modify("shift_offset(s=20)", "OCR_fixes")
#submod.modify("remove_HI", "OCR_fixes", "common", "OCR_fixes", "shift_offset(s=20)", "OCR_fixes", "color(name=white)", "shift_offset(s=-5, ms=-350)")
#srt = Subtitle.pysubs2_to_unicode(submod.f)
#content = fix_text(Subtitle.pysubs2_to_unicode(submod.f, format=format), **ftfy_defaults)\
# .encode(encoding="utf-8")
#print submod.f.to_string("srt", encoding="utf-8")
#print repr(content)
f = codecs.open("testout.srt", "w+")
f.write(content)
f.close()
#print submod.f.to_string("srt")
#submod.modify("OCR_fixes")
#submod.modify("change_FPS(from=24,to=25)")
#submod.modify("common")
#print Subtitle.pysubs2_to_unicode(submod.f)