CouchPotatoServer/couchpotato/core/plugins/dashboard/main.py

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.orm import joinedload_all
import random
import time

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)

        return jsonified({
            'result': True,
            'suggestions': suggestions
        })

    def getSoonView(self):

        params = getParams()
        db = get_session()
        now = time.time()

        # 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
        active_status = fireEvent('status.get', 'active', single = True)
        subq = db.query(Movie).filter(Movie.status_id == active_status.get('id')).subquery()

        q = db.query(Movie).join((subq, subq.c.id == Movie.id)) \
            .options(joinedload_all('releases')) \
            .options(joinedload_all('profile.types')) \
            .options(joinedload_all('library.titles')) \
            .options(joinedload_all('library.files')) \
            .options(joinedload_all('status')) \
            .options(joinedload_all('files'))

        # 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', {}) or {}
            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': {},
                })

                # Don't list older movies
                if ((not params.get('late') and (not eta.get('dvd') or (eta.get('dvd') and eta.get('dvd') > (now - 2419200)))) or \
                        (params.get('late') and eta.get('dvd') and eta.get('dvd') < (now - 2419200))):
                    movies.append(temp)

                if len(movies) >= limit:
                    break

        return jsonified({
            'success': True,
            'empty': len(movies) == 0,
            'movies': movies,
        })

    getLateView = getSoonView