root / pykota / trunk / pykota / accounter.py @ 2994

Revision 2947, 4.1 kB (checked in by jerome, 18 years ago)

Fixed path problem.

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
Line 
1# PyKota
2# -*- coding: ISO-8859-15 -*-
3#
4# PyKota : Print Quotas for CUPS and LPRng
5#
6# (c) 2003, 2004, 2005, 2006 Jerome Alet <alet@librelogiciel.com>
7# This program is free software; you can redistribute it and/or modify
8# it under the terms of the GNU General Public License as published by
9# the Free Software Foundation; either version 2 of the License, or
10# (at your option) any later version.
11#
12# This program is distributed in the hope that it will be useful,
13# but WITHOUT ANY WARRANTY; without even the implied warranty of
14# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15# GNU General Public License for more details.
16#
17# You should have received a copy of the GNU General Public License
18# along with this program; if not, write to the Free Software
19# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
20#
21# $Id$
22#
23#
24
25import sys
26import os
27import imp
28
29class PyKotaAccounterError(Exception):
30    """An exception for Accounter related stuff."""
31    def __init__(self, message = ""):
32        self.message = message
33        Exception.__init__(self, message)
34    def __repr__(self):
35        return self.message
36    __str__ = __repr__
37   
38class AccounterBase :   
39    """A class to account print usage by querying printers."""
40    def __init__(self, kotafilter, arguments, ispreaccounter=0) :
41        """Sets instance vars depending on the current printer."""
42        self.filter = kotafilter
43        self.arguments = arguments
44        self.onerror = self.filter.config.getPrinterOnAccounterError(self.filter.PrinterName)
45        self.isSoftware = 1 # by default software accounting
46        self.isPreAccounter = ispreaccounter 
47       
48    def getLastPageCounter(self) :   
49        """Returns last internal page counter value (possibly faked)."""
50        try :
51            return self.LastPageCounter or 0
52        except :   
53            return 0
54           
55    def beginJob(self, printer) :   
56        """Saves the computed job size."""
57        # computes job's size
58        self.JobSize = self.computeJobSize()
59        if self.filter.InputFile is not None :
60            self.JobSize *= self.filter.Copies
61       
62        # get last job information for this printer
63        if not self.isPreAccounter :
64            # TODO : check if this code is still needed
65            if not printer.LastJob.Exists :
66                # The printer hasn't been used yet, from PyKota's point of view
67                self.LastPageCounter = 0
68            else :   
69                # get last job size and page counter from Quota Storage
70                # Last lifetime page counter before actual job is
71                # last page counter + last job size
72                self.LastPageCounter = int(printer.LastJob.PrinterPageCounter or 0) + int(printer.LastJob.JobSize or 0)
73       
74    def fakeBeginJob(self) :   
75        """Do nothing."""
76        pass
77       
78    def endJob(self, printer) :   
79        """Do nothing."""
80        pass
81       
82    def getJobSize(self, printer) :   
83        """Returns the actual job size."""
84        try :
85            return self.JobSize
86        except AttributeError :   
87            return 0
88       
89    def computeJobSize(self) :   
90        """Must be overriden in children classes."""
91        raise RuntimeError, "AccounterBase.computeJobSize() must be overriden !"
92       
93def openAccounter(kotafilter, ispreaccounter=0) :
94    """Returns a connection handle to the appropriate accounter."""
95    if ispreaccounter :
96        (backend, args) = kotafilter.config.getPreAccounterBackend(kotafilter.PrinterName)
97    else :
98        (backend, args) = kotafilter.config.getAccounterBackend(kotafilter.PrinterName)
99    try :
100        accounterbackend = imp.load_source("accounterbackend", 
101                                            os.path.join(os.path.dirname(__file__),
102                                                         "accounters",
103                                                         "%s.py" % backend.lower()))
104    except ImportError :
105        raise PyKotaAccounterError, _("Unsupported accounter backend %s") % backend
106    else :   
107        return accounterbackend.Accounter(kotafilter, args, ispreaccounter)
Note: See TracBrowser for help on using the browser.