mirror of
https://github.com/osm-search/Nominatim.git
synced 2026-02-14 18:37:58 +00:00
100 lines
2.9 KiB
Python
100 lines
2.9 KiB
Python
# SPDX-License-Identifier: GPL-2.0-only
|
|
#
|
|
# This file is part of Nominatim. (https://nominatim.org)
|
|
#
|
|
# Copyright (C) 2025 by the Nominatim developer community.
|
|
# For a full list of authors see the git log.
|
|
"""
|
|
Collection of assertion functions used for the steps.
|
|
"""
|
|
import json
|
|
import math
|
|
import re
|
|
|
|
|
|
OSM_TYPE = {'N': 'node', 'W': 'way', 'R': 'relation',
|
|
'n': 'node', 'w': 'way', 'r': 'relation',
|
|
'node': 'n', 'way': 'w', 'relation': 'r'}
|
|
|
|
|
|
class OsmType:
|
|
""" Compares an OSM type, accepting both N/R/W and node/way/relation.
|
|
"""
|
|
|
|
def __init__(self, value):
|
|
self.value = value
|
|
|
|
def __eq__(self, other):
|
|
return other == self.value or other == OSM_TYPE[self.value]
|
|
|
|
def __str__(self):
|
|
return f"{self.value} or {OSM_TYPE[self.value]}"
|
|
|
|
|
|
class Field:
|
|
""" Generic comparator for fields, which looks at the type of the
|
|
value compared.
|
|
"""
|
|
def __init__(self, value, **extra_args):
|
|
self.value = value
|
|
self.extra_args = extra_args
|
|
|
|
def __eq__(self, other):
|
|
if isinstance(self.value, float):
|
|
return math.isclose(self.value, float(other), **self.extra_args)
|
|
|
|
if self.value.startswith('^'):
|
|
return re.fullmatch(self.value, str(other))
|
|
|
|
if isinstance(other, dict):
|
|
return other == eval('{' + self.value + '}')
|
|
|
|
return str(self.value) == str(other)
|
|
|
|
def __str__(self):
|
|
return str(self.value)
|
|
|
|
|
|
class Bbox:
|
|
""" Comparator for bounding boxes.
|
|
"""
|
|
def __init__(self, bbox_string):
|
|
self.coord = [float(x) for x in bbox_string.split(',')]
|
|
|
|
def __contains__(self, item):
|
|
if isinstance(item, str):
|
|
item = item.split(',')
|
|
item = list(map(float, item))
|
|
|
|
if len(item) == 2:
|
|
return self.coord[0] <= item[0] <= self.coord[2] \
|
|
and self.coord[1] <= item[1] <= self.coord[3]
|
|
|
|
if len(item) == 4:
|
|
return item[0] >= self.coord[0] and item[1] <= self.coord[1] \
|
|
and item[2] >= self.coord[2] and item[3] <= self.coord[3]
|
|
|
|
raise ValueError("Not a coordinate or bbox.")
|
|
|
|
def __str__(self):
|
|
return str(self.coord)
|
|
|
|
|
|
def check_for_attributes(obj, attrs, presence='present'):
|
|
""" Check that the object has the given attributes. 'attrs' is a
|
|
string with a comma-separated list of attributes. If 'presence'
|
|
is set to 'absent' then the function checks that the attributes do
|
|
not exist for the object
|
|
"""
|
|
def _dump_json():
|
|
return json.dumps(obj, sort_keys=True, indent=2, ensure_ascii=False)
|
|
|
|
for attr in attrs.split(','):
|
|
attr = attr.strip()
|
|
if presence == 'absent':
|
|
assert attr not in obj, \
|
|
f"Unexpected attribute {attr}. Full response:\n{_dump_json()}"
|
|
else:
|
|
assert attr in obj, \
|
|
f"No attribute '{attr}'. Full response:\n{_dump_json()}"
|