43 lines
1.8 KiB
Python
43 lines
1.8 KiB
Python
import cx_Oracle
|
|
|
|
from django.db.backends.oracle.introspection import DatabaseIntrospection
|
|
|
|
|
|
class OracleIntrospection(DatabaseIntrospection):
|
|
# Associating any OBJECTVAR instances with GeometryField. Of course,
|
|
# this won't work right on Oracle objects that aren't MDSYS.SDO_GEOMETRY,
|
|
# but it is the only object type supported within Django anyways.
|
|
data_types_reverse = DatabaseIntrospection.data_types_reverse.copy()
|
|
data_types_reverse[cx_Oracle.OBJECT] = 'GeometryField'
|
|
|
|
def get_geometry_type(self, table_name, description):
|
|
with self.connection.cursor() as cursor:
|
|
# Querying USER_SDO_GEOM_METADATA to get the SRID and dimension information.
|
|
try:
|
|
cursor.execute(
|
|
'SELECT "DIMINFO", "SRID" FROM "USER_SDO_GEOM_METADATA" '
|
|
'WHERE "TABLE_NAME"=%s AND "COLUMN_NAME"=%s',
|
|
(table_name.upper(), description.name.upper())
|
|
)
|
|
row = cursor.fetchone()
|
|
except Exception as exc:
|
|
raise Exception(
|
|
'Could not find entry in USER_SDO_GEOM_METADATA '
|
|
'corresponding to "%s"."%s"' % (table_name, description.name)
|
|
) from exc
|
|
|
|
# TODO: Research way to find a more specific geometry field type for
|
|
# the column's contents.
|
|
field_type = 'GeometryField'
|
|
|
|
# Getting the field parameters.
|
|
field_params = {}
|
|
dim, srid = row
|
|
if srid != 4326:
|
|
field_params['srid'] = srid
|
|
# Size of object array (SDO_DIM_ARRAY) is number of dimensions.
|
|
dim = dim.size()
|
|
if dim != 2:
|
|
field_params['dim'] = dim
|
|
return field_type, field_params
|