Simplify source file layout.
baztian
parent
6ab7812e96
commit
7358271599
|
|
@ -1 +1,602 @@
|
|||
from dbapi2 import *
|
||||
#-*- coding: utf-8 -*-
|
||||
|
||||
# Copyright 2010-2015 Bastian Bowe
|
||||
#
|
||||
# This file is part of JayDeBeApi.
|
||||
# JayDeBeApi is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU Lesser General Public License as
|
||||
# published by the Free Software Foundation, either version 3 of the
|
||||
# License, or (at your option) any later version.
|
||||
#
|
||||
# JayDeBeApi 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
|
||||
# Lesser General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU Lesser General Public
|
||||
# License along with JayDeBeApi. If not, see
|
||||
# <http://www.gnu.org/licenses/>.
|
||||
|
||||
__version_info__ = (0, 1, 6)
|
||||
__version__ = ".".join(str(i) for i in __version_info__)
|
||||
|
||||
import datetime
|
||||
import exceptions
|
||||
import glob
|
||||
import os
|
||||
import time
|
||||
import re
|
||||
import sys
|
||||
import warnings
|
||||
|
||||
# Mapping from java.sql.Types attribute name to attribute value
|
||||
_jdbc_name_to_const = None
|
||||
|
||||
# Mapping from java.sql.Types attribute constant value to it's attribute name
|
||||
_jdbc_const_to_name = None
|
||||
|
||||
_jdbc_connect = None
|
||||
|
||||
_java_array_byte = None
|
||||
|
||||
_handle_sql_exception = None
|
||||
|
||||
def _handle_sql_exception_jython():
|
||||
from java.sql import SQLException
|
||||
exc_info = sys.exc_info()
|
||||
if isinstance(exc_info[1], SQLException):
|
||||
exc_type = DatabaseError
|
||||
else:
|
||||
exc_type = InterfaceError
|
||||
raise exc_type, exc_info[1], exc_info[2]
|
||||
|
||||
def _jdbc_connect_jython(jclassname, jars, libs, *args):
|
||||
if _jdbc_name_to_const is None:
|
||||
from java.sql import Types
|
||||
types = Types
|
||||
types_map = {}
|
||||
const_re = re.compile('[A-Z][A-Z_]*$')
|
||||
for i in dir(types):
|
||||
if const_re.match(i):
|
||||
types_map[i] = getattr(types, i)
|
||||
_init_types(types_map)
|
||||
global _java_array_byte
|
||||
if _java_array_byte is None:
|
||||
import jarray
|
||||
def _java_array_byte(data):
|
||||
return jarray.array(data, 'b')
|
||||
# register driver for DriverManager
|
||||
jpackage = jclassname[:jclassname.rfind('.')]
|
||||
dclassname = jclassname[jclassname.rfind('.') + 1:]
|
||||
# print jpackage
|
||||
# print dclassname
|
||||
# print jpackage
|
||||
from java.lang import Class
|
||||
from java.lang import ClassNotFoundException
|
||||
try:
|
||||
Class.forName(jclassname).newInstance()
|
||||
except ClassNotFoundException:
|
||||
if not jars:
|
||||
raise
|
||||
_jython_set_classpath(jars)
|
||||
Class.forName(jclassname).newInstance()
|
||||
from java.sql import DriverManager
|
||||
return DriverManager.getConnection(*args)
|
||||
|
||||
def _jython_set_classpath(jars):
|
||||
'''
|
||||
import a jar at runtime (needed for JDBC [Class.forName])
|
||||
|
||||
adapted by Bastian Bowe from
|
||||
http://stackoverflow.com/questions/3015059/jython-classpath-sys-path-and-jdbc-drivers
|
||||
'''
|
||||
from java.net import URL, URLClassLoader
|
||||
from java.lang import ClassLoader
|
||||
from java.io import File
|
||||
m = URLClassLoader.getDeclaredMethod("addURL", [URL])
|
||||
m.accessible = 1
|
||||
urls = [File(i).toURL() for i in jars]
|
||||
m.invoke(ClassLoader.getSystemClassLoader(), urls)
|
||||
|
||||
def _prepare_jython():
|
||||
global _jdbc_connect
|
||||
_jdbc_connect = _jdbc_connect_jython
|
||||
global _handle_sql_exception
|
||||
_handle_sql_exception = _handle_sql_exception_jython
|
||||
|
||||
def _handle_sql_exception_jpype():
|
||||
import jpype
|
||||
SQLException = jpype.java.sql.SQLException
|
||||
exc_info = sys.exc_info()
|
||||
if issubclass(exc_info[1].__javaclass__, SQLException):
|
||||
exc_type = DatabaseError
|
||||
else:
|
||||
exc_type = InterfaceError
|
||||
raise exc_type, exc_info[1], exc_info[2]
|
||||
|
||||
def _jdbc_connect_jpype(jclassname, jars, libs, *driver_args):
|
||||
import jpype
|
||||
if not jpype.isJVMStarted():
|
||||
args = []
|
||||
class_path = []
|
||||
if jars:
|
||||
class_path.extend(jars)
|
||||
class_path.extend(_get_classpath())
|
||||
if class_path:
|
||||
args.append('-Djava.class.path=%s' %
|
||||
os.path.pathsep.join(class_path))
|
||||
if libs:
|
||||
# path to shared libraries
|
||||
libs_path = os.path.pathsep.join(libs)
|
||||
args.append('-Djava.library.path=%s' % libs_path)
|
||||
# jvm_path = ('/usr/lib/jvm/java-6-openjdk'
|
||||
# '/jre/lib/i386/client/libjvm.so')
|
||||
jvm_path = jpype.getDefaultJVMPath()
|
||||
jpype.startJVM(jvm_path, *args)
|
||||
if not jpype.isThreadAttachedToJVM():
|
||||
jpype.attachThreadToJVM()
|
||||
if _jdbc_name_to_const is None:
|
||||
types = jpype.java.sql.Types
|
||||
types_map = {}
|
||||
for i in types.__javaclass__.getClassFields():
|
||||
types_map[i.getName()] = i.getStaticAttribute()
|
||||
_init_types(types_map)
|
||||
global _java_array_byte
|
||||
if _java_array_byte is None:
|
||||
def _java_array_byte(data):
|
||||
return jpype.JArray(jpype.JByte, 1)(data)
|
||||
# register driver for DriverManager
|
||||
jpype.JClass(jclassname)
|
||||
return jpype.java.sql.DriverManager.getConnection(*driver_args)
|
||||
|
||||
def _get_classpath():
|
||||
"""Extract CLASSPATH from system environment as JPype doesn't seem
|
||||
to respect that variable.
|
||||
"""
|
||||
try:
|
||||
orig_cp = os.environ['CLASSPATH']
|
||||
except KeyError:
|
||||
return []
|
||||
expanded_cp = []
|
||||
for i in orig_cp.split(os.path.pathsep):
|
||||
expanded_cp.extend(_jar_glob(i))
|
||||
return expanded_cp
|
||||
|
||||
def _jar_glob(item):
|
||||
if item.endswith('*'):
|
||||
return glob.glob('%s.[jJ][aA][rR]' % item)
|
||||
else:
|
||||
return [item]
|
||||
|
||||
def _prepare_jpype():
|
||||
global _jdbc_connect
|
||||
_jdbc_connect = _jdbc_connect_jpype
|
||||
global _handle_sql_exception
|
||||
_handle_sql_exception = _handle_sql_exception_jpype
|
||||
|
||||
if sys.platform.lower().startswith('java'):
|
||||
_prepare_jython()
|
||||
else:
|
||||
_prepare_jpype()
|
||||
|
||||
apilevel = '2.0'
|
||||
threadsafety = 1
|
||||
paramstyle = 'qmark'
|
||||
|
||||
class DBAPITypeObject(object):
|
||||
_mappings = {}
|
||||
def __init__(self, *values):
|
||||
"""Construct new DB-API 2.0 type object.
|
||||
values: Attribute names of java.sql.Types constants"""
|
||||
self.values = values
|
||||
for type_name in values:
|
||||
if type_name in DBAPITypeObject._mappings:
|
||||
raise ValueError, "Non unique mapping for type '%s'" % type_name
|
||||
DBAPITypeObject._mappings[type_name] = self
|
||||
def __cmp__(self, other):
|
||||
if other in self.values:
|
||||
return 0
|
||||
if other < self.values:
|
||||
return 1
|
||||
else:
|
||||
return -1
|
||||
def __repr__(self):
|
||||
return 'DBAPITypeObject(%s)' % ", ".join([repr(i) for i in self.values])
|
||||
@classmethod
|
||||
def _map_jdbc_type_to_dbapi(cls, jdbc_type_const):
|
||||
try:
|
||||
type_name = _jdbc_const_to_name[jdbc_type_const]
|
||||
except KeyError:
|
||||
warnings.warn("Unknown JDBC type with constant value %d. "
|
||||
"Using None as a default type_code." % jdbc_type_const)
|
||||
return None
|
||||
try:
|
||||
return cls._mappings[type_name]
|
||||
except KeyError:
|
||||
warnings.warn("No type mapping for JDBC type '%s' (constant value %d). "
|
||||
"Using None as a default type_code." % (type_name, jdbc_type_const))
|
||||
return None
|
||||
|
||||
|
||||
STRING = DBAPITypeObject('CHAR', 'NCHAR', 'NVARCHAR', 'VARCHAR', 'OTHER')
|
||||
|
||||
TEXT = DBAPITypeObject('CLOB', 'LONGVARCHAR', 'LONGNVARCHAR', 'NCLOB', 'SQLXML')
|
||||
|
||||
BINARY = DBAPITypeObject('BINARY', 'BLOB', 'LONGVARBINARY', 'VARBINARY')
|
||||
|
||||
NUMBER = DBAPITypeObject('BOOLEAN', 'BIGINT', 'INTEGER', 'SMALLINT')
|
||||
|
||||
FLOAT = DBAPITypeObject('FLOAT', 'REAL', 'DOUBLE')
|
||||
|
||||
DECIMAL = DBAPITypeObject('DECIMAL', 'NUMERIC')
|
||||
|
||||
DATE = DBAPITypeObject('DATE')
|
||||
|
||||
TIME = DBAPITypeObject('TIME')
|
||||
|
||||
DATETIME = DBAPITypeObject('TIMESTAMP')
|
||||
|
||||
ROWID = DBAPITypeObject('ROWID')
|
||||
|
||||
# DB-API 2.0 Module Interface Exceptions
|
||||
class Error(exceptions.StandardError):
|
||||
pass
|
||||
|
||||
class Warning(exceptions.StandardError):
|
||||
pass
|
||||
|
||||
class InterfaceError(Error):
|
||||
pass
|
||||
|
||||
class DatabaseError(Error):
|
||||
pass
|
||||
|
||||
class InternalError(DatabaseError):
|
||||
pass
|
||||
|
||||
class OperationalError(DatabaseError):
|
||||
pass
|
||||
|
||||
class ProgrammingError(DatabaseError):
|
||||
pass
|
||||
|
||||
class IntegrityError(DatabaseError):
|
||||
pass
|
||||
|
||||
class DataError(DatabaseError):
|
||||
pass
|
||||
|
||||
class NotSupportedError(DatabaseError):
|
||||
pass
|
||||
|
||||
# DB-API 2.0 Type Objects and Constructors
|
||||
|
||||
def _java_sql_blob(data):
|
||||
return _java_array_byte(data)
|
||||
|
||||
Binary = _java_sql_blob
|
||||
|
||||
def _str_func(func):
|
||||
def to_str(*parms):
|
||||
return str(func(*parms))
|
||||
return to_str
|
||||
|
||||
Date = _str_func(datetime.date)
|
||||
|
||||
Time = _str_func(datetime.time)
|
||||
|
||||
Timestamp = _str_func(datetime.datetime)
|
||||
|
||||
def DateFromTicks(ticks):
|
||||
return apply(Date, time.localtime(ticks)[:3])
|
||||
|
||||
def TimeFromTicks(ticks):
|
||||
return apply(Time, time.localtime(ticks)[3:6])
|
||||
|
||||
def TimestampFromTicks(ticks):
|
||||
return apply(Timestamp, time.localtime(ticks)[:6])
|
||||
|
||||
# DB-API 2.0 Module Interface connect constructor
|
||||
def connect(jclassname, driver_args, jars=None, libs=None):
|
||||
"""Open a connection to a database using a JDBC driver and return
|
||||
a Connection instance.
|
||||
|
||||
jclassname: Full qualified Java class name of the JDBC driver.
|
||||
driver_args: Argument or sequence of arguments to be passed to the
|
||||
Java DriverManager.getConnection method. Usually the
|
||||
database URL. See
|
||||
http://docs.oracle.com/javase/6/docs/api/java/sql/DriverManager.html
|
||||
for more details
|
||||
jars: Jar filename or sequence of filenames for the JDBC driver
|
||||
libs: Dll/so filenames or sequence of dlls/sos used as shared
|
||||
library by the JDBC driver
|
||||
"""
|
||||
if isinstance(driver_args, basestring):
|
||||
driver_args = [ driver_args ]
|
||||
if jars:
|
||||
if isinstance(jars, basestring):
|
||||
jars = [ jars ]
|
||||
else:
|
||||
jars = []
|
||||
if libs:
|
||||
if isinstance(libs, basestring):
|
||||
libs = [ libs ]
|
||||
else:
|
||||
libs = []
|
||||
jconn = _jdbc_connect(jclassname, jars, libs, *driver_args)
|
||||
return Connection(jconn, _converters)
|
||||
|
||||
# DB-API 2.0 Connection Object
|
||||
class Connection(object):
|
||||
|
||||
Error = Error
|
||||
Warning = Warning
|
||||
InterfaceError = InterfaceError
|
||||
DatabaseError = DatabaseError
|
||||
InternalError = InternalError
|
||||
OperationalError = OperationalError
|
||||
ProgrammingError = ProgrammingError
|
||||
IntegrityError = IntegrityError
|
||||
DataError = DataError
|
||||
NotSupportedError = NotSupportedError
|
||||
|
||||
def __init__(self, jconn, converters):
|
||||
self.jconn = jconn
|
||||
self._closed = False
|
||||
self._converters = converters
|
||||
|
||||
def close(self):
|
||||
if self._closed:
|
||||
raise Error
|
||||
self.jconn.close()
|
||||
self._closed = True
|
||||
|
||||
def commit(self):
|
||||
try:
|
||||
self.jconn.commit()
|
||||
except:
|
||||
_handle_sql_exception()
|
||||
|
||||
def rollback(self):
|
||||
try:
|
||||
self.jconn.rollback()
|
||||
except:
|
||||
_handle_sql_exception()
|
||||
|
||||
def cursor(self):
|
||||
return Cursor(self, self._converters)
|
||||
|
||||
# DB-API 2.0 Cursor Object
|
||||
class Cursor(object):
|
||||
|
||||
rowcount = -1
|
||||
_meta = None
|
||||
_prep = None
|
||||
_rs = None
|
||||
_description = None
|
||||
|
||||
def __init__(self, connection, converters):
|
||||
self._connection = connection
|
||||
self._converters = converters
|
||||
|
||||
@property
|
||||
def description(self):
|
||||
if self._description:
|
||||
return self._description
|
||||
m = self._meta
|
||||
if m:
|
||||
count = m.getColumnCount()
|
||||
self._description = []
|
||||
for col in range(1, count + 1):
|
||||
size = m.getColumnDisplaySize(col)
|
||||
jdbc_type = m.getColumnType(col)
|
||||
if jdbc_type == 0:
|
||||
# PEP-0249: SQL NULL values are represented by the
|
||||
# Python None singleton
|
||||
dbapi_type = None
|
||||
else:
|
||||
dbapi_type = DBAPITypeObject._map_jdbc_type_to_dbapi(jdbc_type)
|
||||
col_desc = ( m.getColumnName(col),
|
||||
dbapi_type,
|
||||
size,
|
||||
size,
|
||||
m.getPrecision(col),
|
||||
m.getScale(col),
|
||||
m.isNullable(col),
|
||||
)
|
||||
self._description.append(col_desc)
|
||||
return self._description
|
||||
|
||||
# optional callproc(self, procname, *parameters) unsupported
|
||||
|
||||
def close(self):
|
||||
self._close_last()
|
||||
self._connection = None
|
||||
|
||||
def _close_last(self):
|
||||
"""Close the resultset and reset collected meta data.
|
||||
"""
|
||||
if self._rs:
|
||||
self._rs.close()
|
||||
self._rs = None
|
||||
if self._prep:
|
||||
self._prep.close()
|
||||
self._prep = None
|
||||
self._meta = None
|
||||
self._description = None
|
||||
|
||||
# TODO: this is a possible way to close the open result sets
|
||||
# but I'm not sure when __del__ will be called
|
||||
__del__ = _close_last
|
||||
|
||||
def _set_stmt_parms(self, prep_stmt, parameters):
|
||||
for i in range(len(parameters)):
|
||||
# print (i, parameters[i], type(parameters[i]))
|
||||
prep_stmt.setObject(i + 1, parameters[i])
|
||||
|
||||
def execute(self, operation, parameters=None):
|
||||
if self._connection._closed:
|
||||
raise Error
|
||||
if not parameters:
|
||||
parameters = ()
|
||||
self._close_last()
|
||||
self._prep = self._connection.jconn.prepareStatement(operation)
|
||||
self._set_stmt_parms(self._prep, parameters)
|
||||
try:
|
||||
is_rs = self._prep.execute()
|
||||
except:
|
||||
_handle_sql_exception()
|
||||
if is_rs:
|
||||
self._rs = self._prep.getResultSet()
|
||||
self._meta = self._rs.getMetaData()
|
||||
self.rowcount = -1
|
||||
else:
|
||||
self.rowcount = self._prep.getUpdateCount()
|
||||
# self._prep.getWarnings() ???
|
||||
|
||||
def executemany(self, operation, seq_of_parameters):
|
||||
self._close_last()
|
||||
self._prep = self._connection.jconn.prepareStatement(operation)
|
||||
for parameters in seq_of_parameters:
|
||||
self._set_stmt_parms(self._prep, parameters)
|
||||
self._prep.addBatch()
|
||||
update_counts = self._prep.executeBatch()
|
||||
# self._prep.getWarnings() ???
|
||||
self.rowcount = sum(update_counts)
|
||||
self._close_last()
|
||||
|
||||
def fetchone(self):
|
||||
if not self._rs:
|
||||
raise Error
|
||||
if not self._rs.next():
|
||||
return None
|
||||
row = []
|
||||
for col in range(1, self._meta.getColumnCount() + 1):
|
||||
sqltype = self._meta.getColumnType(col)
|
||||
converter = self._converters.get(sqltype, _unknownSqlTypeConverter)
|
||||
v = converter(self._rs, col)
|
||||
row.append(v)
|
||||
return tuple(row)
|
||||
|
||||
def fetchmany(self, size=None):
|
||||
if not self._rs:
|
||||
raise Error
|
||||
if size is None:
|
||||
size = self.arraysize
|
||||
# TODO: handle SQLException if not supported by db
|
||||
self._rs.setFetchSize(size)
|
||||
rows = []
|
||||
row = None
|
||||
for i in xrange(size):
|
||||
row = self.fetchone()
|
||||
if row is None:
|
||||
break
|
||||
else:
|
||||
rows.append(row)
|
||||
# reset fetch size
|
||||
if row:
|
||||
# TODO: handle SQLException if not supported by db
|
||||
self._rs.setFetchSize(0)
|
||||
return rows
|
||||
|
||||
def fetchall(self):
|
||||
rows = []
|
||||
while True:
|
||||
row = self.fetchone()
|
||||
if row is None:
|
||||
break
|
||||
else:
|
||||
rows.append(row)
|
||||
return rows
|
||||
|
||||
# optional nextset() unsupported
|
||||
|
||||
arraysize = 1
|
||||
|
||||
def setinputsizes(self, sizes):
|
||||
pass
|
||||
|
||||
def setoutputsize(self, size, column=None):
|
||||
pass
|
||||
|
||||
def _unknownSqlTypeConverter(rs, col):
|
||||
return rs.getObject(col)
|
||||
|
||||
def _to_datetime(rs, col):
|
||||
java_val = rs.getTimestamp(col)
|
||||
if not java_val:
|
||||
return
|
||||
d = datetime.datetime.strptime(str(java_val)[:19], "%Y-%m-%d %H:%M:%S")
|
||||
d = d.replace(microsecond=int(str(java_val.getNanos())[:6]))
|
||||
return str(d)
|
||||
|
||||
def _to_time(rs, col):
|
||||
java_val = rs.getTime(col)
|
||||
if not java_val:
|
||||
return
|
||||
return str(java_val)
|
||||
|
||||
def _to_date(rs, col):
|
||||
java_val = rs.getDate(col)
|
||||
if not java_val:
|
||||
return
|
||||
d = datetime.datetime.strptime(str(java_val)[:10], "%Y-%m-%d")
|
||||
return d.strftime("%Y-%m-%d")
|
||||
|
||||
def _to_binary(rs, col):
|
||||
java_val = rs.getObject(col)
|
||||
if java_val is None:
|
||||
return
|
||||
return str(java_val)
|
||||
|
||||
def _java_to_py(java_method):
|
||||
def to_py(rs, col):
|
||||
java_val = rs.getObject(col)
|
||||
if java_val is None:
|
||||
return
|
||||
if isinstance(java_val, (basestring, int, long, float, bool)):
|
||||
return java_val
|
||||
return getattr(java_val, java_method)()
|
||||
return to_py
|
||||
|
||||
_to_double = _java_to_py('doubleValue')
|
||||
|
||||
_to_int = _java_to_py('intValue')
|
||||
|
||||
def _init_types(types_map):
|
||||
global _jdbc_name_to_const
|
||||
_jdbc_name_to_const = types_map
|
||||
global _jdbc_const_to_name
|
||||
_jdbc_const_to_name = dict((y,x) for x,y in types_map.iteritems())
|
||||
_init_converters(types_map)
|
||||
|
||||
def _init_converters(types_map):
|
||||
"""Prepares the converters for conversion of java types to python
|
||||
objects.
|
||||
types_map: Mapping of java.sql.Types field name to java.sql.Types
|
||||
field constant value"""
|
||||
global _converters
|
||||
_converters = {}
|
||||
for i in _DEFAULT_CONVERTERS:
|
||||
const_val = types_map[i]
|
||||
_converters[const_val] = _DEFAULT_CONVERTERS[i]
|
||||
|
||||
# Mapping from java.sql.Types field to converter method
|
||||
_converters = None
|
||||
|
||||
_DEFAULT_CONVERTERS = {
|
||||
# see
|
||||
# http://download.oracle.com/javase/6/docs/api/java/sql/Types.html
|
||||
# for possible keys
|
||||
'TIMESTAMP': _to_datetime,
|
||||
'TIME': _to_time,
|
||||
'DATE': _to_date,
|
||||
'BINARY': _to_binary,
|
||||
'DECIMAL': _to_double,
|
||||
'NUMERIC': _to_double,
|
||||
'DOUBLE': _to_double,
|
||||
'FLOAT': _to_double,
|
||||
'INTEGER': _to_int,
|
||||
'SMALLINT': _to_int,
|
||||
'BOOLEAN': _java_to_py('booleanValue'),
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,602 +0,0 @@
|
|||
#-*- coding: utf-8 -*-
|
||||
|
||||
# Copyright 2010-2015 Bastian Bowe
|
||||
#
|
||||
# This file is part of JayDeBeApi.
|
||||
# JayDeBeApi is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU Lesser General Public License as
|
||||
# published by the Free Software Foundation, either version 3 of the
|
||||
# License, or (at your option) any later version.
|
||||
#
|
||||
# JayDeBeApi 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
|
||||
# Lesser General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU Lesser General Public
|
||||
# License along with JayDeBeApi. If not, see
|
||||
# <http://www.gnu.org/licenses/>.
|
||||
|
||||
__version_info__ = (0, 1, 6)
|
||||
__version__ = ".".join(str(i) for i in __version_info__)
|
||||
|
||||
import datetime
|
||||
import exceptions
|
||||
import glob
|
||||
import os
|
||||
import time
|
||||
import re
|
||||
import sys
|
||||
import warnings
|
||||
|
||||
# Mapping from java.sql.Types attribute name to attribute value
|
||||
_jdbc_name_to_const = None
|
||||
|
||||
# Mapping from java.sql.Types attribute constant value to it's attribute name
|
||||
_jdbc_const_to_name = None
|
||||
|
||||
_jdbc_connect = None
|
||||
|
||||
_java_array_byte = None
|
||||
|
||||
_handle_sql_exception = None
|
||||
|
||||
def _handle_sql_exception_jython():
|
||||
from java.sql import SQLException
|
||||
exc_info = sys.exc_info()
|
||||
if isinstance(exc_info[1], SQLException):
|
||||
exc_type = DatabaseError
|
||||
else:
|
||||
exc_type = InterfaceError
|
||||
raise exc_type, exc_info[1], exc_info[2]
|
||||
|
||||
def _jdbc_connect_jython(jclassname, jars, libs, *args):
|
||||
if _jdbc_name_to_const is None:
|
||||
from java.sql import Types
|
||||
types = Types
|
||||
types_map = {}
|
||||
const_re = re.compile('[A-Z][A-Z_]*$')
|
||||
for i in dir(types):
|
||||
if const_re.match(i):
|
||||
types_map[i] = getattr(types, i)
|
||||
_init_types(types_map)
|
||||
global _java_array_byte
|
||||
if _java_array_byte is None:
|
||||
import jarray
|
||||
def _java_array_byte(data):
|
||||
return jarray.array(data, 'b')
|
||||
# register driver for DriverManager
|
||||
jpackage = jclassname[:jclassname.rfind('.')]
|
||||
dclassname = jclassname[jclassname.rfind('.') + 1:]
|
||||
# print jpackage
|
||||
# print dclassname
|
||||
# print jpackage
|
||||
from java.lang import Class
|
||||
from java.lang import ClassNotFoundException
|
||||
try:
|
||||
Class.forName(jclassname).newInstance()
|
||||
except ClassNotFoundException:
|
||||
if not jars:
|
||||
raise
|
||||
_jython_set_classpath(jars)
|
||||
Class.forName(jclassname).newInstance()
|
||||
from java.sql import DriverManager
|
||||
return DriverManager.getConnection(*args)
|
||||
|
||||
def _jython_set_classpath(jars):
|
||||
'''
|
||||
import a jar at runtime (needed for JDBC [Class.forName])
|
||||
|
||||
adapted by Bastian Bowe from
|
||||
http://stackoverflow.com/questions/3015059/jython-classpath-sys-path-and-jdbc-drivers
|
||||
'''
|
||||
from java.net import URL, URLClassLoader
|
||||
from java.lang import ClassLoader
|
||||
from java.io import File
|
||||
m = URLClassLoader.getDeclaredMethod("addURL", [URL])
|
||||
m.accessible = 1
|
||||
urls = [File(i).toURL() for i in jars]
|
||||
m.invoke(ClassLoader.getSystemClassLoader(), urls)
|
||||
|
||||
def _prepare_jython():
|
||||
global _jdbc_connect
|
||||
_jdbc_connect = _jdbc_connect_jython
|
||||
global _handle_sql_exception
|
||||
_handle_sql_exception = _handle_sql_exception_jython
|
||||
|
||||
def _handle_sql_exception_jpype():
|
||||
import jpype
|
||||
SQLException = jpype.java.sql.SQLException
|
||||
exc_info = sys.exc_info()
|
||||
if issubclass(exc_info[1].__javaclass__, SQLException):
|
||||
exc_type = DatabaseError
|
||||
else:
|
||||
exc_type = InterfaceError
|
||||
raise exc_type, exc_info[1], exc_info[2]
|
||||
|
||||
def _jdbc_connect_jpype(jclassname, jars, libs, *driver_args):
|
||||
import jpype
|
||||
if not jpype.isJVMStarted():
|
||||
args = []
|
||||
class_path = []
|
||||
if jars:
|
||||
class_path.extend(jars)
|
||||
class_path.extend(_get_classpath())
|
||||
if class_path:
|
||||
args.append('-Djava.class.path=%s' %
|
||||
os.path.pathsep.join(class_path))
|
||||
if libs:
|
||||
# path to shared libraries
|
||||
libs_path = os.path.pathsep.join(libs)
|
||||
args.append('-Djava.library.path=%s' % libs_path)
|
||||
# jvm_path = ('/usr/lib/jvm/java-6-openjdk'
|
||||
# '/jre/lib/i386/client/libjvm.so')
|
||||
jvm_path = jpype.getDefaultJVMPath()
|
||||
jpype.startJVM(jvm_path, *args)
|
||||
if not jpype.isThreadAttachedToJVM():
|
||||
jpype.attachThreadToJVM()
|
||||
if _jdbc_name_to_const is None:
|
||||
types = jpype.java.sql.Types
|
||||
types_map = {}
|
||||
for i in types.__javaclass__.getClassFields():
|
||||
types_map[i.getName()] = i.getStaticAttribute()
|
||||
_init_types(types_map)
|
||||
global _java_array_byte
|
||||
if _java_array_byte is None:
|
||||
def _java_array_byte(data):
|
||||
return jpype.JArray(jpype.JByte, 1)(data)
|
||||
# register driver for DriverManager
|
||||
jpype.JClass(jclassname)
|
||||
return jpype.java.sql.DriverManager.getConnection(*driver_args)
|
||||
|
||||
def _get_classpath():
|
||||
"""Extract CLASSPATH from system environment as JPype doesn't seem
|
||||
to respect that variable.
|
||||
"""
|
||||
try:
|
||||
orig_cp = os.environ['CLASSPATH']
|
||||
except KeyError:
|
||||
return []
|
||||
expanded_cp = []
|
||||
for i in orig_cp.split(os.path.pathsep):
|
||||
expanded_cp.extend(_jar_glob(i))
|
||||
return expanded_cp
|
||||
|
||||
def _jar_glob(item):
|
||||
if item.endswith('*'):
|
||||
return glob.glob('%s.[jJ][aA][rR]' % item)
|
||||
else:
|
||||
return [item]
|
||||
|
||||
def _prepare_jpype():
|
||||
global _jdbc_connect
|
||||
_jdbc_connect = _jdbc_connect_jpype
|
||||
global _handle_sql_exception
|
||||
_handle_sql_exception = _handle_sql_exception_jpype
|
||||
|
||||
if sys.platform.lower().startswith('java'):
|
||||
_prepare_jython()
|
||||
else:
|
||||
_prepare_jpype()
|
||||
|
||||
apilevel = '2.0'
|
||||
threadsafety = 1
|
||||
paramstyle = 'qmark'
|
||||
|
||||
class DBAPITypeObject(object):
|
||||
_mappings = {}
|
||||
def __init__(self, *values):
|
||||
"""Construct new DB-API 2.0 type object.
|
||||
values: Attribute names of java.sql.Types constants"""
|
||||
self.values = values
|
||||
for type_name in values:
|
||||
if type_name in DBAPITypeObject._mappings:
|
||||
raise ValueError, "Non unique mapping for type '%s'" % type_name
|
||||
DBAPITypeObject._mappings[type_name] = self
|
||||
def __cmp__(self, other):
|
||||
if other in self.values:
|
||||
return 0
|
||||
if other < self.values:
|
||||
return 1
|
||||
else:
|
||||
return -1
|
||||
def __repr__(self):
|
||||
return 'DBAPITypeObject(%s)' % ", ".join([repr(i) for i in self.values])
|
||||
@classmethod
|
||||
def _map_jdbc_type_to_dbapi(cls, jdbc_type_const):
|
||||
try:
|
||||
type_name = _jdbc_const_to_name[jdbc_type_const]
|
||||
except KeyError:
|
||||
warnings.warn("Unknown JDBC type with constant value %d. "
|
||||
"Using None as a default type_code." % jdbc_type_const)
|
||||
return None
|
||||
try:
|
||||
return cls._mappings[type_name]
|
||||
except KeyError:
|
||||
warnings.warn("No type mapping for JDBC type '%s' (constant value %d). "
|
||||
"Using None as a default type_code." % (type_name, jdbc_type_const))
|
||||
return None
|
||||
|
||||
|
||||
STRING = DBAPITypeObject('CHAR', 'NCHAR', 'NVARCHAR', 'VARCHAR', 'OTHER')
|
||||
|
||||
TEXT = DBAPITypeObject('CLOB', 'LONGVARCHAR', 'LONGNVARCHAR', 'NCLOB', 'SQLXML')
|
||||
|
||||
BINARY = DBAPITypeObject('BINARY', 'BLOB', 'LONGVARBINARY', 'VARBINARY')
|
||||
|
||||
NUMBER = DBAPITypeObject('BOOLEAN', 'BIGINT', 'INTEGER', 'SMALLINT')
|
||||
|
||||
FLOAT = DBAPITypeObject('FLOAT', 'REAL', 'DOUBLE')
|
||||
|
||||
DECIMAL = DBAPITypeObject('DECIMAL', 'NUMERIC')
|
||||
|
||||
DATE = DBAPITypeObject('DATE')
|
||||
|
||||
TIME = DBAPITypeObject('TIME')
|
||||
|
||||
DATETIME = DBAPITypeObject('TIMESTAMP')
|
||||
|
||||
ROWID = DBAPITypeObject('ROWID')
|
||||
|
||||
# DB-API 2.0 Module Interface Exceptions
|
||||
class Error(exceptions.StandardError):
|
||||
pass
|
||||
|
||||
class Warning(exceptions.StandardError):
|
||||
pass
|
||||
|
||||
class InterfaceError(Error):
|
||||
pass
|
||||
|
||||
class DatabaseError(Error):
|
||||
pass
|
||||
|
||||
class InternalError(DatabaseError):
|
||||
pass
|
||||
|
||||
class OperationalError(DatabaseError):
|
||||
pass
|
||||
|
||||
class ProgrammingError(DatabaseError):
|
||||
pass
|
||||
|
||||
class IntegrityError(DatabaseError):
|
||||
pass
|
||||
|
||||
class DataError(DatabaseError):
|
||||
pass
|
||||
|
||||
class NotSupportedError(DatabaseError):
|
||||
pass
|
||||
|
||||
# DB-API 2.0 Type Objects and Constructors
|
||||
|
||||
def _java_sql_blob(data):
|
||||
return _java_array_byte(data)
|
||||
|
||||
Binary = _java_sql_blob
|
||||
|
||||
def _str_func(func):
|
||||
def to_str(*parms):
|
||||
return str(func(*parms))
|
||||
return to_str
|
||||
|
||||
Date = _str_func(datetime.date)
|
||||
|
||||
Time = _str_func(datetime.time)
|
||||
|
||||
Timestamp = _str_func(datetime.datetime)
|
||||
|
||||
def DateFromTicks(ticks):
|
||||
return apply(Date, time.localtime(ticks)[:3])
|
||||
|
||||
def TimeFromTicks(ticks):
|
||||
return apply(Time, time.localtime(ticks)[3:6])
|
||||
|
||||
def TimestampFromTicks(ticks):
|
||||
return apply(Timestamp, time.localtime(ticks)[:6])
|
||||
|
||||
# DB-API 2.0 Module Interface connect constructor
|
||||
def connect(jclassname, driver_args, jars=None, libs=None):
|
||||
"""Open a connection to a database using a JDBC driver and return
|
||||
a Connection instance.
|
||||
|
||||
jclassname: Full qualified Java class name of the JDBC driver.
|
||||
driver_args: Argument or sequence of arguments to be passed to the
|
||||
Java DriverManager.getConnection method. Usually the
|
||||
database URL. See
|
||||
http://docs.oracle.com/javase/6/docs/api/java/sql/DriverManager.html
|
||||
for more details
|
||||
jars: Jar filename or sequence of filenames for the JDBC driver
|
||||
libs: Dll/so filenames or sequence of dlls/sos used as shared
|
||||
library by the JDBC driver
|
||||
"""
|
||||
if isinstance(driver_args, basestring):
|
||||
driver_args = [ driver_args ]
|
||||
if jars:
|
||||
if isinstance(jars, basestring):
|
||||
jars = [ jars ]
|
||||
else:
|
||||
jars = []
|
||||
if libs:
|
||||
if isinstance(libs, basestring):
|
||||
libs = [ libs ]
|
||||
else:
|
||||
libs = []
|
||||
jconn = _jdbc_connect(jclassname, jars, libs, *driver_args)
|
||||
return Connection(jconn, _converters)
|
||||
|
||||
# DB-API 2.0 Connection Object
|
||||
class Connection(object):
|
||||
|
||||
Error = Error
|
||||
Warning = Warning
|
||||
InterfaceError = InterfaceError
|
||||
DatabaseError = DatabaseError
|
||||
InternalError = InternalError
|
||||
OperationalError = OperationalError
|
||||
ProgrammingError = ProgrammingError
|
||||
IntegrityError = IntegrityError
|
||||
DataError = DataError
|
||||
NotSupportedError = NotSupportedError
|
||||
|
||||
def __init__(self, jconn, converters):
|
||||
self.jconn = jconn
|
||||
self._closed = False
|
||||
self._converters = converters
|
||||
|
||||
def close(self):
|
||||
if self._closed:
|
||||
raise Error
|
||||
self.jconn.close()
|
||||
self._closed = True
|
||||
|
||||
def commit(self):
|
||||
try:
|
||||
self.jconn.commit()
|
||||
except:
|
||||
_handle_sql_exception()
|
||||
|
||||
def rollback(self):
|
||||
try:
|
||||
self.jconn.rollback()
|
||||
except:
|
||||
_handle_sql_exception()
|
||||
|
||||
def cursor(self):
|
||||
return Cursor(self, self._converters)
|
||||
|
||||
# DB-API 2.0 Cursor Object
|
||||
class Cursor(object):
|
||||
|
||||
rowcount = -1
|
||||
_meta = None
|
||||
_prep = None
|
||||
_rs = None
|
||||
_description = None
|
||||
|
||||
def __init__(self, connection, converters):
|
||||
self._connection = connection
|
||||
self._converters = converters
|
||||
|
||||
@property
|
||||
def description(self):
|
||||
if self._description:
|
||||
return self._description
|
||||
m = self._meta
|
||||
if m:
|
||||
count = m.getColumnCount()
|
||||
self._description = []
|
||||
for col in range(1, count + 1):
|
||||
size = m.getColumnDisplaySize(col)
|
||||
jdbc_type = m.getColumnType(col)
|
||||
if jdbc_type == 0:
|
||||
# PEP-0249: SQL NULL values are represented by the
|
||||
# Python None singleton
|
||||
dbapi_type = None
|
||||
else:
|
||||
dbapi_type = DBAPITypeObject._map_jdbc_type_to_dbapi(jdbc_type)
|
||||
col_desc = ( m.getColumnName(col),
|
||||
dbapi_type,
|
||||
size,
|
||||
size,
|
||||
m.getPrecision(col),
|
||||
m.getScale(col),
|
||||
m.isNullable(col),
|
||||
)
|
||||
self._description.append(col_desc)
|
||||
return self._description
|
||||
|
||||
# optional callproc(self, procname, *parameters) unsupported
|
||||
|
||||
def close(self):
|
||||
self._close_last()
|
||||
self._connection = None
|
||||
|
||||
def _close_last(self):
|
||||
"""Close the resultset and reset collected meta data.
|
||||
"""
|
||||
if self._rs:
|
||||
self._rs.close()
|
||||
self._rs = None
|
||||
if self._prep:
|
||||
self._prep.close()
|
||||
self._prep = None
|
||||
self._meta = None
|
||||
self._description = None
|
||||
|
||||
# TODO: this is a possible way to close the open result sets
|
||||
# but I'm not sure when __del__ will be called
|
||||
__del__ = _close_last
|
||||
|
||||
def _set_stmt_parms(self, prep_stmt, parameters):
|
||||
for i in range(len(parameters)):
|
||||
# print (i, parameters[i], type(parameters[i]))
|
||||
prep_stmt.setObject(i + 1, parameters[i])
|
||||
|
||||
def execute(self, operation, parameters=None):
|
||||
if self._connection._closed:
|
||||
raise Error
|
||||
if not parameters:
|
||||
parameters = ()
|
||||
self._close_last()
|
||||
self._prep = self._connection.jconn.prepareStatement(operation)
|
||||
self._set_stmt_parms(self._prep, parameters)
|
||||
try:
|
||||
is_rs = self._prep.execute()
|
||||
except:
|
||||
_handle_sql_exception()
|
||||
if is_rs:
|
||||
self._rs = self._prep.getResultSet()
|
||||
self._meta = self._rs.getMetaData()
|
||||
self.rowcount = -1
|
||||
else:
|
||||
self.rowcount = self._prep.getUpdateCount()
|
||||
# self._prep.getWarnings() ???
|
||||
|
||||
def executemany(self, operation, seq_of_parameters):
|
||||
self._close_last()
|
||||
self._prep = self._connection.jconn.prepareStatement(operation)
|
||||
for parameters in seq_of_parameters:
|
||||
self._set_stmt_parms(self._prep, parameters)
|
||||
self._prep.addBatch()
|
||||
update_counts = self._prep.executeBatch()
|
||||
# self._prep.getWarnings() ???
|
||||
self.rowcount = sum(update_counts)
|
||||
self._close_last()
|
||||
|
||||
def fetchone(self):
|
||||
if not self._rs:
|
||||
raise Error
|
||||
if not self._rs.next():
|
||||
return None
|
||||
row = []
|
||||
for col in range(1, self._meta.getColumnCount() + 1):
|
||||
sqltype = self._meta.getColumnType(col)
|
||||
converter = self._converters.get(sqltype, _unknownSqlTypeConverter)
|
||||
v = converter(self._rs, col)
|
||||
row.append(v)
|
||||
return tuple(row)
|
||||
|
||||
def fetchmany(self, size=None):
|
||||
if not self._rs:
|
||||
raise Error
|
||||
if size is None:
|
||||
size = self.arraysize
|
||||
# TODO: handle SQLException if not supported by db
|
||||
self._rs.setFetchSize(size)
|
||||
rows = []
|
||||
row = None
|
||||
for i in xrange(size):
|
||||
row = self.fetchone()
|
||||
if row is None:
|
||||
break
|
||||
else:
|
||||
rows.append(row)
|
||||
# reset fetch size
|
||||
if row:
|
||||
# TODO: handle SQLException if not supported by db
|
||||
self._rs.setFetchSize(0)
|
||||
return rows
|
||||
|
||||
def fetchall(self):
|
||||
rows = []
|
||||
while True:
|
||||
row = self.fetchone()
|
||||
if row is None:
|
||||
break
|
||||
else:
|
||||
rows.append(row)
|
||||
return rows
|
||||
|
||||
# optional nextset() unsupported
|
||||
|
||||
arraysize = 1
|
||||
|
||||
def setinputsizes(self, sizes):
|
||||
pass
|
||||
|
||||
def setoutputsize(self, size, column=None):
|
||||
pass
|
||||
|
||||
def _unknownSqlTypeConverter(rs, col):
|
||||
return rs.getObject(col)
|
||||
|
||||
def _to_datetime(rs, col):
|
||||
java_val = rs.getTimestamp(col)
|
||||
if not java_val:
|
||||
return
|
||||
d = datetime.datetime.strptime(str(java_val)[:19], "%Y-%m-%d %H:%M:%S")
|
||||
d = d.replace(microsecond=int(str(java_val.getNanos())[:6]))
|
||||
return str(d)
|
||||
|
||||
def _to_time(rs, col):
|
||||
java_val = rs.getTime(col)
|
||||
if not java_val:
|
||||
return
|
||||
return str(java_val)
|
||||
|
||||
def _to_date(rs, col):
|
||||
java_val = rs.getDate(col)
|
||||
if not java_val:
|
||||
return
|
||||
d = datetime.datetime.strptime(str(java_val)[:10], "%Y-%m-%d")
|
||||
return d.strftime("%Y-%m-%d")
|
||||
|
||||
def _to_binary(rs, col):
|
||||
java_val = rs.getObject(col)
|
||||
if java_val is None:
|
||||
return
|
||||
return str(java_val)
|
||||
|
||||
def _java_to_py(java_method):
|
||||
def to_py(rs, col):
|
||||
java_val = rs.getObject(col)
|
||||
if java_val is None:
|
||||
return
|
||||
if isinstance(java_val, (basestring, int, long, float, bool)):
|
||||
return java_val
|
||||
return getattr(java_val, java_method)()
|
||||
return to_py
|
||||
|
||||
_to_double = _java_to_py('doubleValue')
|
||||
|
||||
_to_int = _java_to_py('intValue')
|
||||
|
||||
def _init_types(types_map):
|
||||
global _jdbc_name_to_const
|
||||
_jdbc_name_to_const = types_map
|
||||
global _jdbc_const_to_name
|
||||
_jdbc_const_to_name = dict((y,x) for x,y in types_map.iteritems())
|
||||
_init_converters(types_map)
|
||||
|
||||
def _init_converters(types_map):
|
||||
"""Prepares the converters for conversion of java types to python
|
||||
objects.
|
||||
types_map: Mapping of java.sql.Types field name to java.sql.Types
|
||||
field constant value"""
|
||||
global _converters
|
||||
_converters = {}
|
||||
for i in _DEFAULT_CONVERTERS:
|
||||
const_val = types_map[i]
|
||||
_converters[const_val] = _DEFAULT_CONVERTERS[i]
|
||||
|
||||
# Mapping from java.sql.Types field to converter method
|
||||
_converters = None
|
||||
|
||||
_DEFAULT_CONVERTERS = {
|
||||
# see
|
||||
# http://download.oracle.com/javase/6/docs/api/java/sql/Types.html
|
||||
# for possible keys
|
||||
'TIMESTAMP': _to_datetime,
|
||||
'TIME': _to_time,
|
||||
'DATE': _to_date,
|
||||
'BINARY': _to_binary,
|
||||
'DECIMAL': _to_double,
|
||||
'NUMERIC': _to_double,
|
||||
'DOUBLE': _to_double,
|
||||
'FLOAT': _to_double,
|
||||
'INTEGER': _to_int,
|
||||
'SMALLINT': _to_int,
|
||||
'BOOLEAN': _java_to_py('booleanValue'),
|
||||
}
|
||||
Loading…
Reference in New Issue