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")