U g=@sdZddlZddlZddlZddlZddlmZddlmZdZ dZ ddd d d d gZ d dddiddiddd ddddiddiddd ddddiddiddd ddd d!iddiddd d"dd dd#iddiddd d$dddiddiddd d%d&diddiddd d'd(ddidd)iddd d*d+ddidd,iddd d-d.diddiddd d/d0ddidd1iddd d2d3ddidd4iddd d5d6diddiddd d7d8diddiddd9Z d:Z eejd;Zejed<ZGd=d>d>ZdS)?zThis module is designed to interact with the innertube API. This module is NOT intended to be used directly by end users, as each of the interfaces returns raw results. These should instead be parsed to extract the useful information for the end user. N)parse)requestzH861556708454-d6dlm3lh05idd8npek18k6be8ba3oc68.apps.googleusercontent.comZSboVhoG9s0rNafixCSGGKXATZ'AIzaSyAO_FJ2SlqU8Q4STEHLGCilw_Y9_11qcW8Z'AIzaSyCtkvNIR1HCEwzsqK6JuE6KqpyjusIRI30z'AIzaSyA8eiZmM1FaDVjRy-df2KTyQ_vz_yYM39wZ'AIzaSyC8UYZpvA2eknNex0Pjid0_eTLJoDu6losZ'AIzaSyCjc_pVEDi4qsv5MtC2dMXzpIaDoRFLsxwZ'AIzaSyDHQ9ipnphqTzDqZsbtd8_Ru4_kiKVQe2kclientWEBz2.20200720.00.02) clientName clientVersionz User-Agentz Mozilla/5.0)contextheaderapi_keyANDROIDz17.31.35)rrandroidSdkVersionzcom.google.android.youtube/IOSz17.33.2z iPhone14,3)rrZ deviceModelzcom.google.ios.youtube/ZWEB_EMBEDDED_PLAYERz2.20210721.00.00ZEMBED)rr clientScreenZANDROID_EMBEDDED_PLAYER)rrrr ZIOS_MESSAGES_EXTENSIONZ WEB_REMIXz1.20220727.01.00 ANDROID_MUSICz5.16.51z&com.google.android.apps.youtube.music/ IOS_MUSICz5.21zcom.google.ios.youtubemusic/ WEB_CREATORz1.20220726.00.00ANDROID_CREATORz 22.30.100z(com.google.android.apps.youtube.creator/ IOS_CREATORz 22.33.101zcom.google.ios.ytcreator/MWEBz2.20220801.00.00ZTVHTML5_SIMPLY_EMBEDDED_PLAYERz2.0)rr rZ WEB_EMBEDZ ANDROID_EMBEDZ IOS_EMBEDZ WEB_MUSICrrrrrrZTV_EMBEDiZ __cache__z tokens.jsonc@seZdZdZd&ddZddZd'd d Zd d Zed dZ eddZ eddZ ddZ ddZ ddZddZddZddZd(d d!Zd"d#Zd$d%ZdS)) InnerTubez.Object for interacting with the innertube API.rFTc Cst|d|_t|d|_t|d|_d|_d|_||_||_d|_|jr|jrt j t rt t 6}t|}|d|_|d|_|d|_|W5QRXdS)agInitialize an InnerTube object. :param str client: Client to use for the object. Default to web because it returns the most playback types. :param bool use_oauth: Whether or not to authenticate to YouTube. :param bool allow_cache: Allows caching of oauth tokens on the machine. rr r N access_token refresh_tokenexpires)_default_clientsrr r rr use_oauth allow_cacherospathexists _token_fileopenjsonloadrefresh_bearer_token)selfrrrfdatar(4/tmp/pip-unpacked-wheel-1a9f0fi6/pytube/innertube.py__init__s        zInnerTube.__init__c CsX|js dS|j|j|jd}tjts2ttt t d}t ||W5QRXdS)z Cache tokens to file if allowed.N)rrrw) rrrrrrr _cache_dirmkdirr!r r"dump)r%r'r&r(r(r) cache_tokenss   zInnerTube.cache_tokenscCs|js dS|jtkr |s dSttd}ttd|jd}tjddddi|d }t | }|d |_ ||d |_| dS) zxRefreshes the OAuth token if necessary. :param bool force: Force-refresh the bearer token. Nr r) client_id client_secret grant_typer#https://oauth2.googleapis.com/tokenPOST Content-Typeapplication/jsonheadersr'r expires_in)rrtimeint _client_id_client_secretrr_execute_requestr"loadsreadrr/)r%force start_timer'response response_datar(r(r)r$s, zInnerTube.refresh_bearer_tokencCsttd}tdd}tjddddi|d}t|}|d }|d }td |d |t d tt |ddd}tjddddi|d}t|}|d|_ |d|_ ||d|_ |dS)zFetch an OAuth token.r z'https://www.googleapis.com/auth/youtube)r0Zscopez)https://oauth2.googleapis.com/device/coder4r5r6r7verification_url user_codez Please open z and input code z.Press enter when you have completed this step. device_codez,urn:ietf:params:oauth:grant-type:device_code)r0r1rGr2r3rrr9N)r;r:r<rr>r"r?r@printinputr=rrrr/)r%rBr'rCrDrErFr(r(r)fetch_bearer_token1sF  zInnerTube.fetch_bearer_tokencCsdS)z3Return the base url endpoint for the innertube API.z#https://www.youtube.com/youtubei/v1r(r%r(r(r)base_url\szInnerTube.base_urlcCs d|jiS)z;Return the base json data to transmit to the innertube API.r)rrKr(r(r) base_dataaszInnerTube.base_datacCs|jdddS)zBReturn the base query parameters to transmit to the innertube API.T)keyZcontentCheckOkZ racyCheckOk)r rKr(r(r) base_paramshszInnerTube.base_paramscCs|jr |d=|dt|}ddi}|jrf|jrN|d|j|d<n|d|j|d<||jtj |d||d}t | S) zOMake a request to a given endpoint with the provided query parameters and data.rN?r5r6zBearer Authorizationr4r7) rr urlencoderr$rJupdater rr>r"r?r@)r%endpointqueryr'Z endpoint_urlr8rCr(r(r) _call_apiqs( zInnerTube._call_apicCsdS)z]Make a request to the browse endpoint. TODO: Figure out how we can use this Nr(rKr(r(r)browseszInnerTube.browsecCsdS)z]Make a request to the config endpoint. TODO: Figure out how we can use this Nr(rKr(r(r)configszInnerTube.configcCsdS)z\Make a request to the guide endpoint. TODO: Figure out how we can use this Nr(rKr(r(r)guideszInnerTube.guidecCsdS)z[Make a request to the next endpoint. TODO: Figure out how we can use this Nr(rKr(r(r)nextszInnerTube.nextcCs0|jd}d|i}||j||||jS)zMake a request to the player endpoint. :param str video_id: The video id to get player info for. :rtype: dict :returns: Raw player info results. z/playervideoIdrLrSrOrVrM)r%video_idrTrUr(r(r)players  zInnerTube.playerNcCsJ|jd}d|i}||ji}|r0||d<||j||||S)zMake a request to the search endpoint. :param str search_query: The query to search. :rtype: dict :returns: Raw search query results. z/searchrU continuation)rLrSrOrMrV)r%Z search_queryr_rTrUr'r(r(r)searchs   zInnerTube.searchcCsD|jd}ddd|iidd}||j|||j|}|S)aMake a request to the age_verify endpoint. Notable examples of the types of video this verification step is for: * https://www.youtube.com/watch?v=QLdAhwSBZ3w * https://www.youtube.com/watch?v=hc0ZDaAZQT0 :param str video_id: The video id to get player info for. :rtype: dict :returns: Returns information that includes a URL for bypassing certain restrictions. z /verify_ageZ urlEndpointurlz /watch?v=T)Z nextEndpointZsetControvercy)rLrSrMrVrO)r%r]rTr'resultr(r(r) verify_ages  zInnerTube.verify_agecCs4|jd}d|i}||j||||j}|S)zMake a request to the get_transcript endpoint. This is likely related to captioning for videos, but is currently untested. z/get_transcriptr[r\)r%r]rTrUrbr(r(r)get_transcripts  zInnerTube.get_transcript)rFT)F)N)__name__ __module__ __qualname____doc__r*r/r$rJpropertyrLrMrOrVrWrXrYrZr^r`rcrdr(r(r(r)rs( ! "+        r)rhr"rpathlibr:urllibrZpytuberr<r=Z _api_keysrZ_token_timeoutPath__file__parentresolver,rjoinr rr(r(r(r)sX    :