Ruud
12 years ago
17 changed files with 654 additions and 219 deletions
@ -0,0 +1,6 @@ |
|||||
|
from .main import Dashboard |
||||
|
|
||||
|
def start(): |
||||
|
return Dashboard() |
||||
|
|
||||
|
config = [] |
||||
@ -0,0 +1,120 @@ |
|||||
|
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, |
||||
|
}) |
||||
@ -0,0 +1,91 @@ |
|||||
|
Page.Home = new Class({ |
||||
|
|
||||
|
Extends: PageBase, |
||||
|
|
||||
|
name: 'home', |
||||
|
title: 'Manage new stuff for things and such', |
||||
|
|
||||
|
indexAction: function(param){ |
||||
|
var self = this; |
||||
|
|
||||
|
if(self.soon_list) |
||||
|
return |
||||
|
|
||||
|
// Snatched
|
||||
|
self.available_list = new MovieList({ |
||||
|
'navigation': false, |
||||
|
'identifier': 'snatched', |
||||
|
'load_more': false, |
||||
|
'view': 'list', |
||||
|
'actions': MovieActions, |
||||
|
'title': 'Snatched & Available', |
||||
|
'filter': { |
||||
|
'release_status': 'snatched,available' |
||||
|
} |
||||
|
}); |
||||
|
|
||||
|
// Downloaded
|
||||
|
// self.downloaded_list = new MovieList({
|
||||
|
// 'navigation': false,
|
||||
|
// 'identifier': 'downloaded',
|
||||
|
// 'load_more': false,
|
||||
|
// 'view': 'titles',
|
||||
|
// 'filter': {
|
||||
|
// 'release_status': 'done',
|
||||
|
// 'order': 'release_order'
|
||||
|
// }
|
||||
|
// });
|
||||
|
// self.el.adopt(
|
||||
|
// new Element('h2', {
|
||||
|
// 'text': 'Just downloaded'
|
||||
|
// }),
|
||||
|
// $(self.downloaded_list)
|
||||
|
// );
|
||||
|
|
||||
|
// Comming Soon
|
||||
|
self.soon_list = new MovieList({ |
||||
|
'navigation': false, |
||||
|
'identifier': 'soon', |
||||
|
'limit': 24, |
||||
|
'title': 'Soon', |
||||
|
'filter': { |
||||
|
'random': true |
||||
|
}, |
||||
|
'load_more': false, |
||||
|
'view': 'thumbs', |
||||
|
'api_call': 'dashboard.soon' |
||||
|
}); |
||||
|
|
||||
|
self.el.adopt( |
||||
|
$(self.available_list), |
||||
|
$(self.soon_list) |
||||
|
); |
||||
|
|
||||
|
// Suggest
|
||||
|
// self.suggestion_list = new MovieList({
|
||||
|
// 'navigation': false,
|
||||
|
// 'identifier': 'suggestions',
|
||||
|
// 'limit': 6,
|
||||
|
// 'load_more': false,
|
||||
|
// 'view': 'thumbs',
|
||||
|
// 'api_call': 'suggestion.suggest'
|
||||
|
// });
|
||||
|
// self.el.adopt(
|
||||
|
// new Element('h2', {
|
||||
|
// 'text': 'You might like'
|
||||
|
// }),
|
||||
|
// $(self.suggestion_list)
|
||||
|
// );
|
||||
|
|
||||
|
// Recent
|
||||
|
// Snatched
|
||||
|
// Renamed
|
||||
|
// Added
|
||||
|
|
||||
|
// Free space
|
||||
|
|
||||
|
// Shortcuts
|
||||
|
|
||||
|
} |
||||
|
|
||||
|
}) |
||||
Loading…
Reference in new issue