Changeset 639 for tea4cups

Timestamp:

06/07/05 10:58:06 (19 years ago)

Author:

jerome

Message:

Integrated Peter Stuge's second patch

Files:

• tea4cups/trunk/tea4cups (modified) (8 diffs)

tea4cups/trunk/tea4cups

@@ -5 +5 @@
 #
 # (c) 2005 Jerome Alet <alet@librelogiciel.com>
+# (c) 2005 Peter Stuge <stuge-tea4cups@cdy.org>
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -38 +39 @@
 from struct import unpack
 
-version = "2.12alpha2_unofficial"
+version = "2.12alpha3_unofficial"
 
 class TeeError(Exception):
@@ -320 +321 @@
         branchbasename = "%s_" % branchtype.lower()
         try :
-            # globalbranches = [ (k, v) for (k, v) in self.config.items("global") if k.startswith(branchbasename) ]
             globalbranches = [ (k, self.config.get("global", k)) for k in self.config.options("global") if k.startswith(branchbasename) ]
         except ConfigParser.NoSectionError, msg :    
             raise ConfigError, "Invalid configuration file : %s" % msg
         try :
-            # sectionbranches = [ (k, v) for (k, v) in self.config.items(printqueuename) if k.startswith(branchbasename) ]
             sectionbranches = [ (k, self.config.get(printqueuename, k)) for k in self.config.options(printqueuename) if k.startswith(branchbasename) ]
         except ConfigParser.NoSectionError, msg :    
@@ -581 +580 @@
         signal.signal(signal.SIGTERM, self.sigtermHandler)
         serialize = self.isTrue(self.getPrintQueueOption(self.PrinterName, "serialize", ignore=1))
-        self.pipe = os.pipe()
+        self.pipes = { 0: os.pipe() }
         for branchtype in ["prehook", "tee", "posthook"] :
             if branchtype == "posthook" :
-                os.close(self.pipe[1])
+                os.close(self.pipes[0][1])
             branches = self.enumBranches(self.PrinterName, branchtype)
+            for b in branches :
+                bname = b.split("_", 1)[1]
+                if branchtype != "posthook" and bname not in self.pipes :
+                    self.pipes[bname] = os.pipe()
+                else :
+                    if bname in self.pipes :
+                        os.close(self.pipes[bname][1])
             status = self.runCommands(branchtype, branches, serialize)
             if status :
@@ -596 +602 @@
                 break # We don't want to execute tees or posthooks in this case
         signal.signal(signal.SIGTERM, signal.SIG_IGN)
-        try :
-            os.close(self.pipe[1])
-        except OSError :
-            pass
-        os.close(self.pipe[0])
+        for p in self.pipes :
+            os.close(self.pipes[p][0])
+            try :
+                os.close(self.pipes[p][1])
+            except OSError :
+                pass
         if not exitcode :
             self.logInfo("OK")
@@ -607 +614 @@
         return exitcode
         
-    def pipedSystem(self, cmd, stdin=0, stdout=1) :
-        """Launches a command making a pipe available to it."""
-        # Code contributed by Peter Stuge on May 23rd 2005
+    def stdioRedirSystem(self, cmd, stdin=0, stdout=1) :
+        """Launches a command with stdio redirected."""
+        # Code contributed by Peter Stuge on May 23rd and June 7th 2005
         pid = os.fork()
         if pid == 0 :
-            os.dup2(stdin, 0)
-            os.dup2(stdout, 1)
-            os.execl("/bin/sh", "sh", "-c", cmd)
-            os.exit(1)
+            if stdin != 0 :
+                os.dup2(stdin, 0)
+                os.close(stdin)
+            if stdout != 1 :
+                os.dup2(stdout, 1)
+                os.close(stdout)
+            try :
+                os.execl("/bin/sh", "sh", "-c", cmd)
+            except OSError , msg :
+                self.logDebug("execl() failed: %s" % msg)
+            os._exit(1)
         return os.waitpid(pid, 0)[1]
         
+    def runCommand(self, branch, command) :
+        """Runs a particular branch command."""
+        # Code contributed by Peter Stuge on June 7th 2005
+        self.logDebug("Launching %s : %s" % (branch, command))
+        btype, bname = branch.split("_", 1)
+        if bname not in self.pipes :
+          bname = 0
+        if btype != "posthook" :
+            return self.stdioRedirSystem(command, 0, self.pipes[bname][1])
+        else :
+            return self.stdioRedirSystem(command, self.pipes[bname][0])
+
     def runCommands(self, btype, branches, serialize) :    
         """Runs the commands for a particular branch type."""
@@ -638 +664 @@
                 if self.gotSigTerm :
                     break
-                self.logDebug("Launching %s : %s" % (branch, command))
-                if btype != "posthook" :
-                    retcode = self.pipedSystem(command, 0, self.pipe[1])
-                else :
-                    retcode = self.pipedSystem(command, self.pipe[0])
+                retcode = self.runCommand(branch, command)
                 self.logDebug("Exit code for %s %s on printer %s is %s" % (btype, branch, self.PrinterName, retcode))
                 if os.WIFEXITED(retcode) :
@@ -672 +694 @@
                         sys.exit(self.runOriginalBackend())
                     else :
-                        self.logDebug("Launching %s : %s" % (branch, command))
-                        retcode = os.system(command)
+                        retcode = self.runCommand(branch, command)
                         if os.WIFEXITED(retcode) :
                             retcode = os.WEXITSTATUS(retcode)