root / pykota / trunk / bin / pykota @ 825

#! /usr/bin/env python

# PyKota accounting filter
#
# PyKota - Print Quotas for CUPS
#
# (c) 2003 Jerome Alet <alet@librelogiciel.com>
# You're welcome to redistribute this software under the
# terms of the GNU General Public Licence version 2.0
# or, at your option, any higher version.
#
# You can read the complete GNU GPL in the file COPYING
# which should come along with this software, or visit
# the Free Software Foundation's WEB site http://www.fsf.org
#
# $Id$
#
# $Log$
# Revision 1.12  2003/02/27 23:48:41  jalet
# Correctly maps PyKota's log levels to syslog log levels
#
# Revision 1.11  2003/02/27 22:55:20  jalet
# WARN log priority doesn't exist.
#
# Revision 1.10  2003/02/27 22:43:21  jalet
# Missing import
#
# Revision 1.9  2003/02/27 22:40:26  jalet
# Correctly handles cases where the printer is off.
#
# Revision 1.8  2003/02/09 12:56:53  jalet
# Internationalization begins...
#
# Revision 1.7  2003/02/07 10:23:48  jalet
# Avoid a possible future name clash
#
# Revision 1.6  2003/02/06 22:54:33  jalet
# warnpykota should be ok
#
# Revision 1.5  2003/02/05 22:45:25  jalet
# Forgotten import
#
# Revision 1.4  2003/02/05 22:42:51  jalet
# Typo
#
# Revision 1.3  2003/02/05 22:38:39  jalet
# Typo
#
# Revision 1.2  2003/02/05 22:16:20  jalet
# DEVICE_URI is undefined outside of CUPS, i.e. for normal command line tools
#
# Revision 1.1  2003/02/05 21:28:17  jalet
# Initial import into CVS
#
#
#

import sys
import os

from pykota.tool import PyKotaTool, PyKotaToolError
from pykota.requester import openRequester, PyKotaRequesterError

class PyKotaFilter(PyKotaTool) :    
    """Class for the PyKota filter."""
    def __init__(self, username) :
        PyKotaTool.__init__(self, isfilter=1)
        self.username = username
        self.requester = openRequester(self.config, self.printername)
        self.printerhostname = self.getPrinterHostname()
    
    def getPrinterHostname(self) :
        """Returns the printer hostname."""
        device_uri = os.environ.get("DEVICE_URI", "")
        # TODO : check this for more complex urls than ipp://myprinter.dot.com:631/printers/lp
        try :
            (backend, destination) = device_uri.split(":", 1) 
        except ValueError :    
            raise PyKotaToolError, "Invalid DEVICE_URI : %s\n" % device_uri
        while destination.startswith("/") :
            destination = destination[1:]
        return destination.split("/")[0].split(":")[0]
        
    def filterInput(self, inputfile) :
        """Transparent filter."""
        mustclose = 0    
        if inputfile is not None :    
            infile = open(inputfile, "rb")
            mustclose = 1
        else :    
            infile = sys.stdin
        data = infile.read(256*1024)    
        while data :
            sys.stdout.write(data)
            data = infile.read(256*1024)
        if mustclose :    
            infile.close()
            
def main() :    
    """Do it, and do it right !"""
    #
    # This is a CUPS filter, so we should act and die like a CUPS filter when needed
    narg = len(sys.argv)
    if narg not in (6, 7) :    
        sys.stderr.write("ERROR: %s job-id user title copies options [file]\n" % sys.argv[0])
        return 1
    elif narg == 7 :    
        # input file
        inputfile = sys.argv[6]
    else :    
        # stdin
        inputfile = None
        
    #    
    # According to CUPS documentation, the username is the third command line argument
    username = sys.argv[2].strip()    
    
    # Initializes the current tool
    kotafilter = PyKotaFilter(username)    
    
    # Get the page counter directly from the printer itself
    try :
        counterbeforejob = kotafilter.requester.getPrinterPageCounter(kotafilter.printerhostname) # TODO use printername instead, make them match from CUPS' config files
    except PyKotaRequesterError, msg :
        # can't get actual page counter, assume printer is off, but warns in log
        kotafilter.logger.log_message(msg, "warn")
        printerIsOff = 1
    else :    
        printerIsOff = 0
        
    # Get the last page counter and last username from the Quota Storage backend
    pgc = kotafilter.storage.getPrinterPageCounter(kotafilter.printername)    
    if pgc is None :
        # The printer is unknown from the Quota Storage perspective
        # we let the job pass through, but log a warning message
        kotafilter.logger.log_message(_("Printer %s not registered in the PyKota system") % kotafilter.printername)
    else :    
        # get last values from Quota Storage
        (lastpagecounter, lastusername) = (pgc["pagecounter"], pgc["lastusername"])
        
        # if printer is off then we assume the correct counter value is the last one
        if printerIsOff :
            counterbeforejob = lastpagecounter
            
        # Update the last page counter and last username in the Quota Storage backend
        # set them to current user and 
        kotafilter.storage.updatePrinterPageCounter(kotafilter.printername, username, counterbeforejob) # TODO : allow or deny users not in quota system, and die cleanly if needed
        
        # Is the current user allowed to print at all ?
        action = kotafilter.warnUserPQuota(username)
        if action == "DENY" :
            # No, just die cleanly
            return 1
            
        # Yes     
        if (lastpagecounter is None) or (lastusername is None) :
            lastusername = username
            lastpagecounter = counterbeforejob
        jobsize = (counterbeforejob - lastpagecounter)    
        if jobsize >= 0:
            kotafilter.storage.updateUserPQuota(lastusername, kotafilter.printername, jobsize)
            kotafilter.warnUserPQuota(lastusername)
        else :    
            kotafilter.logger.log_message(_("Error in page count value %i for user %s on printer %s") % (jobsize, kotafilter.printername, lastusername), "error")
        
    # pass the job untouched to the underlying layer
    kotafilter.filterInput(inputfile)      
    
    return 0

if __name__ == "__main__" :    
    sys.exit(main() or 0)