raoul/openlp
1
0
Fork 0
forked from openlp/openlp
Code Issues Pull Requests Projects Releases Wiki Activity

Reapply OpenLP's changes to pymediainfo

Browse Source
This commit is contained in:
Bastian Germann 2018-10-07 01:00:18 +02:00
parent 7fb3ad4e9e
commit 4784cdf152
1 changed files with 19 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
openlp/core/ui/media/vendor/mediainfoWrapper.py vendored
View File

@ -1,6 +1,10 @@
# -*- coding: utf-8 -*-
# vim: autoindent shiftwidth=4 expandtab textwidth=120 tabstop=4 softtabstop=4
# The MIT License # The MIT License
# #
# Copyright (c) 2010-2014, Patrick Altman <paltman@gmail.com> # Copyright (c) 2010-2014, Patrick Altman <paltman@gmail.com>
# Copyright (c) 2016-2018, OpenLP Developers
# #
# Permission is hereby granted, free of charge, to any person obtaining a copy # Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal # of this software and associated documentation files (the "Software"), to deal
@ -22,16 +26,16 @@
# #
# http://www.opensource.org/licenses/mit-license.php # http://www.opensource.org/licenses/mit-license.php
"""
The :mod:`~openlp.core.ui.media.mediainfo` module contains code to run mediainfo
on a media file and obtain information related to the requested media.
"""
import json import json
import os import os
from subprocess import Popen from subprocess import check_output
from tempfile import mkstemp
import six
from bs4 import BeautifulSoup, NavigableString from bs4 import BeautifulSoup, NavigableString
__version__ = '1.4.1'
ENV_DICT = os.environ ENV_DICT = os.environ
@ -77,27 +81,23 @@ class Track(object):
pass pass
def __repr__(self): def __repr__(self):
return("<Track track_id='{0}', track_type='{1}'>".format(self.track_id, self.track_type)) return "<Track track_id='{0}', track_type='{1}'>".format(self.track_id, self.track_type)
def to_data(self): def to_data(self):
data = {} data = {}
for k, v in six.iteritems(self.__dict__): for k, v in self.__dict__.items():
if k != 'xml_dom_fragment': if k != 'xml_dom_fragment':
data[k] = v data[k] = v
return data return data
class MediaInfo(object): class MediaInfoWrapper(object):
def __init__(self, xml): def __init__(self, xml):
self.xml_dom = xml self.xml_dom = xml
if six.PY3: xml_types = (str,) # no unicode type in python3
xml_types = (str,) # no unicode type in python3
else:
xml_types = (str, unicode)
if isinstance(xml, xml_types): if isinstance(xml, xml_types):
self.xml_dom = MediaInfo.parse_xml_data_into_dom(xml) self.xml_dom = MediaInfoWrapper.parse_xml_data_into_dom(xml)
@staticmethod @staticmethod
def parse_xml_data_into_dom(xml_data): def parse_xml_data_into_dom(xml_data):
@ -105,21 +105,11 @@ class MediaInfo(object):
@staticmethod @staticmethod
def parse(filename, environment=ENV_DICT): def parse(filename, environment=ENV_DICT):
command = ["mediainfo", "-f", "--Output=XML", filename] xml = check_output(['mediainfo', '-f', '--Output=XML', '--Inform=OLDXML', filename])
fileno_out, fname_out = mkstemp(suffix=".xml", prefix="media-") if not xml.startswith(b'<?xml'):
fileno_err, fname_err = mkstemp(suffix=".err", prefix="media-") xml = check_output(['mediainfo', '-f', '--Output=XML', filename])
fp_out = os.fdopen(fileno_out, 'r+b') xml_dom = MediaInfoWrapper.parse_xml_data_into_dom(xml)
fp_err = os.fdopen(fileno_err, 'r+b') return MediaInfoWrapper(xml_dom)
p = Popen(command, stdout=fp_out, stderr=fp_err, env=environment)
p.wait()
fp_out.seek(0)
xml_dom = MediaInfo.parse_xml_data_into_dom(fp_out.read())
fp_out.close()
fp_err.close()
os.unlink(fname_out)
os.unlink(fname_err)
return MediaInfo(xml_dom)
def _populate_tracks(self): def _populate_tracks(self):
if self.xml_dom is None: if self.xml_dom is None: