def _translate_tag(self, tag):
# check to see if tag is just a string
try:
tag_name = tag.name
except AttributeError:
# if no more tags found, strip text
self.line.append(CaptionData.create_text(tag.strip()))
return
# convert line breaks
if tag.name == 'br':
self.line.append(CaptionData.create_break())
# convert italics
elif tag.name == 'i':
self.line.append(CaptionData.create_style(True, {'italics': True}))
# recursively call function for any children elements
for a in tag.contents:
self._translate_tag(a)
self.line.append(CaptionData.create_style(False, {'italics': True}))
elif tag.name == 'span':
self._translate_span(tag)
else:
# recursively call function for any children elements
for a in tag.contents:
self._translate_tag(a)
def _convert_to_caption(self, buffer, start):
# check to see if previous caption needs an end-time
if self.scc and self.scc[-1].end == 0:
self.scc[-1].end = start
# initial variables
caption = Caption()
caption.start = start
caption.end = 0 # Not yet known; filled in later
self.open_italic = False
self.first_element = True
# split into elements (e.g. break, italics, text)
for element in buffer.split('<$>'):
# skip empty elements
if element.strip() == '':
continue
# handle line breaks
elif element == '{break}':
self._translate_break(caption)
# handle open italics
elif element == '{italic}':
# add italics
caption.nodes.append(CaptionData.create_style(True, {'italics': True}))
# open italics, no longer first element
self.open_italic = True
self.first_element = False
# handle clone italics
elif element == '{end-italic}' and self.open_italic:
caption.nodes.append(CaptionData.create_style(False, {'italics': True}))
self.open_italic = False
# handle text
else:
# add text
caption.nodes.append(CaptionData.create_text(' '.join(element.decode("utf-8").split())))
# no longer first element
self.first_element = False
# close any open italics left over
if self.open_italic == True:
caption.nodes.append(CaptionData.create_style(False, {'italics': True}))
# remove extraneous italics tags in the same caption
self._remove_italics(caption)
# only add captions to list if content inside exists
if caption.nodes:
self.scc.append(caption)
def _translate_tag(self, tag):
# ensure that tag is not just text
try:
tag_name = tag.name
# if no more tags found, strip text
except AttributeError:
if tag.strip() != '':
node = CaptionData.create_text(tag.strip())
self.nodes.append(node)
return
# convert line breaks
if tag_name == 'br':
self.nodes.append(CaptionData.create_break())
# convert italics
elif tag_name == 'span':
# convert span
self._translate_span(tag)
else:
# recursively call function for any children elements
for a in tag.contents:
self._translate_tag(a)
def read(self, content, lang="en"):
captions = CaptionSet()
inlines = content.splitlines()
start_line = 0
subdata = []
while start_line < len(inlines):
if not inlines[start_line].isdigit():
break
caption = Caption()
end_line = self._find_text_line(start_line, inlines)
timing = inlines[start_line + 1].split("-->")
caption.start = self._srttomicro(timing[0].strip(" \r\n"))
caption.end = self._srttomicro(timing[1].strip(" \r\n"))
for line in inlines[start_line + 2 : end_line - 1]:
caption.nodes.append(CaptionData.create_text(line))
caption.nodes.append(CaptionData.create_break())
caption.nodes.pop() # remove last line break from end of caption list
subdata.append(caption)
start_line = end_line
captions.set_captions(lang, subdata)
return captions
