From ae038bc4164a0457f962b775b43d10cba8fb2e90 Mon Sep 17 00:00:00 2001
From: Ruud <ruud@crashdummy.nl>
Date: Sun, 29 Jan 2012 16:10:30 +0100
Subject: [PATCH] Mark movie as done when renamer is disabled

---
 couchpotato/core/plugins/searcher/main.py | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/couchpotato/core/plugins/searcher/main.py b/couchpotato/core/plugins/searcher/main.py
index ad570f6..36a2f6c 100644
--- a/couchpotato/core/plugins/searcher/main.py
+++ b/couchpotato/core/plugins/searcher/main.py
@@ -104,6 +104,7 @@ class Searcher(Plugin):
                     return self.download(data = nzb, movie = movie)
             else:
                 log.info('Better quality (%s) already available or snatched for %s' % (type['quality']['label'], default_title))
+                fireEvent('movie.restatus', movie['id'])
                 break
 
             # Break if CP wants to shut down
@@ -125,10 +126,33 @@ class Searcher(Plugin):
             rls.status_id = snatched_status.get('id')
             db.commit()
 
-            snatch_message = 'Snatched "%s": %s (%s) in %s' % (data.get('name'), movie['library']['titles'][0]['title'], movie['library']['year'], rls.quality.label)
+            log_movie = '%s (%s) in %s' % (movie['library']['titles'][0]['title'], movie['library']['year'], rls.quality.label)
+            snatch_message = 'Snatched "%s": %s' % (data.get('name'), log_movie)
             log.info(snatch_message)
             fireEvent('movie.snatched', message = snatch_message, data = rls.to_dict())
 
+
+            # If renamer isn't used, mark movie done
+            if not Env.setting('enabled', 'renamer'):
+                active_status = fireEvent('status.get', 'active', single = True)
+                done_status = fireEvent('status.get', 'done', single = True)
+                try:
+                    if movie['status_id'] == active_status.get('id'):
+                        for profile_type in movie['profile']['types']:
+                            if profile_type['quality_id'] == rls.quality.id and profile_type['finish']:
+                                log.info('Renamer disabled, marking movie as finished: %s' % log_movie)
+
+                                # Mark release done
+                                rls.status_id = done_status.get('id')
+                                db.commit()
+
+                                # Mark movie done
+                                mvie = db.query(Movie).filter_by(id = movie['id']).first()
+                                mvie.status_id = done_status.get('id')
+                                db.commit()
+                except Exception, e:
+                    log.error('Failed marking movie finished: %s %s' % (e, traceback.format_exc()))
+
             return True
 
         log.error('Tried to download, but none of the downloaders are enabled')