def main(): parser = argparse.ArgumentParser(prog='SrtModPy',\ description='Just adjusts the timing of subtitle files') parser.add_argument('file', help='the subtitle to be modified') parser.add_argument('time_amount', type=float,\ help='the amount of time to be added or subtracted to subtitle') parser.add_argument('-t',\ help='it\'s the part of time to be modified', choices=['S', 'M'], default='S') parser.add_argument('-o',\ help='add or discount time to subtitle', choices=['A', 'D'], default='A') arg = parser.parse_args() time_amount = arg.time_amount file_ = arg.file time_part = arg.t operation = arg.o try: srt = SrtMod(file_, time_amount, time_part, operation) if srt.process(): print 'subtitle file was created successfully' print 'file saved at: ' + os.path.splitext(file_)[0] +'(modified).srt' else: print '\nsubtitle can not be processed' except OverflowError: print 'Exception: Invalid time amount for this operation'
def test_file_cannot_be_processed(self): srt = SrtMod('lalala.srt', 1, 'm', 'd') result = srt.process() self.assertFalse(result)
def test_file_can_be_processed(self): srt = SrtMod('sub.srt', 1, 's', 'a') result = srt.process() self.assertTrue(result)
def test_when_file_is_load_content_must_be_greater_than_zero(self): srt = SrtMod('sub.srt') srt.process() self.assertGreater(len(srt.content), 0)