diff --git a/couchpotato/core/providers/info/_modifier/__init__.py b/couchpotato/core/providers/info/_modifier/__init__.py index 3bdf5e0d..6242e963 100644 --- a/couchpotato/core/providers/info/_modifier/__init__.py +++ b/couchpotato/core/providers/info/_modifier/__init__.py @@ -1,7 +1,7 @@ -from .main import MovieResultModifier +from .main import InfoResultModifier def start(): - return MovieResultModifier() + return InfoResultModifier() config = [] diff --git a/couchpotato/core/providers/info/_modifier/main.py b/couchpotato/core/providers/info/_modifier/main.py index e652bc62..083fc3e7 100644 --- a/couchpotato/core/providers/info/_modifier/main.py +++ b/couchpotato/core/providers/info/_modifier/main.py @@ -3,6 +3,7 @@ from couchpotato.core.event import addEvent, fireEvent from couchpotato.core.helpers.variable import mergeDicts, randomString from couchpotato.core.logger import CPLog from couchpotato.core.plugins.base import Plugin +from couchpotato.core.providers.base import MultiProvider from couchpotato.core.settings.model import Library import copy import traceback @@ -10,7 +11,17 @@ import traceback log = CPLog(__name__) -class MovieResultModifier(Plugin): +class InfoResultModifier(MultiProvider): + + def getTypes(self): + return [Movie, Show] + + +class ModifierBase(Plugin): + pass + + +class Movie(ModifierBase): default_info = { 'tmdb_id': 0, @@ -93,3 +104,7 @@ class MovieResultModifier(Plugin): if result and result.get('imdb'): return mergeDicts(result, self.getLibraryTags(result['imdb'])) return result + + +class Show(ModifierBase): + pass \ No newline at end of file