U 5d;@sdZddlZddlZddlmZddlmZddlmZmZm Z m Z ddl m Z ddl mZddlmZmZdd lmZmZdd lmZdd lmZeeZGd d d ZdS)a/ This module contains a container for stream manifest data. A container object for the media stream (video only / audio only / video+audio combined). This was referred to as ``Video`` in the legacy pytube version, but has been renamed to accommodate DASH (which serves the audio and video separately). N)ceil)datetime)BinaryIODictOptionalTuple) HTTPError)parse_qs)extractrequest) safe_filenametarget_directory)get_format_profile) Monostatec @seZdZdZeedddZeedddZ eeddd Z eedd d Z eedd d Z e eeeefdddZeedddZeedddZeedddZeedddZeedddZeedddZeedddZeedddZd7eeeeeeeeeeeed#d$d%Zd8eeeeeeed&d'd(Zeed)d*d+Zed d,d-d.Z e!eed/d0d1Z"eed2d3d4Z#edd5d6Z$d S)9Streamz#Container for stream manifest data.)stream monostatecCsZ||_|d|_t|d|_t|d\|_|_|jd\|_ |_ | \|_ |_ |d|_|d|_t|dd|_ttt|ddd d d |_ttt|ddd d d d |_ttt|ddd d d d d |_t|j}|d |_|d |_d |kr.|d |_|d|_|d|_|d|_|d|_dS)aConstruct a :class:`Stream `. :param dict stream: The unscrambled data extracted from YouTube. :param dict monostate: Dictionary of data shared across all instances of :class:`Stream `. urlitagZmimeType/is_otfbitrateZ contentLengthris_dashabrfps resolutionis_3dis_hdris_liveN) _monostaterintrr Zmime_type_codecZ mime_typecodecssplittypesubtype parse_codecsZ video_codecZ audio_codecrrget _filesizefloatr _filesize_kb _filesize_mb _filesize_gbrrrrrrrr )selfrrZ itag_profiler/2/tmp/pip-unpacked-wheel-qmwpt_in/pytube/streams.py__init__s2    &*.       zStream.__init__)returncCstt|jdS)z:Whether the stream is DASH. :rtype: bool )boollenr#r.r/r/r0 is_adaptiveZszStream.is_adaptivecCs|j S)zAWhether the stream is progressive. :rtype: bool )r7r6r/r/r0is_progressivedszStream.is_progressivecCs|jp|jdkS)zFWhether the stream only contains audio. :rtype: bool audior8r%r6r/r/r0includes_audio_tracklszStream.includes_audio_trackcCs|jp|jdkS)zFWhether the stream only contains video. :rtype: bool videor:r6r/r/r0includes_video_tracktszStream.includes_video_trackcCsDd}d}|js|j\}}n"|jr,|jd}n|jr<|jd}||fS)aGet the video/audio codecs from list of codecs. Parse a variable length sized list of codecs and returns a constant two element tuple, with the video codec as the first element and audio as the second. Returns None if one is not available (adaptive only). :rtype: tuple :returns: A two element tuple with audio and video codecs. Nr)r7r#r=r;)r.r<r9r/r/r0r'|s    zStream.parse_codecsc Cs`|jdkrZzt|j|_Wn<tk rX}z|jdkr:t|j|_W5d}~XYnX|jS)zFile size of the media stream in bytes. :rtype: int :returns: Filesize (in bytes) of the stream. rN)r)r filesizerrcode seq_filesizer.er/r/r0r?s   zStream.filesizec Cs|jdkrz&ttt|jddd|_WnPtk r}z2|jdkrNttt|jddd|_W5d}~XYnX|jS)zFile size of the media stream in kilobytes. :rtype: float :returns: Rounded filesize (in kilobytes) of the stream. rrrr>N) r+r*rr r?rrr@rArBr/r/r0 filesize_kbs & 4zStream.filesize_kbc Cs|jdkrz*ttt|jdddd|_WnTtk r}z6|jdkrRttt|jdddd|_W5d}~XYnX|jS)zFile size of the media stream in megabytes. :rtype: float :returns: Rounded filesize (in megabytes) of the stream. rrrr>N) r,r*rr r?rrr@rArBr/r/r0 filesize_mbs * 8zStream.filesize_mbc Cs|jdkrz.ttt|jddddd|_WnXtk r}z:|jdkrVttt|jddddd|_W5d}~XYnX|jS)zFile size of the media stream in gigabytes. :rtype: float :returns: Rounded filesize (in gigabytes) of the stream. rrrr>N) r-r*rr r?rrr@rArBr/r/r0 filesize_gbs . <zStream.filesize_gbcCs |jjp dS)zbGet title of video :rtype: str :returns: Youtube video title zUnknown YouTube Video Title)r!titler6r/r/r0rGsz Stream.titlecCs.|jjr(|jr(d}t|jj|j|S|jS)zGet approximate filesize of the video Falls back to HTTP call if there is not sufficient information to approximate :rtype: int :returns: size of video in bytes )r!durationrr"r?)r.Z bits_in_byter/r/r0filesize_approxs zStream.filesize_approxcCs*t|jdddd}tt|S)N?expirer)r rr$rutcfromtimestampr")r.rMr/r/r0 expirationszStream.expirationcCst|j}|d|jS)zGenerate filename based on the video title. :rtype: str :returns: An os file system compatible filename. .)r rGr&)r.filenamer/r/r0default_filenames zStream.default_filenameNTr) output_pathrQfilename_prefix skip_existingtimeout max_retriesr2c Cs|j|||d}|r>||r>td|d|||S|j}td|jd|t|d} z6tj|j ||dD]} |t | 8}| | | |q|Wn`t k r} zB| j dkrtj|j ||dD]} |t | 8}| | | |qW5d } ~ XYnXW5QRX|||S) a-Write the media stream to disk. :param output_path: (optional) Output path for writing media file. If one is not specified, defaults to the current working directory. :type output_path: str or None :param filename: (optional) Output filename (stem only) for writing media file. If one is not specified, the default filename is used. :type filename: str or None :param filename_prefix: (optional) A string that will be prepended to the filename. For example a number in a playlist or the name of a series. If one is not specified, nothing will be prepended This is separate from filename so you can use the default filename but still add a prefix. :type filename_prefix: str or None :param skip_existing: (optional) Skip existing files, defaults to True :type skip_existing: bool :param timeout: (optional) Request timeout length in seconds. Uses system default. :type timeout: int :param max_retries: (optional) Number of retries to attempt after socket timeout. Defaults to 0. :type max_retries: int :returns: Path to the saved video :rtype: str )rQrSrTzfile z already exists, skippingz downloading (z total bytes) file to wb)rVrWr>N) get_file_pathexists_at_pathloggerdebug on_completer?openr rrr5 on_progressrr@Z seq_stream) r.rSrQrTrUrVrW file_pathbytes_remainingfhchunkrCr/r/r0downloads@(       , zStream.download)rQrSrTr2cCs,|s |j}|r||}tjt||SN)rRospathjoinr )r.rQrSrTr/r/r0rYSs  zStream.get_file_path)r`r2cCstj|otj||jkSre)rfrgisfilegetsizer?)r.r`r/r/r0rZ_s zStream.exists_at_path)bufferr2cCsN|j}td|jt|jD]}|t|8}||||q |ddS)zLWrite the media stream to buffer :rtype: io.BytesIO buffer z+downloading (%s total bytes) file to bufferN) r?r[infor rrr5r_r])r.rkrarcr/r/r0stream_to_bufferes zStream.stream_to_buffer)rc file_handlerracCs2||td||jjr.|j|||dS)aOn progress callback function. This function writes the binary data to the file, then checks if an additional callback is defined in the monostate. This is exposed to allow things like displaying a progress bar. :param bytes chunk: Segment of media file binary data, not yet written to disk. :param file_handler: The file handle where the media is being written to. :type file_handler: :py:class:`io.BufferedWriter` :param int bytes_remaining: The delta between the total file size in bytes and amount already downloaded. :rtype: None zdownload remaining: %sN)writer[r\r!r_)r.rcrnrar/r/r0r_vs  zStream.on_progress)r`cCs0td|jj}|r,td||||dS)zOn download complete handler function. :param file_path: The file handle where the media is being written to. :type file_path: str :rtype: None zdownload finishedzcalling on_complete callback %sN)r[r\r!r])r.r`r]r/r/r0r]s  zStream.on_completecCsvddg}|jr@|ddg|js2|ddgqN|dgn|ddg|dd gd d |j|d d S)zPrintable object representation. :rtype: str :returns: A string representation of a :class:`Stream ` object. zitag="{s.itag}"zmime_type="{s.mime_type}"zres="{s.resolution}"zfps="{s.fps}fps"zvcodec="{s.video_codec}"zacodec="{s.audio_codec}"z abr="{s.abr}"z progressive="{s.is_progressive}"ztype="{s.type}"z )r=extendr7rhformat)r.partsr/r/r0__repr__szStream.__repr__)NNNTNr)NNN)%__name__ __module__ __qualname____doc__rrr1propertyr4r7r8r;r=rrstrr'r"r?r*rDrErFrGrJrrOrRrdrYrZrrmbytesr_r]rvr/r/r/r0rsz =    S  r)rzloggingrfmathrrtypingrrrr urllib.errorr urllib.parser Zpytuber r Zpytube.helpersr r Z pytube.itagsrZpytube.monostater getLoggerrwr[rr/r/r/r0s