75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
import datetime
|
|
|
|
from .base import Database
|
|
|
|
|
|
class InsertVar:
|
|
"""
|
|
A late-binding cursor variable that can be passed to Cursor.execute
|
|
as a parameter, in order to receive the id of the row created by an
|
|
insert statement.
|
|
"""
|
|
types = {
|
|
'AutoField': int,
|
|
'BigAutoField': int,
|
|
'SmallAutoField': int,
|
|
'IntegerField': int,
|
|
'BigIntegerField': int,
|
|
'SmallIntegerField': int,
|
|
'PositiveSmallIntegerField': int,
|
|
'PositiveIntegerField': int,
|
|
'FloatField': Database.NATIVE_FLOAT,
|
|
'DateTimeField': Database.TIMESTAMP,
|
|
'DateField': Database.Date,
|
|
'DecimalField': Database.NUMBER,
|
|
}
|
|
|
|
def __init__(self, field):
|
|
internal_type = getattr(field, 'target_field', field).get_internal_type()
|
|
self.db_type = self.types.get(internal_type, str)
|
|
|
|
def bind_parameter(self, cursor):
|
|
param = cursor.cursor.var(self.db_type)
|
|
cursor._insert_id_var = param
|
|
return param
|
|
|
|
|
|
class Oracle_datetime(datetime.datetime):
|
|
"""
|
|
A datetime object, with an additional class attribute
|
|
to tell cx_Oracle to save the microseconds too.
|
|
"""
|
|
input_size = Database.TIMESTAMP
|
|
|
|
@classmethod
|
|
def from_datetime(cls, dt):
|
|
return Oracle_datetime(
|
|
dt.year, dt.month, dt.day,
|
|
dt.hour, dt.minute, dt.second, dt.microsecond,
|
|
)
|
|
|
|
|
|
class BulkInsertMapper:
|
|
BLOB = 'TO_BLOB(%s)'
|
|
DATE = 'TO_DATE(%s)'
|
|
INTERVAL = 'CAST(%s as INTERVAL DAY(9) TO SECOND(6))'
|
|
NUMBER = 'TO_NUMBER(%s)'
|
|
TIMESTAMP = 'TO_TIMESTAMP(%s)'
|
|
|
|
types = {
|
|
'BigIntegerField': NUMBER,
|
|
'BinaryField': BLOB,
|
|
'BooleanField': NUMBER,
|
|
'DateField': DATE,
|
|
'DateTimeField': TIMESTAMP,
|
|
'DecimalField': NUMBER,
|
|
'DurationField': INTERVAL,
|
|
'FloatField': NUMBER,
|
|
'IntegerField': NUMBER,
|
|
'NullBooleanField': NUMBER,
|
|
'PositiveIntegerField': NUMBER,
|
|
'PositiveSmallIntegerField': NUMBER,
|
|
'SmallIntegerField': NUMBER,
|
|
'TimeField': TIMESTAMP,
|
|
}
|