class AbstractSearch(abc.ABC):
""" Encapuslation of a single lookup in the database.
"""
+ SEARCH_PRIO: int = 2
def __init__(self, penalty: float) -> None:
self.penalty = penalty
class CountrySearch(AbstractSearch):
""" Search for a country name or country code.
"""
+ SEARCH_PRIO = 0
+
def __init__(self, sdata: SearchData) -> None:
super().__init__(sdata.penalty)
self.countries = sdata.countries
class PlaceSearch(AbstractSearch):
""" Generic search for an address or named place.
"""
+ SEARCH_PRIO = 1
+
def __init__(self, extra_penalty: float, sdata: SearchData, expected_count: int) -> None:
super().__init__(sdata.penalty + extra_penalty)
self.countries = sdata.countries
log().table_dump('Searches for assignment',
_dump_searches(searches, query, num_searches))
num_searches = len(searches)
- searches.sort(key=lambda s: s.penalty)
+ searches.sort(key=lambda s: (s.penalty, s.SEARCH_PRIO))
return query, searches