178 lines
4.9 KiB
Python
178 lines
4.9 KiB
Python
from couchpotato import get_session
|
|
from couchpotato.api import addApiView
|
|
from couchpotato.core.event import fireEvent, fireEventAsync, addEvent
|
|
from couchpotato.core.helpers.request import getParams, jsonified
|
|
from couchpotato.core.plugins.base import Plugin
|
|
from couchpotato.core.settings.model import Movie
|
|
from couchpotato.environment import Env
|
|
from sqlalchemy.sql.expression import or_
|
|
from urllib import urlencode
|
|
|
|
|
|
class MoviePlugin(Plugin):
|
|
|
|
default_dict = {
|
|
'profile': {'types': {'quality': {}}},
|
|
'releases': {'status': {}, 'quality': {}, 'files':{}, 'info': {}},
|
|
'library': {'titles': {}, 'files':{}},
|
|
'files': {},
|
|
'status': {}
|
|
}
|
|
|
|
def __init__(self):
|
|
addApiView('movie.search', self.search)
|
|
addApiView('movie.list', self.listView)
|
|
addApiView('movie.refresh', self.refresh)
|
|
|
|
addApiView('movie.add', self.add)
|
|
addApiView('movie.edit', self.edit)
|
|
addApiView('movie.delete', self.delete)
|
|
|
|
addEvent('movie.get', self.get)
|
|
addEvent('movie.list', self.list)
|
|
|
|
def get(self, movie_id):
|
|
|
|
db = get_session()
|
|
m = db.query(Movie).filter_by(movie_id = movie_id).first()
|
|
|
|
return m.to_dict(self.default_dict)
|
|
|
|
def list(self, status = ['active']):
|
|
|
|
db = get_session()
|
|
|
|
# Make a list from string
|
|
if not isinstance(status, (list, tuple)):
|
|
status = [status]
|
|
|
|
results = db.query(Movie).filter(or_(*[Movie.status.has(identifier = s) for s in status])).all()
|
|
|
|
movies = []
|
|
for movie in results:
|
|
temp = movie.to_dict(self.default_dict)
|
|
movies.append(temp)
|
|
|
|
return movies
|
|
|
|
def listView(self):
|
|
|
|
params = getParams()
|
|
status = params.get('status', ['active'])
|
|
movies = self.list(status)
|
|
|
|
return jsonified({
|
|
'success': True,
|
|
'empty': len(movies) == 0,
|
|
'movies': movies,
|
|
})
|
|
|
|
def refresh(self):
|
|
|
|
params = getParams()
|
|
db = get_session()
|
|
|
|
movie = db.query(Movie).filter_by(id = params.get('id')).first()
|
|
|
|
# Get current selected title
|
|
default_title = ''
|
|
for title in movie.library.titles:
|
|
if title.default: default_title = title.title
|
|
|
|
if movie:
|
|
#addEvent('library.update.after', )
|
|
fireEventAsync('library.update', identifier = movie.library.identifier, default_title = default_title, force = True)
|
|
fireEventAsync('searcher.single', movie.to_dict(self.default_dict))
|
|
|
|
return jsonified({
|
|
'success': True,
|
|
})
|
|
|
|
def search(self):
|
|
|
|
params = getParams()
|
|
cache_key = '%s/%s' % (__name__, urlencode(params))
|
|
movies = Env.get('cache').get(cache_key)
|
|
|
|
if not movies:
|
|
movies = fireEvent('movie.search', q = params.get('q'), merge = True)
|
|
Env.get('cache').set(cache_key, movies)
|
|
|
|
return jsonified({
|
|
'success': True,
|
|
'empty': len(movies) == 0,
|
|
'movies': movies,
|
|
})
|
|
|
|
def add(self):
|
|
|
|
params = getParams()
|
|
db = get_session();
|
|
|
|
library = fireEvent('library.add', single = True, attrs = params)
|
|
|
|
# Status
|
|
status_active = fireEvent('status.add', 'active', single = True)
|
|
status_snatched = fireEvent('status.add', 'snatched', single = True)
|
|
|
|
m = db.query(Movie).filter_by(library_id = library.get('id')).first()
|
|
if not m:
|
|
m = Movie(
|
|
library_id = library.get('id'),
|
|
profile_id = params.get('profile_id')
|
|
)
|
|
db.add(m)
|
|
else:
|
|
# Clean snatched history
|
|
for release in m.releases:
|
|
if release.status_id == status_snatched.get('id'):
|
|
release.delete()
|
|
|
|
m.profile_id = params.get('profile_id')
|
|
|
|
m.status_id = status_active.get('id')
|
|
db.commit()
|
|
|
|
movie_dict = m.to_dict(self.default_dict)
|
|
|
|
fireEventAsync('searcher.single', movie_dict)
|
|
|
|
return jsonified({
|
|
'success': True,
|
|
'added': True,
|
|
'movie': movie_dict,
|
|
})
|
|
|
|
def edit(self):
|
|
|
|
params = getParams()
|
|
db = get_session();
|
|
|
|
m = db.query(Movie).filter_by(id = params.get('id')).first()
|
|
m.profile_id = params.get('profile_id')
|
|
|
|
# Default title
|
|
for title in m.library.titles:
|
|
title.default = params.get('default_title').lower() == title.title.lower()
|
|
|
|
db.commit()
|
|
|
|
return jsonified({
|
|
'success': True,
|
|
})
|
|
|
|
def delete(self):
|
|
|
|
params = getParams()
|
|
db = get_session()
|
|
|
|
status = fireEvent('status.add', 'deleted', single = True)
|
|
|
|
movie = db.query(Movie).filter_by(id = params.get('id')).first()
|
|
movie.status_id = status.get('id')
|
|
db.commit()
|
|
|
|
return jsonified({
|
|
'success': True,
|
|
})
|