def _output_srt(self, output_file):
""" Output to SRT """
i = 1
for fragment in self.fragments:
text = fragment.text_fragment
output_file.write("%d\n" % i)
output_file.write("%s --> %s\n" % (
gf.time_to_srt(fragment.begin),
gf.time_to_srt(fragment.end)
))
output_file.write("%s\n" % text.text)
output_file.write("\n")
i += 1
def _write_srt(self, output_file):
"""
Write to SRT file
"""
i = 1
for fragment in self.fragments:
text = fragment.text_fragment
output_file.write("%d\n" % i)
output_file.write("%s --> %s\n" % (
gf.time_to_srt(fragment.begin),
gf.time_to_srt(fragment.end)
))
for line in text.lines:
output_file.write("%s\n" % line)
output_file.write("\n")
i += 1
def test_time_to_srt(self):
tests = [
[None, "00:00:00,000"],
[0.000, "00:00:00,000"],
[12.000, "00:00:12,000"],
[12.345, "00:00:12,345"],
[60, "00:01:00,000"],
[83.000, "00:01:23,000"],
[83.456, "00:01:23,456"],
[3600.000, "01:00:00,000"],
[3612.000, "01:00:12,000"],
[3612.340, "01:00:12,340"], # numerical issues
[4980.000, "01:23:00,000"],
[5025.000, "01:23:45,000"],
[5025.670, "01:23:45,670"], # numerical issues
]
for test in tests:
self.assertEqual(gf.time_to_srt(test[0]), test[1])
def test_time_to_srt(self):
tests = [
(None, "00:00:00,000"),
(0.000, "00:00:00,000"),
(12.000, "00:00:12,000"),
(12.345, "00:00:12,345"),
(60, "00:01:00,000"),
(83.000, "00:01:23,000"),
(83.456, "00:01:23,456"),
(3600.000, "01:00:00,000"),
(3612.000, "01:00:12,000"),
(3612.340, "01:00:12,340"), # numerical issues
(4980.000, "01:23:00,000"),
(5025.000, "01:23:45,000"),
(5025.670, "01:23:45,670"), # numerical issues
]
for test in tests:
self.assertEqual(gf.time_to_srt(test[0]), test[1])
def test_time_to_srt(self):
tests = [
[None, "00:00:00,000"],
[0.000, "00:00:00,000"],
[12.000, "00:00:12,000"],
[12.345, "00:00:12,345"],
[60, "00:01:00,000"],
[83.000, "00:01:23,000"],
[83.456, "00:01:23,456"],
[3600.000, "01:00:00,000"],
[3612.000, "01:00:12,000"],
[3612.340, "01:00:12,340"], # numerical issues
[4980.000, "01:23:00,000"],
[5025.000, "01:23:45,000"],
[5025.670, "01:23:45,670"], # numerical issues
]
for test in tests:
self.assertEqual(gf.time_to_srt(test[0]), test[1])
def test_time_to_srt(self):
tests = [
(None, "00:00:00,000"),
(0.000, "00:00:00,000"),
(12.000, "00:00:12,000"),
(12.345, "00:00:12,345"),
(60, "00:01:00,000"),
(83.000, "00:01:23,000"),
(83.456, "00:01:23,456"),
(3600.000, "01:00:00,000"),
(3612.000, "01:00:12,000"),
(3612.340, "01:00:12,340"), # numerical issues
(4980.000, "01:23:00,000"),
(5025.000, "01:23:45,000"),
(5025.670, "01:23:45,670"), # numerical issues
]
for test in tests:
self.assertEqual(gf.time_to_srt(test[0]), test[1])
