X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/9f6f12cfeb5ab51d2604b459f668c21530d7ad26..566ac7af993f487e249185ddae59bc38c38f84de:/nominatim/db/sqlalchemy_types.py diff --git a/nominatim/db/sqlalchemy_types.py b/nominatim/db/sqlalchemy_types.py index e34dc7c1..f31966cd 100644 --- a/nominatim/db/sqlalchemy_types.py +++ b/nominatim/db/sqlalchemy_types.py @@ -7,14 +7,17 @@ """ Custom types for SQLAlchemy. """ -from typing import Callable, Any +from typing import Callable, Any, cast +import sys import sqlalchemy as sa -import sqlalchemy.types as types +from sqlalchemy import types -from nominatim.typing import SaColumn +from nominatim.typing import SaColumn, SaBind -class Geometry(types.UserDefinedType[Any]): +#pylint: disable=all + +class Geometry(types.UserDefinedType): # type: ignore[type-arg] """ Simplified type decorator for PostGIS geometry. This type only supports geometries in 4326 projection. """ @@ -28,29 +31,29 @@ class Geometry(types.UserDefinedType[Any]): return f'GEOMETRY({self.subtype}, 4326)' - def bind_processor(self, dialect: sa.Dialect) -> Callable[[Any], str]: + 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 result_processor(self, dialect: sa.Dialect, coltype: object) -> Callable[[Any], str]: + def result_processor(self, dialect: 'sa.Dialect', coltype: object) -> Callable[[Any], str]: def process(value: Any) -> str: assert isinstance(value, str) return value return process - def bind_expression(self, bindvalue: sa.BindParameter[Any]) -> SaColumn: - return sa.func.ST_GeomFromText(bindvalue, type_=self) + def bind_expression(self, bindvalue: SaBind) -> SaColumn: + 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) -> SaColumn: + def intersects(self, other: SaColumn) -> 'sa.Operators': return self.op('&&')(other) def is_line_like(self) -> SaColumn: