U gA@s6dZddlZddlZddlZddlZddlZddlZddlZddlZ ddl Z ddl m Z m Z ddlmZddlmZddlmZmZmZmZddlmZmZeeZddZeejdd d d Zd9eje e ejd d dZ eddddZ!d:e"e"e#e$ddddZ%ee&e"ddddZ'd;ee e#e e#ddddZ(e#e#e#e#e#dddZ)dee#e e#dd d)d*Z-d?ee#e e#dd d+d,Z.eddd-d.Z/edd/d0d1Z0d@ee e#e e#dd2d3d4Z1dAee#e e#dd5d6d7Z2ed8kr2edS)Bz=A simple command line application to download youtube videos.N)ListOptional) __version__) CaptionQueryPlaylistStreamYouTube) safe_filename setup_loggerc Cstjtjd}t|}|jrJd}|jr,|j}ttj |dt dt |j rZd|j krl|tdd|j krtdt|j }|jst|j|_|jD]N}zt||Wqtjk r}ztd |t|W5d}~XYqXqntd t|j }t||dS) z4Command line application to download youtube videos.) descriptionN) log_filenamezPytube version: Zyoutuz /playlistzLoading playlist...zThere was an error with video: zLoading video...)argparseArgumentParsermain__doc__ _parse_argsverboselogfiler loggingDEBUGloggerdebugrurl print_helpsysexitprintrtargetr titleZvideos_perform_args_on_youtube exceptionsZ PytubeErrorr)parserargsr ZplaylistZ youtube_videoeyoutuber&./tmp/pip-unpacked-wheel-1a9f0fi6/pytube/cli.pyrs2      r)r%r#returncCsttjdkrt|d|jd|jr.t|j|jrz-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)rAr>z-vz --verboserz$Set logger output to verbose output.)rAdestr>z --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)r>z-az--audiomp4zpDownload the audio for a given URL at the highest bitrate available. Defaults to mp4 format if none is specified)constr?r>z-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    r)r%r(c Csttj}tjtd|j d|d}|j }|j }|j }t |d(}|t|j|||ddW5QRXdS)zuSerialize the request data to json for offline debugging. :param YouTube youtube: A YouTube object. z yt-video--z.json.gzwb)rjs watch_htmlZ video_infoutf8N)rJdtdatetimeutcnow timestampospathjoingetcwdZvideo_idrOrPvid_infogzipopenwritejsondumpsZ watch_urlencode)r%tsfprOrPrZfhr&r&r'r5s""r5█皙?)bytes_receivedfilesizechscaler(c 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@r u ↳ |z| z% N) shutilget_terminal_sizecolumnsrJroundfloatrstdoutr]flush) rfrgrhrirm max_widthZfilled remaining progress_barpercenttextr&r&r'display_progress_bars   rw)streamchunkbytes_remainingr(cCs|j}||}t||dSN)rgrw)rxryrzrgrfr&r&r' on_progresssr|)rxrfilenamer(cCsl|jd}t|p|jd|d|j||d}||rNtd|dS|j||dtjddS)Niz | z MB)r} output_pathzAlready downloaded at: )r~r} ) rgrdefault_filenameZ get_file_pathZexists_at_pathdownloadrrpr])rxrr}Zfilesize_megabytes file_pathr&r&r' _downloads  r)basesubtype media_typerr(cCsNd}|d|d|}tj||d|}tj|s@|S|d7}qdS)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. r_.r N)rVrWrXexists)rrrrcounter file_namerr&r&r' _unique_name s  r)r%r*rr(cCs|t|pt}|dkrh|jjddd}|jjdddd}|j|jkrb|}q|}n.|jjd|dd }|s|jjd|d }|d krt d |t d t |t |j|j}|s|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 rHF) progressiver*rF)rr)rr*r)rr*N)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_callbackr|rVrYstreamsfilterorder_bylastr*firstrr4rrZget_audio_onlyr_ffmpeg_downloader)r%r*rZhighest_quality_streamZ mp4_streamrrr&r&r'r<!sR     r<)rrrr(c 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 videorr9)rxrr}zLoading audio...rr;z-iz-codeccopyN) rr rrrrrVrWrX subprocessrununlink)rrrZvideo_unique_nameZaudio_unique_nameZ video_pathZ audio_pathZ final_pathr&r&r'r[sP r)r%r+rr(cCsv|j|}|dkr:td|tdt|t|tzt||dWnt k rptYnXdS)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) rZ get_by_itagrr4rrrr|rKeyboardInterrupt)r%r+rrxr&r&r'r6s  r6cCsv|j|}|dkr:td|tdt|t|tzt||dWnt k rptYnXdS)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) rZget_by_resolutionrr4rrrr|rr)r%r*rrxr&r&r'r8s  r8c Cs~|tz|j}Wn2tjk rJ}ztd|W5d}~XYn0Xzt||dWntk rxt YnXdS)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) rr|rZget_highest_resolutionr!ZVideoUnavailablerrrrr)r%r*rrxerrr&r&r'r0s  r0cCs|jD] }t|qdS)zzProbe YouTube video and lists its available formats. :param YouTube youtube: A valid YouTube watch URL. N)rr)r%rxr&r&r'r4s r4)r2r(cCs"tdddd|DdS)NzAvailable caption codes are: z, css|] }|jVqdSr{)code).0cr&r&r' sz,_print_available_captions..)rrX)r2r&r&r'r1sr1)r%r,rr(cCs^z,|j|}|j|j|d}td|Wn,tk rXtd|t|jYnXdS)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 )rr~zSaved caption file to: z"Unable to find caption with code: N)r2rrrKeyErrorr1)r%r,rcaptionZdownloaded_pathr&r&r'r7s r7)r%r-rr(cCsv|jjd|dd}|dkr:tdt|t|t zt ||dWnt k rptYnXdS)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) rrrrrr4rrrr|rr)r%r-rr9r&r&r'r:s r:__main__)N)rdre)NN)N)N)N)N)N)N)3rrr[r^rrVrkrrSrRrtypingrrZpytube.exceptionsr!ZpytuberrrrrZpytube.helpersr r getLogger__name__rr Namespacer rrr5rJrKrorwbytesr|rrr<rr6r8r0r4r1r7r:r&r&r&r's   " " ^ #   ; ;      !