From a848a18d373dcecf9a8482e97dc3509561dc3718 Mon Sep 17 00:00:00 2001
From: Ruud <ruud@crashdummy.nl>
Date: Fri, 27 Jan 2012 21:19:00 +0100
Subject: [PATCH] Cleanup on terminate

---
 CouchPotato.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/CouchPotato.py b/CouchPotato.py
index f0ba0b4..707cfd9 100755
--- a/CouchPotato.py
+++ b/CouchPotato.py
@@ -1,8 +1,10 @@
 #!/usr/bin/env python
 from os.path import dirname
+from signal import signal, SIGTERM
+import atexit
 import os
-import sys
 import subprocess
+import sys
 import time
 
 
@@ -53,17 +55,21 @@ def main():
             if not restart:
                 break
 
+    sys.exit()
+
+def cleanup():
     from couchpotato.core.event import fireEvent
     fireEvent('app.crappy_shutdown', single = True)
     time.sleep(1)
-    sys.exit()
-
 
 if __name__ == '__main__':
 
+    signal(SIGTERM, lambda signum, stack_frame: sys.exit(1))
+
     if options.daemon and options.pid_file and not os.environ.get('cp_main'):
         from daemon import Daemon
         daemon = Daemon(options.pid_file)
         daemon.daemonize()
 
+    atexit.register(cleanup)
     main()