def test_should_read_subtitles_utf8(self): subtitle_1 = Subtitle(1249, '01:24:09,860', '01:24:11,300', u'Te escuché.\n') subtitle_2 = Subtitle( 1250, '01:24:15,020', '01:24:17,900', u'¡Qué más da, es hora de\nser Sherlock Holmes!\n') subtitles_reference = [subtitle_1, subtitle_2] with SrtReader(self.file_srt_2_subtitles_utf8_path) as reader: subtitles = reader.read_subtitles() self.assertEqual(subtitles_reference, subtitles)
def test_should_read_subtitles(self): subtitle_1 = Subtitle(1249, '01:24:09,860', '01:24:11,300', u'I heard you.\n') subtitle_2 = Subtitle( 1250, '01:24:15,020', '01:24:17,900', u'Anyway, time to go and\nbe Sherlock Holmes.\n') subtitles_reference = [subtitle_1, subtitle_2] with SrtReader(self.file_srt_2_subtitles_path) as reader: subtitles = reader.read_subtitles() self.assertEqual(subtitles_reference, subtitles)
def test_should_write_all_subtitles_utf8(self): subtitle_1 = Subtitle(1249, '01:24:09,860', '01:24:11,300', u'Te escuché.\n') subtitle_1_serialised = '1249\n01:24:09,860 --> 01:24:11,300\nTe escuché.\n\n' subtitle_2 = Subtitle( 1250, '01:24:15,020', '01:24:17,900', u'¡Qué más da, es hora de\nser Sherlock Holmes!\n') subtitle_2_serialised = '1250\n01:24:15,020 --> 01:24:17,900\n¡Qué más da, es hora de\nser Sherlock Holmes!\n\n' subtitles = [subtitle_1, subtitle_2] with SrtWriter(self.file_srt_temp_path) as writer: writer.write(subtitles) with open(self.file_srt_temp_path) as file_str_temp: self.assertEqual(subtitle_1_serialised + subtitle_2_serialised, file_str_temp.read())
def test_should_write_all_subtitles(self): subtitle_1 = Subtitle(1249, '01:24:09,860', '01:24:11,300', u'I heard you.\n') subtitle_1_serialised = '1249\n01:24:09,860 --> 01:24:11,300\nI heard you.\n\n' subtitle_2 = Subtitle( 1250, '01:24:15,020', '01:24:17,900', u'Anyway, time to go and\nbe Sherlock Holmes.\n') subtitle_2_serialised = '1250\n01:24:15,020 --> 01:24:17,900\nAnyway, time to go and\nbe Sherlock Holmes.\n\n' subtitles = [subtitle_1, subtitle_2] with SrtWriter(self.file_srt_temp_path) as writer: writer.write(subtitles) with open(self.file_srt_temp_path) as file_str_temp: self.assertEqual(subtitle_1_serialised + subtitle_2_serialised, file_str_temp.read())
def test_should_reset_srt_file(self): subtitle_1 = Subtitle(1, '00:00:07,130', '00:00:09,131', u'Uh, yeah, no.\n') subtitle_2 = Subtitle(2, '00:00:09,132', '00:00:11,133', u'I mean, you people called me.\n') with SrtReader(self.file_srt_with_bom, False) as reader: subtitle = reader.read_next_subtitle() self.assertEqual(subtitle, subtitle_1) subtitle = reader.read_next_subtitle() self.assertEqual(subtitle, subtitle_2) reader.reset() subtitle = reader.read_next_subtitle() self.assertEqual(subtitle, subtitle_1) subtitle = reader.read_next_subtitle() self.assertEqual(subtitle, subtitle_2)
def test_should_read_subtitles_iso_8859_1(self): subtitle_1 = Subtitle( 1, '00:00:00,490', '00:00:02,830', u"Anteriormente en Marvel's\nAgents of S.H.I.E.L.D....\n") subtitle_2 = Subtitle( 2, '00:00:02,860', '00:00:05,090', u"Se supone que Malick es\nla última cabeza de Hydra,\n") subtitles_reference = [subtitle_1, subtitle_2] with SrtReader(self.file_srt_2_subtitles_iso_8859_1_path) as reader: subtitles = reader.read_subtitles() self.assertEqual(subtitles_reference, subtitles) with SrtReader(open( self.file_srt_2_subtitles_iso_8859_1_path)) as reader: subtitles = reader.read_subtitles() self.assertEqual(subtitles_reference, subtitles)
def test_should_write_1_subtitle(self): subtitle = Subtitle(1249, '01:24:09,860', '01:24:11,300', u'I heard you.\n') subtitle_serialised = '1249\n01:24:09,860 --> 01:24:11,300\nI heard you.\n\n' with SrtWriter(self.file_srt_temp_path) as writer: writer.write(subtitle) with open(self.file_srt_temp_path) as file_str_temp: self.assertEqual(subtitle_serialised, file_str_temp.read())
def get_subtitles(): if 'subtitles_mongo_document_id' not in session: raise ValueError("No subtitles found") else: dao = dao_subtitles.Subtitles() subtitles = dao.find_by_id(session['subtitles_mongo_document_id']) if subtitles and 'list' in subtitles: return [Subtitle.get_instance_from_dict(subtitle) for subtitle in subtitles['list']] else: raise ValueError("No subtitles found")
def get_subtitles(): if 'subtitles_mongo_document_id' not in session: raise ValueError("No subtitles found") else: dao = dao_subtitles.Subtitles() subtitles = dao.find_by_id(session['subtitles_mongo_document_id']) if subtitles and 'list' in subtitles: return [ Subtitle.get_instance_from_dict(subtitle) for subtitle in subtitles['list'] ] else: raise ValueError("No subtitles found")