root / pykota / trunk / bin / autopykota @ 2512

Revision 2512, 5.8 kB (checked in by jerome, 19 years ago)

Ensure that human made errors (like incorrect command line options)
don't produce a traceback anymore. No need to frighten users with
such complete tracebacks and email reporting each time they mistype
some command.
Makes pykosd check more carefully the values of its command line options.

  • Property svn:eol-style set to native
  • Property svn:executable set to *
  • Property svn:keywords set to Author Date Id Revision
RevLine 
[1758]1#! /usr/bin/env python
2# -*- coding: ISO-8859-15 -*-
3
4# autopykota : script to automate user creation in PyKota
5#
6# PyKota - Print Quotas for CUPS and LPRng
7#
[2028]8# (c) 2003, 2004, 2005 Jerome Alet <alet@librelogiciel.com>
[1758]9# This program is free software; you can redistribute it and/or modify
10# it under the terms of the GNU General Public License as published by
11# the Free Software Foundation; either version 2 of the License, or
12# (at your option) any later version.
13#
14# This program is distributed in the hope that it will be useful,
15# but WITHOUT ANY WARRANTY; without even the implied warranty of
16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17# GNU General Public License for more details.
18#
19# You should have received a copy of the GNU General Public License
20# along with this program; if not, write to the Free Software
[2303]21# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
[1758]22#
23# $Id$
24#
[2028]25#
[1758]26
27import sys
28import os
29
[2512]30from pykota.tool import PyKotaTool, PyKotaToolError, PyKotaCommandLineError, crashed, N_
[1758]31
[2344]32__doc__ = N_("""autopykota v%(__version__)s (c) %(__years__)s %(__author__)s
[2267]33
[1758]34A tool to automate user account creation and initial balance setting.
35
36THIS TOOL MUST NOT BE USED IF YOU WANT TO LIMIT YOUR USERS BY PAGE QUOTA !
37
38command line usage :
39
40  THIS TOOL MUST NOT BE USED FROM THE COMMAND LINE BUT ONLY AS PART
41  OF AN external policy IN pykota.conf
42 
43  autopykota { -i | --initbalance value }
44
45options :
46
[1801]47  -v | --version       Prints autopykota's version number then exits.
[1758]48  -h | --help          Prints this message then exits.
49 
50  -i | --initbalance b Sets the user's account initial balance value to b.
51                       If the user already exists, actual balance is left
[1780]52                       unmodified. If unset, the default value is 0.
[2344]53""")
[1758]54
55class AutoPyKota(PyKotaTool) :
56    """A class for the automat."""
57    def main(self, arguments, options) :
58        """Main entry point."""
[1814]59        os.environ["PATH"] = "%s:/bin:/usr/bin:/usr/local/bin:/opt/bin:/sbin:/usr/sbin" % os.environ.get("PATH", "")
[1758]60        username = os.environ.get("PYKOTAUSERNAME")
61        printername = os.environ.get("PYKOTAPRINTERNAME")
62        if (username is None) or (printername is None) :
63            raise PyKotaToolError, "Either the username or the printername is undefined. Fatal Error."
64        else :
[2447]65            printer = self.storage.getPrinter(printername)
[1758]66            user = self.storage.getUser(username)
67            if user.Exists :
[1780]68                self.logdebug("User %s already exits." % username) 
[1758]69                if printer.Exists :
70                    userpquota = self.storage.getUserPQuota(user, printer)
71                    if userpquota.Exists :
[2447]72                        # we should never get there, unless autopykota was launched manually
[1758]73                        self.logdebug("User %s's quota entry on printer %s already exists. Nothing to do." % (username, printername))
74                        return 0
75                    else :   
[2447]76                        printersnames = [printer.Name] + [p.Name for p in self.storage.getParentPrinters(printer)]
77                        self.logdebug("Creating a quota entry for user %s on printers %s" % (username, printersnames))
78                        return os.system('edpykota --add --printer "%s" "%s"' % (','.join(printersnames), username))
[1758]79                else :
[1780]80                    self.logdebug("Printer %s doesn't exist. Creating printer %s and a quota entry for user %s on printer %s." % (printername, printername, username, printername))
[1759]81                    return os.system('edpykota --add --printer "%s" "%s"' % (printername, username))
[1758]82            else :
[2447]83                if printer.Exists :
84                    printersnames = [printer.Name] + [p.Name for p in self.storage.getParentPrinters(printer)]
85                else :   
86                    printersnames = [printer.Name]
[1780]87                self.logdebug("User %s doesn't exist yet." % username)
[2447]88                self.logdebug("Creating user %s's account with balance %s and quota entries on printers %s" % (username, options["initbalance"], printersnames))
89                return os.system('edpykota --add --limitby balance --balance "%s" --printer "%s" "%s"' % (options["initbalance"], ','.join(printersnames), username))
[1758]90   
91if __name__ == "__main__" :   
92    retcode = 0
93    try :
94        defaults = { \
[1780]95                     "initbalance" : 0.0
[1758]96                   }
97        short_options = "vhi:"
98        long_options = ["help", "version", "initbalance="]
99       
100        # Initializes the command line tool
101        automat = AutoPyKota(doc=__doc__)
[2210]102        automat.deferredInit()
[1758]103       
104        # parse and checks the command line
105        (options, args) = automat.parseCommandline(sys.argv[1:], short_options, long_options)
106       
107        # sets long options
108        options["help"] = options["h"] or options["help"]
109        options["version"] = options["v"] or options["version"]
[1780]110        options["initbalance"] = options["i"] or options["initbalance"] or defaults["initbalance"]
[1758]111       
112        if options["help"] :
113            automat.display_usage_and_quit()
114        elif options["version"] :
115            automat.display_version_and_quit()
116        elif args :   
[2512]117            raise PyKotaCommandLineError, "autopykota doesn't accept non option arguments !"
[1758]118        else :
119            retcode = automat.main(args, options)
[2216]120    except KeyboardInterrupt :       
121        sys.stderr.write("\nInterrupted with Ctrl+C !\n")
[2512]122    except PyKotaCommandLineError, msg :   
123        sys.stderr.write("%s : %s\n" % (sys.argv[0], msg))
[1758]124    except SystemExit :       
125        pass
126    except :
127        try :
128            automat.crashed("autopykota failed")
129        except :   
130            crashed("autopykota failed")
131        retcode = -1
132
133    try :
134        automat.storage.close()
135    except (TypeError, NameError, AttributeError) :   
136        pass
137       
138    sys.exit(retcode)   
Note: See TracBrowser for help on using the browser.