root / pykota / trunk / pykota / storages / pgstorage.py @ 2960

Revision 2960, 5.8 kB (checked in by jerome, 18 years ago)

Fixed a database client encoding problem : if the database already
exists and its encoding is not compatible with UTF-8 client encoding
(for example MULE_INTERNAL), an error occured in the SET CLIENT_ENCODING query.

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
RevLine 
[1021]1# PyKota
[1144]2# -*- coding: ISO-8859-15 -*-
[1021]3#
4# PyKota : Print Quotas for CUPS and LPRng
5#
[2622]6# (c) 2003, 2004, 2005, 2006 Jerome Alet <alet@librelogiciel.com>
[1021]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
[2302]19# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
[1021]20#
21# $Id$
22#
[2033]23#
[1021]24
[2741]25import time
[2599]26from types import StringType
27
[2830]28from pykota.storage import PyKotaStorageError, BaseStorage
[1327]29from pykota.storages.sql import SQLStorage
[1021]30
31try :
32    import pg
33except ImportError :   
34    import sys
35    # TODO : to translate or not to translate ?
36    raise PyKotaStorageError, "This python version (%s) doesn't seem to have the PygreSQL module installed correctly." % sys.version.split()[0]
[2031]37else :   
38    try :
39        PGError = pg.Error
40    except AttributeError :   
41        PGError = pg.error
[1021]42
[1327]43class Storage(BaseStorage, SQLStorage) :
[1021]44    def __init__(self, pykotatool, host, dbname, user, passwd) :
45        """Opens the PostgreSQL database connection."""
[1130]46        BaseStorage.__init__(self, pykotatool)
[1021]47        try :
48            (host, port) = host.split(":")
49            port = int(port)
50        except ValueError :   
[2874]51            port = 5432         # Use PostgreSQL's default tcp/ip port (5432).
[1021]52       
[2418]53        self.tool.logdebug("Trying to open database (host=%s, port=%s, dbname=%s, user=%s)..." % (host, port, dbname, user))
[2210]54        self.database = pg.connect(host=host, port=port, dbname=dbname, user=user, passwd=passwd)
55        self.closed = 0
[2960]56        try :
57            self.database.query("SET CLIENT_ENCODING TO 'UTF-8';")
58        except PGError, msg :   
59            self.tool.printInfo("Impossible to set database client encoding to UTF-8 : %s" % msg, "error")
[2210]60        self.tool.logdebug("Database opened (host=%s, port=%s, dbname=%s, user=%s)" % (host, port, dbname, user))
[1021]61           
[1113]62    def close(self) :   
[1021]63        """Closes the database connection."""
64        if not self.closed :
65            self.database.close()
66            self.closed = 1
[1130]67            self.tool.logdebug("Database closed.")
[1021]68       
[1041]69    def beginTransaction(self) :   
70        """Starts a transaction."""
[2741]71        self.before = time.time()
[1041]72        self.database.query("BEGIN;")
[1130]73        self.tool.logdebug("Transaction begins...")
[1041]74       
75    def commitTransaction(self) :   
76        """Commits a transaction."""
77        self.database.query("COMMIT;")
[2741]78        after = time.time()
[1130]79        self.tool.logdebug("Transaction committed.")
[2741]80        self.tool.logdebug("Transaction duration : %.4f seconds" % (after - self.before))
[1041]81       
82    def rollbackTransaction(self) :     
83        """Rollbacks a transaction."""
84        self.database.query("ROLLBACK;")
[2741]85        after = time.time()
[1130]86        self.tool.logdebug("Transaction aborted.")
[2741]87        self.tool.logdebug("Transaction duration : %.4f seconds" % (after - self.before))
[1041]88       
[1717]89    def doRawSearch(self, query) :
90        """Does a raw search query."""
[1021]91        query = query.strip()   
92        if not query.endswith(';') :   
93            query += ';'
94        try :
[2741]95            before = time.time()
[1130]96            self.tool.logdebug("QUERY : %s" % query)
[1021]97            result = self.database.query(query)
[2031]98        except PGError, msg :   
[2033]99            raise PyKotaStorageError, str(msg)
[1021]100        else :   
[2741]101            after = time.time()
102            self.tool.logdebug("Query Duration : %.4f seconds" % (after - before))
[1717]103            return result
[1041]104           
[1717]105    def doSearch(self, query) :       
106        """Does a search query."""
107        result = self.doRawSearch(query)
108        if (result is not None) and (result.ntuples() > 0) : 
109            return result.dictresult()
110       
[1041]111    def doModify(self, query) :
112        """Does a (possibly multiple) modify query."""
113        query = query.strip()   
114        if not query.endswith(';') :   
115            query += ';'
116        try :
[2741]117            before = time.time()
[1130]118            self.tool.logdebug("QUERY : %s" % query)
[1041]119            result = self.database.query(query)
[2031]120        except PGError, msg :   
[2773]121            self.tool.logdebug("Query failed : %s" % repr(msg))
[2033]122            raise PyKotaStorageError, str(msg)
[1068]123        else :   
[2741]124            after = time.time()
125            self.tool.logdebug("Query Duration : %.4f seconds" % (after - before))
[1068]126            return result
[1041]127           
[1021]128    def doQuote(self, field) :
129        """Quotes a field for use as a string in SQL queries."""
130        if type(field) == type(0.0) : 
131            typ = "decimal"
132        elif type(field) == type(0) :   
133            typ = "int"
[1520]134        elif type(field) == type(0L) :   
135            typ = "int"
[1021]136        else :   
137            typ = "text"
138        return pg._quote(field, typ)
[2593]139       
140    def prepareRawResult(self, result) :
141        """Prepares a raw result by including the headers."""
142        if result.ntuples() > 0 :
143            entries = [result.listfields()]
144            entries.extend(result.getresult())
145            nbfields = len(entries[0])
146            for i in range(1, len(entries)) :
147                fields = list(entries[i])
148                for j in range(nbfields) :
149                    field = fields[j]
150                    if type(field) == StringType :
151                        fields[j] = self.databaseToUserCharset(field) 
152                entries[i] = tuple(fields)   
153            return entries
154       
Note: See TracBrowser for help on using the browser.