Changeset 719 for pykota/trunk/bin/edpykota

Timestamp:

02/06/03 15:28:59 (21 years ago)

Author:

jalet

Message:

edpykota should be ok, minus some typos

Files:

• pykota/trunk/bin/edpykota (modified) (5 diffs)

pykota/trunk/bin/edpykota

@@ -17 +17 @@
 #
 # $Log$
+# Revision 1.4  2003/02/06 14:28:59  jalet
+# edpykota should be ok, minus some typos
+#
 # Revision 1.3  2003/02/06 10:47:21  jalet
 # Documentation string and command line options didn't match.
@@ -52 +55 @@
   -u | --users         Edit users print quotas, this is the default.
   
-  -P | --printer p     Edit quotas on printer p only.
+  -P | --printer p     Edit quotas on printer p only. Actually p can
+                       use wildcards characters to select only
+                       some printers. The default value is *, meaning
+                       all printers.
   
   -g | --groups        Edit groups print quotas instead of users.
@@ -105 +111 @@
 Please e-mail bugs to: %s""" % (version.__version__, version.__author__)
         
-def main(arguments, options) :
-    """Edit user or group quotas."""
-    pass
-    
+class EdPyKota(PyKotaTool) :        
+    """A class for edpykota."""
+    def main(self, names, options) :
+        """Edit user or group quotas."""
+        printernames = self.storage.getMatchingPrinters(options["printer"])
+        if not printernames :
+            raise PyKotaToolError, "There's no printer matching %s" % options["printer"]
+        softlimit = hardlimit = None    
+        if options["softlimit"] :
+            try :
+                softlimit = int(options["softlimit"].strip())
+            except ValueError :    
+                raise PyKotaToolError, "Invalid softlimit value %s." % options["softlimit"]
+        if options["hardlimit"] :
+            try :
+                hardlimit = int(options["hardlimit"].strip())
+            except ValueError :    
+                raise PyKotaToolError, "Invalid hardlimit value %s." % options["hardlimit"]
+        if hardlimit < softlimit :        
+            # error, exchange them
+            self.logger.log_message("Hard limit %i is less than soft limit %i, values will be exchanged." % (hardlimit, softlimit), "warn")
+            (softlimit, hardlimit) = (hardlimit, softlimit)
+        for printer in printernames :
+            if options["prototype"] :
+                if options["users"] :
+                    prototype = self.storage.getUserPQuota(options["prototype"], printer)
+                else :     
+                    prototype = self.storage.getGroupPQuota(options["prototype"], printer)
+                if prototype is None :
+                    self.logger.message("Prototype %s not found in Quota Storage for printer %s." % (options["prototype"], printer))
+                    continue    # skip this printer
+                else :    
+                    (softlimit, hardlimit) = (prototype["softlimit"], prototype["hardlimit"])
+            if (hardlimit is None) or (softlimit is None) :        
+                raise PyKotaToolError, "Both hard and soft limits must be set !  Aborting."
+            if hardlimit is None :    
+                hardlimit = softlimit
+                self.logger.message("Undefined hard limit set to %i on printer %s." % (hardlimit, printer))
+            if softlimit is None :    
+                softlimit = hardlimit
+                self.logger.message("Undefined soft limit set to %i on printer %s." % (softlimit, printer))
+            for name in names :
+                if options["users"] :
+                    quota = self.storage.getUserPQuota(name, printer)
+                else :
+                    quota = self.storage.getGroupPQuota(name, printer)
+                if quota is None :
+                    # not found
+                    if options["add"] :
+                        if options["users"] :
+                            self.storage.addUserPQuota(name, printer)
+                            quota = self.storage.getUserPQuota(name, printer)
+                        else :
+                            self.storage.addGroupPQuota(name, printer)
+                            quota = self.storage.getGroupPQuota(name, printer)
+                if quota is None :     
+                    self.logger.log_message("Quota not found for object %s on printer %s." % (name, printer))
+                else :    
+                    if options["users"] :
+                        self.storage.setUserPQuota(name, printer, softlimit, hardlimit)
+                    else :
+                        self.storage.setGroupPQuota(name, printer, softlimit, hardlimit)
+                     
 if __name__ == "__main__" : 
     try :
-        tool = PyKotaTool(doc=__doc__)
-        
         defaults = { \
                      "users"  : 1, \
                      "groups" : 0, \
+                     "printer" : "*", \
                    }
         short_options = "vhaugp:P:S:H:"
@@ -121 +185 @@
         
         # Initializes the command line tool
-        tool = PyKotaTool(doc=__doc__)
+        tool = EdPyKota(doc=__doc__)
         
         # parse and checks the command line
@@ -143 +207 @@
         elif options["users"] and options["groups"] :    
             raise PyKotaToolError, "edpykota: options --users and --groups are incompatible.\n"
-        elif options["groups"] : # TODO : add support for group quotas    
-            raise PyKotaToolError, "edpykota: group quotas are currently not implemented, sorry.\n"
+        elif (options["softlimit"] or options["hardlimit"]) and optiions["prototype"] :    
+            raise PyKotaToolError, "edpykota: options --softlimit or --hardlimit and --prototype are incompatible.\n"
+        elif options["groups"] :    
+            raise PyKotaToolError, "edpykota: options --groups is currently not implemented.\n"
         else :
-            apply(main, args, options)
+            sys.exit(apply(tool.main, args, options))
     except PyKotaToolError, msg :            
         sys.stderr.write("%s\n" % msg)