X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/c3788d765ed4e7ddf00794085af757714bc102cf..bdded69ab636ab8e3bb46322ebbe4d2f9ec41614:/nominatim/tools/special_phrases/special_phrase.py diff --git a/nominatim/tools/special_phrases/special_phrase.py b/nominatim/tools/special_phrases/special_phrase.py index dc7f69fe..40f6a9e4 100644 --- a/nominatim/tools/special_phrases/special_phrase.py +++ b/nominatim/tools/special_phrases/special_phrase.py @@ -10,17 +10,28 @@ This class is a model used to transfer a special phrase through the process of load and importation. """ -import re +from typing import Any -class SpecialPhrase(): +class SpecialPhrase: """ Model representing a special phrase. """ - def __init__(self, p_label, p_class, p_type, p_operator): + def __init__(self, p_label: str, p_class: str, p_type: str, p_operator: str) -> None: self.p_label = p_label.strip() self.p_class = p_class.strip() - # Hack around a bug where building=yes was imported with quotes into the wiki - self.p_type = re.sub(r'\"|"', '', p_type.strip()) + self.p_type = p_type.strip() # Needed if some operator in the wiki are not written in english p_operator = p_operator.strip().lower() self.p_operator = '-' if p_operator not in ('near', 'in') else p_operator + + def __eq__(self, other: Any) -> bool: + if not isinstance(other, SpecialPhrase): + return False + + return self.p_label == other.p_label \ + and self.p_class == other.p_class \ + and self.p_type == other.p_type \ + and self.p_operator == other.p_operator + + def __hash__(self) -> int: + return hash((self.p_label, self.p_class, self.p_type, self.p_operator))