mirror of
https://github.com/moparisthebest/SickRage
synced 2024-11-15 13:55:05 -05:00
64 lines
2.4 KiB
Python
64 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
# enzyme - Video metadata parser
|
|
# Copyright 2011-2012 Antoine Bertin <diaoulael@gmail.com>
|
|
# Copyright 2003-2006 Thomas Schueppel <stain@acm.org>
|
|
# Copyright 2003-2006 Dirk Meyer <dischi@freevo.org>
|
|
#
|
|
# This file is part of enzyme.
|
|
#
|
|
# enzyme is free software; you can redistribute it and/or modify it under
|
|
# the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# enzyme is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with enzyme. If not, see <http://www.gnu.org/licenses/>.
|
|
import mimetypes
|
|
import os
|
|
import sys
|
|
from exceptions import *
|
|
|
|
|
|
PARSERS = [('asf', ['video/asf'], ['asf', 'wmv', 'wma']),
|
|
('flv', ['video/flv'], ['flv']),
|
|
('mkv', ['video/x-matroska', 'application/mkv'], ['mkv', 'mka', 'webm']),
|
|
('mp4', ['video/quicktime', 'video/mp4'], ['mov', 'qt', 'mp4', 'mp4a', 'm4v', '3gp', '3gp2', '3g2', 'mk2']),
|
|
('mpeg', ['video/mpeg'], ['mpeg', 'mpg', 'mp4', 'ts']),
|
|
('ogm', ['application/ogg'], ['ogm', 'ogg', 'ogv']),
|
|
('real', ['video/real'], ['rm', 'ra', 'ram']),
|
|
('riff', ['video/avi'], ['wav', 'avi'])
|
|
]
|
|
|
|
|
|
def parse(path):
|
|
"""Parse metadata of the given video
|
|
|
|
:param string path: path to the video file to parse
|
|
:return: a parser corresponding to the video's mimetype or extension
|
|
:rtype: :class:`~enzyme.core.AVContainer`
|
|
|
|
"""
|
|
if not os.path.isfile(path):
|
|
raise ValueError('Invalid path')
|
|
extension = os.path.splitext(path)[1][1:]
|
|
mimetype = mimetypes.guess_type(path)[0]
|
|
parser_ext = None
|
|
parser_mime = None
|
|
for (parser_name, parser_mimetypes, parser_extensions) in PARSERS:
|
|
if mimetype in parser_mimetypes:
|
|
parser_mime = parser_name
|
|
if extension in parser_extensions:
|
|
parser_ext = parser_name
|
|
parser = parser_mime or parser_ext
|
|
if not parser:
|
|
raise NoParserError()
|
|
mod = __import__(parser, globals=globals(), locals=locals(), fromlist=[], level=-1)
|
|
with open(path, 'rb') as f:
|
|
p = mod.Parser(f)
|
|
return p
|