U 5dN1@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@sZeZdZdZddZd6ddZeedddd Ze dd d d Z dd ddZ dd ddZ e eedddZe eedddZeed ddZeed ddZd7e eedddZd8edd!d"d#Zeed d$d%Zd&d'Zed(d9ee e d)d*d+Zed,eed d-d.Zeee fd/d0d1Ze d d2d3Ze d d4d5Z 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 0/tmp/pip-unpacked-wheel-qmwpt_in/pytube/query.py sz(StreamQuery.__init__..N) fmt_streams itag_index)selfrr r r__init__ szStreamQuery.__init__NcsFg}s r |fddr6|fddrL|fdd rb| fdd sjr~| fddsr|fdd r| fddr|fd d| r|d d| r|d d|r|d d|r|d d|r ||dk r<|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 cs|jp kSN) resolutionr)resrr r~z$StreamQuery.filter..cs |jkSrfpsrrr rrrcs |jkSr mime_typerr r rrrcs |jkSrtyperr"r rrrcs|jp kSr)subtyper)file_extensionr$r rrrcs|jp kSr)abrr)r&bitrater rrrcs |jkSr video_codecrr(r rrrcs |jkSr audio_codecrr*r rrrcSs|jo |j Sr)includes_audio_trackincludes_video_trackrr r rrscSs|jo |j Sr)r-r,rr r rrscSs|jSr)Zis_progressiverr r rrrcSs|jSr)Z is_adaptiverr r rrrNcs |jkSris_dashrr.r rrr)appendextend_filter)rrrrr!r#r$r%r&r'r)r+ only_audioZ only_video progressiveZadaptiver/Zcustom_filter_functionsfiltersr ) r&r+r'r%rr/r!rrr$r#r)rfiltersDg  zStreamQuery.filter)r5returncCs&|j}|D]}t||}q tt|Sr)rr6r list)rr5rZ filter_lambdar r rr2s zStreamQuery._filter)attribute_namer7cstfdd|jD}|r\tt|dtr\ztt|fdddWStk rZYnXtt|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|dk r|qSrgetattrrr9r r sz(StreamQuery.order_by..rcstdttjt|S)N)r joinr6strisdigitr;rr<r rrsz&StreamQuery.order_by..)keycs t|Srr:rr<r rrr)r isinstancer;r@r sorted ValueError)rr9Z has_attributer r<rorder_bys&    zStreamQuery.order_byr7cCst|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 rIr r rascszStreamQuery.asc)rr7cCs|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_itag)rr7cCs|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)r4r$r)r6first)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. TrN)r4r$r)r6rFrOrIr r rget_lowest_resolutions 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)r4rr6rFlastrIr r rget_highest_resolutions z"StreamQuery.get_highest_resolutionrN)r$r7cCs|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)r3r$r&rR)rr$r r rget_audio_only"s zStreamQuery.get_audio_onlyF)is_otfr7cs|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 cs |jkSrrVrrWr rr9rz!StreamQuery.otf..)r2)rrVr rWrotf2szStreamQuery.otfcCs(z |jdWStk r"YdSXdS)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 IndexErrorrIr r rrO;s  zStreamQuery.firstcCs&z |jdWStk r YnXdS)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. rHNrYrIr r rrSIs  zStreamQuery.lastz.Get the size of this list directly using len())valuer7cCs|r|j|St|S)zAGet the count of items in the list. :rtype: int )rcountlen)rr[r r rr\Ws zStreamQuery.countz6This object can be treated as a list, all() is uselesscCs|jSzXGet all the results represented by this query as a list. :rtype: list rrIr r rallbszStreamQuery.allicCs |j|Srr_rrbr r r __getitem__kszStreamQuery.__getitem__cCs t|jSr)r]rrIr r r__len__nszStreamQuery.__len__cCs|jSrr_rIr r r__repr__qszStreamQuery.__repr__)NNNNNNNNNNNNNNNNN)rN)F)N)!__name__ __module__ __qualname____doc__rr6rrr2r@rFrJrKr rr rMrPrQrTrUboolrXrOrSr r\r`rslicerdrerfr r r rr sP  #    r c@seZdZdZeedddZedee edddZ ed eed d d Z ed ddZ e d ddZddZed ddZdS) 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)rrnr r rrxszCaptionQuery.__init__z?This object can be treated as a dictionary, i.e. captions['en']) lang_coder7cCs |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. )rrrL)rrsr r rget_by_language_codesz!CaptionQuery.get_by_language_codez*This object can be treated as a dictionaryrGcCst|jSr^)r8rrvaluesrIr r rr`szCaptionQuery.allracCs |j|Srrqrcr r rrdszCaptionQuery.__getitem__cCs t|jSr)r]rrrIr r rreszCaptionQuery.__len__cCst|jSr)iterrrrurIr r r__iter__szCaptionQuery.__iter__cCs|jSrrqrIr r rrfszCaptionQuery.__repr__N)rgrhrirjrrrr r@rrtr`rdr rerwrfr r r rrmus rmN)rjcollections.abcrrtypingrrrrZpytuberr Zpytube.helpersr r rmr r r rs n