osm-search/Nominatim
1
1
Fork 0
mirror of https://github.com/osm-search/Nominatim.git synced 2024-11-22 12:06:27 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #3590 from lonvia/lookup-per-osm-type

Browse Source 
Look up different OSM types in placex separately
This commit is contained in:
Sarah Hoffmann 2024-11-15 09:44:16 +01:00 committed by GitHub
commit 689bcbd6ea
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
src/nominatim_api/lookup.py
View File

@ -201,22 +201,24 @@ async def find_in_placex(conn: SearchConnection, collector: Collector) -> bool:
t.c.geometry.ST_Expand(0).label('bbox'), t.c.geometry.ST_Expand(0).label('bbox'),
t.c.centroid) t.c.centroid)
osm_ids = [{'i': i, 'ot': p.osm_type, 'oi': p.osm_id, 'oc': p.osm_class or ''} for osm_type in ('N', 'W', 'R'):
for i, p in collector.enumerate_free_osm_ids()] osm_ids = [{'i': i, 'oi': p.osm_id, 'oc': p.osm_class or ''}
for i, p in collector.enumerate_free_osm_ids()
if p.osm_type == osm_type]
if osm_ids: if osm_ids:
oid_tab = sa.func.JsonArrayEach(sa.type_coerce(osm_ids, sa.JSON))\ oid_tab = sa.func.JsonArrayEach(sa.type_coerce(osm_ids, sa.JSON))\
.table_valued(sa.column('value', type_=sa.JSON)) .table_valued(sa.column('value', type_=sa.JSON))
psql = sql.add_columns(oid_tab.c.value['i'].as_integer().label('_idx'))\ psql = sql.add_columns(oid_tab.c.value['i'].as_integer().label('_idx'))\
.where(t.c.osm_type == oid_tab.c.value['ot'].as_string())\ .where(t.c.osm_type == osm_type)\
.where(t.c.osm_id == oid_tab.c.value['oi'].as_string().cast(sa.BigInteger))\ .where(t.c.osm_id == oid_tab.c.value['oi'].as_string().cast(sa.BigInteger))\
.where(sa.or_(oid_tab.c.value['oc'].as_string() == '', .where(sa.or_(oid_tab.c.value['oc'].as_string() == '',
oid_tab.c.value['oc'].as_string() == t.c.class_))\ oid_tab.c.value['oc'].as_string() == t.c.class_))\
.order_by(t.c.class_) .order_by(t.c.class_)
if await collector.add_rows_from_sql(conn, psql, t.c.geometry, if await collector.add_rows_from_sql(conn, psql, t.c.geometry,
nres.create_from_placex_row): nres.create_from_placex_row):
return True return True
place_ids = [{'i': i, 'id': p.place_id} place_ids = [{'i': i, 'id': p.place_id}
for i, p in collector.enumerate_free_place_ids()] for i, p in collector.enumerate_free_place_ids()]