o {#i|7@sdZddlZddlZddlmZddlmZmZddlmZm Z m Z m Z m Z m Z ddlmZmZmZddlmZmZmZmZeeZGdd d eZdS) z>Module to download a complete playlist from a youtube channel.N)Sequence)datedatetime)DictIterableListOptionalTupleUnion)extractrequestYouTube)cacheDeferredGeneratorList install_proxy uniqueifyc@seZdZdZdDdedeeeeffddZeddZ ed d Z ed d Z ed dZ eddZ eddZeddZ dDdeedeeefddZdedeeeeffddZededeeeeeffddZdedeefd d!Zd"d#Zeedefd$d%Zd&d'Zedeefd(d)Zd*e e!e"fde eeeffd+d,Z#de"fd-d.Z$defd/d0Z%eedee&fd1d2Z'eedeefd3d4Z(edefd5d6Z)ed7d8Z*ed9d:Z+ed;d<Z,ed=d>Z-ed?d@Z.edAefdBdCZ/dS)EPlaylistz Load a YouTube playlist with URLNurlproxiescCs4|rt|||_d|_d|_d|_d|_d|_dSN)r _input_url_html_ytcfg _initial_data _sidebar_info _playlist_id)selfrrr^/var/www/GraceOrthoHospital/venv_grace/lib/python3.10/site-packages/pytube/contrib/playlist.py__init__s zPlaylist.__init__cC |jr|jSt|j|_|jS)z2Get the playlist id. :rtype: str )rr playlist_idrrrrrr!zPlaylist.playlist_idcC d|jS)z8Get the base playlist url. :rtype: str z&https://www.youtube.com/playlist?list=)r!r"rrr playlist_url*s zPlaylist.playlist_urlcCr )z9Get the playlist page html. :rtype: str )rr getr%r"rrrhtml2r#z Playlist.htmlcCr )zMExtract the ytcfg from the playlist page html. :rtype: dict )rr get_ytcfgr'r"rrrytcfg=r#zPlaylist.ytcfgcCr )zTExtract the initial data from the playlist page html. :rtype: dict )rr initial_datar'r"rrrr*HszPlaylist.initial_datacCs&|jr|jS|jddd|_|jS)zTExtract the sidebar info from the playlist page html. :rtype: dict sidebarplaylistSidebarRendereritems)rr*r"rrr sidebar_infoTszPlaylist.sidebar_infocCs |jdS)zTExtract the INNERTUBE_API_KEY from the playlist ytcfg. :rtype: str INNERTUBE_API_KEY)r)r"rrr yt_api_keyas zPlaylist.yt_api_keyuntil_watch_idreturnc cs<|tt|j\}}|r-z|d|}|d|VWdSty,Ynw|V|r;||\}}}nd\}}}|r|r|rt d|t j |||d}||\}}|ryz|d|}|d|VWdStyxYnw|V|r||\}}}nd\}}}|r|r|sFdSdSdSdSdSdS)a>Parse the video links from the page source, yields the /watch?v= part from video link :param until_watch_id Optional[str]: YouTube Video watch id until which the playlist should be read. :rtype: Iterable[List[str]] :returns: Iterable of lists of YouTube watch ids /watch?v=N)NNNzload more url: %s) extra_headersdata) _extract_videosjsondumpsr r*r'index ValueError_build_continuation_urlloggerdebugr post) rr1 videos_urls continuation trim_index load_more_urlheadersr5reqrrr _paginateisF       $zPlaylist._paginater@cCs(d|jddd|ddddidfS) aXHelper method to build the url and headers required to request the next page of videos :param str continuation: Continuation extracted from the json response of the last page :rtype: Tuple[str, dict, dict] :returns: Tuple of an url and required headers for the next http request z/https://www.youtube.com/youtubei/v1/browse?key=1z2.20200720.00.02)zX-YouTube-Client-NamezX-YouTube-Client-VersionclientWEB) clientName clientVersion)r@context)r0)rr@rrrr;sz Playlist._build_continuation_urlraw_jsonc Cs^t|}z?|dddddddd}z|ddddd }Wntttfy>|d dddd }Ynw|d}Wn;tttfyz|d dd d }|}Wn tttfy|}zt|gdfWYd}~YSd}~wwYnwz|ddddd}|dd}Wn ttfyd}Ynwttt dd||fS)aPExtracts videos from a raw json page :param str raw_json: Input json extracted from the page or the last server response :rtype: Tuple[List[str], Optional[str]] :returns: Tuple containing a list of up to 100 video watch ids and a continuation token, if more videos are available contentstwoColumnBrowseResultsRenderertabsr tabRenderercontentsectionListRendereritemSectionRendererplaylistVideoListRendereronResponseReceivedActionsappendContinuationItemsActioncontinuationItemsNcontinuationItemRenderercontinuationEndpointcontinuationCommandtokencSsd|ddS)Nr3playlistVideoRenderervideoIdr)xrrrs z*Playlist._extract_videos..) r7loadsKeyError IndexError TypeErrorr<inforlistmap)rLr*section_contentsimportant_contentvideospr@rrrr6s      zPlaylist._extract_videosvideo_idc#s0j|dD]}fdd|DEdHqdS)a}Retrieve a list of YouTube video URLs trimmed at the given video ID i.e. if the playlist has video IDs 1,2,3,4 calling trimmed(3) returns [1,2] :type video_id: str video ID to trim the returned list of playlist URLs at :rtype: List[str] :returns: List of video URLs from the playlist trimmed at the given ID )r1c3s|]}|VqdSr) _video_url).0 watch_pathr"rr sz#Playlist.trimmed..N)rE)rrmpagerr"rtrimmeds zPlaylist.trimmedccs*|D] }|D]}||Vq qdS)zGGenerator that yields video URLs. :Yields: Video URLs N)rErn)rrrvideorrr url_generators  zPlaylist.url_generatorcC t|S)zuComplete links of all the videos in playlist :rtype: List[str] :returns: List of video URLs )rrur"rrr video_urlss zPlaylist.video_urlsccs|jD]}t|VqdSr)rwr )rrrrrvideos_generator's  zPlaylist.videos_generatorcCrv)z{Yields YouTube objects of videos in this playlist :rtype: List[YouTube] :returns: List of YouTube )rrxr"rrrrk+ zPlaylist.videosicCs |j|Sr)rw)rrzrrr __getitem__4 zPlaylist.__getitem__cCs t|jSr)lenrwr"rrr__len__7r|zPlaylist.__len__cCs t|jSr)reprrwr"rrr__repr__:s zPlaylist.__repr__c Cs|jddddddd}z%|}|d}|dd}|d}t|d |d d |d WSttfyC|YSw) a Extract the date that the playlist was last updated. For some playlists, this will be a specific date, which is returned as a datetime object. For other playlists, this is an estimate such as "1 week ago". Due to the fact that this value is returned as a string, pytube does a best-effort parsing where possible, and returns the raw string where it is not possible. :return: Date of last playlist update where possible, else the string provided :rtype: datetime.date r"playlistSidebarPrimaryInfoRendererstatsrunsrUtext, z0>2z%b %d %Y)r.splitstriprstrptimerrdrc)rlast_updated_textdate_componentsmonthdayyearrrr last_updated=s0 zPlaylist.last_updatedcCs|jddddddS)zeExtract playlist title :return: playlist title (name) :rtype: Optional[str] rrtitlerrr.r"rrrrWs zPlaylist.titlecCs|jddddS)Nrr description simpleTextrr"rrrrbs zPlaylist.descriptioncCs6|jddddddd}|dd}t|S)zqExtract the number of videos in the playlist. :return: Playlist video count :rtype: int rrrrrr)r.replaceint)r count_textrrrlengthgs  zPlaylist.lengthcCs:|jddddd}|d}|dd}t|S)zcExtract view count for playlist. :return: Playlist view count :rtype: int rrrrUrrr)r.rrr)r views_textrrrrviewsss   zPlaylist.viewscCs&|jddddddddS) zfExtract the owner of the playlist. :return: Playlist owner name. :rtype: str rU$playlistSidebarSecondaryInfoRenderer videoOwnervideoOwnerRendererrrrrrr"rrrowners zPlaylist.ownercCs.|jddddddddd d S) zExtract the channel_id of the owner of the playlist. :return: Playlist owner's channel ID. :rtype: str rUrrrrrrnavigationEndpointbrowseEndpointbrowseIdrr"rrrowner_ids" zPlaylist.owner_idcCr$)zCreate the channel url of the owner of the playlist. :return: Playlist owner's channel url. :rtype: str z https://www.youtube.com/channel/)rr"rrr owner_urlryzPlaylist.owner_urlrpcCs d|S)Nzhttps://www.youtube.comr)rprrrrns zPlaylist._video_urlr)0__name__ __module__ __qualname____doc__strrrrpropertyr!r%r'r)r*r.r0rrrEr dictr; staticmethodr6rsrurrrwrxr rkr slicerr{r~rrrrrrrrrrrnrrrrrsr          9!$B &      r)rr7loggingcollections.abcrrrtypingrrrrr r pytuber r r pytube.helpersrrrr getLoggerrr<rrrrrs