Move movie to new media type folder

This commit is contained in:
Ruud
2013-08-15 22:22:45 +02:00
parent 92b08bb5d5
commit 3e93983f6e
12 changed files with 38 additions and 10 deletions

View File

@@ -31,6 +31,13 @@ class Loader(object):
if os.path.isdir(path) and provider[:2] != '__':
self.paths[provider + '_provider'] = (25, 'couchpotato.core.providers.' + provider, path)
# Add media to loader
media_dir = os.path.join(root, 'couchpotato', 'core', 'media')
for media in os.listdir(media_dir):
path = os.path.join(media_dir, media)
if os.path.isdir(path) and media[:2] != '__':
self.paths[media + '_media'] = (25, 'couchpotato.core.media.' + media, path)
for plugin_type, plugin_tuple in self.paths.iteritems():
priority, module, dir_name = plugin_tuple

View File

@@ -0,0 +1,17 @@
from couchpotato.core.event import addEvent
from couchpotato.core.logger import CPLog
from couchpotato.core.plugins.base import Plugin
log = CPLog(__name__)
class MediaBase(Plugin):
identifier = None
def __init__(self):
addEvent('media.types', self.getType)
def getType(self):
return self.identifier

View File

View File

@@ -0,0 +1,6 @@
from .main import MovieBase
def start():
return MovieBase()
config = []

View File

@@ -4,7 +4,7 @@ from couchpotato.core.event import fireEvent, fireEventAsync, addEvent
from couchpotato.core.helpers.encoding import toUnicode, simplifyString
from couchpotato.core.helpers.variable import getImdb, splitString, tryInt
from couchpotato.core.logger import CPLog
from couchpotato.core.plugins.base import Plugin
from couchpotato.core.media import MediaBase
from couchpotato.core.settings.model import Library, LibraryTitle, Movie, \
Release
from couchpotato.environment import Env
@@ -16,7 +16,9 @@ import time
log = CPLog(__name__)
class MoviePlugin(Plugin):
class MovieBase(MediaBase):
identifier = 'movie'
default_dict = {
'profile': {'types': {'quality': {}}},
@@ -27,6 +29,8 @@ class MoviePlugin(Plugin):
}
def __init__(self):
super(MovieBase, self).__init__()
addApiView('movie.search', self.search, docs = {
'desc': 'Search the movie providers for a movie',
'params': {
@@ -476,7 +480,7 @@ class MoviePlugin(Plugin):
fireEvent('movie.restatus', m.id)
movie_dict = m.to_dict(self.default_dict)
fireEventAsync('searcher.single', movie_dict, on_complete = self.createNotifyFront(movie_id))
fireEventAsync('movie.searcher.single', movie_dict, on_complete = self.createNotifyFront(movie_id))
db.expire_all()
return {
@@ -574,7 +578,7 @@ class MoviePlugin(Plugin):
def onComplete():
db = get_session()
movie = db.query(Movie).filter_by(id = movie_id).first()
fireEventAsync('searcher.single', movie.to_dict(self.default_dict), on_complete = self.createNotifyFront(movie_id))
fireEventAsync('movie.searcher.single', movie.to_dict(self.default_dict), on_complete = self.createNotifyFront(movie_id))
db.expire_all()
return onComplete

View File

@@ -1,6 +0,0 @@
from .main import MoviePlugin
def start():
return MoviePlugin()
config = []