Changeset 1917

Timestamp:

11/15/04 16:54:03 (20 years ago)

Author:

jalet

Message:

Continued integration of Matt's patch for banners

Files:

• pykota/trunk/pykota/tool.py (modified) (3 diffs)

pykota/trunk/pykota/tool.py

@@ -22 +22 @@
 #
 # $Log$
+# Revision 1.139  2004/11/15 15:54:03  jalet
+# Continued integration of Matt's patch for banners
+#
 # Revision 1.138  2004/11/12 23:46:44  jalet
 # Heavy work on pkbanner. Not finished yet though, but mostly works.
@@ -513 +516 @@
 import md5
 import ConfigParser
+import popen2
+import cStringIO
 
 from mx import DateTime
@@ -1209 +1214 @@
             self.logdebug("Executing post-hook [%s]" % posthook)
             os.system(posthook)
+
+    def genBanner(self, bannerfileorcommand) :
+        """Reads a banner or generates one through an external command.
+        
+           Returns the banner's content in a format which MUST be accepted
+           by the printer.
+        """
+        banner = "" # no banner by default
+        if bannerfileorcommand :
+            if (os.access(bannerfileorcommand, os.X_OK)) :
+                self.logdebug("Launching %s to generate a banner." % bannerfileorcommand)
+                child = popen2.Popen3(bannerfileorcommand, capturestderr=1)
+                child.tochild.close()
+                child.childerr.close()
+                banner = child.fromchild.read()
+                child.fromchild.close()
+                status = child.wait()
+                if os.WIFEXITED(status) :
+                    status = os.WEXITSTATUS(status)
+                self.printInfo(_("Banner generator %s exit code is %s") % (bannerfileorcommand, str(status)))
+            else :
+                self.logdebug("Using %s as the banner." % bannerfileorcommand)
+                try :
+                    fh = open(bannerfileorcommand, 'r')
+                except IOError, msg :    
+                    self.printInfo("Impossible to open %s : %s" % (bannerfileorcommand, msg), "error")
+                else :    
+                    banner = fh.read()
+                    fh.close()
+        return cStringIO.StringIO(banner)
+    
+    def startingBanner(self, printername) :
+        """Retrieves a starting banner for current printer and returns its content."""
+        self.logdebug("Retrieving starting banner...")
+        return self.genBanner(self.config.getStartingBanner(printername))
+    
+    def endingBanner(self, printername) :
+        """Retrieves an ending banner for current printer and returns its content."""
+        self.logdebug("Retrieving ending banner...")
+        return self.genBanner(self.config.getEndingBanner(printername))
         
     def printInfo(self, message, level="info") :