#! /usr/bin/python # -*- coding: ISO-8859-15 -*- # PyKota Print Quota Reports generator # # PyKota - Print Quotas for CUPS and LPRng # # (c) 2003, 2004, 2005, 2006, 2007 Jerome Alet # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. # # $Id$ # # import sys import os import cgi import urllib from pykota import version from pykota.tool import PyKotaToolError from pykota.dumper import DumPyKota from pykota.cgifuncs import getLanguagePreference, getCharsetPreference header = """Content-type: text/html;charset=%s %s

PyKota's Logo
PyKota v%s

%s

%s

""" footer = """


%s © %s %s

%s
        

""" class PyKotaDumperGUI(DumPyKota) : """PyKota Dumper GUI""" def guiDisplay(self) : """Displays the dumper interface.""" global header, footer print header % (self.charset, _("PyKota Data Dumper"), \ self.language, self.charset, \ self.config.getLogoLink(), \ self.config.getLogoURL(), version.__version__, \ self.config.getLogoLink(), \ version.__version__, _("PyKota Data Dumper"), \ _("Dump"), _("Please click on the above button")) print self.htmlListDataTypes(self.options.get("data", "")) print "
" print self.htmlListFormats(self.options.get("format", "")) print "
" print self.htmlFilterInput(" ".join(self.arguments)) print "
" print self.htmlSumCheckbox(self.options.get("sum", "")) print footer % (_("Dump"), version.__doc__, version.__years__, version.__author__, version.__gplblurb__) def htmlListDataTypes(self, selected="") : """Displays the datatype selection list.""" message = '
%s :
' return message def htmlListFormats(self, selected="") : """Displays the formats selection list.""" message = '
%s :
' return message def htmlFilterInput(self, value="") : """Input the optional dump filter.""" return _("Filter") + (' : e.g. username=jerome printername=HP2100 start=today-30' % (value or "")) def htmlSumCheckbox(self, checked="") : """Input the optional Sum option.""" return _("Summarize") + (' : %s' % ((checked and 'checked="checked"'), _("only for payments or history"))) def guiAction(self) : """Main function""" try : wantreport = self.form.has_key("report") except TypeError : pass # WebDAV request probably, seen when trying to open a csv file in OOo else : if wantreport : try : if self.form.has_key("datatype") : self.options["data"] = self.form["datatype"].value if self.form.has_key("format") : self.options["format"] = self.form["format"].value if self.form.has_key("filter") : self.arguments = self.form["filter"].value.split() if self.form.has_key("sum") : self.options["sum"] = self.form["sum"].value # when no authentication is done, or when the remote username # is 'root' (even if not run as root of course), then unrestricted # dump is allowed. remuser = os.environ.get("REMOTE_USER", "root") # special hack to accomodate mod_auth_ldap Apache module try : remuser = remuser.split("=")[1].split(",")[0] except IndexError : pass if remuser != "root" : # non-'root' users when the script is password protected # can not dump any data as they like, we restrict them # to their own datas. if self.options["data"] not in ["printers", "pmembers", "groups", "gpquotas"] : self.arguments.append("username=%s" % remuser) fname = "error" ctype = "text/plain" if self.options["format"] in ("csv", "ssv") : #ctype = "application/vnd.sun.xml.calc" # OpenOffice.org ctype = "text/comma-separated-values" fname = "dump.csv" elif self.options["format"] == "tsv" : #ctype = "application/vnd.sun.xml.calc" # OpenOffice.org ctype = "text/tab-separated-values" fname = "dump.tsv" elif self.options["format"] == "xml" : ctype = "text/xml" fname = "dump.xml" elif self.options["format"] == "cups" : ctype = "text/plain" fname = "page_log" print "Content-type: %s" % ctype print "Content-disposition: attachment; filename=%s" % fname print self.main(self.arguments, self.options, restricted=0) except : print 'Content-type: text/html\n\nCGI Error

%s

' % self.crashed("CGI Error").replace("\n", "
") else : self.guiDisplay() if __name__ == "__main__" : admin = PyKotaDumperGUI(lang=getLanguagePreference(), charset=getCharsetPreference()) admin.deferredInit() admin.form = cgi.FieldStorage() admin.options = { "output" : "-", "data" : "history", "format" : "cups", "sum" : None, } admin.arguments = [] admin.guiAction() try : admin.storage.close() except (TypeError, NameError, AttributeError) : pass sys.exit(0)