U 5d?<@sdZddlZddlmZmZmZmZmZddlZddl m Z ddlm Z m Z ddlm Z mZddlmZddlmZddlmZdd lmZeeZGd d d ZdS) a This module implements the core developer interface for pytube. The problem domain of the :class:`YouTube class focuses almost exclusively on the developer interface. Pytube offloads the heavy lifting to smaller peripheral modules and functions. N)AnyCallableDictListOptional)extractrequest)Stream StreamQuery) install_proxy) InnerTube)YouTubeMetadata) Monostatec@s|eZdZdZdMeeeeee gdfeeeeegdfe eefe e dddZ dd Z ee d d d Zed dZeddZeddZeddZeddZeddZeddZeddZddZedd Zd!d"Zeeejd#d$d%Zeej d#d&d'Z!ee"d#d(d)Z#eed#d*d+Z$ed,d-Z%e%j&d.d-Z%eed#d/d0Z'e'j&d1d0Z'eed#d2d3Z(ee)d#d4d5Z*ee d#d6d7Z+ee d#d8d9Z,eed#d:d;Z-e-j&dZ.eed#d?d@Z/eed#dAdBZ0eee1d#dCdDZ2eeee gdfdEdFdGZ3eeeegdfdEdHdIZ4e5eddJdKdLZ6dS)NYouTubez$Core developer interface for pytube.NFT)urlon_progress_callbackon_complete_callbackproxies use_oauthallow_oauth_cachecCsd|_d|_d|_d|_d|_d|_d|_d|_d|_d|_ t ||_ d|j |_ d|j |_ t||d|_|r~t|d|_d|_d|_||_||_dS)aConstruct a :class:`YouTube `. :param str url: A valid YouTube watch URL. :param func on_progress_callback: (Optional) User defined callback function for stream download progress events. :param func on_complete_callback: (Optional) User defined callback function for stream download complete events. :param dict proxies: (Optional) A dict mapping protocol to proxy address which will be used by pytube. :param bool use_oauth: (Optional) Prompt the user to authenticate to YouTube. If allow_oauth_cache is set to True, the user should only be prompted once. :param bool allow_oauth_cache: (Optional) Cache OAuth tokens locally on the machine. Defaults to True. These tokens are only generated if use_oauth is set to True as well. Nzhttps://youtube.com/watch?v=zhttps://www.youtube.com/embed/) on_progress on_complete)_js_js_url _vid_info _watch_html _embed_htmlZ_player_config_args_age_restricted _fmt_streams _initial_data _metadatarvideo_id watch_url embed_urlrstream_monostater _author_title _publish_daterr)selfrrrrrrr)3/tmp/pip-unpacked-wheel-qmwpt_in/pytube/__main__.py__init__s0 zYouTube.__init__cCsd|jdS)Nz)r!r(r)r)r*__repr__[szYouTube.__repr__)oreturncCst|t|ko|j|jkSN)typer")r(r0r)r)r*__eq__^szYouTube.__eq__cCs"|jr |jStj|jd|_|jSN)r)rrgetr"r.r)r)r* watch_htmlbszYouTube.watch_htmlcCs"|jr |jStj|jd|_|jSr5)rrr6r#r.r)r)r* embed_htmliszYouTube.embed_htmlcCs |jr |jSt|j|_|jSr2)rrZis_age_restrictedr7r.r)r)r*age_restrictedpszYouTube.age_restrictedcCs6|jr |jS|jr"t|j|_nt|j|_|jSr2)rr9rjs_urlr8r7r.r)r)r*r:ws zYouTube.js_urlcCsF|jr |jStj|jkr8t|j|_|jt_|jt_ntj|_|jSr2)rpytube __js_url__r:rr6__js__r.r)r)r*jss  z YouTube.jscCs |jr |jSt|j|_|jSr2)rr initial_datar7r.r)r)r*r?szYouTube.initial_datacCs*d|jkr|jdS||jdSdS)z%Return streamingData from video info.Z streamingDataN)vid_infobypass_age_gater.r)r)r*streaming_datas  zYouTube.streaming_datacCs||jr|jSg|_t|j}zt||j|jWn@tj k r|d|_ d|_ dt _ dt _t||j|jYnX|D]}t||jd}|j|q|j|j_|j|j_|jS)zReturns a list of streams if they have been initialized. If the streams have not been initialized, finds all relevant streams and initializes them. N)streamZ monostate)check_availabilityrrZapply_descramblerrBZapply_signaturer@r> exceptions ExtractErrorrrr;r=r<r r$appendtitlelengthduration)r(Zstream_manifestrCvideor)r)r* fmt_streamss,   zYouTube.fmt_streamscCst|j\}}|D]}|dkr`|dkr8tj|jdq|dkrPtj|jdqtj|jdq|dkr|dkrtj|jdq|dkr|dkrtj|jdq|d krtj |jdqd S) zCheck whether the video is available. Raises different exceptions based on why the video is unavailable, otherwise does nothing. UNPLAYABLEzcJoin this channel to get access to members-only content like this video, and other exclusive perks.r-z,This live stream recording is not available.ZLOGIN_REQUIREDzFThis is a private video. Please sign in to verify that you may see it.ERRORzVideo unavailableZ LIVE_STREAMN) rplayability_statusr7rEZ MembersOnlyr!ZRecordingUnavailableZVideoUnavailableZ VideoPrivateZLiveStreamError)r(statusmessagesreasonr)r)r*rDs(zYouTube.check_availabilitycCs4|jr |jSt|j|jd}||j}||_|jS)z]Parse the raw vid info and return the parsed result. :rtype: Dict[Any, Any] )r allow_cache)rr rrplayerr!)r( innertubeinnertube_responser)r)r*r@s  zYouTube.vid_infocCsLtd|j|jd}||j}|ddd}|dkrBt|j||_dS)z9Attempt to update the vid_info by bypassing the age gate.Z ANDROID_EMBED)clientrrSZplayabilityStatusrPNrM) r rrrTr!r6rEZAgeRestrictedErrorr)r(rUrVrOr)r)r*rAs  zYouTube.bypass_age_gate)r1cCs,|jdididg}dd|DS)zQGet a list of :class:`Caption `. :rtype: List[Caption] captionsZplayerCaptionsTracklistRendererZ captionTrackscSsg|]}t|qSr))r;Caption).0trackr)r)r* sz*YouTube.caption_tracks..r@r6)r(Z raw_tracksr)r)r*caption_tracks szYouTube.caption_trackscCs t|jS)zbInterface to query caption tracks. :rtype: :class:`CaptionQuery `. )r; CaptionQueryr^r.r)r)r*rXszYouTube.captionscCs|t|jS)z~Interface to query both adaptive (DASH) and progressive streams. :rtype: :class:`StreamQuery `. )rDr rLr.r)r)r*streams!szYouTube.streamscCs>|jdidid}|r0|d}|dSd|jdS)z:Get the thumbnail url image. :rtype: str videoDetailsZ thumbnailZ thumbnailsrzhttps://img.youtube.com/vi/z/maxresdefault.jpg)r@r6r!)r(Zthumbnail_detailsr)r)r* thumbnail_url*szYouTube.thumbnail_urlcCs |jr |jSt|j|_|jS)z8Get the publish date. :rtype: datetime )r'r publish_dater7r.r)r)r*rd;szYouTube.publish_datecCs ||_dS)zSets the publish date.N)r'r(valuer)r)r*rdFscCsX|jr |jSz|jdd|_Wn0tk rP|td|jdYnX|jS)z2Get the video title. :rtype: str rarHz#Exception while accessing title of z>. Please file a bug report at https://github.com/pytube/pytube)r&r@KeyErrorrDrEZ PytubeErrorr"r.r)r)r*rHKs z YouTube.titlecCs ||_dS)zSets the title value.N)r&rer)r)r*rHcscCs|jdidS)z8Get the video description. :rtype: str raZshortDescriptionr]r.r)r)r* descriptionhszYouTube.descriptioncCs|jdidS)z>Get the video average rating. :rtype: float raZ averageRatingr]r.r)r)r*ratingpszYouTube.ratingcCst|jdidS)z>Get the video length in seconds. :rtype: int raZ lengthSecondsintr@r6r.r)r)r*rIyszYouTube.lengthcCst|jdidS)zTGet the number of the times the video has been viewed. :rtype: int raZ viewCountrjr.r)r)r*viewssz YouTube.viewscCs*|jr |jS|jdidd|_|jS)z2Get the video author. :rtype: str raauthorunknown)r%r@r6r.r)r)r*rmszYouTube.authorcCs ||_dS)zSet the video author.N)r%rer)r)r*rmscCs|jdidgS)z;Get the video keywords. :rtype: List[str] rakeywordsr]r.r)r)r*roszYouTube.keywordscCs|jdiddS)z@Get the video poster's channel id. :rtype: str raZ channelIdNr]r.r)r)r* channel_idszYouTube.channel_idcCs d|jS)zcConstruct the channel url for the video's poster from the channel id. :rtype: str z https://www.youtube.com/channel/)rpr.r)r)r* channel_urlszYouTube.channel_urlcCs$|jr |jSt|j|_|jSdS)zIGet the metadata for the video. :rtype: YouTubeMetadata N)r rmetadatar?r.r)r)r*rrszYouTube.metadata)funccCs ||j_dS)zRegister a download progress callback function post initialization. :param callable func: A callback function that takes ``stream``, ``chunk``, and ``bytes_remaining`` as parameters. :rtype: None N)r$rr(rsr)r)r*register_on_progress_callbacks z%YouTube.register_on_progress_callbackcCs ||j_dS)zRegister a download complete callback function post initialization. :param callable func: A callback function that takes ``stream`` and ``file_path``. :rtype: None N)r$rrtr)r)r*register_on_complete_callbacks z%YouTube.register_on_complete_callback)r!r1cCstd|S)zConstruct a :class:`YouTube ` object from a video id. :param str video_id: The video id of the YouTube video. :rtype: :class:`YouTube ` z https://www.youtube.com/watch?v=)rr-r)r)r*from_ids zYouTube.from_id)NNNFT)7__name__ __module__ __qualname____doc__strrrrbytesrkrboolr+r/objectr4propertyr7r8r9r:r>r?rBrLrDr@rArr;rYr^r_rXr r`rcrdsetterrHrhfloatrirIrlrmrorprqr rrrurv staticmethodrwr)r)r)r*rs  @        (          r)r{loggingtypingrrrrrr;Zpytube.exceptionsrErrr r Zpytube.helpersr Zpytube.innertuber Zpytube.metadatar Zpytube.monostater getLoggerrxloggerrr)r)r)r*s