o {#iA @sdZddlZddlZddlZddlZddlZddlZddlZddlZ ddl Z ddl m Z m Z ddlmZddlmZddlmZmZmZmZddlmZmZeeZddZd ed ejd dfd d Z dCdejd e e d ejfddZ d ed dfddZ! dDde"de"de#de$d df ddZ%dede&de"d dfddZ'  dEded e e#d!e e#d dfd"d#Z(d$e#d%e#d&e#d e#d e#f d'd(Z) dCd ed)e#d e e#d dfd*d+Z*d,ed-ed e#d dfd.d/Z+ dCd ed0e"d e e#d dfd1d2Z, dCd ed)e#d e e#d dfd3d4Z- dCd ed)e#d e e#d dfd5d6Z.d ed dfd7d8Z/d9ed dfd:d;Z0 dCd edZ1 dCd ed?e#d e e#d dfd@dAZ2edBkrOedSdS)Fz=A simple command line application to download youtube videos.N)ListOptional) __version__) CaptionQueryPlaylistStreamYouTube) safe_filename setup_loggerc Cstjtjd}t|}|jr%d}|jr|j}ttj |dt dt |j r-d|j vr6|tdd|j vrztdt|j }|jsMt|j|_|jD]'}zt||WqPtjyw}ztd |t|WYd}~qPd}~wwdStd t|j }t||dS) z4Command line application to download youtube videos.) descriptionN) log_filenamezPytube version: youtuz /playlistzLoading playlist...zThere was an error with video: zLoading video...)argparseArgumentParsermain__doc__ _parse_argsverboselogfiler loggingDEBUGloggerdebugrurl print_helpsysexitprintrtargetr titlevideos_perform_args_on_youtube exceptions PytubeErrorr)parserargsr playlist youtube_videoeyoutuber+Q/var/www/GraceOrthoHospital/venv_grace/lib/python3.10/site-packages/pytube/cli.pyrs8      rr*r&returncCsttjdkrt|d|jd|jrt|j|jrt ||j r%t ||j r1t ||j |jd|j r=t||j |jd|jrIt||j|jd|jrUt||j|jd|jrct||j|jddSdS)Nhighest)r* resolutionr)r*itagr)r* lang_coder)r*filetyper)lenrargv'download_highest_resolution_progressiver list_captions_print_available_captionscaptionslistdisplay_streamsbuild_playback_reportr1download_by_itag caption_codedownload_captionr0download_by_resolutionaudiodownload_audioffmpegffmpeg_process)r*r&r+r+r,r"8s:      r"r%cCs|jdddd|jdddtd|jd td d |jd d tdd |jddddd|jdddddd|jdddd|jdddd|jddtd d |jd!d"dd#d|jd$d%d&d'|jd(d)d*dd+d,|jd-d.d/dd0d,||S)1Nrz#The YouTube /watch or /playlist url?)helpnargsz --versionversionz %(prog)s )actionrHz--itagzThe itag for the desired stream)typerFz-rz --resolutionz%The resolution for the desired streamz-lz--list store_truezSThe list option causes pytube cli to return a list of streams available to download)rIrFz-vz --verboserz$Set logger output to verbose output.)rIdestrFz --logfilestorez0logging debug and error messages into a log filez--build-playback-reportzSave the html and js to diskz-cz--caption-codezcDownload srt captions for given language code. Prints available language codes if no argument givenz-lcz--list-captionsz(List available caption codes for a videoz-tz--targetzTThe output directory for the downloaded stream. Default is current working directory)rFz-az--audiomp4zpDownload the audio for a given URL at the highest bitrate available. Defaults to mp4 format if none is specified)constrGrFz-fz--ffmpegbestzDownloads the audio and video stream for resolution provided. If no resolution is provided, downloads the best resolution. Runs the command line program ffmpeg to combine the audio and video) add_argumentrintstr parse_args)r%r&r+r+r,rYs     rc Csttj}tjtd|j d|d}|j }|j }|j }t |d}|t|j|||ddWddS1sFwYdS)zuSerialize the request data to json for offline debugging. :param YouTube youtube: A YouTube object. z yt-video--z.json.gzwb)rjs watch_html video_infoutf8N)rRdtdatetimeutcnow timestampospathjoingetcwdvideo_idrWrXvid_infogzipopenwritejsondumps watch_urlencode)r*tsfprWrXrdfhr+r+r,r<s"""r<█皙?bytes_receivedfilesizechscalec Cstj}t||}tt||t|}||}||d|}td|t|d} d|d| d} tj| tj dS)u{Display a simple, pretty progress bar. Example: ~~~~~~~~ PSY - GANGNAM STYLE(강남스타일) MV.mp4 ↳ |███████████████████████████████████████| 100.0% :param int bytes_received: The delta between the total file size (bytes) and bytes already written to disk. :param int filesize: File size of the media stream in bytes. :param str ch: Character to use for presenting progress segment. :param float scale: Scale multiplier to reduce progress bar size.  gY@ru ↳ |z| z% N) shutilget_terminal_sizecolumnsrRroundfloatrstdoutrgflush) rqrrrsrtrx max_widthfilled remaining progress_barpercenttextr+r+r,display_progress_bars   rstreamchunkbytes_remainingcCs|j}||}t||dSN)rrr)rrrrrrqr+r+r, on_progresssrrfilenamecCsl|jd}t|p |jd|d|j||d}||r'td|dS|j||dtjddS)Niz | z MB)r output_pathzAlready downloaded at: )rr ) rrrdefault_filename get_file_pathexists_at_pathdownloadrr{rg)rrrfilesize_megabytes file_pathr+r+r, _downloads  rbasesubtype media_typecCsLd} |d|d|}tj||d|}tj|s!|S|d7}q)a Given a base name, the file format, and the target directory, will generate a filename unique for that directory and file format. :param str base: The given base-name. :param str subtype: The filetype of the video which will be downloaded. :param str media_type: The media_type of the file, ie. "audio" or "video" :param Path target: Target directory for download. rT_.r)r_r`raexists)rrrrcounter file_namerr+r+r, _unique_name s  rr0cCs|t|p t}|dkr4|jjddd}|jjdddd}|j|jkr1|}n|}n|jjd|dd }|sK|jjd|d }|d urbt d |t d t |t |j|j}|sw|jjd d d}|st dt t|||dd S)a  Decides the correct video stream to download, then calls _ffmpeg_downloader. :param YouTube youtube: A valid YouTube object. :param str resolution: YouTube video resolution. :param str target: Target directory for download rPF) progressiver0rN)rr)rr0r)rr0N)Could not find a stream with resolution: Try one of these:T) only_audioabrz#Could not find an audio only stream) audio_stream video_streamr)register_on_progress_callbackrr_rbstreamsfilterorder_bylastr0firstrr;rrget_audio_onlyr_ffmpeg_downloader)r*r0rhighest_quality_stream mp4_streamrrr+r+r,rD!sP    rDrrc Cstt|j|jd|d}tt|j|jd|d}t|||dtdt|||dtj||d|j}tj||d|j}tj|t|jd|j}t dd|d|d d |gt |t |d S) a$ Given a YouTube Stream object, finds the correct audio stream, downloads them both giving them a unique name, them uses ffmpeg to create a new file with the audio and video from the previously downloaded files. Then deletes the original adaptive streams, leaving the combination. :param Stream audio_stream: A valid Stream object representing the audio to download :param Stream video_stream: A valid Stream object representing the video to download :param Path target: A valid Path object videorrA)rrrzLoading audio...rrCz-iz-codeccopyN) rr r rrrr_r`ra subprocessrununlink)rrrvideo_unique_nameaudio_unique_name video_path audio_path final_pathr+r+r,r[sJ rr1cCt|j|}|durtd|tdt|t|tz t||dWdSt y9tYdSw)zStart downloading a YouTube video. :param YouTube youtube: A valid YouTube object. :param int itag: YouTube format identifier code. :param str target: Target directory for download Nz#Could not find a stream with itag: rr) r get_by_itagrr;rrrrrKeyboardInterrupt)r*r1rrr+r+r,r=s   r=cCr)zStart downloading a YouTube video. :param YouTube youtube: A valid YouTube object. :param str resolution: YouTube video resolution. :param str target: Target directory for download Nrrr) rget_by_resolutionrr;rrrrrr)r*r0rrr+r+r,r@s   r@c Cs|tz|j}Wntjy'}ztd|WYd}~dSd}~wwz t||dWdSty?t YdSw)zStart downloading the highest resolution progressive stream. :param YouTube youtube: A valid YouTube object. :param str resolution: YouTube video resolution. :param str target: Target directory for download zNo video streams available: Nr) rrrget_highest_resolutionr#VideoUnavailablerrrrr)r*r0rrerrr+r+r,r6s  r6cCs|jD]}t|qdS)zzProbe YouTube video and lists its available formats. :param YouTube youtube: A valid YouTube watch URL. N)rr)r*rr+r+r,r;s  r;r9cCs"tdddd|DdS)NzAvailable caption codes are: z, css|]}|jVqdSr)code).0cr+r+r, sz,_print_available_captions..)rra)r9r+r+r,r8sr8r2cCs\z|j|}|j|j|d}td|WdSty-td|t|jYdSw)aTDownload a caption for the YouTube video. :param YouTube youtube: A valid YouTube object. :param str lang_code: Language code desired for caption file. Prints available codes if the value is None or the desired code is not available. :param str target: Target directory for download )r rzSaved caption file to: z"Unable to find caption with code: N)r9rr rKeyErrorr8)r*r2rcaptiondownloaded_pathr+r+r,r?s  r?r3cCst|jjd|dd}|durtdt|t|t z t ||dWdSt y9tYdSw)a% Given a filetype, downloads the highest quality available audio stream for a YouTube video. :param YouTube youtube: A valid YouTube object. :param str filetype: Desired file format to download. :param str target: Target directory for download T)rrrNz-No audio only stream found. Try one of these:r) rrrrrr;rrrrrr)r*r3rrAr+r+r,rBs  rB__main__r)rorp)NN)3rrrerhrr_rvrr\r[rtypingrrpytube.exceptionsr#pytuberrrrrpytube.helpersr r getLogger__name__rr Namespacer"rrr<rRrSrzrbytesrrrrDrr=r@r6r;r8r?rBr+r+r+r,s   ! " ^ "    : ;      !