prioritize country searches when penaly is equal

This commit is contained in:
Sarah Hoffmann 2024-01-07 11:11:11 +01:00
parent dc52d0954e
commit 4305160c91
2 changed files with 6 additions and 1 deletions

View File

@ -245,6 +245,7 @@ async def _get_tiger(conn: SearchConnection, place_ids: List[int],
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
@ -448,6 +449,8 @@ class PoiSearch(AbstractSearch):
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
@ -604,6 +607,8 @@ class PostcodeSearch(AbstractSearch):
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

View File

@ -64,7 +64,7 @@ class ForwardGeocoder:
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