U g,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@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-1a9f0fi6/pytube/query.py sz(StreamQuery.__init__..N) fmt_streams itag_index)selfrr r r__init__ szStreamQuery.__init__Ncsg}s r^tts ttr6|fddn(ttsJttr^|fddrt|fddr|fdd r| fdd sr| fddsĈr|fdd r| fd dr|fd d| r|d d| r.|d d|rB|d d|rV|dd|rf||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 resolutionrresrr rz$StreamQuery.filter..cs|jp kSrrrrr rrrcs |jkSrfpsrr r 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_dashrr0r rrr) isinstancestrappendlistextend_filter)rr!rrr#r%r&r'r(r)r+r- only_audioZ only_video progressiveZadaptiver1Zcustom_filter_functionsfiltersr ) r(r-r)r'r!r1r#rrr&r%r+rfiltersJg  zStreamQuery.filter)r:returncCs&|j}|D]}t||}q tt|Sr)rr;r r5)rr:rZ filter_lambdar r rr7s zStreamQuery._filter)attribute_namer<cstfdd|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|qSrgetattrrr=r r sz(StreamQuery.order_by..rcstdttjt|S)N)r joinr;r3isdigitr?rr@r rrsz&StreamQuery.order_by..)keycs t|Srr>rr@r rrr)rr2r?r3r sorted ValueError)rr=Z has_attributer r@rorder_bys&    zStreamQuery.order_byr<cCst|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 rKr r rascszStreamQuery.asc)rr<cCs|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)rr<cCs|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)r9r&r)r;first)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. TrP)r9r&r)r;rHrQrKr 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)r9rr;rHlastrKr r rget_highest_resolutions z"StreamQuery.get_highest_resolutionrP)r&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)r8r&r(rT)rr&r r rget_audio_only%s zStreamQuery.get_audio_onlyF)is_otfr<cs|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 |jkSrrXrrYr rr<rz!StreamQuery.otf..)r7)rrXr rYrotf5szStreamQuery.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 IndexErrorrKr r rrQ>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. rJNr[rKr r rrULs  zStreamQuery.lastz.Get the size of this list directly using len())valuer<cCs|r|j|St|S)zAGet the count of items in the list. :rtype: int )rcountlen)rr]r r rr^Zs 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 rrKr r ralleszStreamQuery.allicCs |j|Srrarrdr r r __getitem__nszStreamQuery.__getitem__cCs t|jSr)r_rrKr r r__len__qszStreamQuery.__len__cCs|jSrrarKr r r__repr__tszStreamQuery.__repr__)NNNNNNNNNNNNNNNNN)rP)F)N)!__name__ __module__ __qualname____doc__rr;rrr7r3rHrLrMr rr rOrRrSrVrWboolrZrQrUr r^rbrslicerfrgrhr 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)rrpr r rr{szCaptionQuery.__init__z?This object can be treated as a dictionary, i.e. captions['en']) lang_coder<cCs |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. )rtrN)rrur r rget_by_language_codesz!CaptionQuery.get_by_language_codez*This object can be treated as a dictionaryrIcCst|jSr`)r5rtvaluesrKr r rrbszCaptionQuery.allrccCs |j|Srrsrer r rrfszCaptionQuery.__getitem__cCs t|jSr)r_rtrKr r rrgszCaptionQuery.__len__cCst|jSr)iterrtrwrKr r r__iter__szCaptionQuery.__iter__cCs|jSrrsrKr r rrhszCaptionQuery.__repr__N)rirjrkrlrrrr r3rrvrbrfr rgryrhr r r rroxs roN)rlcollections.abcrrtypingrrrrZpytuberr Zpytube.helpersr r ror r r rs q