}i
E d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlZd dl Z d dl
m Z m Z m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z d dl m Z m Z de defd Z!de fd
Z" G d d e Z#y)
N)Path)TestCase)patch)YouTubeTranscriptApiTranscriptsDisabledNoTranscriptFoundVideoUnavailable IpBlockedNotTranslatable TranslationLanguageNotAvailableCookiePathInvalid
CookieInvalidFailedToCreateConsentCookieYouTubeRequestFailedInvalidVideoIdFetchedTranscriptFetchedTranscriptSnippet
AgeRestrictedRequestBlockedVideoUnplayablePoTokenRequired)GenericProxyConfigWebshareProxyConfigfilenamereturnc | t dj t j j t
| S )Nz{dirname}/assets/{filename})dirnamer )r formatospathr __file__)r s O/usr/local/lib/python3.12/dist-packages/youtube_transcript_api/test/test_api.pyget_asset_pathr# " s4 %,,GGOOH- -
c z t t | d 5 }|j cd d d S # 1 sw Y y xY w)Nrb)mode)openr# read)r files r"
load_assetr+ * s1
nX&T 2 dyy{ s 1:c P e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Z d Z d Z e d d Z e d d Z! e d d Z" e d d! Z# e d d" Z$e%jL jO d#$ d% Z(e%jL jO d#$ d& Z)e%jL jO d#$ d' Z*e%jL jO d#$ d( Z+y))*TestYouTubeTranscriptApic t t ddd t ddd t ddd gd
d d d
| _ | j j | _ t j t j t
j dt d t j t
j dt d t j t
j dt d y )NzHey, this is just a testg gp=
ף?)textstartdurationz#this is not the original transcriptgp=
ף@z-just something shorter, I made up for testingg@gx @EnglishenF GJLlxj_dtq8)snippetslanguage
language_code is_generatedvideo_id*https://www.youtube.com/youtubei/v1/playerz youtube.innertube.json.staticbody https://www.youtube.com/watchzyoutube.html.static%https://www.youtube.com/api/timedtextztranscript.xml.static) r r ref_transcript to_raw_dataref_transcript_raw httprettyenable register_uriPOSTr+ GETselfs r" setUpz TestYouTubeTranscriptApi.setUp0 s /(3 !
)> !
)H "" "+
. #'"5"5"A"A"C NN 8 ;<
MM + 12
MM 3 34
r$ c T t j t j y )N)rB resetdisablerG s r" tearDownz!TestYouTubeTranscriptApi.tearDownZ s r$ c n t j d }| j || j y )Nr4 )r fetch assertEqualr? rH
transcripts r"
test_fetchz#TestYouTubeTranscriptApi.test_fetch^ s/ )+11-@
r$ c t j dd }d| j d _ | j || j y )Nr4 T)preserve_formattingz*this is not the original transcript )r rO r? r/ rP rQ s r" test_fetch_formattedz-TestYouTubeTranscriptApi.test_fetch_formattedf sO )+11 t 2
'SA #
r$ c t j t j dt d t j d }| j
|| j y )Nr: z0youtube_altered_user_agent.innertube.json.staticr; r4 )rB rD rE r+ r rO rP r? rQ s r" #test_fetch__with_altered_user_agentz csder3 eshijakozh)r listr7 rP )rH transcript_listrR language_codess r" test_listz"TestYouTubeTranscriptApi.test_list sH .055mDETUz*22UU R
Vs Ac t j d }|j dg }| j |j y )Nr4 r[ )r rb find_manually_created_transcript assertFalser8 rH rc rR s r" test_list__find_manually_createdz9TestYouTubeTranscriptApi.test_list__find_manually_created s; .055mD$EEtfM
001r$ c t j d }| j t 5 |j dg d d d |j dg }| j |j y # 1 sw Y 7xY w)Nr4 r[ r3 )r rb assertRaisesr find_generated_transcript
assertTruer8 ri s r" test_list__find_generatedz2TestYouTubeTranscriptApi.test_list__find_generated so .055mD
0
1 > 5 5tf = > %>>vF
//0 > >s A88Bc t j t j dt d | j t
5 t
j d d d d y # 1 sw Y y xY w)Nr: /youtube_video_unavailable.innertube.json.staticr; z8https://www.youtube.com/youtubei/v1/player?v=GJLlxj_dtq8)rB rD rE r+ rl r r rb rG s r" test_list__url_as_video_idz3TestYouTubeTranscriptApi.test_list__url_as_video_id s\ NN 8 MN
~
. " ' 'J
A((A1c t j d j dg }|j d }| j |j
d | j
d|j y )Nr4 r3 arz &tlang=ar)r rb find_transcript translaterP r7 assertIn_url)rH rR translated_transcripts r" test_translate_transcriptz2TestYouTubeTranscriptApi.test_translate_transcript sa )+00 ?OOQUPVW
* 4 4T : .<r$ c t j d j dg }| j t 5 |j d d d d y # 1 sw Y y xY w)Nr4 r3 xyz)r rb rv rl r rw rQ s r" =test_translate_transcript__translation_language_not_availablezVTestYouTubeTranscriptApi.test_translate_transcript__translation_language_not_available sV )+00 ?OOQUPVW
>
? ( ' ( ( (s AA#c t j d j dg }g |_ | j t
5 |j
d d d d y # 1 sw Y y xY w)Nr4 r3 af)r rb rv translation_languagesrl r rw rQ s r" +test_translate_transcript__not_translatablezDTestYouTubeTranscriptApi.test_translate_transcript__not_translatable s] )+00 ?OOQUPVW
+-
(
/ ' & ' ' 's A!!A*c t j dddg t j j }| j d| | j
t |d d | j
|d d d y )Nr4 r\ r3 langrV r )r rO rB last_request querystringrx rP lenrH query_strings r" $test_fetch__correct_language_is_usedz=TestYouTubeTranscriptApi.test_fetch__correct_language_is_used so $$]T4LA --/;;
fl+ \& 12A6 f-a0$7r$ c f t j t j dt d t j dddg t j j }| j d| | j t |d d | j |d d d y )
Nr: z'youtube_ww1_nl_en.innertube.json.staticr; F1xioXWb8CYr\ r3 r rV r ) rB rD rE r+ r rO r r rx rP r r s r" %test_fetch__fallback_language_is_usedz>TestYouTubeTranscriptApi.test_fetch__fallback_language_is_used s NN 8 EF
$$]T4LA --/;;
fl+ \& 12A6 f-a0$7r$ c d t j t j dt d t j d | j
t t j d t j dd D ]! }| j
|j d d # y ) Nr= youtube_consent_page.html.staticr; r rV cookiez'CONSENT=YES+cb.20210328-17-p0.de+FX+119)
rB rD rF r+ r rO rP r latest_requestsheaders)rH requests r" +test_fetch__create_consent_cookie_if_neededzDTestYouTubeTranscriptApi.test_fetch__create_consent_cookie_if_needed s MM + >?
$$]3 Y 6 6 891= 002126 G )+T
r$ c t d D ]1 }t j t j dt d 3 | j t 5 t j d d d d y # 1 sw Y y xY w)N r= r r; r ) rangerB rD rF r+ rl r r rO )rH _s r" 5test_fetch__exception_if_create_consent_cookie_failedzNTestYouTubeTranscriptApi.test_fetch__exception_if_create_consent_cookie_failed sm q A " "
/ BC
:
; 8 " ( ( 7 8 8 8s A88Bc t j t j dt d | j t
5 t
j d d d d y # 1 sw Y y xY w)Nr= z(youtube_consent_page_invalid.html.staticr; r )rB rD rF r+ rl r r rO rG s r" 3test_fetch__exception_if_consent_cookie_age_invalidzLTestYouTubeTranscriptApi.test_fetch__exception_if_consent_cookie_age_invalid sZ MM + FG
:
; 8 " ( ( 7 8 8 8rs c t j t j dt d | j t
5 t
j d d d d y # 1 sw Y y xY w)Nr: rq r; abc)rB rD rE r+ rl r r rO rG s r" *test_fetch__exception_if_video_unavailablezCTestYouTubeTranscriptApi.test_fetch__exception_if_video_unavailable sZ NN 8 MN
/
0 0 " ( ( / 0 0 0rs c t j t j dd | j t 5 }t j
d d d d | j dt j y # 1 sw Y /xY w)Nr: i statusr zRequest to YouTube failed: )
rB rD rE rl r r rO rx str exceptionrH cms r" .test_fetch__exception_if_youtube_request_failszGTestYouTubeTranscriptApi.test_fetch__exception_if_youtube_request_fails sp NN HQT
3
4 0 " ( ( / 0
3S 5FG 0 0s BB
c t j t j dt d | j t
5 t
j d d d d y # 1 sw Y y xY wNr= z%youtube_too_many_requests.html.staticr; r rB rD rF r+ rl r
r rO rG s r" 6test_fetch__exception_if_youtube_request_limit_reachedzOTestYouTubeTranscriptApi.test_fetch__exception_if_youtube_request_limit_reached s[ MM + CD
y
) 0 " ( ( / 0 0 0rs c t j t j dd | j t 5 t j
d d d d y # 1 sw Y y xY w)Nr> i r r )rB rD rF rl r
r rO rG s r" 8test_fetch__exception_if_timedtext_request_limit_reachedzQTestYouTubeTranscriptApi.test_fetch__exception_if_timedtext_request_limit_reached sU MM 3
y
) 0 " ( ( / 0 0 0s A A(c t j t j dt d | j t
5 t
j d d d d y # 1 sw Y y xY w)Nr: z,youtube_age_restricted.innertube.json.staticr; Njp5uhTorCo)rB rD rE r+ rl r r rO rG s r" 'test_fetch__exception_if_age_restrictedz@TestYouTubeTranscriptApi.test_fetch__exception_if_age_restricted sY NN 8 JK
}
- 8 " ( ( 7 8 8 8rs c t j t j dt d | j t
5 t
j d d d d y # 1 sw Y y xY wr r rG s r" #test_fetch__exception_if_ip_blockedzTestYouTubeTranscriptApi.test_fetch__exception_request_blocked= su NN 8 KL
~
. 8" " ( ( 7 8
A3r||CTU 8 8s B
Bc t j t j dt d | j t
5 }t
j d d d d j }| j |j d | j |j ddg | j dt | y # 1 sw Y kxY w)Nr: z(youtube_unplayable.innertube.json.staticr; r z
Custom Reasonz Sub Reason 1z Sub Reason 2)rB rD rE r+ rl r r rO r rP reason sub_reasonsrx r )rH r r s r" test_fetch__exception_unplayablez9TestYouTubeTranscriptApi.test_fetch__exception_unplayableI s NN 8 FG
/ 82 " ( ( 7 8LL ))?; .. 0PQ
os9~6 8 8s C Cc t j t j dt d | j t
5 t
j d d d d t j t j dt d | j t
5 t
j d d d d y # 1 sw Y pxY w# 1 sw Y y xY w)Nr: z2youtube_transcripts_disabled.innertube.json.staticr; dsMFmonKDD4z3youtube_transcripts_disabled2.innertube.json.static Fjg5lYqvzUs)rB rD rE r+ rl r r rO rG s r" -test_fetch__exception_if_transcripts_disabledzFTestYouTubeTranscriptApi.test_fetch__exception_if_transcripts_disabledW s NN 8 PQ
2
3 8 " ( ( 7 8 NN 8 QR
2
3 8 " ( ( 7 8 8 8 8 8 8s C+CCC#c | j t 5 }t j ddg d d d | j dt j y # 1 sw Y /xY w)Nr4 cz) languagesz No transcripts were found for)rl r r rO rx r r r s r" -test_fetch__exception_if_language_unavailablezFTestYouTubeTranscriptApi.test_fetch__exception_if_language_unavailablei s^
0
1 JR " ( ( 4& ( I J
5s2<<7HI J Js A!!A*zByoutube_transcript_api.proxies.GenericProxyConfig.to_requests_dictc t dd }t | j d }| j || j |j y )Nhttp://localhost:8080http_url https_url proxy_configr4 )r r rO rP r? assert_any_call)rH to_requests_dictr rR s r" test_fetch__with_proxyz/TestYouTubeTranscriptApi.test_fetch__with_proxyo sV ),-
*|DJJ
((*r$ c t dd }t | j d t j }| j |j j d d y )Nusernamepasswordproxy_usernameproxy_passwordr r4
Connectionclose)r r rO rB r rP r get)rH r r r s r" 0test_fetch__with_proxy_prevent_alive_connectionszITestYouTubeTranscriptApi.test_fetch__with_proxy_prevent_alive_connections~ sS *%j
,7==mL((* ,,\:GDr$ c , t d D ]1 }t j t j dt d 3 t dd }t
| j d | j t t j d
y ) N r: r r; r r r r r ) r rB rD rE r+ r r rO rP r r )rH r r r s r" )test_fetch__with_proxy_retry_when_blockedzBTestYouTubeTranscriptApi.test_fetch__with_proxy_retry_when_blocked s{ q A " "< OP
+%%
,7==mL Y 6 6 899Er$ c " d}t | D ]1 }t j t j dt d 3 t dd| }| j
t 5 }t | j d d d d | j t t j |d
z | j j j | | j d t! |j y # 1 sw Y xY w) N r: r r; r r )r r retries_when_blockedr r r Webshare)r rB rD rE r+ r rl r r rO rP r r r
_proxy_configrx r )rH r retriesr r r s r" 4test_fetch__with_webshare_proxy_reraise_when_blockedzMTestYouTubeTranscriptApi.test_fetch__with_webshare_proxy_reraise_when_blocked s w A " "< OP
+%%!(
~
. Q" l ; A A- P Q
Y 6 6 897Q;G 33\B
j#bll"34 Q Qs % DDc t j t j dt d t dd }| j t 5 }t | j d d d d | j t t j d | j j j | | j d t |j y # 1 sw Y xY w)
Nr: r r; r r r r r z!YouTube is blocking your requests)rB rD rE r+ r rl r r rO rP r r r r rx r )rH r r r s r" 3test_fetch__with_generic_proxy_reraise_when_blockedzLTestYouTubeTranscriptApi.test_fetch__with_generic_proxy_reraise_when_blocked s NN 8 KL
*,-
~
. Q" l ; A A- P Q
Y 6 6 891= 33\B
93r||;LM Q Qs C//C8zfThis test is temporarily disabled because cookie auth is currently not working due to YouTube changes.)r c t d }t | j d }| j || j y )Nexample_cookies.txt cookie_pathr4 )r# r rO rP r? )rH r rR s r" test_fetch__with_cookiesz1TestYouTubeTranscriptApi.test_fetch__with_cookies s>
%%:; )kBHH W
r$ c t d }t | }|j j j }| j ddit j j | y )Nr r
TEST_FIELD
TEST_VALUE) r# r _fetcher _http_clientcookiesrP requestsutilsdict_from_cookiejar)rH r ytt_apisession_cookiess r" test_load_cookiesz*TestYouTubeTranscriptApi.test_load_cookies sX
%%:; &;?!**77??
< ( NN . . ?
r$ c t d }| j t 5 t | d d d y # 1 sw Y y xY w)Nznonexistent_cookies.txtr )r# rl r
r rH r s r" test_load_cookies__bad_file_pathz9TestYouTubeTranscriptApi.test_load_cookies__bad_file_path s;
%%>?
0
1 : [ 9 : : :
7A c t d }| j t 5 t | d d d y # 1 sw Y y xY w)Nzexpired_example_cookies.txtr )r# rl r r r s r" #test_load_cookies__no_valid_cookieszr sf
( SS T
@:x @:r$