}i9WdZddlZddlZddlmZddlZddlmZmZddlm Z m Z m Z m Z m Z mZddlmZddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZddl m!Z!ejDe#Z$GddZ%y)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)datetimetimezone)BinaryIODictOptionalTupleIteratorCallable) HTTPError)parse_qs)Path)extractrequest)target_directory)get_format_profile) Monostate)file_system_verify)ServerAbrStreamceZdZdZdedededefdZede fdZ ede fd Z ede fd Z e jd Z ede fd Zede fd ZdeeeeeffdZedefdZedefdZedefdZedefdZedefdZedefdZedefdZedefdZedefdZedefdZ d1deedeedeede deeded ee ge fdeefd!Z! d2deedeedeed"edef d#Z"d$ede fd%Z#d&e$ddfd'Z%d(e&d)e$d*efd+Z'd$eefd,Z(defd-Z)d(e&d*efd.Z*d3d/eede+e&fd0Z,y)4Streamz#Container for stream manifest data.stream monostatepo_tokenvideo_playback_ustreamer_configc ||_|d|_t|d|_d|vr|dnd|_t j |d\|_|_|jjd\|_ |_ |j\|_ |_|d|_|d|_t|j#d d |_t't)t'|j#d d d z d zd z |_t't)t'|j#d d d z d z d zd z |_t't)t'|j#d d d z d z d z d zd z |_t1|j}|d |_|d|_d|vr |d|_|d|_d|vr|dnd|_d|vr|dnd|_|d|_|d|_ |d|_!|j#dd|_"|j#dd|_#|d|_$|d|_%||_&||_'d|v|_(|jPrd|ddv|_)tU|ddjWdd|_,|jXjd d |_-tU|dd!jd"d |_.|j\jd#d |_/y|j`xr |jb |_)d|_,d|_-d|_.d|_/y)$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 `. urlitagxtagsNmimeType/is_otfbitrate contentLengthris_dashabrfps resolutionwidthheightis_3dis_hdris_liveisDrcFis_sabrapproxDurationMs lastModified audioTrackoriginal displayNamez original id.-)2 _monostaterintrrrmime_type_codec mime_typecodecssplittypesubtype parse_codecs video_codec audio_codecr"r#get _filesizefloatr _filesize_kb _filesize_mb _filesize_gbrr'r(r)r*_width_heightr-r.r/is_drc_is_sabr durationMs last_Modifiedrrincludes_multiple_audio_tracksis_default_audio_trackstrreplaceaudio_track_name_regionalizedaudio_track_name$audio_track_language_id_regionalizedaudio_track_language_idincludes_audio_trackincludes_video_track)selfrrrr itag_profiles P&Q# #'.."6"6s"; 4<.2->->-@*$*"8, &,Y&7 ),FJJ,J(K.34fjjZ[>\8]`d8dgk8k3los3s-t.34fjjZ[>\8]`d8dgk8knr8r3svz3z-{.34fjjZ[>\8]`d8dgk8knr8ruy8y3z~B4B.C*$))4 #I. & F?e}DH&  *1F):fWo +3v+=vh'4 !'* "8, #I. jj%0  9e4  !34#N3  /N,4@F4J+  . .*4|8L]8[*[D '14VL5I-5X1Y1a1abmoq1rD .$($F$F$L$LS$QRS$TD !7:6,;OPT;U7V7\7\]`7abc7dD 5*.*S*S*Y*YZ]*^_`*aD (*.*C*C*eDLeLeHeD '15D .$(D !8>Q  @!(!1!1$((!; ~~t~~  @66S=!(!5!5dhh!?~~  @s$A B4BBc|jdk(rL tttj|j dz dzdz |_|jS|jS#t $rd}|jdk7rtttj|j dz dzdz |_Yd}~|jSd}~wwxYw)zFile size of the media stream in kilobytes. :rtype: float :returns: Rounded filesize (in kilobytes) of the stream. rr%r&rzN) rJrIrrr{rr r|r}r~s r_ filesize_kbzStream.filesize_kbs    ! c$)$w/?/?/I$/NQU/U*VY]*]$^!    t    c66S=$)$w/C/CDHH/Md/RUY/Y*Z]a*a$b!!    cs?A(( C1ACCc|jdk(rO tttj|j dz dz dzdz |_|jS|jS#t $rg}|jdk7rtttj|j dz dz dzdz |_Yd}~|jSd}~wwxYw)zFile size of the media stream in megabytes. :rtype: float :returns: Rounded filesize (in megabytes) of the stream. rr%r&rzN) rKrIrrr{rr r|r}r~s r_ filesize_mbzStream.filesize_mbs    ! h$)$w/?/?/I$/Nt/SVZ/Z*[^b*b$c!    t    h66S=$)$w/C/CDHH/Md/RSW/WZ^/^*_bf*f$g!!    hsAA++ C4ACCc|jdk(rR tttj|j dz dz dz dzdz |_|jS|jS#t $rj}|jdk7rtttj|j dz dz dz dzdz |_Yd}~|jSd}~wwxYw)zFile size of the media stream in gigabytes. :rtype: float :returns: Rounded filesize (in gigabytes) of the stream. rr%r&rzN) rLrIrrr{rr r|r}r~s r_ filesize_gbzStream.filesize_gbs    ! m$)$w/?/?/I$/Nt/STX/X[_/_*`cg*g$h!    t    m66S=$)$w/C/CDHH/Md/RSW/WX\/\_c/c*dgk*k$l!!    msAA.. C!7ACC!c6|jjxsdS)zbGet title of video :rtype: str :returns: Youtube video title zUnknown YouTube Video Title)r<titlergs r_rz Stream.titles$$E(EErac|jjr=|jr1d}t|jj|jz|z 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<durationr#r=r{)r] bits_in_bytes r_filesize_approxzStream.filesize_approx sP ?? # # L))DLL8LH }}ract|jjdddd}tjt |t jS)N?expirer)r rrAr fromtimestampr=rutc)r]rs r_ expirationzStream.expiration1sC$((..-a01(;A>%%c&k8<<@@rac~d|jvrd|jvrd|_|jd|jS)zGenerate filename based on the video title. :rtype: str :returns: An os file system compatible filename. rqrum4ar:)r?rCrrgs r_default_filenamezStream.default_filename6s; dnn $)F DL**Qt||n--raN output_pathfilenamefilename_prefix skip_existingtimeout max_retriesinterrupt_checkerchtj}|dk(rd} n |dk(rd} nd} t| } |jj | }|r|j | }j |||| } |r=j | r,tjd| d j| | Sj} tjd jd | fd } t| d 5 jsjtjj||D]D}|)|dk(rtjddddy| t!|z} | || Fn;tjdt#| j$j'j| | cdddS#t($r}|j*dk7rYd}~:d}~wt,$rjsktj.j||D]E}|*|dk(r tjdYdddy| t!|z} | || Gn;tjdt#| j$j'YwxYw#1swYyxYw)a Downloads a file from the URL provided by `self.url` and saves it locally with optional configurations. Args: output_path (Optional[str]): Directory path where the downloaded file will be saved. Defaults to the current directory if not specified. filename (Optional[str]): Custom name for the downloaded file. If not provided, a default name is used. filename_prefix (Optional[str]): Prefix to be added to the filename (if provided). skip_existing (bool): Whether to skip the download if the file already exists at the target location. Defaults to True. timeout (Optional[int]): Maximum time, in seconds, to wait for the download request. Defaults to None for no timeout. max_retries (int): The number of times to retry the download if it fails. Defaults to 0 (no retries). interrupt_checker (Optional[Callable[[], bool]]): A callable function that is checked periodically during the download. If it returns True, the download will stop without errors. Returns: Optional[str]: The full file path of the downloaded file, or None if the download was skipped or failed. Raises: HTTPError: Raised if there is an error with the HTTP request during the download process. Note: - The `skip_existing` flag avoids redownloading if the file already exists in the target location. - The `interrupt_checker` allows for the download to be halted cleanly if certain conditions are met during the download process. - Download progress can be monitored using the `on_progress` callback, and the `on_complete` callback is triggered once the download is finished. linuxext4darwinAPFSNTFSN)rrr file_systemzfile z already exists, skippingz downloading (z total bytes) file to c,j||yrn) on_progress)chunk_bytes_remaining_fhr]s r_ write_chunkz$Stream.download..write_chunks   VR)9 :rawb)rrTzFinterrupt_checker returned True, causing to force stop the downloadingz-This stream is SABR. Starting ServerAbrStream)rrrrz)sysplatformrr translate get_file_pathexists_at_pathloggerdebug on_completer{openr1rrrrfrr<startr r| StopIteration seq_stream)r]rrrrrrrkernelrtranslation_table file_pathbytes_remainingrchunkrrs` @r_downloadzStream.downloadCsF W  K x  K K.{;  ,,667HIH ))*;` 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}"zsabr="{s.is_sabr}"ztype="{s.type}"z )r\extendrhjoinformat)r]partss r___repr__zStream.__repr__s#$?@  $ $ LL02DE F## /1KM 89: LL/+EF G bc388E?11D19:!<pc]p3-p"# p  p # pp$HRX$67p #ph#'%))-! C3-Cc]C"# C  C C.   xD"FF*2FEHF:)Xc]) =#=*FEFCF()hsm)x)rar)&rloggingrmathrrrrtypingrrrr r r urllib.errorr urllib.parser pathlibr pytubefixrrpytubefix.helpersrpytubefix.itagsrpytubefix.monostaterpytubefix.file_systemr%pytubefix.sabr.core.server_abr_streamr getLoggerrrrrrar_rsZ  'FF"!&..)4A   8 $C C ra