[!] Fix post migration changes.

werwolfby · Jul 26, 2015 · 67d7a8a · 67d7a8a
1 parent 99c300d
commit 67d7a8a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/plugins/trackers/lostfilm.py b/plugins/trackers/lostfilm.py
@@ -275,7 +275,11 @@ def check_connection(self):
 
     def remove_watch(self, url):
         with DBSession() as db:
-            return db.query(LostFilmTVSeries).filter(LostFilmTVSeries.url == url).delete()
+            topic = db.query(LostFilmTVSeries).filter(LostFilmTVSeries.url == url).first()
+            if topic is None:
+                return False
+            db.delete(topic)
+            return True
 
     def get_watching_torrents(self):
         with DBSession() as db:

diff --git a/plugins/trackers/rutor.py b/plugins/trackers/rutor.py
@@ -119,15 +119,19 @@ def add_watch(self, url, display_name=None):
         if not display_name:
             display_name = title['original_name']
         hash = self.tracker.get_hash(url)
-        entry = RutorOrgTopic(name=display_name, url=url, hash=hash)
+        entry = RutorOrgTopic(display_name=display_name, url=url, hash=hash)
         with DBSession() as db:
             db.add(entry)
             db.commit()
             return entry.id
 
     def remove_watch(self, url):
         with DBSession() as db:
-            return db.query(RutorOrgTopic).filter(RutorOrgTopic.url == url).delete()
+            topic = db.query(RutorOrgTopic).filter(RutorOrgTopic.url == url).first()
+            if topic is None:
+                return False
+            db.delete(topic)
+            return True
 
     def get_watching_torrents(self):
         with DBSession() as db: