Changeset 55 for pykoticon

Show
Ignore:
Timestamp:
01/14/05 20:17:47 (19 years ago)
Author:
jerome
Message:

Transformed skeleton into a networking class

Files:
1 modified

Legend:

Unmodified
Added
Removed
  • pykoticon/trunk/bin/pykoticon

    r54 r55  
    3333USERNAME = "jerome" 
    3434 
    35 def getPrinterNames() : 
    36     """Retrieve the printer's names.""" 
    37     arguments = { "report" : 1, 
    38                   "format" : "csv", 
    39                   "datatype" : "printers", 
    40                 }   
    41     return retrieveDatas(arguments, ["printername"])["printername"] 
    42      
    43 def getUserInfo(username) : 
    44     """Retrieve the user's information.""" 
    45     arguments = { "datatype" : "users", 
    46                   "filter" : "username=%s" % username, 
    47                 }   
    48     return retrieveDatas(arguments, ("limitby", "balance", "lifetimepaid")) 
    49      
    50 def getUserPQuotas(username) : 
    51     """Retrieve the user's print quota information.""" 
    52     arguments = { "datatype" : "upquotas", 
    53                   "filter" : "username=%s" % username, 
    54                 }   
    55     return retrieveDatas(arguments, ("printername", "pagecounter", "softlimit", "datelimit")) 
    56      
    57 def retrieveDatas(args, fieldnames) : 
    58     """Retrieve datas from the CGI script.""" 
    59     arguments = { "report" : 1, 
    60                   "format" : "csv", 
    61                 }   
    62     arguments.update(args)             
    63     answer = {} 
    64     try :             
    65         url = "%s?%s" % (DUMPYKOTA_URL, urllib.urlencode(arguments)) 
    66         u = urllib2.urlopen(url) 
    67         lines = u.readlines() 
    68     except :     
    69         sys.stderr.write("Unable to retrieve %s\n" % url) 
    70     else :     
    71         u.close() 
    72         try : 
    73             lines = [ line.strip().split(",") for line in lines ] 
    74             fields = [field[1:-1] for field in lines[0]] 
    75             indices = [fields.index(fname) for fname in fieldnames] 
    76             answer = dict([ (fieldname, \ 
    77                             [ line[fields.index(fieldname)][1:-1] for line in lines[1:] ]) \ 
    78                             for fieldname in fieldnames ]) 
    79         except :     
    80             sys.stderr.write("Invalid datas retrieved from %s\n" % url) 
    81     return answer 
     35class CGINetworkInterface : 
     36    """A class for all network interactions.""" 
     37    def __init__(self, url, username=None, password=None) : 
     38        """Initialize CGI connection datas.""" 
     39        self.url = url 
     40        self.username = username 
     41        self.password = password 
     42         
     43    def retrieveDatas(self, arguments, fieldnames) : 
     44        """Retrieve datas from the CGI script.""" 
     45        args = { "report" : 1, 
     46                 "format" : "csv", 
     47               }   
     48        args.update(arguments)             
     49        answer = {} 
     50        try :             
     51            url = "%s?%s" % (self.url, urllib.urlencode(args)) 
     52            u = urllib2.urlopen(url) 
     53            lines = u.readlines() 
     54        except IOError, msg :     
     55            raise IOError, "Unable to retrieve %s : %s" % (url, msg) 
     56        else :     
     57            u.close() 
     58            try : 
     59                lines = [ line.strip().split(",") for line in lines ] 
     60                fields = [field[1:-1] for field in lines[0]] 
     61                indices = [fields.index(fname) for fname in fieldnames] 
     62                answer = dict([ (fieldname, \ 
     63                                  [ line[fields.index(fieldname)][1:-1] for line in lines[1:] ]) \ 
     64                                for fieldname in fieldnames ]) 
     65            except :     
     66                raise ValueError, "Invalid datas retrieved from %s" % url 
     67        return answer 
     68         
     69    def getPrinterNames(self) : 
     70        """Retrieve the printer's names.""" 
     71        arguments = { "report" : 1, 
     72                      "format" : "csv", 
     73                      "datatype" : "printers", 
     74                    }   
     75        return self.retrieveDatas(arguments, ["printername"])["printername"] 
     76         
     77    def getUserInfo(self, username) : 
     78        """Retrieve the user's information.""" 
     79        arguments = { "datatype" : "users", 
     80                      "filter" : "username=%s" % username, 
     81                    }   
     82        return self.retrieveDatas(arguments, ("limitby", "balance", "lifetimepaid")) 
     83         
     84    def getUserPQuotas(self, username) : 
     85        """Retrieve the user's print quota information.""" 
     86        arguments = { "datatype" : "upquotas", 
     87                      "filter" : "username=%s" % username, 
     88                    }   
     89        return self.retrieveDatas(arguments, ("printername", "pagecounter", "softlimit", "datelimit")) 
    8290     
    8391if __name__ == "__main__" : 
    84     #printernames = getPrinterNames() 
    85     #print "List of printers : ", printernames 
     92    net = CGINetworkInterface(DUMPYKOTA_URL) 
     93    print "List of printers : ", net.getPrinterNames() 
    8694     
    87     userinfo = getUserInfo("jerome") 
    88     print "User jerome : ", userinfo 
     95    print "User : ", net.getUserInfo("jerome") 
    8996     
    90     userpquotas = getUserPQuotas("jerome") 
    91     print "User jerome's print quotas : ", userpquotas 
     97    print "User print quotas : ", net.getUserPQuotas("jerome") 
    9298