Movie add

JSONP

couchpotato/core/plugins/file_browser/main.py

@@ -1,6 +1,5 @@
 from couchpotato.api import addApiView
-from couchpotato.environment import Env
-from flask.helpers import jsonify
+from couchpotato.core.helpers.request import getParam, jsonified
 import os
 import string
 
@@ -45,12 +44,12 @@ class FileBrowser():
     def view(self):
 
         try:
-            fb = FileBrowser(Env.getParam('path', '/'))
+            fb = FileBrowser(getParam('path', '/'))
             dirs = fb.getDirectories()
         except:
             dirs = []
 
-        return jsonify({
+        return jsonified({
             'empty': len(dirs) == 0,
             'dirs': dirs,
         })

couchpotato/core/plugins/movie_add/main.py

@@ -1,30 +1,36 @@
 from couchpotato.api import addApiView
-from couchpotato.core.event import getEvent, fireEvent
+from couchpotato.core.event import fireEvent
+from couchpotato.core.helpers.request import getParams, jsonified
 from couchpotato.core.plugins.base import Plugin
-from couchpotato.environment import Env
-from flask.helpers import jsonify
 
 class MovieAdd(Plugin):
 
     def __init__(self):
         addApiView('movie.add.search', self.search)
+        addApiView('movie.add.select', self.select)
 
     def search(self):
 
-        a = Env.getParams()
+        a = getParams()
 
-        print fireEvent('provider.movie.search', q = a.get('q'))
+        results = fireEvent('provider.movie.search', q = a.get('q'))
 
-        movies = [
-            {'id': 1, 'name': 'test'}
-        ]
+        # Combine movie results
+        movies = []
+        for r in results:
+            movies += r
 
-        return jsonify({
+        return jsonified({
             'success': True,
             'empty': len(movies) == 0,
             'movies': movies,
         })
 
-
     def select(self):
-        pass
+
+        a = getParams()
+
+        return jsonified({
+            'success': True,
+            'added': True,
+        })

couchpotato/core/plugins/movie_list/__init__.py

@@ -0,0 +1,6 @@
+from couchpotato.core.plugins.movie_list.main import MovieList
+
+def start():
+    return MovieList()
+
+config = []

couchpotato/core/plugins/movie_list/main.py

@@ -0,0 +1,41 @@
+from couchpotato import get_session
+from couchpotato.api import addApiView
+from couchpotato.core.helpers.request import getParams, jsonified
+from couchpotato.core.plugins.base import Plugin
+from couchpotato.core.settings.model import Movie, Release
+
+class MovieList(Plugin):
+
+    def __init__(self):
+        addApiView('movie.list', self.list)
+
+    def list(self):
+
+        a = getParams()
+
+        results = get_session().query(Movie).filter(
+                Movie.releases.any(
+                    Release.status.has(identifier = 'wanted')
+                )
+            ).all()
+
+        movies = []
+        for movie in results:
+            temp = {
+                'id': movie.id,
+                'name': movie.id,
+                'releases': [],
+            }
+            for release in movie.releases:
+                temp['releases'].append({
+                    'status': release.status.label,
+                    'quality': release.quality.label
+                })
+
+            movies.append(temp)
+
+        return jsonified({
+            'success': True,
+            'empty': len(movies) == 0,
+            'movies': movies,
+        })