Downloading of single releases and more

couchpotato/core/plugins/release/main.py

@@ -1,8 +1,10 @@
 from couchpotato import get_session
+from couchpotato.api import addApiView
 from couchpotato.core.event import fireEvent, addEvent
+from couchpotato.core.helpers.request import getParam, jsonified
 from couchpotato.core.logger import CPLog
 from couchpotato.core.plugins.base import Plugin
-from couchpotato.core.settings.model import File, Release, Movie
+from couchpotato.core.settings.model import File, Release as Relea, Movie
 from sqlalchemy.sql.expression import and_, or_
 
 log = CPLog(__name__)
@@ -13,6 +15,8 @@ class Release(Plugin):
     def __init__(self):
         addEvent('release.add', self.add)
 
+        addApiView('release.download', self.download)
+
     def add(self, group):
         db = get_session()
 
@@ -30,22 +34,22 @@ class Release(Plugin):
             db.add(movie)
             db.commit()
 
-        # Add release
+        # Add Release
         snatched_status = fireEvent('status.get', 'snatched', single = True)
-        release = db.query(Release).filter(
+        rel = db.query(Relea).filter(
             or_(
-                Release.identifier == identifier,
-                and_(Release.identifier.startswith(group['library']['identifier'], Release.status_id == snatched_status.get('id')))
+                Relea.identifier == identifier,
+                and_(Relea.identifier.startswith(group['library']['identifier'], Relea.status_id == snatched_status.get('id')))
             )
         ).first()
-        if not release:
-            release = Release(
+        if not rel:
+            rel = Relea(
                 identifier = identifier,
                 movie = movie,
                 quality_id = group['meta_data']['quality'].get('id'),
                 status_id = done_status.get('id')
             )
-            db.add(release)
+            db.add(rel)
             db.commit()
 
         # Add each file type
@@ -54,10 +58,10 @@ class Release(Plugin):
                 added_file = self.saveFile(file, type = type, include_media_info = type is 'movie')
                 try:
                     added_file = db.query(File).filter_by(id = added_file.get('id')).one()
-                    release.files.append(added_file)
+                    Relea.files.append(added_file)
                     db.commit()
                 except Exception, e:
-                    log.debug('Failed to attach "%s" to release: %s' % (file, e))
+                    log.debug('Failed to attach "%s" to Relea: %s' % (file, e))
 
         db.remove()
 
@@ -73,3 +77,34 @@ class Release(Plugin):
         # Check database and update/insert if necessary
         return fireEvent('file.add', path = file, part = self.getPartNumber(file), type = self.file_types[type], properties = properties, single = True)
 
+    def download(self):
+
+        db = get_session()
+        id = getParam('id')
+
+        rel = db.query(Relea).filter_by(id = id).first()
+        if rel:
+            item = {}
+            for info in rel.info:
+                item[info.identifier] = info.value
+
+            # Get matching provider
+            provider = fireEvent('provider.belongs_to', item['url'], single = True)
+            item['download'] = provider.download
+
+            fireEvent('searcher.download', data = item, movie = rel.movie.to_dict({
+                'profile': {'types': {'quality': {}}},
+                'releases': {'status': {}, 'quality': {}},
+                'library': {'titles': {}, 'files':{}},
+                'files': {}
+            }))
+
+            return jsonified({
+                'success': True
+            })
+        else:
+            log.error('Couldn\'t find release with id: %s' % id)
+
+        return jsonified({
+            'success': False
+        })