o ,jT@sdZddlZddlZddlZddlZddlZddlZddlZddlm m Z ddl m Z mZmZmZmZmZzddlmZWn eyLddlZYnwzedZWn ejy`dZYnwGdddZGd d d ZdS) z8 This module is a wrapper around the MediaInfo library. N)AnyDictListOptionalTupleUnion)metadata pymediainfoc@s^eZdZdZddZddZddZdd Zd ej fd d Z d dZ de e effddZdS)Tracka An object associated with a media file track. Each :class:`Track` attribute corresponds to attributes parsed from MediaInfo's output. All attributes are lower case. Attributes that are present several times such as `Duration` yield a second attribute starting with `other_` which is a list of all alternative attribute values. When a non-existing attribute is accessed, `None` is returned. Example: >>> t = mi.tracks[0] >>> t >>> t.duration 3000 >>> t.other_duration ['3 s 0 ms', '3 s 0 ms', '3 s 0 ms', '00:00:03.000', '00:00:03.000'] >>> type(t.non_existing) NoneType All available attributes can be obtained by calling :func:`to_data`. cC |j|jkSN__dict__selfotherrQ/var/www/kodo/Anonymous/send/lib/python3.10/site-packages/pymediainfo/__init__.py__eq__5 z Track.__eq__cCs$zt||WStyYdSwr )object__getattribute__AttributeError)rnamerrrr8s  zTrack.__getattribute__cC|jSr rrrrr __getstate__?szTrack.__getstate__cCs ||_dSr r)rstaterrr __setstate__Bs zTrack.__setstate__xml_dom_fragmentc Cs2|jd|_g}|D]F}|jd}|dkrd}|j}t||dur-t|||q d|}|||ft||durHt|||gq t|||q |D]C\}}z t||t t||WqSt yt||D]#} zt||} t||t | t||| Wn t yYqpwYqSwdS)Ntype_idtrack_idother_) attrib track_typetaglowerstriptextgetattrsetattrappendint ValueError) rr repeated_attributeselem node_name node_valueother_node_name primary_key other_key other_valuecurrentrrr__init__Es<      zTrack.__init__cCsd|j|jS)Nz&)formatr$r'rrrr__repr__kszTrack.__repr__returncCr)a Returns a dict representation of the track attributes. Example: >>> sorted(track.to_data().keys())[:3] ['codec', 'codec_extensions_usually_used', 'codec_url'] >>> t.to_data()["file_size"] 5988 :rtype: dict rrrrrto_datansz Track.to_dataN)__name__ __module__ __qualname____doc__rrrrETElementr:r<rstrrr>rrrrr s&r c@seZdZdZddZdded&eed/eded0ed1ed2ed3eeeefd4eed5eed e edffd6d7Z!d eeeffd8d9Z"d efd:d;Z#d%S)? MediaInfoa An object containing information about a media file. :class:`MediaInfo` objects can be created by directly calling code from libmediainfo (in this case, the library must be present on the system): >>> pymediainfo.MediaInfo.parse("/path/to/file.mp4") Alternatively, objects may be created from MediaInfo's XML output. Such output can be obtained using the ``XML`` output format on versions older than v17.10 and the ``OLDXML`` format on newer versions. Using such an XML file, we can create a :class:`MediaInfo` object: >>> with open("output.xml") as f: ... mi = pymediainfo.MediaInfo(f.read()) :param str xml: XML output obtained from MediaInfo. :param str encoding_errors: option to pass to :func:`str.encode`'s `errors` parameter before parsing `xml`. :raises xml.etree.ElementTree.ParseError: if passed invalid XML. :var tracks: A list of :py:class:`Track` objects which the media file contains. For instance: >>> mi = pymediainfo.MediaInfo.parse("/path/to/file.mp4") >>> for t in mi.tracks: ... print(t) cCr r tracksrrrrrrzMediaInfo.__eq__strictxmlencoding_errorscCsPt|d|}g|_|jdkrd}nd}||D] }|jt|qdS)Nzutf-8Filetrackz File/track)rC fromstringencoderHr(iterfindr.r )rrJrKxml_domxpath xml_trackrrrr:s zMediaInfo.__init__r'r=csfdd|jDS)Ncsg|] }|jkr|qSrr').0rMrTrr sz%MediaInfo._tracks..rG)rr'rrTr_tracksszMediaInfo._trackscC |dS)zm :return: All :class:`Track`\s of type ``General``. :rtype: list of :class:`Track`\s GeneralrWrrrrgeneral_tracks zMediaInfo.general_trackscCrX)zk :return: All :class:`Track`\s of type ``Video``. :rtype: list of :class:`Track`\s VideorZrrrr video_tracksr\zMediaInfo.video_trackscCrX)zk :return: All :class:`Track`\s of type ``Audio``. :rtype: list of :class:`Track`\s AudiorZrrrr audio_tracksr\zMediaInfo.audio_trackscCrX)zj :return: All :class:`Track`\s of type ``Text``. :rtype: list of :class:`Track`\s TextrZrrrr text_tracksr\zMediaInfo.text_trackscCrX)zk :return: All :class:`Track`\s of type ``Other``. :rtype: list of :class:`Track`\s OtherrZrrrr other_tracksr\zMediaInfo.other_trackscCrX)zk :return: All :class:`Track`\s of type ``Image``. :rtype: list of :class:`Track`\s ImagerZrrrr image_tracksr\zMediaInfo.image_trackscCrX)zj :return: All :class:`Track`\s of type ``Menu``. :rtype: list of :class:`Track`\s MenurZrrrr menu_tracksr\zMediaInfo.menu_tracksfilenamecCs@ttdrt|tjrt|Stdurt|tjrt|S|S)NPathLike)hasattros isinstancerjfspathpathlibPurePathrE)rirrr_normalize_filenames  zMediaInfo._normalize_filenamelibcCstj|j_g|j_tj|j_tjtjtjg|j_tj|j_tjtjg|j_tj|j_tjtjg|j _tj|j _tjtj tj g|j _tj|j _tjtj tjg|j _tj|j _tjg|j_tj |j_tjg|j_tj|j_tjg|j_d|j_tjg|j_d|j_dSr )ctypes c_wchar_pMediaInfo_Informrestype MediaInfo_Newargtypesc_void_pMediaInfo_Optionc_size_tMediaInfo_Openc_uint64MediaInfo_Open_Buffer_Initc_char_pMediaInfo_Open_Buffer_Continue'MediaInfo_Open_Buffer_Continue_GoTo_GetMediaInfo_Open_Buffer_FinalizeMediaInfo_DeleteMediaInfo_Close)clsrrrrr_define_library_prototypess<              z$MediaInfo._define_library_prototypesos_is_nt.cCs^|rd}n tjdkr d}nd}tjt}|D]}tj||}tj|r,|f}|Sq|S)N)z MediaInfo.dlldarwin)zlibmediainfo.0.dylibzlibmediainfo.dylib)zlibmediainfo.so.0)sysplatformrlpathdirname__file__joinisfile)r library_paths script_dirlibraryabsolute_library_pathrrr_get_library_pathss   zMediaInfo._get_library_pathsN library_filec Cstjdv}|r tj}ntj}|dur||}n|f}g}|D]V}z;||}|||}||dd} t d| } | rP| d} t dd| dD} ntd ||| | fWStyu} z |t| WYd} ~ qd} ~ wwtd d |d |) N)ntdosos2ce Info_Versionr z^MediaInfoLib - v(\S+)css|]}t|VqdSr )r/rUr"rrr Dsz)MediaInfo._get_library...z#Could not determine library versionz#Failed to load library from {} - {}z, )rlrrsWinDLLCDLLrrrwrzresearchgrouptuplesplit RuntimeErrorOSErrorr.rEr;r)rrrlib_typer exceptions library_pathrrhandleversionmatchlib_version_str lib_versionexcrrr _get_library*s:     zMediaInfo._get_librarycCsFz||dd\}}||||WdSty"YdSw)z Checks whether media files can be analyzed using libmediainfo. :param str library_file: path to the libmediainfo library, this should only be used if the library cannot be auto-detected. :rtype: bool NTF)rrr Exception)rrrrrrrr can_parsePs    zMediaInfo.can_parseF?T cover_data parse_speedfulllegacy_stream_displaymediainfo_optionsoutput buffer_sizec Csz||\} } } }|dkrd}nd}|dkr!| | d|rdnd| | dd | | d | d ur2|n| | | d |r=d nd| | dt|| | d|rQd nd|d uru|dkretd| t|D] \}}| | ||qiz|dd| }|dWn t yd }Ynw|d urdt |ddvrt d| | |d || }|r| | |t|d@rn| | }|tdjkr||| | || nnq| | n/||}| | |dkr| | | | d|vr tj|s t|td|| | d}|d ur'|dkr'| | dd| | | | | d ur;|||S|S)u Analyze a media file using libmediainfo. .. note:: Because of the way the underlying library works, this method should not be called simultaneously from multiple threads *with different arguments*. Doing so will cause inconsistencies or failures by changing library options that are shared across threads. :param filename: path to the media file or file-like object which will be analyzed. A URL can also be used if libmediainfo was compiled with CURL support. :param str library_file: path to the libmediainfo library, this should only be used if the library cannot be auto-detected. :param bool cover_data: whether to retrieve cover data as base64. :param str encoding_errors: option to pass to :func:`str.encode`'s `errors` parameter before parsing MediaInfo's XML output. :param float parse_speed: passed to the library as `ParseSpeed`, this option takes values between 0 and 1. A higher value will yield more precise results in some cases but will also increase parsing time. :param bool full: display additional tags, including computer-readable values for sizes and durations, corresponds to the CLI's ``--Full``/``-f`` parameter. :param bool legacy_stream_display: display additional information about streams. :param dict mediainfo_options: additional options that will be passed to the `MediaInfo_Option` function, for example: ``{"Language": "raw"}``. Do not use this parameter when running the method simultaneously from multiple threads, it will trigger a reset of all options which will cause inconsistencies or failures. :param str output: custom output format for MediaInfo, corresponds to the CLI's ``--Output`` parameter. Setting this causes the method to return a `str` instead of a :class:`MediaInfo` object. Useful values include: * the empty `str` ``""`` (corresponds to the default text output, obtained when running ``mediainfo`` with no additional parameters) * ``"XML"`` * ``"JSON"`` * ``%``-delimited templates (see ``mediainfo --Info-Parameters``) :param int buffer_size: size of the buffer used to read the file, in bytes. This is only used when `filename` is a file-like object. :type filename: str or pathlib.Path or os.PathLike or file-like object. :rtype: str if `output` is set. :rtype: :class:`MediaInfo` otherwise. :raises FileNotFoundError: if passed a non-existent file. :raises ValueError: if passed a file-like object opened in text mode. :raises OSError: if the library file could not be loaded. :raises RuntimeError: if parsing fails, this should not happen unless libmediainfo itself fails. Examples: >>> pymediainfo.MediaInfo.parse("tests/data/sample.mkv") >>> import json >>> mi = pymediainfo.MediaInfo.parse("tests/data/sample.mkv", ... output="JSON") >>> json.loads(mi)["media"]["track"][0] {'@type': 'General', 'TextCount': '1', 'FileExtension': 'mkv', 'FileSize': '5904', … } ) OLDXMLXML) Cover_Database64r CharSetzUTF-8InformNComplete1 ParseSpeedLegacyStreamDisplay) zThis version of MediaInfo (v{}) does not support resetting all options to their default values, passing it custom options is not recommended and may result in unpredictable behavior, see https://github.com/MediaArea/MediaInfoLib/issues/1128rrbmodez$File should be opened in binary modeTz://z3An error occured while opening {} with libmediainfoReset) rrzrEwarningswarnr;RuntimeWarningitemsseektellrr,r0r~readrlenrrsr}valuerrqr|rrrlrexistsFileNotFoundErrorrru)rrirrrKrrrrrrrrrrr xml_option option_name option_value file_sizebufferrinforrrparseaszR              zMediaInfo.parsecCsddd|jDiS)zq Returns a dict representation of the object's :py:class:`Tracks `. :rtype: dict rHcSsg|]}|qSr)r>rrrrrV sz%MediaInfo.to_data..rGrrrrr>szMediaInfo.to_datacCst|S)zp Returns a JSON representation of the object's :py:class:`Tracks `. :rtype: str )jsondumpsr>rrrrto_jsonszMediaInfo.to_json)rIr ) NFrIrTFNNr)$r?r@rArBrrEr:rr rWpropertyr[r^r`rbrdrfrh staticmethodrrq classmethodrboolrrrr/rrfloatrrrr>rrrrrrFs  # %       %rF)rBrsrrlrorrrxml.etree.ElementTreeetree ElementTreerCtypingrrrrrr importlibr ImportErrorimportlib_metadatar __version__PackageNotFoundErrorr rFrrrrs,   e