Changeset 196 for pkpgcounter/trunk/pdlanalyzer/escp2.py

Timestamp:

04/04/05 09:29:41 (20 years ago)

Author:

jerome

Message:

Added test functions

Files:

• pkpgcounter/trunk/pdlanalyzer/escp2.py (modified) (2 diffs)

pkpgcounter/trunk/pdlanalyzer/escp2.py

@@ -24 +24 @@
 import sys
 
-from pdlanalyzer.pdlparser import PDLParser
+from pdlanalyzer.pdlparser import PDLParser, PDLParserError
 
 class ESCP2Parser(PDLParser) :
@@ -65 +65 @@
 def test() :        
     """Test function."""
-    raise RuntimeError, "Not implemented !"
+    if (len(sys.argv) < 2) or ((not sys.stdin.isatty()) and ("-" not in sys.argv[1:])) :
+        sys.argv.append("-")
+    totalsize = 0    
+    for arg in sys.argv[1:] :
+        if arg == "-" :
+            infile = sys.stdin
+            mustclose = 0
+        else :    
+            infile = open(arg, "rb")
+            mustclose = 1
+        try :
+            parser = ESCP2Parser(infile, debug=1)
+            totalsize += parser.getJobSize()
+        except PDLParserError, msg :    
+            sys.stderr.write("ERROR: %s\n" % msg)
+            sys.stderr.flush()
+        if mustclose :    
+            infile.close()
+    print "%s" % totalsize
     
 if __name__ == "__main__" :