You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
4.4 KiB
Python
118 lines
4.4 KiB
Python
import os, sys, importlib, glob, inspect, itertools
|
|
|
|
from ffx.track_type import TrackType
|
|
|
|
from ffx.track_descriptor import TrackDescriptor
|
|
from ffx.media_descriptor import MediaDescriptor
|
|
|
|
from .media_combinator import MediaCombinator
|
|
|
|
from .disposition_combinator_2 import DispositionCombinator2
|
|
from .track_tag_combinator_2 import TrackTagCombinator2
|
|
from .jellyfin_combinator import JellyfinCombinator
|
|
from .media_tag_combinator import MediaTagCombinator
|
|
|
|
|
|
class MediaCombinator5(MediaCombinator):
|
|
|
|
VARIANT = 'VAAS'
|
|
|
|
# def __init__(self, SubCombinators: dict = {}, context = None):
|
|
def __init__(self, context = None):
|
|
|
|
self._context = context
|
|
self._logger = context['logger']
|
|
self._reportLogger = context['report_logger']
|
|
|
|
# self._SubCombinators = SubCombinations
|
|
|
|
def getVariant(self):
|
|
return MediaCombinator5.VARIANT
|
|
|
|
|
|
def getPayload(self, audioDispositionTuple = (set(), set()), audioTagTuple = ({}, {})):
|
|
|
|
kwargs = {}
|
|
kwargs[TrackDescriptor.CONTEXT_KEY] = self._context
|
|
kwargs[TrackDescriptor.TRACK_TYPE_KEY] = TrackType.VIDEO
|
|
trackDescriptor0 = TrackDescriptor(**kwargs)
|
|
|
|
kwargs = {}
|
|
kwargs[TrackDescriptor.CONTEXT_KEY] = self._context
|
|
kwargs[TrackDescriptor.TRACK_TYPE_KEY] = TrackType.AUDIO
|
|
kwargs[TrackDescriptor.DISPOSITION_SET_KEY] = audioDispositionTuple[0]
|
|
kwargs[TrackDescriptor.TAGS_KEY] = audioTagTuple[0]
|
|
trackDescriptor1 = TrackDescriptor(**kwargs)
|
|
|
|
kwargs = {}
|
|
kwargs[TrackDescriptor.CONTEXT_KEY] = self._context
|
|
kwargs[TrackDescriptor.TRACK_TYPE_KEY] = TrackType.AUDIO
|
|
kwargs[TrackDescriptor.DISPOSITION_SET_KEY] = audioDispositionTuple[1]
|
|
kwargs[TrackDescriptor.TAGS_KEY] = audioTagTuple[1]
|
|
trackDescriptor2 = TrackDescriptor(**kwargs)
|
|
|
|
kwargs = {}
|
|
kwargs[TrackDescriptor.CONTEXT_KEY] = self._context
|
|
kwargs[TrackDescriptor.TRACK_TYPE_KEY] = TrackType.SUBTITLE
|
|
trackDescriptor3 = TrackDescriptor(**kwargs)
|
|
|
|
|
|
kwargs = {}
|
|
kwargs[MediaDescriptor.CONTEXT_KEY] = self._context
|
|
kwargs[MediaDescriptor.TRACK_DESCRIPTOR_LIST_KEY] = [trackDescriptor0,
|
|
trackDescriptor1,
|
|
trackDescriptor2,
|
|
trackDescriptor3]
|
|
|
|
mediaDescriptor = MediaDescriptor(**kwargs)
|
|
mediaDescriptor.reindexSubIndices()
|
|
|
|
return mediaDescriptor
|
|
|
|
|
|
def assertFunc(self, testObj = {}):
|
|
pass
|
|
|
|
def shouldFail(self):
|
|
return False
|
|
|
|
def getYield(self):
|
|
|
|
for MTC in MediaTagCombinator.getAllClassReferences():
|
|
for DC2 in DispositionCombinator2.getAllClassReferences():
|
|
for TC2 in TrackTagCombinator2.getAllClassReferences():
|
|
for J in JellyfinCombinator.getAllClassReferences():
|
|
|
|
j = J(self._context)
|
|
self._context['use_jellyfin'] = j.getPayload()
|
|
|
|
dc2 = DC2(self._context)
|
|
tc2 = TC2(self._context)
|
|
mtc = MTC(self._context)
|
|
|
|
yieldObj = {}
|
|
|
|
yieldObj['identifier'] = self.getIdentifier()
|
|
yieldObj['variants'] = [self.getVariant(),
|
|
f"A:{dc2.getVariant()}",
|
|
f"A:{tc2.getVariant()}",
|
|
mtc.getVariant(),
|
|
j.getVariant()]
|
|
|
|
yieldObj['payload'] = self.getPayload(dc2.getPayload(), tc2.getPayload())
|
|
|
|
yieldObj['assertSelectors'] = ['M', 'AD', 'AT', 'MT', 'J']
|
|
yieldObj['assertFuncs'] = [self.assertFunc,
|
|
dc2.assertFunc,
|
|
tc2.assertFunc,
|
|
mtc.assertFunc,
|
|
j.assertFunc]
|
|
|
|
yieldObj['shouldFail'] = (self.shouldFail()
|
|
| dc2.shouldFail()
|
|
| tc2.shouldFail()
|
|
| mtc.shouldFail()
|
|
| j.shouldFail())
|
|
|
|
yield yieldObj
|