mirror of
https://github.com/psycopg/psycopg2.git
synced 2024-11-26 02:43:43 +03:00
203 lines
6.7 KiB
Python
203 lines
6.7 KiB
Python
# ZPsycopgDA/DA.py - ZPsycopgDA Zope product: Database Connection
|
||
#
|
||
# Copyright (C) 2004 Federico Di Gregorio <fog@initd.org>
|
||
#
|
||
# 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, or (at your option) any later
|
||
# version.
|
||
#
|
||
# Or, at your option this program (ZPsycopgDA) can be distributed under the
|
||
# Zope Public License (ZPL) Version 1.0, as published on the Zope web site,
|
||
# http://www.zope.org/Resources/ZPL.
|
||
#
|
||
# This program is distributed in the hope that it will be useful, but
|
||
# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTIBILITY
|
||
# or FITNESS FOR A PARTICULAR PURPOSE.
|
||
#
|
||
# See the LICENSE file for details.
|
||
|
||
|
||
ALLOWED_PSYCOPG_VERSIONS = ('1.99.9',)
|
||
|
||
import sys
|
||
import db
|
||
import DABase
|
||
import Shared.DC.ZRDB.Connection
|
||
|
||
from db import DB
|
||
from Globals import DTMLFile
|
||
from Globals import HTMLFile
|
||
from ImageFile import ImageFile
|
||
from ExtensionClass import Base
|
||
from DateTime import DateTime
|
||
|
||
# import psycopg and functions/singletons needed for date/time conversions
|
||
|
||
import psycopg
|
||
from psycopg import DATETIME
|
||
from psycopg.extensions import TIME, DATE, INTERVAL
|
||
from psycopg.extensions import new_type, register_type
|
||
|
||
|
||
|
||
# add a new connection to a folder
|
||
|
||
manage_addZPsycopgConnectionForm = DTMLFile('dtml/add',globals())
|
||
|
||
def manage_addZPsycopgConnection(self, id, title, connection_string,
|
||
zdatetime=None, tilevel=2,
|
||
check=None, REQUEST=None):
|
||
"""Add a DB connection to a folder."""
|
||
self._setObject(id, Connection(id, title, connection_string,
|
||
zdatetime, check, tilevel))
|
||
if REQUEST is not None: return self.manage_main(self, REQUEST)
|
||
|
||
|
||
|
||
# the connection object
|
||
|
||
class Connection(DABase.Connection):
|
||
"""ZPsycopg Connection."""
|
||
id = 'Psycopg_database_connection'
|
||
database_type = 'Psycopg'
|
||
meta_type = title = 'Z Psycopg Database Connection'
|
||
icon = 'misc_/ZPsycopg/conn'
|
||
|
||
def __init__(self, id, title, connection_string,
|
||
zdatetime, check=None, tilevel=2, encoding=''):
|
||
self.zdatetime = zdatetime
|
||
self.id = str(id)
|
||
self.edit(title, connection_string, zdatetime,
|
||
check=check, tilevel=tilevel, encoding=encoding)
|
||
|
||
def factory(self):
|
||
return DB
|
||
|
||
def table_info(self):
|
||
return self._v_database_connection.table_info()
|
||
|
||
def edit(self, title, connection_string,
|
||
zdatetime, check=None, tilevel=2, encoding=''):
|
||
self.title = title
|
||
self.connection_string = connection_string
|
||
self.zdatetime = zdatetime
|
||
self.tilevel = tilevel
|
||
self.encoding = encoding
|
||
|
||
self.set_type_casts()
|
||
|
||
if check: self.connect(self.connection_string)
|
||
|
||
manage_properties = DTMLFile('dtml/edit', globals())
|
||
|
||
def manage_edit(self, title, connection_string,
|
||
zdatetime=None, check=None, tilevel=2, encoding='UTF-8',
|
||
REQUEST=None):
|
||
"""Edit the DB connection."""
|
||
self.edit(title, connection_string, zdatetime,
|
||
check=check, tilevel=tilevel, encoding=encoding)
|
||
if REQUEST is not None:
|
||
msg = "Connection edited."
|
||
return self.manage_main(self,REQUEST,manage_tabs_message=msg)
|
||
|
||
def connect(self, s):
|
||
try:
|
||
self._v_database_connection.close()
|
||
except:
|
||
pass
|
||
|
||
# check psycopg version and raise exception if does not match
|
||
if psycopg.__version__ not in ALLOWED_PSYCOPG_VERSIONS:
|
||
raise ImportError("psycopg version mismatch (imported %s)" +
|
||
psycopg.__version__)
|
||
|
||
self.set_type_casts()
|
||
self._v_connected = ''
|
||
dbf = self.factory()
|
||
|
||
# TODO: let the psycopg exception propagate, or not?
|
||
self._v_database_connection = dbf(
|
||
self.connection_string, self.tilevel, self.encoding)
|
||
self._v_database_connection.open()
|
||
self._v_connected = DateTime()
|
||
|
||
return self
|
||
|
||
def set_type_casts(self):
|
||
# note that in both cases order *is* important
|
||
if self.zdatetime:
|
||
# use zope internal datetime routines
|
||
register_type(ZDATETIME)
|
||
register_type(ZDATE)
|
||
register_type(ZTIME)
|
||
register_type(ZINTERVAL)
|
||
else:
|
||
# use the standard
|
||
register_type(DATETIME)
|
||
register_type(DATE)
|
||
register_type(TIME)
|
||
register_type(INTERVAL)
|
||
|
||
# database connection registration data
|
||
|
||
classes = (Connection,)
|
||
|
||
meta_types = ({'name':'Z Psycopg Database Connection',
|
||
'action':'manage_addZPsycopgConnectionForm'},)
|
||
|
||
folder_methods = {
|
||
'manage_addZPsycopgConnection': manage_addZPsycopgConnection,
|
||
'manage_addZPsycopgConnectionForm': manage_addZPsycopgConnectionForm}
|
||
|
||
__ac_permissions__ = (
|
||
('Add Z Psycopg Database Connections',
|
||
('manage_addZPsycopgConnectionForm', 'manage_addZPsycopgConnection')),)
|
||
|
||
# add icons
|
||
|
||
misc_={'conn': ImageFile('Shared/DC/ZRDB/www/DBAdapterFolder_icon.gif')}
|
||
|
||
for icon in ('table', 'view', 'stable', 'what', 'field', 'text', 'bin',
|
||
'int', 'float', 'date', 'time', 'datetime'):
|
||
misc_[icon] = ImageFile('icons/%s.gif' % icon, globals())
|
||
|
||
# zope-specific psycopg typecasters
|
||
|
||
# convert an ISO timestamp string from postgres to a Zope DateTime object
|
||
def _cast_DateTime(str):
|
||
if str:
|
||
# this will split us into [date, time, GMT/AM/PM(if there)]
|
||
dt = split(str, ' ')
|
||
if len(dt) > 1:
|
||
# we now should split out any timezone info
|
||
dt[1] = split(dt[1], '-')[0]
|
||
dt[1] = split(dt[1], '+')[0]
|
||
return DateTime(join(dt[:2], ' '))
|
||
else:
|
||
return DateTime(dt[0])
|
||
|
||
# convert an ISO date string from postgres to a Zope DateTime object
|
||
def _cast_Date(str):
|
||
if str:
|
||
return DateTime(str)
|
||
|
||
# Convert a time string from postgres to a Zope DateTime object.
|
||
# NOTE: we set the day as today before feeding to DateTime so
|
||
# that it has the same DST settings.
|
||
def _cast_Time(str):
|
||
if str:
|
||
return DateTime(time.strftime('%Y-%m-%d %H:%M:%S',
|
||
time.localtime(time.time())[:3]+
|
||
time.strptime(str[:8], "%H:%M:%S")[3:]))
|
||
|
||
# TODO: DateTime does not support intervals: what's the best we can do?
|
||
def _cast_Interval(str):
|
||
return str
|
||
|
||
ZDATETIME = new_type((1184, 1114), "ZDATETIME", _cast_DateTime)
|
||
ZINTERVAL = new_type((1186,), "ZINTERVAL", _cast_Interval)
|
||
ZDATE = new_type((1082,), "ZDATE", _cast_Date)
|
||
ZTIME = new_type((1083,), "ZTIME", _cast_Time)
|
||
|