"""
Custom types for SQLAlchemy.
"""
-from typing import Callable, Any
+from typing import Callable, Any, cast
+import sys
import sqlalchemy as sa
from sqlalchemy import types
#pylint: disable=all
-class Geometry(types.UserDefinedType[Any]):
+class Geometry(types.UserDefinedType): # type: ignore[type-arg]
""" Simplified type decorator for PostGIS geometry. This type
only supports geometries in 4326 projection.
"""
def bind_processor(self, dialect: 'sa.Dialect') -> Callable[[Any], str]:
def process(value: Any) -> str:
if isinstance(value, str):
- return 'SRID=4326;' + value
+ return value
- return 'SRID=4326;' + value.to_wkt()
+ return cast(str, value.to_wkt())
return process
def bind_expression(self, bindvalue: SaBind) -> SaColumn:
- return sa.func.ST_GeomFromText(bindvalue, type_=self)
+ return sa.func.ST_GeomFromText(bindvalue, sa.text('4326'), type_=self)
- class comparator_factory(types.UserDefinedType.Comparator):
+ class comparator_factory(types.UserDefinedType.Comparator): # type: ignore[type-arg]
def intersects(self, other: SaColumn) -> 'sa.Operators':
return self.op('&&')(other)