Add ProcessManager.restart() method to block until the restart is done.

[software/pymin.git] / pymin / procman.py

diff --git a/pymin/procman.py b/pymin/procman.py
index 9242d86723f5ae2a2562aaaf148986797ffc0a66..5cea4d4e1760989126cb77d94483cc493b551e45 100644 (file)
--- a/pymin/procman.py
+++ b/pymin/procman.py
@@ -84,7 +84,16 @@ class ProcessManager:
 
     def stop(self, name):
         assert name in self.namemap
-        self.namemap[name].stop(name)
+        self.namemap[name].stop()
+
+    def restart(self, name):
+        logging.debug(u'ProcessManager.restart(%s)', name)
+        if name in self.namemap:
+            self.namemap[name].stop()
+            self.namemap[name].wait()
+            self.namemap[name].restart()
+        else:
+            self.namemap[name].start()
 
     def kill(self, name, signum):
         assert name in self.namemap
@@ -131,7 +140,7 @@ class ProcessManager:
         else: # is a pid
             if name in self.pidmap:
                 return self.pidmap[name]
-        return KeyError, name
+        raise KeyError, name
 
     def has(self, name):
         if isinstance(name, basestring): # is a name