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 _translate_break(self, caption): # if break appears at start of caption, skip break if self.first_element == True: return # if the last caption was a break, skip this break elif caption.nodes[-1].type == CaptionData.BREAK: return # close any open italics elif self.open_italic == True: caption.nodes.append(CaptionData.create_style(False, {'italics': True})) self.open_italic = False # add line break caption.nodes.append(CaptionData.create_break())
def _combine_matching_captions(self, captionset): for lang in captionset.get_languages(): captions = captionset.get_captions(lang) new_caps = captions[:1] for caption in captions[1:]: if caption.start == new_caps[-1].start and caption.end == new_caps.end: new_caps[-1].nodes.append(CaptionData.create_break()) new_caps[-1].nodes.extend(caption.nodes) else: new_caps.append(caption) captionset.set_captions(lang, new_caps) return captionset
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