X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/2f4342810d4bf1d12cb6064468ad8dcd9a7f62c3..513504140512d17ddbe0769b2ec3887ee16f7e31:/nominatim/api/types.py diff --git a/nominatim/api/types.py b/nominatim/api/types.py index aa3256cd..60495dd0 100644 --- a/nominatim/api/types.py +++ b/nominatim/api/types.py @@ -14,9 +14,9 @@ import dataclasses import enum import math from struct import unpack +from binascii import unhexlify -from geoalchemy2 import WKTElement -import geoalchemy2.functions +import sqlalchemy as sa from nominatim.errors import UsageError @@ -73,11 +73,13 @@ class Point(NamedTuple): @staticmethod - def from_wkb(wkb: bytes) -> 'Point': + def from_wkb(wkb: Union[str, bytes]) -> 'Point': """ Create a point from EWKB as returned from the database. """ + if isinstance(wkb, str): + wkb = unhexlify(wkb) if len(wkb) != 25: - raise ValueError("Point wkb has unexpected length") + raise ValueError(f"Point wkb has unexpected length {len(wkb)}") if wkb[0] == 0: gtype, srid, x, y = unpack('>iidd', wkb[1:]) elif wkb[0] == 1: @@ -122,10 +124,10 @@ class Point(NamedTuple): return Point(x, y) - def sql_value(self) -> WKTElement: - """ Create an SQL expression for the point. + def to_wkt(self) -> str: + """ Return the WKT representation of the point. """ - return WKTElement(f'POINT({self.x} {self.y})', srid=4326) + return f'POINT({self.x} {self.y})' @@ -179,12 +181,6 @@ class Bbox: return (self.coords[2] - self.coords[0]) * (self.coords[3] - self.coords[1]) - def sql_value(self) -> Any: - """ Create an SQL expression for the box. - """ - return geoalchemy2.functions.ST_MakeEnvelope(*self.coords, 4326) - - def contains(self, pt: Point) -> bool: """ Check if the point is inside or on the boundary of the box. """ @@ -192,14 +188,24 @@ class Bbox: and self.coords[2] >= pt[0] and self.coords[3] >= pt[1] + def to_wkt(self) -> str: + """ Return the WKT representation of the Bbox. This + is a simple polygon with four points. + """ + return 'POLYGON(({0} {1},{0} {3},{2} {3},{2} {1},{0} {1}))'.format(*self.coords) + + @staticmethod - def from_wkb(wkb: Optional[bytes]) -> 'Optional[Bbox]': + def from_wkb(wkb: Union[None, str, bytes]) -> 'Optional[Bbox]': """ Create a Bbox from a bounding box polygon as returned by the database. Return s None if the input value is None. """ if wkb is None: return None + if isinstance(wkb, str): + wkb = unhexlify(wkb) + if len(wkb) != 97: raise ValueError("WKB must be a bounding box polygon") if wkb.startswith(WKB_BBOX_HEADER_LE): @@ -306,7 +312,7 @@ def format_excluded(ids: Any) -> List[int]: (isinstance(i, str) and (not i or i.isdigit())) for i in plist): raise UsageError("Parameter 'excluded' only takes place IDs.") - return [int(id) for id in plist if id] + return [int(id) for id in plist if id] or [0] def format_categories(categories: List[Tuple[str, str]]) -> List[Tuple[str, str]]: @@ -446,6 +452,8 @@ class SearchDetails(LookupDetails): yext = (self.viewbox.maxlat - self.viewbox.minlat)/2 self.viewbox_x2 = Bbox(self.viewbox.minlon - xext, self.viewbox.minlat - yext, self.viewbox.maxlon + xext, self.viewbox.maxlat + yext) + else: + self.viewbox_x2 = None def restrict_min_max_rank(self, new_min: int, new_max: int) -> None: