Changeset 2506 for pykota/trunk/pykota/accounters/snmp.py

Timestamp:

09/26/05 10:47:48 (19 years ago)

Author:

jerome

Message:

Moved test code into a function.

Files:

• pykota/trunk/pykota/accounters/snmp.py (modified) (2 diffs)

pykota/trunk/pykota/accounters/snmp.py

@@ -187 +187 @@
             return self.printerInternalPageCounter
             
+def main(hostname) :
+    """Tries SNMP accounting for a printer host."""
+    class fakeFilter :
+        def __init__(self) :
+            self.PrinterName = "FakePrintQueue"
+            self.JobSizeBytes = 1
+            
+        def printInfo(self, msg, level="info") :
+            sys.stderr.write("%s : %s\n" % (level.upper(), msg))
+            sys.stderr.flush()
+            
+        def logdebug(self, msg) :    
+            self.printInfo(msg, "debug")
+            
+    class fakeAccounter :        
+        def __init__(self) :
+            self.arguments = "snmp:public"
+            self.filter = fakeFilter()
+            self.protocolHandler = Handler(self, hostname)
+        
+    acc = fakeAccounter()            
+    return acc.protocolHandler.retrieveInternalPageCounter()
+        
 if __name__ == "__main__" :            
     if len(sys.argv) != 2 :    
@@ -194 +217 @@
             return msg
             
-        class fakeFilter :
-            def __init__(self) :
-                self.PrinterName = "FakePrintQueue"
-                self.JobSizeBytes = 1
-                
-            def printInfo(self, msg, level="info") :
-                sys.stderr.write("%s : %s\n" % (level.upper(), msg))
-                sys.stderr.flush()
-                
-            def logdebug(self, msg) :    
-                self.printInfo(msg, "debug")
-                
-        class fakeAccounter :        
-            def __init__(self) :
-                self.arguments = "snmp:public"
-                self.filter = fakeFilter()
-                self.protocolHandler = Handler(self, sys.argv[1])
-            
-        acc = fakeAccounter()            
-        print "Internal page counter's value is : %s" % acc.protocolHandler.retrieveInternalPageCounter()
+        pagecounter = main(sys.argv[1])
+        print "Internal page counter's value is : %s" % pagecounter