You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
121 lines
4.2 KiB
121 lines
4.2 KiB
12 years ago
|
from couchpotato import get_session
|
||
|
|
from couchpotato.api import addApiView
|
||
|
|
from couchpotato.core.event import fireEvent
|
||
|
|
from couchpotato.core.helpers.request import jsonified, getParams
|
||
|
|
from couchpotato.core.helpers.variable import splitString, tryInt
|
||
|
|
from couchpotato.core.logger import CPLog
|
||
|
|
from couchpotato.core.plugins.base import Plugin
|
||
|
|
from couchpotato.core.settings.model import Movie
|
||
|
|
from sqlalchemy.sql.expression import or_
|
||
|
|
import random
|
||
|
|
|
||
|
|
log = CPLog(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class Dashboard(Plugin):
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
|
||
|
|
addApiView('dashboard.suggestions', self.suggestView)
|
||
|
|
addApiView('dashboard.soon', self.getSoonView)
|
||
|
|
|
||
|
|
def newSuggestions(self):
|
||
|
|
|
||
|
|
movies = fireEvent('movie.list', status = ['active', 'done'], limit_offset = (20, 0), single = True)
|
||
|
|
movie_identifiers = [m['library']['identifier'] for m in movies[1]]
|
||
|
|
|
||
|
|
ignored_movies = fireEvent('movie.list', status = ['ignored', 'deleted'], limit_offset = (100, 0), single = True)
|
||
|
|
ignored_identifiers = [m['library']['identifier'] for m in ignored_movies[1]]
|
||
|
|
|
||
|
|
suggestions = fireEvent('movie.suggest', movies = movie_identifiers, ignore = ignored_identifiers, single = True)
|
||
|
|
suggest_status = fireEvent('status.get', 'suggest', single = True)
|
||
|
|
|
||
|
|
for suggestion in suggestions:
|
||
|
|
fireEvent('movie.add', params = {'identifier': suggestion}, force_readd = False, search_after = False, status_id = suggest_status.get('id'))
|
||
|
|
|
||
|
|
def suggestView(self):
|
||
|
|
|
||
|
|
db = get_session()
|
||
|
|
|
||
|
|
movies = db.query(Movie).limit(20).all()
|
||
|
|
identifiers = [m.library.identifier for m in movies]
|
||
|
|
|
||
|
|
suggestions = fireEvent('movie.suggest', movies = identifiers, single = True)
|
||
|
|
print suggestions
|
||
|
|
|
||
|
|
return jsonified({
|
||
|
|
'result': True,
|
||
|
|
'suggestions': suggestions
|
||
|
|
})
|
||
|
|
|
||
|
|
def getSoonView(self):
|
||
|
|
|
||
|
|
params = getParams()
|
||
|
|
db = get_session()
|
||
|
|
|
||
|
|
# Get profiles first, determine pre or post theater
|
||
|
|
profiles = fireEvent('profile.all', single = True)
|
||
|
|
qualities = fireEvent('quality.all', single = True)
|
||
|
|
pre_releases = fireEvent('quality.pre_releases', single = True)
|
||
|
|
|
||
|
|
id_pre = {}
|
||
|
|
for quality in qualities:
|
||
|
|
id_pre[quality.get('id')] = quality.get('identifier') in pre_releases
|
||
|
|
|
||
|
|
# See what the profile contain and cache it
|
||
|
|
profile_pre = {}
|
||
|
|
for profile in profiles:
|
||
|
|
contains = {}
|
||
|
|
for profile_type in profile.get('types', []):
|
||
|
|
contains['theater' if id_pre.get(profile_type.get('quality_id')) else 'dvd'] = True
|
||
|
|
|
||
|
|
profile_pre[profile.get('id')] = contains
|
||
|
|
|
||
|
|
# Get all active movies
|
||
|
|
q = db.query(Movie) \
|
||
|
|
.join(Movie.profile, Movie.library) \
|
||
|
|
.filter(or_(*[Movie.status.has(identifier = s) for s in ['active']])) \
|
||
|
|
.group_by(Movie.id)
|
||
|
|
|
||
|
|
# Add limit
|
||
|
|
limit_offset = params.get('limit_offset')
|
||
|
|
limit = 12
|
||
|
|
if limit_offset:
|
||
|
|
splt = splitString(limit_offset) if isinstance(limit_offset, (str, unicode)) else limit_offset
|
||
|
|
limit = tryInt(splt[0])
|
||
|
|
|
||
|
|
all_movies = q.all()
|
||
|
|
|
||
|
|
if params.get('random', False):
|
||
|
|
random.shuffle(all_movies)
|
||
|
|
|
||
|
|
movies = []
|
||
|
|
for movie in all_movies:
|
||
|
|
pp = profile_pre.get(movie.profile.id)
|
||
|
|
eta = movie.library.info.get('release_date', {})
|
||
|
|
coming_soon = False
|
||
|
|
|
||
|
|
# Theater quality
|
||
|
|
if pp.get('theater') and fireEvent('searcher.could_be_released', True, eta, single = True):
|
||
|
|
coming_soon = True
|
||
|
|
if pp.get('dvd') and fireEvent('searcher.could_be_released', False, eta, single = True):
|
||
|
|
coming_soon = True
|
||
|
|
|
||
|
|
if coming_soon:
|
||
|
|
temp = movie.to_dict({
|
||
|
|
'profile': {'types': {}},
|
||
|
|
'releases': {'files':{}, 'info': {}},
|
||
|
|
'library': {'titles': {}, 'files':{}},
|
||
|
|
'files': {},
|
||
|
|
})
|
||
|
|
movies.append(temp)
|
||
|
|
|
||
|
|
if len(movies) >= limit:
|
||
|
|
break
|
||
|
|
|
||
|
|
return jsonified({
|
||
|
|
'success': True,
|
||
|
|
'empty': len(movies) == 0,
|
||
|
|
'movies': movies,
|
||
|
|
})
|