#! /usr/bin/python
# -*- coding: UTF-8 -*-
# PyKota Print Quota Reports generator
#
# PyKota - Print Quotas for CUPS
#
# (c) 2003, 2004, 2005, 2006, 2007, 2008 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 3 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, see .
#
# $Id$
#
#
import sys
import os
import cgi
import urllib
from xml.sax import saxutils
import pykota.appinit
from pykota import version, utils
from pykota.dumper import DumPyKota
header = """Content-type: text/html;charset=%s
%s
%s
© %s %s
%s
"""
class PyKotaDumperGUI(DumPyKota) :
"""PyKota Dumper GUI"""
def guiDisplay(self) :
"""Displays the dumper interface."""
global header, footer
content = [ header % (self.charset, _("PyKota Data Dumper"), \
self.config.getLogoLink(), \
self.config.getLogoURL(), version.__version__, \
self.config.getLogoLink(), \
version.__version__, _("PyKota Data Dumper"), \
_("Dump"), _("Please click on the above button")) ]
content.append(self.htmlListDataTypes(self.options.get("data", "")))
content.append(u" ")
content.append(self.htmlListFormats(self.options.get("format", "")))
content.append(u" ")
content.append(self.htmlFilterInput(" ".join(self.arguments)))
content.append(u" ")
content.append(self.htmlOrderbyInput(self.options.get("orderby", "")))
content.append(u" ")
content.append(self.htmlSumCheckbox(self.options.get("sum", "")))
content.append(footer % (_("Dump"),
version.__doc__,
version.__years__,
version.__author__,
saxutils.escape(version.__gplblurb__)))
for c in content :
sys.stdout.write(c.encode(self.charset, "replace"))
sys.stdout.flush()
def htmlListDataTypes(self, selected="") :
"""Displays the datatype selection list."""
message = '%s : ' % _("Data Type")
for dt in self.validdatatypes.items() :
if dt[0] == selected :
message += '%s (%s) ' % (dt[0], dt[0], dt[1])
else :
message += '%s (%s) ' % (dt[0], dt[0], dt[1])
message += '
'
return message
def htmlListFormats(self, selected="") :
"""Displays the formats selection list."""
message = '%s : ' % _("Output Format")
for fmt in self.validformats.items() :
if fmt[0] == selected :
message += '%s (%s) ' % (fmt[0], fmt[0], fmt[1])
else :
message += '%s (%s) ' % (fmt[0], fmt[0], fmt[1])
message += '
'
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 htmlOrderbyInput(self, value="") :
"""Input the optional ordering."""
return _("Ordering") + (' : e.g. +username,-printername ' % (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
if self.form.has_key("orderby") :
self.options["orderby"] = self.form["orderby"].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__" :
utils.reinitcgilocale()
admin = PyKotaDumperGUI()
admin.deferredInit()
admin.form = cgi.FieldStorage()
admin.options = { "output" : "-",
"data" : "history",
"format" : "cups",
"sum" : None,
"orderby" : None,
}
admin.arguments = []
admin.guiAction()
try :
admin.storage.close()
except (TypeError, NameError, AttributeError) :
pass
sys.exit(0)