}iGddlmZmZddlmZddlmZddlmZddl m Z m Z m Z m Z mZmZddlmZddlZddlmZmZmZd d lmZd d lmZmZmZd d lmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+eGd dZ,eGddZ-eGddZ.Gdde/eZ0Gdde/eZ1dede/defdZ2GddZ3GddZ4Gdd Z5Gd!d"Z6y)#) dataclassasdict)Enum)chain)unescape)ListDictIteratorIterablePatternOptional) ElementTreeN) HTTPErrorSessionResponse) ProxyConfig) WATCH_URLINNERTUBE_CONTEXTINNERTUBE_API_URL)VideoUnavailableYouTubeRequestFailedNoTranscriptFoundTranscriptsDisabledNotTranslatableTranslationLanguageNotAvailableFailedToCreateConsentCookieInvalidVideoId IpBlockedRequestBlocked AgeRestrictedVideoUnplayableYouTubeDataUnparsablePoTokenRequiredc.eZdZUeed<eed< eed<y)FetchedTranscriptSnippettextstartdurationN)__name__ __module__ __qualname__str__annotations__floatN/usr/local/lib/python3.12/dist-packages/youtube_transcript_api/_transcripts.pyr&r&"s I LOr1r&ceZdZUdZeeed<eed<eed<eed<eed<de efdZ defd Z de fd Z deefd Zy ) FetchedTranscriptz Represents a fetched transcript. This object is iterable, which allows you to iterate over the transcript snippets. snippetsvideo_idlanguage language_code is_generatedreturnc,t|jSN)iterr5selfs r2__iter__zFetchedTranscript.__iter__>sDMM""r1c |j|Sr<)r5)r?indexs r2 __getitem__zFetchedTranscript.__getitem__As}}U##r1c,t|jSr<)lenr5r>s r2__len__zFetchedTranscript.__len__Ds4==!!r1c>|Dcgc] }t|c}Scc}wr<)r)r?snippets r2 to_raw_datazFetchedTranscript.to_raw_dataGs/34Gw444sN)r*r+r,__doc__rr&r.r-boolr r@rCintrFr rIr0r1r2r4r41sf +,,MM#(#;<#$$<$""5T$Z5r1r4c"eZdZUeed<eed<y)_TranslationLanguager7r8N)r*r+r,r-r.r0r1r2rNrNKs Mr1rNceZdZdZdZdZy)_PlayabilityStatusOKERRORLOGIN_REQUIREDN)r*r+r,rQrRrSr0r1r2rPrPQs B E%Nr1rPceZdZdZdZdZy)_PlayabilityFailedReasonu%Sign in to confirm you’re not a botz/This video may be inappropriate for some users.zThis video is unavailableN)r*r+r, BOT_DETECTEDAGE_RESTRICTEDVIDEO_UNAVAILABLEr0r1r2rUrUWs:LFN3r1rUresponser6r:c |jdk(r t||j|S#t$r}t ||d}~wwxYw)Ni) status_coderraise_for_statusrr)rYr6errors r2_raise_http_errorsr^]sN4   3 &H% %!!# 4"8U334s+. A AAczeZdZdededededededeefdZdd ed e fd Z d efd Z e d efd Z ded dfdZy) Transcript http_clientr6urlr7r8r9translation_languagesc||_||_||_||_||_||_||_|Dcic]}|j|jc}|_ycc}w)z You probably don't want to initialize this directly. Usually you'll access Transcript objects using a TranscriptList. N) _http_clientr6_urlr7r8r9rc_translation_languages_dict) r?rar6rbr7r8r9rctranslation_languages r2__init__zTranscript.__init__hsm(     *(%:")>, $ ! . .0D0M0M M, (, sApreserve_formattingr:ctd|jvrt|j|jj |j}t |j t||jj}t||j|j|j|jS)z Loads the actual transcript data. :param preserve_formatting: whether to keep select HTML text formatting z&exp=xpe)rj)r5r6r7r8r9) rfr$r6reget_TranscriptParserparser^r'r4r7r8r9)r?rjrYr5s r2fetchzTranscript.fetchs  "!$--0 0$$((3$9LMSS x 7 < < !]]]],,**   r1cxdj|j|j|jrdSdS)Nz7{language_code} ("{language}"){translation_description}z[TRANSLATABLE])r7r8translation_description)formatr7r8is_translatabler>s r2__str__zTranscript.__str__sNHOO]],,8<8L8L$4P  SUP  r1c2t|jdkDS)Nr)rErcr>s r2rtzTranscript.is_translatables4--.22r1c $|jst|j||jvrt |jt |j |jdj|j||j||dgS)Nz{url}&tlang={language_code})rbr8T) rtrr6rgrr`rersrf)r?r8s r2 translatezTranscript.translates##!$--0 0  @ @ @1$--@ @    MM ) 0 0II] 1   , ,] ;     r1NF)r*r+r,rr-rKrrNrir4rorupropertyrtrxr0r1r2r`r`gs           $$89 4  :K &  333 s | r1r`c eZdZdZdedeeefdeeefdeefdZ e de deded dfd Z d e efd Zd eed efd Zd eed efdZd eed efdZd eedeeeefd efdZd efdZdeed efdZy)TranscriptListz This object represents a list of transcripts. It can be iterated over to list all transcripts which are available for a given YouTube video. Also, it provides functionality to search for a transcript in a given language. r6manually_created_transcriptsgenerated_transcriptsrcc<||_||_||_||_y)a The constructor is only for internal use. Use the static build method instead. :param video_id: the id of the video this TranscriptList is for :param manually_created_transcripts: dict mapping language codes to the manually created transcripts :param generated_transcripts: dict mapping language codes to the generated transcripts :param translation_languages: list of languages which can be used for translatable languages N)r6_manually_created_transcripts_generated_transcripts_translation_languages)r?r6r}r~rcs r2rizTranscriptList.__init__s#! -I*&;#&;#r1ra captions_jsonr:c |jdgDcgc]}t|dddd|d }}i}i}|dD]|}|jd d d k(r|}n|}t|||d jd d |dddd|d|jd d d k(|jddr|ng||d<~t ||||Scc}w)a] Factory method for TranscriptList. :param http_client: http client which is used to make the transcript retrieving http calls :param video_id: the id of the video this TranscriptList is for :param captions_json: the JSON parsed from the YouTube pages static HTML :return: the created TranscriptList translationLanguages languageNamerunsrr' languageCoder7r8 captionTrackskindrqasrbaseUrlz &fmt=srv3nameisTranslatableF)rlrNr`replacer|) rar6rrhrcr}r~captiontranscript_dicts r2buildzTranscriptList.builds+")6(9(9:PRT(U ! % !-n=fEaHP2>B ! ! (*$ "$_5 G{{62&%/"7">7A "**;;'*62' FB'50)05Eu)M%SU8OGN3 4     ( ! !   7! s#Cczt|jj|jjSr<)rrvaluesrr>s r2r@zTranscriptList.__iter__s3  . . 5 5 7  ' ' . . 0  r1language_codescR|j||j|jgS)a> Finds a transcript for a given language code. Manually created transcripts are returned first and only if none are found, generated transcripts are used. If you only want generated transcripts use `find_manually_created_transcript` instead. :param language_codes: A list of language codes in a descending priority. For example, if this is set to ['de', 'en'] it will first try to fetch the german transcript (de) and then fetch the english transcript (en) if it fails to do so. :return: the found Transcript )_find_transcriptrrr?rs r2find_transcriptzTranscriptList.find_transcripts.$$   / /1L1L M  r1c<|j||jgS)a Finds an automatically generated transcript for a given language code. :param language_codes: A list of language codes in a descending priority. For example, if this is set to ['de', 'en'] it will first try to fetch the german transcript (de) and then fetch the english transcript (en) if it fails to do so. :return: the found Transcript )rrrs r2find_generated_transcriptz(TranscriptList.find_generated_transcripts $$^d6Q6Q5RSSr1c<|j||jgS)a| Finds a manually created transcript for a given language code. :param language_codes: A list of language codes in a descending priority. For example, if this is set to ['de', 'en'] it will first try to fetch the german transcript (de) and then fetch the english transcript (en) if it fails to do so. :return: the found Transcript )rrrs r2 find_manually_created_transcriptz/TranscriptList.find_manually_created_transcripts%$$ T??@  r1transcript_dictscf|D]}|D]}||vs||ccSt|j||r<)rr6)r?rrr8rs r2rzTranscriptList._find_transcript,sJ , :M#3 : O3*=99 : :   ~tDDr1c 8dj|j|jd|jj D|jd|j j D|jd|j DS)NaFor this video ({video_id}) transcripts are available in the following languages: (MANUALLY CREATED) {available_manually_created_transcript_languages} (GENERATED) {available_generated_transcripts} (TRANSLATION LANGUAGES) {available_translation_languages}c32K|]}t|ywr<r-.0 transcripts r2 z)TranscriptList.__str__..Cs[J[c32K|]}t|ywr<rrs r2rz)TranscriptList.__str__..GsK$.JKrc3jK|]+}dj|j|j-yw)z{language_code} ("{language}")rN)rsr7r8)rrhs r2rz)TranscriptList.__str__..Js>K ) 1771::"6"D"D8Ks13)r6/available_manually_created_transcript_languagesavailable_generated_transcriptsavailable_translation_languages)rsr6_get_language_descriptionrrrrr>s r2ruzTranscriptList.__str__8s 0 &]]<@djd|D}|r|SdS)N c3@K|]}dj|yw)z - {transcript})rN)rsrs r2rz;TranscriptList._get_language_description..Ts&   $ $ $ ; sNone)join)r?r descriptions r2rz(TranscriptList._get_language_descriptionSs,ii 0  *{5v5r1N)r*r+r,rJr-r r`rrNri staticmethodrrr r@r rrrrrurr0r1r2r|r|s4 <<'+3 ?&;< $CO4 < $$89 <(+ + (++ <@+ + + Z (:.  hsm  T T* T  &sm     E   EtCO45 E  E  66HSM6c6r1r|ceZdZdedeefdZdedefdZ ddede de fdZ d ededefd Z d e dede fd Zd e deddfdZd ededdfdZdedefdZdedefdZdedede fdZy)TranscriptListFetcherra proxy_configc ||_||_yr<)re _proxy_config)r?rars r2rizTranscriptListFetcher.__init__\s')r1r6r:cbtj|j||j|Sr<)r|rre_fetch_captions_json)r?r6s r2rozTranscriptListFetcher.fetch`s/##      % %h /  r1 try_numbercx |j|}|j||}|j||}|j||S#t$rg}|j dn|j j }|dz|kr|j||dzcYd}~S|j|j d}~wwxYw)Nrr)r) _fetch_video_html_extract_innertube_api_key_fetch_innertube_data_extract_captions_jsonr rretries_when_blockedrwith_proxy_config)r?r6rhtmlapi_keyinnertube_data exceptionretriess r2rz*TranscriptListFetcher._fetch_captions_jsongs B))(3D55dHEG!77'JN..~xH H B%%-''<<  A~'00jSTn0UU--d.@.@A A Bs%AA B9AB4B9B44B9rcd}tj||}|r-t|jdk(r|j dSd|vr t |t |)Nz)"INNERTUBE_API_KEY":\s*"([a-zA-Z0-9_-]+)"rzclass="g-recaptcha")researchrEgroupsgrouprr#)r?rr6patternmatchs r2rz0TranscriptListFetcher._extract_innertube_api_keywsX> '4( S(A-;;q> ! D (H% %#H--r1rc|j|jd||jdijd}|d|vr t||S)NplayabilityStatuscaptionsplayerCaptionsTracklistRendererr)_assert_playabilityrlr)r?rr6rs r2rz,TranscriptListFetcher._extract_captions_jsons^   !3!34G!H(S&**:r:>> -   O=$H%h/ /r1playability_status_dataNc .|jd}|tjjk7ra|]|jd}|tjjk(rP|t j jk(r t||t jjk(r t||tjjk(rU|t jjk(r8|jds|jdr t|t||jdijdijdijdg}t|||Dcgc]}|jd d c}yycc}w) Nstatusreasonzhttp://zhttps:// errorScreenplayerErrorMessageRenderer subreasonrr'rq)rlrPrQvaluerSrUrVr rWr!rRrX startswithrrr")r?rr6playability_statusr subreasonsruns r2rz)TranscriptListFetcher._assert_playabilitysc488B "4"7"7"="= =".,00:F!%7%F%F%L%LL5BBHHH(225DDJJJ'11"&8&>&>&D&DD6HHNNN&&y1X5H5H5T(22&x00'++M2>126["%VR  "&*"M337762#6"M +/ >.#Ns0F ctjd|}| t||jjj dd|j dzdy)Nzname="v" value="(.*?)"CONSENTzYES+rz .youtube.com)domain)rrrrecookiessetr)r?rr6rs r2_create_consent_cookiez,TranscriptListFetcher._create_consent_cookiesV 2D9 =-h7 7 !!%% v A.~ & r1c|j|}d|vr2|j|||j|}d|vr t||S)Nz&action="https://consent.youtube.com/s") _fetch_htmlrr)r?r6rs r2rz'TranscriptListFetcher._fetch_video_htmlsR) 3t ;  ' 'h 7##H-D74?1(;; r1c|jjtj|}t t ||j S)N)r6)rerlrrsrr^r')r?r6rYs r2rz!TranscriptListFetcher._fetch_htmls<$$(()9)98)LM*8X>CCDDr1rc|jjtj|t|d}t ||j }|S)N)r)contextvideoId)json)repostrrsrr^r)r?r6rrYdatas r2rz+TranscriptListFetcher._fetch_innertube_datasR$$))  $ $W 5,#* "(H5::< r1)r)r*r+r,rr rrir-r|rorLr rrrrrrrrr0r1r2rr[s*G*8K;P* c n BSBcB$B .s.c.c. T S T 43SW: 3 # $ ##ECECE c C D r1rcNeZdZgdZd defdZdedeefdZdede e fdZ y) rm) strongembimarksmalldelinssubsuprjc0|j||_yr<)_get_html_regex _html_regex)r?rjs r2riz_TranscriptParser.__init__s//0CDr1r:c|rIdj|j}d|zdz}tj|tj}|Stjdtj}|S)N|z <\/?(?!\/?(z )\b).*?\b>z<[^>]*>)r_FORMATTING_TAGSrcompile IGNORECASE)r?rj formats_regex html_regexs r2rz!_TranscriptParser._get_html_regexs` HHT%:%:;M*]:]JMM2==AJJ >Jr1raw_datac Xtj|Dcgc]}|jytt j |j dt|jt|jdt|jjddc}Scc}w)Nrqr(durz0.0)r'r(r)) r fromstringr'r&rrrrr/attribrl)r?r xml_elements r2rnz_TranscriptParser.parses +55h?  + %VVD,,b(;;K;K2LMK..w78{1155eUCD    sB B'Nry) r*r+r,rrKrir r-rrr&rnr0r1r2rmrmsH EDE4GCL  c  d+C&D  r1rm)7 dataclassesrrenumr itertoolsrrrtypingrr r r r r defusedxmlrrrequestsrrrproxiesr _settingsrrr_errorsrrrrrrrrrr r!r"r#r$r&r4rNr-rPrUr^r`r|rrmr0r1r2rs)DD" 11 FF$      55 52   &d& 4sD4 44S4X4J J Zd6d6Nkk\# # r1