U
‹¾úg®, ã @ s d Z ddlZddlZddlmZmZ G dd„ dƒZG dd„ deƒZG d d
„ d
eƒZ e
d ¡Z e
d ¡Z e
d
¡Z
e
d¡Ze
d¡Zdd„ Zdd„ ZG dd„ dƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG d d „ d eƒZG d d „ d eƒZG d d!„ d!eƒZd"d#„ ZdS )$u´
A partial parser for WebVTT segments. Interprets enough of the WebVTT stream
to be able to assemble a single stand-alone subtitle file, suitably adjusting
timestamps on the way, while everything else is passed through unmodified.
Regular expressions based on the W3C WebVTT specification
. The X-TIMESTAMP-MAP extension is described
in RFC 8216 §3.5 .
é Né )Ú int_or_noneÚtimetuple_from_msecc @ s8 e Zd ZdZdd„ Zdd„ Zdd„ Zdd „ Zd
d „ Zd S )
Ú _MatchParserz”
An object that maintains the current parsing position and allows
conveniently advancing it as syntax elements are successfully parsed.
c C s || _ d| _d S ©Nr )Ú_dataÚ_pos)ÚselfÚstring© r ú1/tmp/pip-unpacked-wheel-q5ljy6pj/yt_dlp/webvtt.pyÚ__init__ s z_MatchParser.__init__c C sN t |tjƒr | | j| j¡S t |tƒrB| j || j¡r>t|ƒS d S t |ƒ‚d S ©N)
Ú
isinstanceÚreÚPatternÚmatchr r ÚstrÚ
startswithÚlenÚ
ValueError©r Úrr r r r s
z_MatchParser.matchc C sh |d krd}nHt |tjƒr*t| d¡ƒ}n,t |tƒr>t|ƒ}nt |tƒrN|}nt|ƒ‚| j|7 _|S r ) r r ÚMatchr Úgroupr Úintr r )r ZbyÚamtr r r Úadvance$ s
z_MatchParser.advancec C s | | |¡¡S r )r r r r r r Úconsume2 s z_MatchParser.consumec C s t | ƒS r )Ú_MatchChildParser©r r r r Úchild5 s z_MatchParser.childN) Ú__name__Ú
__module__Ú __qualname__Ú__doc__r
r r r r! r r r r r s r c s( e Zd ZdZ‡ fdd„Zdd„ Z‡ ZS )r zÝ
A child parser state, which advances through the same data as
its parent, but has an independent position. This is useful when
advancing through syntax elements we might later want to backtrack
from.
c s t ƒ |j¡ || _|j| _d S r )Úsuperr
r Ú_MatchChildParser__parentr )r Úparent©Ú __class__r r r
A s z_MatchChildParser.__init__c C s | j | j_ | jS )zW
Advance the parent state to the current position of this child state.
)r r' r r r r ÚcommitF s
z_MatchChildParser.commit)r" r# r$ r% r
r+ Ú
__classcell__r r r) r r 9 s r c s e Zd Z‡ fdd„Z‡ ZS )Ú
ParseErrorc s6 |j |j|jd … }tƒ d|j› d|›d¡ d S )Néd zParse error at position z (near ú))r r r& r
)r ÚparserÚdatar) r r r
O s zParseError.__init__)r" r# r$ r
r, r r r) r r- N s r- zL(?x)
(?:([0-9]{1,}):)?
([0-9]{2}):
([0-9]{2})\.
([0-9]{3})?
z\Zz(?:\r\n|[\r\n]|$)z(?:\r\n|[\r\n])+z[ \t]*c C s dt dd„ t| ¡ dƒD ƒƒ S )z•
Convert a parsed WebVTT timestamp (a re.Match obtained from _REGEX_TS)
into an MPEG PES timestamp: a tick counter at 90 kHz resolution.
éZ c s s" | ]\}}t |pd ƒ| V qdS )r N)r )Ú.0ÚpartZmultr r r Ú i s z _parse_ts..)i€î6 i`ê iè r )ÚsumÚzipÚgroups©Útsr r r Ú _parse_tsd s
ÿr; c C s dt t| d d ƒƒ S )zn
Convert an MPEG PES timestamp into a WebVTT timestamp.
This will lose sub-millisecond precision.
z%02u:%02u:%02u.%03ué- r2 )r r r9 r r r Ú
_format_tsm s r= c @ s, e Zd ZdZdd„ Zedd„ ƒZdd„ ZdS ) ÚBlockz#
An abstract WebVTT block.
c K s" | ¡ D ]\}}t| ||ƒ qd S r )ÚitemsÚsetattr)r ÚkwargsÚkeyÚvalr r r r
z s zBlock.__init__c C s. | | j¡}|sd S | |¡ | | d¡dS )Nr )Úraw)r Ú_REGEXr r )Úclsr0 Úmr r r Úparse~ s
z Block.parsec C s | | j¡ d S r )ÚwriterD ©r Ústreamr r r Ú
write_into† s zBlock.write_intoN)r" r# r$ r% r
Ú classmethodrH rL r r r r r> u s
r> c @ s e Zd ZdZdS )Ú HeaderBlockzm
A WebVTT block that may only appear in the header part of the file,
i.e. before any cue blocks.
N)r" r# r$ r% r r r r rN Š s rN c @ sh e Zd Ze d¡Ze d¡Ze d¡Ze d¡Ze d¡Z e d¡Z
e dd„ ƒZ e d d
„ ƒZ
d d „ Zd
S )ÚMagicz,\ufeff?WEBVTT([ \t][^\r\n]*)?(?:\r\n|[\r\n])zX-TIMESTAMP-MAP=zLOCAL:zMPEGTS:([0-9]+)z
[ \t]*,[ \t]*z6(?:(?!-->)[^\r\n])+:(?:(?!-->)[^\r\n])+(?:\r\n|[\r\n])c C s¸ | ¡ }| | j¡}|rL| t¡}|d kr2t|ƒ‚t|ƒ}|d kr„t|ƒ‚n8| | j¡}|r|t| d¡ƒ}|d kr„t|ƒ‚nt|ƒ‚| | j ¡r’q| t
¡ržq¨t|ƒ‚q| ¡ ||fS )Nr ) r! r Ú_REGEX_TSMAP_LOCALÚ _REGEX_TSr- r; Ú_REGEX_TSMAP_MPEGTSr r Ú_REGEX_TSMAP_SEPÚ _REGEX_NLr+ )rF r0 rG ÚlocalÚmpegtsr r r Z
__parse_tsmap© s,
zMagic.__parse_tsmapc C sœ | ¡ }| | j¡}|s t|ƒ‚| d¡}d\}}}| t¡s„| | j¡rZ| |¡\}}q4| | j¡}|rz|| d¡7 }q4t|ƒ‚q4| ¡ | ||||dS )Nr )NNÚ r )ÚextrarV rU Úmeta)
r! r rE r- r rT Ú _REGEX_TSMAPÚ_Magic__parse_tsmapÚ _REGEX_METAr+ )rF r0 rG rX rU rV rY r r r rH Ç s"
z Magic.parsec C s° | d¡ | jd k r | | j¡ | d¡ | js6| jr| d¡ | t| jd k rV| jndƒ¡ | d¡ | t| jd k r~| jndƒ¡ | d¡ | jr¢| | j¡ | d¡ d S )NZWEBVTTÚ
zX-TIMESTAMP-MAP=LOCAL:r z,MPEGTS:)rI rX rU rV r= r rY rJ r r r rL Ý s
zMagic.write_intoN)r" r# r$ r ÚcompilerE rZ rP rR rS r\ rM r[ rH rL r r r r rO ’ s
rO c @ s e Zd Ze d¡ZdS )Ú
StyleBlockzs(?x)
STYLE[\ \t]*(?:\r\n|[\r\n])
((?:(?!-->)[^\r\n])+(?:\r\n|[\r\n]))*
(?:\r\n|[\r\n])
N©r" r# r$ r r^ rE r r r r r_ í s r_ c @ s e Zd Ze d¡ZdS )Ú RegionBlockze(?x)
REGION[\ \t]*
((?:(?!-->)[^\r\n])+(?:\r\n|[\r\n]))*
(?:\r\n|[\r\n])
Nr` r r r r ra õ s ra c @ s e Zd Ze d¡ZdS )Ú CommentBlockzo(?x)
NOTE(?:\r\n|[\ \t\r\n])
((?:(?!-->)[^\r\n])+(?:\r\n|[\r\n]))*
(?:\r\n|[\r\n])
Nr` r r r r rb ý s rb c @ st e Zd ZdZe d¡Ze d¡Ze d¡Ze d¡Z e
dd„ ƒZ dd „ Z e
d
d „ ƒZd d
„ Ze
dd„ ƒZdd„ ZdS )ÚCueBlockz6
A cue block. The payload is not interpreted.
z$((?:(?!-->)[^\r\n])+)(?:\r\n|[\r\n])z[ \t]+-->[ \t]+z[ \t]+((?:(?!-->)[^\r\n])+)z[^\r\n]+(?:\r\n|[\r\n])?c C sî | ¡ }d }| | j¡}|r&| d¡}| t¡}|s8d S | | j¡sHd S | t¡}|sZd S | | j¡}| t¡ | t¡s~d S t |ƒ}t |ƒ}|d k r | d¡nd } t
¡ }
| | j ¡}|s¾qÐ|
| d¡¡ q¬| ¡ | |||| |
¡ dS )Nr r )ÚidÚstartÚendÚsettingsÚtext)r! r Ú _REGEX_IDr rQ Ú _REGEX_ARROWÚ_REGEX_SETTINGSÚ_REGEX_OPTIONAL_WHITESPACErT r; ÚioÚStringIOÚ_REGEX_PAYLOADrI r+ Úgetvalue) rF r0 Zid_rG Zm0Úm1Úm2re rf rg rh r r r rH sB
ýzCueBlock.parsec C sŽ | j d k r | | j ¡ | d¡ | t| jƒ¡ | d¡ | t| jƒ¡ | jd k rj| d¡ | | j¡ | d¡ | | j¡ | d¡ d S )Nr] z --> ú )rd rI r= re rf rg rh rJ r r r rL 7 s
zCueBlock.write_intoc C s | j | j| j| j| jdœS )N©rd re rf rh rg rt r r r r Úas_jsonE s ûzCueBlock.as_jsonc C s | j |j kS r )ru ©r Úotherr r r Ú__eq__O s zCueBlock.__eq__c C s& | |d |d |d |d |d dS )Nrd re rf rh rg rt r )rF Újsonr r r Ú from_jsonR s ûzCueBlock.from_jsonc C sJ | j |j krdS | j|jkr dS | j| j koD|j koD|jkS S )NF)rh rg re rf rv r r r Úhinges\ s
zCueBlock.hingesN)r" r# r$ r% r r^ ri rj rk ro rM rH rL Úpropertyru rx rz r{ r r r r rc s
'
rc c c sÄ t | ¡ ƒ}t |¡V | t¡st| t¡r.qt |¡}|rD|V qt |¡}|rZ|V qt
|¡}|rt|V qqtq| t¡sÀ| t¡rŠqtt
|¡}|r |V qtt |¡}|r¶|V qtt |ƒ‚qtdS )zŽ
A generator that yields (partially) parsed WebVTT blocks when given
a bytes object containing the raw contents of a WebVTT file.
N)
r ÚdecoderO rH r Ú
_REGEX_EOFr Ú _REGEX_BLANKra r_ rb rc r- )Z frag_contentr0 Úblockr r r Úparse_fragmentd s<
r )r% rm r Úutilsr r r r Ú Exceptionr- r^ rQ r~ rT r rl r; r= r> rN rO r_ ra rb rc r r r r r Ú s*
(
[_