102 lines
3.8 KiB
Python
102 lines
3.8 KiB
Python
from django.contrib.gis.db.backends.base.adapter import WKTAdapter
|
|
from django.contrib.gis.db.backends.base.operations import (
|
|
BaseSpatialOperations,
|
|
)
|
|
from django.contrib.gis.db.backends.utils import SpatialOperator
|
|
from django.contrib.gis.db.models import GeometryField, aggregates
|
|
from django.db.backends.mysql.operations import DatabaseOperations
|
|
from django.utils.functional import cached_property
|
|
|
|
|
|
class MySQLOperations(BaseSpatialOperations, DatabaseOperations):
|
|
|
|
mysql = True
|
|
name = 'mysql'
|
|
|
|
Adapter = WKTAdapter
|
|
|
|
@cached_property
|
|
def geom_func_prefix(self):
|
|
return '' if self.is_mysql_5_5 else 'ST_'
|
|
|
|
@cached_property
|
|
def is_mysql_5_5(self):
|
|
return self.connection.mysql_version < (5, 6, 1)
|
|
|
|
@cached_property
|
|
def is_mysql_5_6(self):
|
|
return self.connection.mysql_version < (5, 7, 6)
|
|
|
|
@cached_property
|
|
def uses_invalid_empty_geometry_collection(self):
|
|
return self.connection.mysql_version >= (5, 7, 5)
|
|
|
|
@cached_property
|
|
def select(self):
|
|
return self.geom_func_prefix + 'AsText(%s)'
|
|
|
|
@cached_property
|
|
def from_wkb(self):
|
|
return self.geom_func_prefix + 'GeomFromWKB'
|
|
|
|
@cached_property
|
|
def from_text(self):
|
|
return self.geom_func_prefix + 'GeomFromText'
|
|
|
|
@cached_property
|
|
def gis_operators(self):
|
|
MBREquals = 'MBREqual' if self.is_mysql_5_6 else 'MBREquals'
|
|
return {
|
|
'bbcontains': SpatialOperator(func='MBRContains'), # For consistency w/PostGIS API
|
|
'bboverlaps': SpatialOperator(func='MBROverlaps'), # ...
|
|
'contained': SpatialOperator(func='MBRWithin'), # ...
|
|
'contains': SpatialOperator(func='MBRContains'),
|
|
'disjoint': SpatialOperator(func='MBRDisjoint'),
|
|
'equals': SpatialOperator(func=MBREquals),
|
|
'exact': SpatialOperator(func=MBREquals),
|
|
'intersects': SpatialOperator(func='MBRIntersects'),
|
|
'overlaps': SpatialOperator(func='MBROverlaps'),
|
|
'same_as': SpatialOperator(func=MBREquals),
|
|
'touches': SpatialOperator(func='MBRTouches'),
|
|
'within': SpatialOperator(func='MBRWithin'),
|
|
}
|
|
|
|
@cached_property
|
|
def function_names(self):
|
|
return {'Length': 'GLength'} if self.is_mysql_5_5 else {}
|
|
|
|
disallowed_aggregates = (
|
|
aggregates.Collect, aggregates.Extent, aggregates.Extent3D,
|
|
aggregates.MakeLine, aggregates.Union,
|
|
)
|
|
|
|
@cached_property
|
|
def unsupported_functions(self):
|
|
unsupported = {
|
|
'AsGML', 'AsKML', 'AsSVG', 'Azimuth', 'BoundingCircle', 'ForceRHR',
|
|
'LineLocatePoint', 'MakeValid', 'MemSize', 'Perimeter',
|
|
'PointOnSurface', 'Reverse', 'Scale', 'SnapToGrid', 'Transform',
|
|
'Translate',
|
|
}
|
|
if self.connection.mysql_version < (5, 7, 5):
|
|
unsupported.update({'AsGeoJSON', 'GeoHash', 'IsValid'})
|
|
if self.is_mysql_5_5:
|
|
unsupported.update({'Difference', 'Distance', 'Intersection', 'SymDifference', 'Union'})
|
|
return unsupported
|
|
|
|
def geo_db_type(self, f):
|
|
return f.geom_type
|
|
|
|
def get_db_converters(self, expression):
|
|
converters = super().get_db_converters(expression)
|
|
if isinstance(expression.output_field, GeometryField) and self.uses_invalid_empty_geometry_collection:
|
|
converters.append(self.convert_invalid_empty_geometry_collection)
|
|
return converters
|
|
|
|
# https://dev.mysql.com/doc/refman/en/spatial-function-argument-handling.html
|
|
# MySQL 5.7.5 adds support for the empty geometry collections, but they are represented with invalid WKT.
|
|
def convert_invalid_empty_geometry_collection(self, value, expression, connection):
|
|
if value == b'GEOMETRYCOLLECTION()':
|
|
return b'GEOMETRYCOLLECTION EMPTY'
|
|
return value
|