Sh]ddlZddlmZddlZddlZddlmZmZm Z m Z m Z m Z m Z mZmZmZddlmZddlmZddlmZddlmZmZGd d eeZdd lmZmZmZdS) N)chain) OptionalListDictIterableAnyoverloadIteratorTextIOTupleMutableSequence) IntOrFloat)SSAEvent)SSAStyle) make_time ms_to_strceZdZUdZddddZeeefed<dBd Ze dCd ed ed e ede e de ede ddfdZ e dDded e ede e de ddf dZe dDded e ede e de ddf dZ dCd ed ed e ede e de ede ddfdZdEd ede e de defdZdEded ede e de ddf dZ dFdedededede ede e ddfdZd e d!e ddfd"Zd#ed$eddfd%ZdGd'dd(eddfd)ZdBd*Zdeefd+Zd,ddefd-Zdefd.Z dBd/Z!de"efd0Z#e$d1edefd2Z%e$de&deefd3Z%d1e de fd4Z%e$d5ed6eddfd7Z'e$d8e&d9e(eddfd:Z'd5e d6e ddfd;Z'e$d5eddfd<Z)e$de&ddfd=Z)d5e ddfd>Z)defd?Z*d@ed6eddfdAZ+dS)HSSAFilea Subtitle file in SubStation Alpha format. This class has a list-like interface which exposes :attr:`SSAFile.events`, list of subtitles in the file:: subs = SSAFile.load("subtitles.srt") for line in subs: print(line.text) subs.insert(0, SSAEvent(start=0, end=make_time(s=2.5), text="New first subtitle")) del subs[0] 0yesNormal) WrapStyleScaledBorderAndShadow Collisions DEFAULT_INFOreturnNcg|_dtji|_|j|_i|_i|_i|_ d|_ d|_ dS)NDefault) eventsr DEFAULT_STYLEcopystylesrinfoaegisub_project fonts_opaquegraphics_opaquefpsformatselfs U/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/pysubs2/ssafile.py__init__zSSAFile.__init__%sd&( ,5x7M7R7R7T7T+U $($5$:$:$<$< /1,./1$(%) utf-8pathencodingformat_r(errorskwargsc |t|||5}|j||fd|i|cdddS#1swxYwYdS)a7 Load subtitle file from given path. This method is implemented in terms of :meth:`SSAFile.from_file()`. See also: Specific formats may implement additional loading options, please refer to documentation of the implementation classes (eg. :meth:`pysubs2.formats.subrip.SubripFormat.from_file()`) Arguments: path (str): Path to subtitle file. encoding (str): Character encoding of input file. Defaults to UTF-8, you may need to change this. errors (Optional[str]): Error handling for character encoding of input file. Defaults to ``None``; use the value ``"surrogateescape"`` for pass-through of bytes not supported by selected encoding via `Unicode surrogate pairs `_. See documentation of builtin ``open()`` function for more. .. versionchanged:: 1.7.0 The ``errors`` parameter was introduced to facilitate pass-through of subtitle files with unknown text encoding. Previous versions of the library behaved as if ``errors=None``. format_ (str): Optional, forces use of specific parser (eg. `"srt"`, `"ass"`). Otherwise, format is detected automatically from file contents. This argument should be rarely needed. fps (float): Framerate for frame-based formats (MicroDVD), for other formats this argument is ignored. Framerate might be detected from the file, in which case you don't need to specify it here (when given, this argument overrides autodetection). kwargs: Extra options for the reader. Returns: SSAFile Raises: IOError UnicodeDecodeError pysubs2.exceptions.UnknownFPSError pysubs2.exceptions.UnknownFormatIdentifierError pysubs2.exceptions.FormatAutodetectionError Note: pysubs2 may autodetect subtitle format and/or framerate. These values are set as :attr:`SSAFile.format` and :attr:`SSAFile.fps` attributes. Example: >>> subs1 = pysubs2.load("subrip-subtitles.srt") >>> subs2 = pysubs2.load("microdvd-subtitles.sub",fps=23.976) >>> subs3 = pysubs2.load("subrip-subtitles-with-fancy-tags.srt",keep_unknown_html_tags=True) r1r3r(N)open from_file)clsr0r1r2r(r3r4fps r,loadz SSAFile.load3sx$& 9 9 9 AR 3=W@@#@@@ A A A A A A A A A A A A A A A A A As 155stringc Ltj|}|j||fd|i|S)a) Load subtitle file from string. See :meth:`SSAFile.load()` for full description. Arguments: string (str): Subtitle file in a string. Note that the string must be Unicode (``str``, not ``bytes``). format_ (str): Optional, forces use of specific parser (eg. `"srt"`, `"ass"`). Otherwise, format is detected automatically from file contents. This argument should be rarely needed. fps (float): Framerate for frame-based formats (MicroDVD), for other formats this argument is ignored. Framerate might be detected from the file, in which case you don't need to specify it here (when given, this argument overrides autodetection). Returns: SSAFile Example: >>> text = ''' ... 1 ... 00:00:00,000 --> 00:00:05,000 ... An example SubRip file. ... ''' >>> subs = SSAFile.from_string(text) r()ioStringIOr8)r9r<r2r(r4r:s r, from_stringzSSAFile.from_stringrs5@[ s}R<r?get_format_classr)r(r8) r9r:r2r(r4textfragmentimplsubss r,r8zSSAFile.from_files: ?7799DFUF|H'11GT""B((suu tR<`_. See documentation of builtin ``open()`` function for more. .. versionchanged:: 1.7.0 The ``errors`` parameter was introduced to facilitate pass-through of subtitle files with unknown text encoding. Previous versions of the library behaved as if ``errors=None``. kwargs: Extra options for the writer. Raises: IOError UnicodeEncodeError pysubs2.exceptions.UnknownFPSError pysubs2.exceptions.UnknownFormatIdentifierError pysubs2.exceptions.UnknownFileExtensionError Nrwr6r()osr0splitextlowerget_format_identifierr7to_file) r+r0r1r2r(r3r4extr:s r,savez SSAFile.savesb ?'""4((+1133C+C00G $hv > > > 9" DLW 8 8# 8 8 8 8 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9sA;;A?A?c rtj}|j||fd|i||S)z Get subtitle file as a string. See :meth:`SSAFile.save()` for full description. Returns: str r()r>r?rOgetvalue)r+r2r(r4r:s r, to_stringzSSAFile.to_strings>[]] R44c4V444{{}}r.c Ht|}|j|||fd|i|dS)a Write subtitle file to file object. See :meth:`SSAFile.save()` for full description. Note: This is a low-level method. Usually, one of :meth:`SSAFile.save()` or :meth:`SSAFile.to_string()` is preferable. Arguments: fp (file object): A file object, ie. :class:`TextIO` instance. Note that the file must be opened in text mode (as opposed to binary). r(N)rDrO)r+r:r2r(r4rGs r,rOzSSAFile.to_files: (( T2w::C:6:::::r.rhmsmsframesczt||||||}|D]"}|xj|z c_|xj|z c_#dS)aV Shift all subtitles by constant time amount. Shift may be time-based (the default) or frame-based. In the latter case, specify both frames and fps. h, m, s, ms will be ignored. Arguments: h: Integer or float values, may be positive or negative (hours). m: Integer or float values, may be positive or negative (minutes). s: Integer or float values, may be positive or negative (seconds). ms: Integer or float values, may be positive or negative (milliseconds). frames (int): When specified, must be an integer number of frames. May be positive or negative. fps must be also specified. fps (float): When specified, must be a positive number. Raises: ValueError: Invalid fps or missing number of frames. )rVrWrXrYrZr(N)rstartend) r+rVrWrXrYrZr(deltalines r,shiftz SSAFile.shiftsY*AaBv3GGG  D JJ% JJ HH HHH  r.in_fpsout_fpsc|dks|dkrtd|d|||z }|D]T}tt|j|z|_tt|j|z|_UdS)aJ Rescale all timestamps by ratio of in_fps/out_fps. Can be used to fix files converted from frame-based to time-based with wrongly assumed framerate. Arguments: in_fps (float) out_fps (float) Raises: ValueError: Non-positive framerate given. rz.Framerates must be positive, cannot transform z -> N) ValueErrorintroundr\r])r+rarbratior_s r,transform_frameratezSSAFile.transform_framerate8s Q;;'Q,,cfccZaccdd d  4 4DU4:#56677DJ5E!12233DHH 4 4r.old_namenew_namec"ddlm}||jvrtd|d||jvrt d|||st |d|j||j|<|j|=|D]}|j|kr||_dS)aw Rename a style, including references to it. Arguments: old_name (str): Style to be renamed. new_name (str): New name for the style (must be unused). Raises: KeyError: No style named old_name. ValueError: new_name is not a legal name (cannot use commas) or new_name is taken. r)is_valid_field_contentzStyle z not foundz There is already a style called z is not a valid nameN)formats.substationrlr#KeyErrorrdstyle)r+rirjrlr_s r, rename_stylezSSAFile.rename_styleSs ?>>>>> 4; & &:H:::;; ; t{ " "LLLMM M%%h// B@@@AA A $ H 5 H K ! & &DzX%%%  & &r.TrH overwritect|tstd|jD]\}}||jvs|r ||j|<dS)z Merge in styles from other SSAFile. Arguments: subs (SSAFile): Subtitle file imported from. overwrite (bool): On name conflict, use style from the other file (default: True). zMust supply an SSAFile.N) isinstancer TypeErrorr#items)r+rHrqnameros r, import_styleszSSAFile.import_stylesrso$(( 7566 6;,,.. * *KD%4;&&)&$) D! * *r.cg}t}i}t|D]t\}}||j|jfg}||jdkr||||jut|D]Y\}}|j s|j rt|j dkr?||vrD||Z||_ dS)a] Remove subtitles which appear to be non-essential (the --clean in CLI) Currently, this removes events matching any of these criteria: - SSA event type Comment - SSA drawing tags - Less than two characters of text - Duplicated text with identical time interval (only the first event is kept) rN)set enumerate setdefaultr\r]count plaintextaddappend is_drawing is_commentlenstripr )r+ new_eventsduplicate_text_idstimes_to_textsietmps r,remove_miscellaneous_eventsz#SSAFile.remove_miscellaneous_eventss  UU;=dOO $ $DAq ++QWae,z+SSAFile.get_text_events..s---a19----r.rr*s r,get_text_eventszSSAFile.get_text_eventss.-4----r.otherc t|trtt|j|jdhz D]}|j||j|}}|tjd|dS|tjd|dS||krtjd|||dStt|j |j D]}|j ||j |}}|tjd|dS|tjd|dS||krtjd |||dStt|j |j D]}|j ||j |}}|tjd |dS|tjd |dS||krtjd |||dStt|j |j D]}|j ||j |} } | tjd |dS| tjd|dS| | kr| j D]9} t| | t| | krtjd| :tjd|| | dSt|t|kr2tjdt|t|dSt!t#|j|jD]\} \} }| |s| j D]9} t| | t|| krtjd| :tjd| | |dSdSt)d)ah Equality of two SSAFiles. Compares :attr:`SSAFile.info`, :attr:`SSAFile.styles` and :attr:`SSAFile.events`. Order of entries in OrderedDicts does not matter. "ScriptType" key in info is considered an implementation detail and thus ignored. Useful mostly in unit tests. Differences are logged at DEBUG level. ScriptTypeNz%r missing in self.infoFz%r missing in other.infoz#info %r differs (self=%r, other=%r)z%r missing in self.fonts_opaquez %r missing in other.fonts_opaquez+fonts_opaque %r differs (self=%r, other=%r)z"%r missing in self.graphics_opaquez#%r missing in other.graphics_opaquez.graphics_opaque %r differs (self=%r, other=%r)z%r missing in self.stylesz%r missing in other.styleszdifference in field %rz$style %r differs (self=%r, other=%r)z,different # of subtitles (self=%d, other=%d)z$event %d differs (self=%r, other=%r)Tz$Cannot compare to non-SSAFile object)rsrrzrr$keysgetloggingdebugr&r'r#FIELDSgetattras_dictrr{zipr equalsrt)r+rkey self_info other_info self_font other_font self_image other_image self_style other_stylekr self_event other_events r,rzSSAFile.equalss eW % %B D5!1!15:??3D3DEEFF,W ! !(, c(:(:EJNN3S>X>X>Z>Z[[\\ ! !*.*>*B*B3*G*GI^IbIbcfIgIgK %M"FLLL 55 (M"GMMM 55;..M"RTWYcepqqq 55/5!1!1!3!3U\5F5F5H5HIIJJ ! !*.+//#*>*> @P@PQT@U@UK %M"=sCCC 55 (M">DDD 55;..'.GG":q11W[!5L5LLL#M*BAFFFM"H#zOaOaOcOcepexexezez{{{ 55 /4yyCJJ&& LcRViiY\]bYcYcdddu09#dk5<:X:X0Y0Y ! !,,J !((55!'.GG":q11W[!5L5LLL#M*BAFFFM"H!ZM_M_MaMacncvcvcxcxyyy 55 !4BCC Cr.c|jrRtd|D}dt|dt|jdt |d}ndt|jd}|S)Nc3$K|] }|jV dSN)r])revs r, z#SSAFile.__repr__..s$11b26111111r.zz)r maxrr#r)r+max_timerXs r,__repr__zSSAFile.__repr__s ; I11D11111HxTxxDK8H8HxxbkltbubuxxxAAHc$+.>.>HHHAr.c8|jdS)z#Sort subtitles time-wise, in-place.N)r sortr*s r,rz SSAFile.sort s r.c*t|jSr)iterr r*s r,__iter__zSSAFile.__iter__sDK   r.itemcdSrrr+rs r, __getitem__zSSAFile.__getitem__ r.cdSrrr+rXs r,rzSSAFile.__getitem__rr.c|j|Srr rs r,rzSSAFile.__getitem__s{4  r.rvaluecdSrr)r+rrs r, __setitem__zSSAFile.__setitem__rr.rvaluescdSrr)r+rrs r,rzSSAFile.__setitem__ rr.cZt|tr0t|tr ||j|<dSt dt|t rCt |}td|Dr ||j|<dSt dt d)N1SSAFile.events must contain only SSAEvent objectsc3@K|]}t|tVdSr)rsr)rvs r,rz&SSAFile.__setitem__..,s,;;q:a**;;;;;;r.z Bad key type)rsrerr rtslicelistall)r+rrrs r,rzSSAFile.__setitem__$s c3   ,%** U#( C    STTT U # # ,%[[F;;F;;;;; U#) C    STTTN++ +r.cdSrrr+rs r, __delitem__zSSAFile.__delitem__3rr.cdSrrrs r,rzSSAFile.__delitem__7rr.c|j|=dSrrrs r,rzSSAFile.__delitem__;s K   r.c*t|jSr)rr r*s r,__len__zSSAFile.__len__>s4;r.indexct|tr|j||dSt d)Nr)rsrr insertrt)r+rrs r,rzSSAFile.insertAsB eX & & Q K  ue , , , , ,OPP Pr.)rN)r/NNN)NNr)rrrrNN)T),__name__ __module__ __qualname____doc__rrstr__annotations__r- classmethodrfloatrr;r@r r8rQrTrOrrer`rhrpboolrwrrrrrrrr rr rrrrrrrrr.r,rr s$!&$$L$sCx. ****lp%)<A<A<As<Ax}<AZbchZi<Ac]<A58<A=F<A<A<A[<A| = = =x} =(SX/ =! =&/ = = =[ =D**6*HSM*xPU**$-***[*Xnr%)69696969 69[cdi[j69c]695869=A69696969p   8E? S UX    ;;&;3;Xe_;WZ;_c;;;;,_`CGz*ZQ[sm19%LP44%4%4D44446&S&C&D&&&&>**)*******!!!!@.h.... NDIND$NDNDNDND`#!(8,!!!!     X  U tH~   X !!!!!! s 8    X   x/A d   X  ,s ,3 ,4 , , , , s t   X  U t   X st     QCQQTQQQQQQr.r)rCrDrN)r> itertoolsros.pathrKrtypingrrrrrr r r r r commonrssaeventrssastylertimerrrformatsrCrDrNrr.r,rs/ jjjjjjjjjjjjjjjjjjjjjjjj&&&&&&&&xQxQxQxQxQoh'xQxQxQvPOOOOOOOOOOOr.