Manage running plugins

couchpotato/core/_base/_core/main.py

@@ -35,7 +35,8 @@ class Core(Plugin):
 
             brk = True
             for running in still_running:
-                if running > 0:
+                if len(running) > 0:
+                    log.info('Waiting on plugins to finish: %s' % running)
                     brk = False
 
             if brk: break

couchpotato/core/event.py

@@ -19,9 +19,9 @@ def addEvent(name, handler):
 
         try:
             parent = handler.im_self
-            parent.beforeCall()
+            parent.beforeCall(handler)
             h = handler(*args, **kwargs)
-            parent.afterCall()
+            parent.afterCall(handler)
         except:
             h = handler(*args, **kwargs)
 

couchpotato/core/plugins/base.py

@@ -17,7 +17,7 @@ class Plugin(object):
     auto_register_static = True
 
     needs_shutdown = False
-    running = 0
+    running = []
 
     def registerPlugin(self):
         addEvent('app.shutdown', self.doShutdown)
@@ -67,11 +67,12 @@ class Plugin(object):
         except Exception, e:
             log.error('Unable to create folder "%s": %s' % (path, e))
 
-    def beforeCall(self):
-        self.isRunning(True)
+    def beforeCall(self, handler):
+        log.debug('Calling %s.%s' % (self.getName(), handler.__name__))
+        self.isRunning('%s.%s' % (self.getName(), handler.__name__))
 
-    def afterCall(self):
-        self.isRunning(False)
+    def afterCall(self, handler):
+        self.isRunning('%s.%s' % (self.getName(), handler.__name__), False)
 
     def doShutdown(self):
         self.shuttingDown(True)
@@ -82,12 +83,17 @@ class Plugin(object):
 
         self.needs_shutdown = value
 
-    def isRunning(self, value = None):
+    def isRunning(self, value = None, bool = True):
         if value is None:
             return self.running
 
-        log.debug('Running %s: %s' % (value, self.getName()))
-        self.running += 1 if value else -1
+        if bool:
+            self.running.append(value)
+        else:
+            try:
+                self.running.remove(value)
+            except:
+                log.error("Something went wrong when finishing the plugin function. Could not find the 'is_running' key")
 
     def isDisabled(self):
         return not self.isEnabled()