}i;xdZddlmZmZddlmZmZmZmZddl m Z m Z ddl m Z GddeZGdd eZy ) zFThis module provides a query interface for media streams and captions.)MappingSequence)CallableListOptionalUnion)CaptionStream) deprecatedceZdZdZdZ d$dZdeeddfdZde ddfd Z d%d Z d%d Z d e ee fdeefd Zde deefdZd%dZdedfdZdedfdZd&deefdZd'deefdZd(de deefdZd)deddfdZdeefdZdZedd*dee defdZeddeefdZd e e effd!Z!defd"Z"de fd#Z#y)+ StreamQueryz3Interface for querying the available media streams.cn||_|Dcic]}t|j|c}|_ycc}w)zConstruct a :class:`StreamQuery `. param list fmt_streams: list of :class:`Stream ` instances. N) fmt_streamsintitag itag_index)selfrss :/usr/local/lib/python3.12/dist-packages/pytubefix/query.py__init__zStreamQuery.__init__ s- '3>?a3qvv;>??s2Nc g}srkttsttr|jfdn5ttsttr|jfdr|jfdr|jfdr|jfdsr|jfds r|j fd r|j fd r|j fd | r|jd | r|jd |r|jd |r|jd r|jfd|r|j ||jfd|jfd|j |S)a Apply the given filtering criterion. :param fps: (optional) The frames per second. :type fps: int or None :param resolution: (optional) Alias to ``res``. :type res: str or None :param res: (optional) The video resolution. :type resolution: str or None :param mime_type: (optional) Two-part identifier for file formats and format contents composed of a "type", a "subtype". :type mime_type: str or None :param type: (optional) Type part of the ``mime_type`` (e.g.: audio, video). :type type: str or None :param subtype: (optional) Sub-type part of the ``mime_type`` (e.g.: mp4, mov). :type subtype: str or None :param file_extension: (optional) Alias to ``sub_type``. :type file_extension: str or None :param abr: (optional) Average bitrate (ABR) refers to the average amount of data transferred per unit of time (e.g.: 64kbps, 192kbps). :type abr: str or None :param bitrate: (optional) Alias to ``abr``. :type bitrate: str or None :param video_codec: (optional) Video compression format. :type video_codec: str or None :param audio_codec: (optional) Audio compression format. :type audio_codec: str or None :param bool progressive: Excludes adaptive streams (one file contains both audio and video tracks). :param bool adaptive: Excludes progressive streams (audio and video are on separate tracks). :param bool is_dash: Include/exclude dash streams. :param bool only_audio: Excludes streams with video tracks. :param bool only_video: Excludes streams with audio tracks. :param bool is_drc: Include/exclude stable volume streams. :param audio_track_name: Name of the dubbed audio track :type type: str or None :param custom_filter_functions: (optional) Interface for defining complex filters without subclassing. :type custom_filter_functions: list or None c*|jxsk(SN resolutionrresrs rz$StreamQuery.filter..s#:K)Lc(|jxsvSrrrs rrz$StreamQuery.filter..s#:K)Lrc"|jk(Sr)fps)rr"s rrz$StreamQuery.filter..sQUUc\rc"|jk(Sr) mime_type)rr$s rrz$StreamQuery.filter..sQ[[I%=rc"|jk(Sr)type)rr&s rrz$StreamQuery.filter..sQVVt^rc*|jxsk(Sr)subtype)rfile_extensionr(s rrz$StreamQuery.filter..sQYY73Ln%Mrc*|jxsk(Sr)abr)rr+bitrates rrz$StreamQuery.filter..sQUUs~g%>rc"|jk(Sr) video_codec)rr.s rrz$StreamQuery.filter..Q]]k%Arc"|jk(Sr) audio_codec)rr1s rrz$StreamQuery.filter..r/rc8|jxr |j Sr)includes_audio_trackincludes_video_trackrs rrz$StreamQuery.filter..**I13I3I/Irc8|jxr |j Sr)r4r3r5s rrz$StreamQuery.filter..r6rc|jSr)is_progressiver5s rrz$StreamQuery.filter..s Q%5%5rc|jSr) is_adaptiver5s rrz$StreamQuery.filter..s Q]]rc"|jk(Sraudio_track_name)rr>s rrz$StreamQuery.filter..sQ%7%7;K%Krc"|jk(Sr)is_dash)rr@s rrz$StreamQuery.filter..sQYY'%9rc"|jk(Sr)is_drc)rrBs rrz$StreamQuery.filter..sQXX%7r) isinstancestrappendlistextend_filter)rr"rrr$r&r(r)r+r,r.r1 only_audio only_video progressiveadaptiver@rBr>custom_filter_functionsfilterss ``````````` ``` rfilterzStreamQuery.filtersfb *#s#z*c'BLMC&*Z*FLM  NN1 2  NN= >  NN3 4 n NNM N ' NN> ?  NNA B  NNA B  NN   NN   NN5 6  NN2 3  NNK L " NN2 3   NN9 :   NN7 8||G$$rrNreturnch|j}|D]}t||}tt|Sr)rrOr rF)rrNr filter_lambdas rrHzStreamQuery._filters8&& $ =M  .s-cGG &s{{GA~4N O'r)keyct|Sr)rYrZs rrz&StreamQuery.order_by..s>0Jr)rrYrCrDr sorted ValueError)rrSr has_attributes ` rorder_byzStreamQuery.order_bys%% q.)5   Z M!$n 5s "%   =&J K  3 ,  sA9A>> B  B c8t|jdddS)z_Sort streams in descending order. :rtype: :class:`StreamQuery ` N)r rrs rdesczStreamQuery.descs 4++DbD122rc|S)z^Sort streams in ascending order. :rtype: :class:`StreamQuery ` rcs rasczStreamQuery.ascs  rrct|tr|jj|St|tr5|j r$|jjt|Syy)a6Get the corresponding :class:`Stream ` for a given itag. :param int itag: YouTube format identifier code. :rtype: :class:`Stream ` or None :returns: The :class:`Stream ` matching the given itag or None if not found. N)rCrrgetrDrX)rrs r get_by_itagzStreamQuery.get_by_itagsV dC ??&&t, , c "t||~??&&s4y1 1(6 "rrcF|jdd|jS)aGet the corresponding :class:`Stream ` for a given resolution. Stream must be a progressive mp4. :param str resolution: Video resolution i.e. "720p", "480p", "360p", "240p", "144p" :rtype: :class:`Stream ` or None :returns: The :class:`Stream ` matching the given itag or None if not found. Tmp4)rKr(r)rOfirst)rrs rget_by_resolutionzStreamQuery.get_by_resolution s({{e  %' rc(|jdgS)zTakes the standard audio tracks, will return all audio tracks if there is no dubbing. :rtype: :class:`StreamQuery ` :returns: A StreamQuery object with filtered default dubbing streams. c|jSr)is_default_audio_trackr5s rrz5StreamQuery.get_default_audio_track..$s q'?'?rrHrcs rget_default_audio_trackz#StreamQuery.get_default_audio_tracks ||?@AArc(|jdgS)zGet only dubbed audio tracks. :rtype: :class:`StreamQuery ` or None :returns: A StreamQuery object with filtering only the dubbing streams. cV|j xr|jxr |j Sr)rqr3r4r5s rrz3StreamQuery.get_extra_audio_track..,s2!"!9!99 >rc.|jfdgS)zFilter dubbed audio streams by name :rtype: :class:`StreamQuery ` or None :returns: A StreamQuery object filtering dubbed audio streams by name. c"|jk(Srr=)rnames rrz;StreamQuery.get_extra_audio_track_by_name..7sq'9'9T'Arrr)rrys `rget_extra_audio_track_by_namez)StreamQuery.get_extra_audio_track_by_name1s ||ABCCrcb|j|djdjS)aGet lowest resolution stream that is a progressive mp4. :param bool progressive: Filter only progressive streams (video and audio in the same file), default is True. Set False to get the adaptive stream (separate video and audio) at the lowest resolution :rtype: :class:`Stream ` or None :returns: The :class:`Stream ` matching the given itag or None if not found. rl)rKr(r)rOr`rm)rrKs rget_lowest_resolutionz!StreamQuery.get_lowest_resolution9s* KKKK ? Xl # UW rcb|j||jdjS)a9Get highest resolution stream that is a progressive video. :param bool progressive: Filter only progressive streams (video and audio in the same file), default is True. Set False to get the adaptive stream (separate video and audio) at the highest resolution :param str mime_type: Filter by mime_type. Leave as None to accept any mime_type. :rtype: :class:`Stream ` or None :returns: The :class:`Stream ` matching the given itag or None if not found. )rKr$rrOr`last)rrKr$s rget_highest_resolutionz"StreamQuery.get_highest_resolutionKs,{{{i{HQQR^_ddffrr(cb|jd|jdjS)a9Get highest bitrate audio stream for given codec (defaults to mp4) :param str subtype: Audio subtype, defaults to mp4 :rtype: :class:`Stream ` or None :returns: The :class:`Stream ` matching the given itag or None if not found. T)rIr(r+r~)rr(s rget_audio_onlyzStreamQuery.get_audio_only[s) KK4K 9 Xe_ TV ris_otfc.|jfdgS)aFilter stream by OTF, useful if some streams have 404 URLs :param bool is_otf: Set to False to retrieve only non-OTF streams :rtype: :class:`StreamQuery ` :returns: A StreamQuery object with otf filtered streams c"|jk(Sr)r)rrs rrz!StreamQuery.otf..rsqxx6'9rrr)rrs `rotfzStreamQuery.otfks||9:;;rc@ |jdS#t$rYywxYw)zGet the first :class:`Stream ` in the results. :rtype: :class:`Stream ` or None :returns: the first result of this query or None if the result doesn't contain any streams. rNr IndexErrorrcs rrmzStreamQuery.firstts* ##A& &    c@ |jdS#t$rYywxYw)zGet the last :class:`Stream ` in the results. :rtype: :class:`Stream ` or None :returns: Return the last result of this query or None if the result doesn't contain any streams. rbNrrcs rrzStreamQuery.lasts* ##B' '   rz.Get the size of this list directly using len()valuecR|r|jj|St|S)zAGet the count of items in the list. :rtype: int )rcountlen)rrs rrzStreamQuery.counts& 16t%%e,D3t9Drz6This object can be treated as a list, all() is uselessc|jSzXGet all the results represented by this query as a list. :rtype: list rrcs rallzStreamQuery.allsric |j|Srrrrs r __getitem__zStreamQuery.__getitem__s""rc,t|jSr)rrrcs r__len__zStreamQuery.__len__s4##$$rc|jSrrrcs r__repr__zStreamQuery.__repr__s""#$r)NNNNNNNNNNNNNNNNNNN)rPr )T)TN)rl)Fr)$__name__ __module__ __qualname____doc__rrOrrrHrDr`rdrgrrrr rjrnrsrvrzr|rrboolrrmrr rrslicerrrrfrrr r s=@     $)m%^.tH~.-. ! s! }! F32c3h2HV4D2 CHV4D"B >x '> >DXm5LD &9I $g(SYJZg  c hv6F <$<=< x'  @AE8C=ECEBEHI T&\ J #U5#:.#%%%#%rr ceZdZdZdeefdZeddede efdZ eddeefd Z d efd Z de fd Zd ZdefdZy) CaptionQueryz.Interface for querying the available captions.captionscN|Dcic]}|j|c}|_ycc}w)zConstruct a :class:`Caption `. param list captions: list of :class:`Caption ` instances. N)codelang_code_index)rrcs rrzCaptionQuery.__init__s"4<` for a given ``lang_code``. :param str lang_code: The code that identifies the caption language. :rtype: :class:`Caption ` or None :returns: The :class:`Caption ` matching the given ``lang_code`` or None if it does not exist. )rri)rrs rget_by_language_codez!CaptionQuery.get_by_language_codes##'' 22rz*This object can be treated as a dictionarycHt|jjSr)rFrvaluesrcs rrzCaptionQuery.allsD((//122rrc |j|Srrrs rrzCaptionQuery.__getitem__s##A&&rc,t|jSr)rrrcs rrzCaptionQuery.__len__s4''((rcHt|jjSr)iterrrrcs r__iter__zCaptionQuery.__iter__sD((//122rc|jSrrrcs rrzCaptionQuery.__repr__s&&'(rN)rrrrrr rr rDrrrrrrrrrfrrrrs8=g=I 3 3 '  3 3<=3T']3>3'S'))3)#)rrN)rcollections.abcrrtypingrrrr pytubefixr r pytubefix.helpersr r rrfrrrs2L-22%(_%(_%D 0)70)r