From 24e45b0b345e830c25d68e4bdb5495e3fa5b7a8a Mon Sep 17 00:00:00 2001 From: Ruud Date: Mon, 16 Apr 2012 00:20:39 +0200 Subject: [PATCH] Movie delete event --- couchpotato/core/plugins/movie/main.py | 22 ++++++++++++++-------- 1 file changed, 14 insertions(+), 8 deletions(-) diff --git a/couchpotato/core/plugins/movie/main.py b/couchpotato/core/plugins/movie/main.py index 354e64a..073e2fc 100644 --- a/couchpotato/core/plugins/movie/main.py +++ b/couchpotato/core/plugins/movie/main.py @@ -74,7 +74,7 @@ class MoviePlugin(Plugin): 'default_title': {'desc': 'Movie title to use for searches. Has to be one of the titles returned by movie.search.'}, } }) - addApiView('movie.delete', self.delete, docs = { + addApiView('movie.delete', self.deleteView, docs = { 'desc': 'Delete a movie from the wanted list', 'params': { 'id': {'desc': 'Movie ID(s) you want to delete.', 'type': 'int (comma separated)'}, @@ -82,6 +82,7 @@ class MoviePlugin(Plugin): }) addEvent('movie.add', self.add) + addEvent('movie.delete', self.delete) addEvent('movie.get', self.get) addEvent('movie.list', self.list) addEvent('movie.restatus', self.restatus) @@ -347,23 +348,28 @@ class MoviePlugin(Plugin): 'success': True, }) - def delete(self): + def deleteView(self): params = getParams() - db = get_session() - - status = fireEvent('status.add', 'deleted', single = True) ids = params.get('id').split(',') for movie_id in ids: - movie = db.query(Movie).filter_by(id = movie_id).first() - movie.status_id = status.get('id') - db.commit() + self.delete(movie_id) return jsonified({ 'success': True, }) + def delete(self, movie_id): + + db = get_session() + + movie = db.query(Movie).filter_by(id = movie_id).first() + db.delete(movie) + db.commit() + + return True + def restatus(self, movie_id): active_status = fireEvent('status.get', 'active', single = True)