X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/e828d0d3f79400b3f3541b38d3a7d4de5d9cfc35..6f0eb1797000bb8ee9faa0657917803ed5bd50a6:/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 d9bf9e58..40f6a9e4 100644 --- a/nominatim/tools/special_phrases/special_phrase.py +++ b/nominatim/tools/special_phrases/special_phrase.py @@ -10,15 +10,28 @@ This class is a model used to transfer a special phrase through the process of load and importation. """ -class SpecialPhrase(): +from typing import Any + +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 = 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))