| 123456789101112131415161718192021222324252627282930313233 |
- from MySQLdb.constants import FIELD_TYPE
- from django.contrib.gis.gdal import OGRGeomType
- from django.db.backends.mysql.introspection import DatabaseIntrospection
- class MySQLIntrospection(DatabaseIntrospection):
- # Updating the data_types_reverse dictionary with the appropriate
- # type for Geometry fields.
- data_types_reverse = DatabaseIntrospection.data_types_reverse.copy()
- data_types_reverse[FIELD_TYPE.GEOMETRY] = 'GeometryField'
- def get_geometry_type(self, table_name, geo_col):
- cursor = self.connection.cursor()
- try:
- # In order to get the specific geometry type of the field,
- # we introspect on the table definition using `DESCRIBE`.
- cursor.execute('DESCRIBE %s' %
- self.connection.ops.quote_name(table_name))
- # Increment over description info until we get to the geometry
- # column.
- for column, typ, null, key, default, extra in cursor.fetchall():
- if column == geo_col:
- # Using OGRGeomType to convert from OGC name to Django field.
- # MySQL does not support 3D or SRIDs, so the field params
- # are empty.
- field_type = OGRGeomType(typ).django
- field_params = {}
- break
- finally:
- cursor.close()
- return field_type, field_params
|
