diff options
| author | remitamine <remitamine@gmail.com> | 2016-01-28 12:38:34 +0100 | 
|---|---|---|
| committer | remitamine <remitamine@gmail.com> | 2016-01-28 12:38:34 +0100 | 
| commit | 2b14cb566fde3e5482ce9a63b2be7103cec939e0 (patch) | |
| tree | 5b5a0b127f7323a7844ba4fefe015f68680697b7 | |
| parent | ed7cd1e859cf97e975a28a5e8c58a1d1aca819fe (diff) | |
[utils] fix dfxp2srt text extraction(fixes #8055)
| -rw-r--r-- | youtube_dl/utils.py | 29 | 
1 files changed, 18 insertions, 11 deletions
| diff --git a/youtube_dl/utils.py b/youtube_dl/utils.py index c63b61598..18dbe28bb 100644 --- a/youtube_dl/utils.py +++ b/youtube_dl/utils.py @@ -2017,20 +2017,27 @@ def dfxp2srt(dfxp_data):          'ttaf1': 'http://www.w3.org/2006/10/ttaf1',      }) -    def parse_node(node): -        str_or_empty = functools.partial(str_or_none, default='') +    class TTMLPElementParser: +        out = '' -        out = str_or_empty(node.text) +        def start(self, tag, attrib): +            if tag in (_x('ttml:br'), _x('ttaf1:br'), 'br'): +                self.out += '\n' -        for child in node: -            if child.tag in (_x('ttml:br'), _x('ttaf1:br'), 'br'): -                out += '\n' + str_or_empty(child.tail) -            elif child.tag in (_x('ttml:span'), _x('ttaf1:span'), 'span'): -                out += str_or_empty(parse_node(child)) -            else: -                out += str_or_empty(xml.etree.ElementTree.tostring(child)) +        def end(self, tag): +            pass -        return out +        def data(self, data): +            self.out += data + +        def close(self): +            return self.out.strip() + +    def parse_node(node): +        target = TTMLPElementParser() +        parser = xml.etree.ElementTree.XMLParser(target=target) +        parser.feed(xml.etree.ElementTree.tostring(node)) +        return parser.close()      dfxp = compat_etree_fromstring(dfxp_data.encode('utf-8'))      out = [] | 
