def to_subtitles(response):
subtitles = SSAFile()
subtitles.events = [
SSAEvent(
text=result.alternatives[0].transcript,
start=duration_to_ms(result.alternatives[0].words[0].start_time),
end=duration_to_ms(result.alternatives[0].words[-1].end_time),
) for result in response.results
]
return subtitles
def test_writer_tags():
subs = SSAFile()
subs.styles["italic_style"] = SSAStyle(italic=True)
subs.events = [SSAEvent(start=0, end=10, text=r"Plain."),
SSAEvent(start=0, end=10, text=r"{\i1}Inline."),
SSAEvent(start=0, end=10, text=r"Styled.", style="italic_style"),
SSAEvent(start=0, end=10, text=r"{\i1}Also{\i0} {\ritalic_style}italic."),
SSAEvent(start=0, end=10, text=r"Not {\i1}italic.")]
f = dedent("""\
{0}{0}1000
{0}{10}Plain.
{0}{10}{Y:i}Inline.
{0}{10}{Y:i}Styled.
{0}{10}{Y:i}Also italic.
{0}{10}Not italic.
""")
assert subs.to_string("microdvd", 1000) == f
def test_writer_tags():
subs = SSAFile()
subs.styles["italic_style"] = SSAStyle(italic=True)
subs.events = [
SSAEvent(start=0, end=10, text=r"Plain."),
SSAEvent(start=0, end=10, text=r"{\i1}Inline."),
SSAEvent(start=0, end=10, text=r"Styled.", style="italic_style"),
SSAEvent(start=0,
end=10,
text=r"{\i1}Also{\i0} {\ritalic_style}italic."),
SSAEvent(start=0, end=10, text=r"Not {\i1}italic.")
]
f = dedent("""\
{0}{0}1000
{0}{10}Plain.
{0}{10}{Y:i}Inline.
{0}{10}{Y:i}Styled.
{0}{10}{Y:i}Also italic.
{0}{10}Not italic.
""")
assert subs.to_string("microdvd", 1000) == f
