restrict deduplication to results from placex

All other sources do not have deduplicatable entries.
This commit is contained in:
Sarah Hoffmann
2023-09-06 10:43:55 +02:00
parent 9056c9276f
commit 8b56b55761

View File

@@ -108,6 +108,7 @@ def deduplicate_results(results: SearchResults, max_results: int) -> SearchResul
assert result.names and 'ref' in result.names assert result.names and 'ref' in result.names
if any(_is_postcode_relation_for(r, result.names['ref']) for r in results): if any(_is_postcode_relation_for(r, result.names['ref']) for r in results):
continue continue
if result.source_table == SourceTable.PLACEX:
classification = (result.osm_object[0] if result.osm_object else None, classification = (result.osm_object[0] if result.osm_object else None,
result.category, result.category,
result.display_name, result.display_name,
@@ -117,6 +118,8 @@ def deduplicate_results(results: SearchResults, max_results: int) -> SearchResul
deduped.append(result) deduped.append(result)
osm_ids_done.add(result.osm_object) osm_ids_done.add(result.osm_object)
classification_done.add(classification) classification_done.add(classification)
else:
deduped.append(result)
if len(deduped) >= max_results: if len(deduped) >= max_results:
break break