Changeset 1041 for pykota/trunk/bin/warnpykota

Timestamp:

06/25/03 16:10:01 (21 years ago)

Author:

jalet

Message:

Hey, it may work (edpykota --reset excepted) !

Files:

• pykota/trunk/bin/warnpykota (modified) (6 diffs)

pykota/trunk/bin/warnpykota

@@ -23 +23 @@
 #
 # $Log$
+# Revision 1.20  2003/06/25 14:10:01  jalet
+# Hey, it may work (edpykota --reset excepted) !
+#
 # Revision 1.19  2003/04/29 22:03:38  jalet
 # Better error handling.
@@ -84 +87 @@
 
 import sys
+import os
+import pwd
+import grp
 
 from pykota import version
@@ -96 +102 @@
 command line usage :
 
-  warnpykota [options] 
+  warnpykota  [options]  [names]
 
 options :
@@ -125 +131 @@
   any printer.
 
-  $ warnpykota --groups --printer "laserjet*"
-  
-  This will warn all users of groups which have exceeded 
-  their print quota on any printer which name begins with "laserjet"
+  $ warnpykota --groups --printer "laserjet*" "dev*"
+  
+  This will warn all users of groups which names begins with "dev" and
+  who have exceeded their print quota on any printer which name begins 
+  with "laserjet"
+  
+  If launched by a non-root user, additionnal arguments representing
+  users or groups names are ignored, and only the current user/group
+  is warned.
 
 This program is free software; you can redistribute it and/or modify
@@ -148 +159 @@
 class WarnPyKota(PyKotaTool) :        
     """A class for warnpykota."""
-    def main(self, options) :
+    def main(self, ugnames, options) :
         """Warn users or groups over print quota."""
+        uid = os.geteuid()
+        if not uid :
+            # root user
+            if not ugnames :
+                # no username, means all usernames
+                ugnames = [ "*" ]
+        else :        
+            # not the root user
+            # warns only the current user
+            # the utility of this is discutable, but at least it
+            # protects other users from mail bombing if they are
+            # over quota.
+            if options["groups"] :
+                ugnames = [ grp.getgrgid(pwd.getpwuid(uid)[3])[0] ]
+            else :
+                ugnames = [ pwd.getpwuid(uid)[0] ]
+        
         printers = self.storage.getMatchingPrinters(options["printer"])
         if not printers :
             raise PyKotaToolError, _("There's no printer matching %s") % options["printer"]
-        for (printerid, printer) in printers :
+        for printer in printers :
             if options["groups"] :
-                for (ident, name) in self.storage.getPrinterGroups(printerid) :
-                    self.warnGroupPQuota(name, printer)
+                for (group, grouppquota) in self.storage.getPrinterGroupsAndQuotas(printer, ugnames) :
+                    self.warnGroupPQuota(grouppquota)
             else :
-                for (ident, name) in self.storage.getPrinterUsers(printerid) :
-                    self.warnUserPQuota(name, printer)
+                for (user, userpquota) in self.storage.getPrinterUsersAndQuotas(printer, ugnames) :
+                    self.warnUserPQuota(userpquota)
                      
 if __name__ == "__main__" : 
@@ -188 +216 @@
         elif options["users"] and options["groups"] :    
             raise PyKotaToolError, _("incompatible options, see help.")
-        elif args :    
-            raise PyKotaToolError, _("unused arguments [%s]. Aborting.") % ", ".join(args)
         else :
-            sys.exit(sender.main(options))
+            sys.exit(sender.main(args, options))
     except (PyKotaToolError, PyKotaConfigError, PyKotaStorageError), msg :            
         sys.stderr.write("%s\n" % msg)