from Utils import Utils
from Translator import Translator
## MAIN ##
#make sure we have the correct number of args
if len(sys.argv) == 3 or len(sys.argv) == 4:
desiredSubLang = sys.argv[2]
originalSubLang = sys.argv[3] if len(sys.argv) == 4 else u'auto'
parsedSubInfoList = Utils.ParseSRTFile(sys.argv[1])
if len(parsedSubInfoList
) != 0: #make sure the srt file was correctly translated
subTranslator = Translator(parsedSubInfoList, originalSubLang,
desiredSubLang)
if (subTranslator.LanguageSupported(desiredSubLang)
and subTranslator.LanguageSupported(originalSubLang)):
print("Translating...")
translatedSubInfo = subTranslator.Translate()
print("Writing file to: " + desiredSubLang.upper() + "_" +
sys.argv[1] + "...")
Utils.WriteSRTFile(desiredSubLang.upper() + "_" + sys.argv[1],
translatedSubInfo)
print("Done!")
else:
langs = desiredSubLang if not subTranslator.LanguageSupported(
desiredSubLang) else ""
langs += " " + originalSubLang if not subTranslator.LanguageSupported(
originalSubLang) else ""
print("ERROR: Unrecognized language(s): " + langs)
print("""Supported languages:""")
