o {#i,2@sldZddlmZmZddlmZmZmZmZddl m Z m Z ddl m Z GdddeZGdd d eZd S) zFThis module provides a query interface for media streams and captions.)MappingSequence)CallableListOptionalUnion)CaptionStream) deprecatedc@seZdZdZddZ                 d6ddZdeeddfd d Zd e ddfd d Z d7ddZ d7ddZ de deefddZde deefddZdeefddZdeefddZd8de deefddZd9d!eddfd"d#Zdeefd$d%Zd&d'Zed(d:d)ee de fd*d+Zed,deefd-d.Zd/eee ffd0d1Zde fd2d3Zde fd4d5Z dS); StreamQueryz3Interface for querying the available media streams.cCs||_dd|D|_dS)zConstruct a :class:`StreamQuery `. param list fmt_streams: list of :class:`Stream ` instances. cSsi|]}t|j|qS)intitag.0sr r S/var/www/GraceOrthoHospital/venv_grace/lib/python3.10/site-packages/pytube/query.py sz(StreamQuery.__init__..N) fmt_streams itag_index)selfrr r r__init__ szStreamQuery.__init__Ncs~g}sr/ttsttr|fddntts%ttr/|fddr:|fddrE|fdd rP| fdd sTr^| fddsbrl|fdd rw| fd dr|fd d| r|d d| r|d d|r|d d|r|dd|r||dur|fdd||S)aApply 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 custom_filter_functions: (optional) Interface for defining complex filters without subclassing. :type custom_filter_functions: list or None c|jpkSN resolutionrresrr rz$StreamQuery.filter..cs|jpvSrrrrr rrr c |jkSrfpsrr"r rr cr!r mime_typerr%r rrr$cr!rtyperr'r rrr$cs|jpkSr)subtyper)file_extensionr)r rrr crr)abrr)r+bitrater rrr cr!r video_codecrr-r rrr$cr!r audio_codecrr/r rrr$cS|jo|j Sr)includes_audio_trackincludes_video_trackrr r rrcSr1r)r3r2rr r rrr4cS|jSr)is_progressiverr r rrcSr5r) is_adaptiverr r rrr7Ncr!ris_dashrr9r rrr$) isinstancestrappendlistextend_filter)rr#rrr&r(r)r*r+r,r.r0 only_audio only_video progressiveadaptiver:custom_filter_functionsfiltersr ) r+r0r,r*r#r:r&rrr)r(r.rfiltersJg  zStreamQuery.filterrFreturncCs&|j}|D]}t||}qtt|Sr)rrGr r>)rrFr filter_lambdar r rr@s  zStreamQuery._filterattribute_namecsrfdd|jD}|r-tt|dtr-z tt|fdddWSty,Ynwtt|fdddS)zApply a sort order. Filters out stream the do not have the attribute. :param str attribute_name: The name of the attribute to sort by. csg|] }t|dur|qSrgetattrrrJr r s z(StreamQuery.order_by..rcstdttjt|S)N)r joinrGr<isdigitrLrrMr rrs z&StreamQuery.order_by..)keycs t|SrrKrrMr rrr$)rr;rLr<r sorted ValueError)rrJ has_attributer rMrorder_bys&   zStreamQuery.order_bycCst|jdddS)z_Sort streams in descending order. :rtype: :class:`StreamQuery ` N)r rrr r rdescszStreamQuery.desccCs|S)z^Sort streams in ascending order. :rtype: :class:`StreamQuery ` r rXr r rascszStreamQuery.ascrcCs|jt|S)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. )rgetr )rrr r r get_by_itags zStreamQuery.get_by_itagrcCs|jdd|dS)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)rCr)r)rGfirst)rrr r rget_by_resolutions  zStreamQuery.get_by_resolutioncCs|jddddS)zGet lowest resolution stream that is a progressive mp4. :rtype: :class:`Stream ` or None :returns: The :class:`Stream ` matching the given itag or None if not found. Tr])rCr)r)rGrVr^rXr r rget_lowest_resolution s z!StreamQuery.get_lowest_resolutioncCs|jdddS)zGet highest resolution stream that is a progressive video. :rtype: :class:`Stream ` or None :returns: The :class:`Stream ` matching the given itag or None if not found. T)rCrrGrVlastrXr r rget_highest_resolutions z"StreamQuery.get_highest_resolutionr]r)cCs|jd|ddS)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)rAr)r+ra)rr)r r rget_audio_only%s zStreamQuery.get_audio_onlyFis_otfcs|fddgS)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 cr!rrerrfr rr<r$z!StreamQuery.otf..)r@)rrer rfrotf5zStreamQuery.otfcC"z|jdWStyYdSw)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 IndexErrorrXr r rr^>   zStreamQuery.firstcCri)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. rWNrjrXr r rrbLrlzStreamQuery.lastz.Get the size of this list directly using len()valuecCs|r|j|St|S)zAGet the count of items in the list. :rtype: int )rcountlen)rrmr r rrnZs zStreamQuery.countz6This object can be treated as a list, all() is uselesscCr5zXGet all the results represented by this query as a list. :rtype: list rrXr r ralleszStreamQuery.allicC |j|Srrqrrsr r r __getitem__n zStreamQuery.__getitem__cC t|jSr)rorrXr r r__len__qrwzStreamQuery.__len__cC|jSrrqrXr r r__repr__tzStreamQuery.__repr__)NNNNNNNNNNNNNNNNN)rHr )r])Fr)!__name__ __module__ __qualname____doc__rrGrrr@r<rVrYrZr rr r\r_r`rcrdboolrgr^rbr rnrrrslicervryr{r r r rr sR    #     r c@seZdZdZdeefddZeddede efdd Z ed deefd d Z d efddZ de fddZddZdefddZdS) CaptionQueryz.Interface for querying the available captions.captionscCsdd|D|_dS)zConstruct a :class:`Caption `. param list captions: list of :class:`Caption ` instances. cSsi|]}|j|qSr )code)rcr r rrsz)CaptionQuery.__init__..Nlang_code_index)rrr r rr{rhzCaptionQuery.__init__z?This object can be treated as a dictionary, i.e. captions['en'] lang_coderHcCs |j|S)a[Get the :class:`Caption ` 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. )rr[)rrr r rget_by_language_codes z!CaptionQuery.get_by_language_codez*This object can be treated as a dictionarycCt|jSrp)r>rvaluesrXr r rrrszCaptionQuery.allrscCrtrrrur r rrvrwzCaptionQuery.__getitem__cCrxr)rorrXr r rryrwzCaptionQuery.__len__cCrr)iterrrrXr r r__iter__r4zCaptionQuery.__iter__cCrzrrrXr r rr{r|zCaptionQuery.__repr__N)r}r~rrrrrr r<rrrrrvr ryrr{r r r rrxs"  rN)rcollections.abcrrtypingrrrrpytuberr pytube.helpersr r rr r r rs q