109 lines
4.2 KiB
Python
109 lines
4.2 KiB
Python
# Copyright (C) 2011 Jason Anderson
|
|
# Copyright (C) 2024 Lunatixz
|
|
#
|
|
#
|
|
# This file is part of PseudoTV Live.
|
|
#
|
|
# PseudoTV Live 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.
|
|
#
|
|
# PseudoTV Live 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 PseudoTV Live. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
from globals import *
|
|
from parsers import MP4Parser
|
|
from parsers import AVIParser
|
|
from parsers import MKVParser
|
|
from parsers import FLVParser
|
|
from parsers import TSParser
|
|
from parsers import VFSParser
|
|
from parsers import NFOParser
|
|
from parsers import YTParser
|
|
|
|
EXTERNAL_PARSER = [NFOParser.NFOParser]
|
|
try:
|
|
import pymediainfo
|
|
from parsers import MediaInfo
|
|
EXTERNAL_PARSER.append(MediaInfo.MediaInfo)
|
|
except: pass
|
|
|
|
try:
|
|
import ffmpeg
|
|
from parsers import FFProbe
|
|
EXTERNAL_PARSER.append(FFProbe.FFProbe)
|
|
except: pass
|
|
|
|
try:
|
|
import hachoir
|
|
from parsers import Hachoir
|
|
EXTERNAL_PARSER.append(Hachoir.Hachoir)
|
|
except: pass
|
|
|
|
try:
|
|
import moviepy
|
|
from parsers import MoviePY
|
|
from numpy.core._multiarray_umath import *
|
|
EXTERNAL_PARSER.append(MoviePY.MoviePY)
|
|
except: pass
|
|
|
|
try:
|
|
import cv2
|
|
from parsers import OpenCV
|
|
EXTERNAL_PARSER.append(OpenCV.OpenCV)
|
|
except: pass
|
|
|
|
class VideoParser:
|
|
def __init__(self):
|
|
self.AVIExts = ['.avi']
|
|
self.MP4Exts = ['.mp4', '.m4v', '.3gp', '.3g2', '.f4v', '.mov']
|
|
self.MKVExts = ['.mkv']
|
|
self.FLVExts = ['.flv']
|
|
self.TSExts = ['.ts', '.m2ts']
|
|
self.STRMExts = ['.strm']
|
|
self.VFSPaths = ['resource://','plugin://','upnp://','pvr://']
|
|
self.YTPaths = ['plugin://plugin.video.youtube','plugin://plugin.video.tubed','plugin://plugin.video.invidious']
|
|
|
|
|
|
def getVideoLength(self, filename: str, fileitem: dict={}, jsonRPC=None) -> int and float:
|
|
duration = jsonRPC._getDuration(filename)
|
|
if duration == 0:
|
|
if not filename: log("VideoParser: getVideoLength, no filename.")
|
|
elif filename.lower().startswith(tuple(self.VFSPaths)):
|
|
if filename.lower().startswith(tuple(self.YTPaths)):
|
|
duration = YTParser.YTParser().determineLength(filename)
|
|
if duration == 0:
|
|
duration = VFSParser.VFSParser().determineLength(filename, fileitem, jsonRPC)
|
|
else:
|
|
ext = os.path.splitext(filename)[1].lower()
|
|
if not FileAccess.exists(filename):
|
|
log("VideoParser: getVideoLength, Unable to find the file")
|
|
duration = 0
|
|
elif ext in self.AVIExts:
|
|
duration = AVIParser.AVIParser().determineLength(filename)
|
|
elif ext in self.MP4Exts:
|
|
duration = MP4Parser.MP4Parser().determineLength(filename)
|
|
elif ext in self.MKVExts:
|
|
duration = MKVParser.MKVParser().determineLength(filename)
|
|
elif ext in self.FLVExts:
|
|
duration = FLVParser.FLVParser().determineLength(filename)
|
|
elif ext in self.TSExts:
|
|
duration = TSParser.TSParser().determineLength(filename)
|
|
elif ext in self.STRMExts:
|
|
duration = NFOParser.NFOParser().determineLength(filename)
|
|
else:
|
|
duration = 0
|
|
|
|
if duration == 0:
|
|
for parser in EXTERNAL_PARSER:
|
|
if MONITOR().waitForAbort(0.0001) or duration > 0: break
|
|
duration = parser().determineLength(filename)
|
|
if duration > 0: duration = jsonRPC._setDuration(filename, fileitem, int(duration))
|
|
log("VideoParser: getVideoLength duration = %s, filename = %s"%(duration,filename))
|
|
return duration |