Skip more tags

This commit is contained in:
Edward Betts 2021-10-15 09:49:08 +01:00
parent 8efbc8a9c1
commit db7b13b706

View file

@ -6,7 +6,6 @@ from sqlalchemy.types import Float
from sqlalchemy.dialects import postgresql from sqlalchemy.dialects import postgresql
from matcher.planet import point, line, polygon from matcher.planet import point, line, polygon
from matcher import model, database, wikidata_api, wikidata from matcher import model, database, wikidata_api, wikidata
from matcher.data import extra_keys
from collections import Counter, defaultdict from collections import Counter, defaultdict
from flask import g, current_app from flask import g, current_app
import re import re
@ -17,6 +16,15 @@ srid = 4326
re_point = re.compile(r'^POINT\((.+) (.+)\)$') re_point = re.compile(r'^POINT\((.+) (.+)\)$')
entity_keys = {"labels", "sitelinks", "aliases", "claims", "descriptions", "lastrevid"} entity_keys = {"labels", "sitelinks", "aliases", "claims", "descriptions", "lastrevid"}
tag_prefixes = {
"disused",
"was",
"abandoned",
"demolished",
"destroyed",
"ruins",
"historic",
}
def get_country_iso3166_1(lat, lon): def get_country_iso3166_1(lat, lon):
point = func.ST_SetSRID(func.ST_MakePoint(lon, lat), srid) point = func.ST_SetSRID(func.ST_MakePoint(lon, lat), srid)
@ -110,6 +118,8 @@ def get_and_save_item(qid):
item = model.Item.query.get(entity_qid[1:]) item = model.Item.query.get(entity_qid[1:])
return item return item
if "claims" not in entity:
return
coords = wikidata.get_entity_coords(entity["claims"]) coords = wikidata.get_entity_coords(entity["claims"])
item_id = int(qid[1:]) item_id = int(qid[1:])
@ -196,6 +206,17 @@ WHERE tags ? 'wikidata'
conn = database.session.connection() conn = database.session.connection()
result = conn.execute(text(sql)) result = conn.execute(text(sql))
print(sql)
point_sql = f'''
SELECT 'point' as tbl, osm_id, tags, ST_AsText(ST_Centroid(way)) as centroid, ST_AsGeoJSON(way) as geojson
FROM planet_osm_point
WHERE ST_Intersects(ST_MakeEnvelope({bbox_str}, {srid}), way) and tags ? 'wikidata'
''' + extra_sql
print("point")
print(point_sql)
tagged = [] tagged = []
for tbl, osm_id, tags, centroid, geojson in result: for tbl, osm_id, tags, centroid, geojson in result:
if tbl == 'point': if tbl == 'point':
@ -233,22 +254,34 @@ def get_items(item_ids):
def get_item_tags(item): def get_item_tags(item):
skip_tags = {"Key:addr", "Key:addr:street", "Key:lit"} skip_tags = {
"Key:addr",
"Key:addr:street",
"Key:lit",
"Key:image",
"Key:name",
"Key:symbol",
"Key:brand",
}
isa_items = [] isa_items = []
isa_list = [v["numeric-id"] for v in item.get_claim("P31")] isa_list = [v["numeric-id"] for v in item.get_isa()]
isa_items = [(isa, []) for isa in get_items(isa_list)] isa_items = [(isa, []) for isa in get_items(isa_list)]
osm_list = defaultdict(list) osm_list = defaultdict(list)
skip_isa = {row[0] for row in database.session.query(model.SkipIsA.item_id)} skip_isa = {row[0] for row in database.session.query(model.SkipIsA.item_id)}
if item.is_tram_stop():
tram_stop_id = 41176
airport_id = 1248784
aerodrome_id = 62447
if {tram_stop_id, airport_id, aerodrome_id} & set(isa_list):
skip_isa.add(41176) # building (Q41176) skip_isa.add(41176) # building (Q41176)
seen = set(isa_list) | skip_isa seen = set(isa_list) | skip_isa
stop = { stop = {
"Q11799049", # public institution "Q11799049", # public institution
"Q7075", # library "Q7075", # library
} }
while isa_items: while isa_items:
isa, isa_path = isa_items.pop() isa, isa_path = isa_items.pop()
@ -256,8 +289,9 @@ def get_item_tags(item):
continue continue
isa_path = isa_path + [{'qid': isa.qid, 'label': isa.label()}] isa_path = isa_path + [{'qid': isa.qid, 'label': isa.label()}]
osm = [v for v in isa.get_claim("P1282") if v not in skip_tags] osm = [v for v in isa.get_claim("P1282") if v not in skip_tags]
if isa.qid in extra_keys:
osm += extra_keys[isa.qid] osm += [extra.tag_or_key
for extra in model.ItemExtraKeys.query.filter_by(item_id=isa.item_id)]
for i in osm: for i in osm:
osm_list[i].append(isa_path[:]) osm_list[i].append(isa_path[:])
@ -306,7 +340,7 @@ def wikidata_items_count(bounds, isa_filter=None):
if isa_filter: if isa_filter:
q = add_isa_filter(q, isa_filter) q = add_isa_filter(q, isa_filter)
print(q.statement.compile(compile_kwargs={"literal_binds": True})) # print(q.statement.compile(compile_kwargs={"literal_binds": True}))
return q.count() return q.count()
@ -345,11 +379,16 @@ def get_tag_filter(tags, tag_list):
tag_filter = [] tag_filter = []
for tag_or_key in tag_list: for tag_or_key in tag_list:
if tag_or_key.startswith("Key:"): if tag_or_key.startswith("Key:"):
tag_filter.append(and_(tags.has_key(tag_or_key[4:]), key = tag_or_key[4:]
tags[tag_or_key[4:]] != 'no')) tag_filter.append(and_(tags.has_key(key), tags[key] != 'no'))
for prefix in tag_prefixes:
tag_filter.append(tags.has_key(f"{prefix}:{key}"))
if tag_or_key.startswith("Tag:"): if tag_or_key.startswith("Tag:"):
k, _, v = tag_or_key.partition("=") k, _, v = tag_or_key[4:].partition("=")
tag_filter.append(tags[k[4:]] == v) tag_filter.append(tags[k] == v)
for prefix in tag_prefixes:
tag_filter.append(tags[f"{prefix}:{k}"] == v)
return tag_filter return tag_filter
@ -367,10 +406,18 @@ def get_preset_translations():
translation_dir = os.path.join(ts_dir, "dist", "translations") translation_dir = os.path.join(ts_dir, "dist", "translations")
for code in g.alpha2_codes: for code in g.alpha2_codes:
if code not in country_language: lang_code = country_language.get("code")
if not lang_code:
continue continue
filename = os.path.join(translation_dir, country_language[code] + ".json") filename = os.path.join(translation_dir, lang_code + ".json")
return json.load(open(filename))["en-GB"]["presets"]["presets"] json_data = json.load(open(filename))
if lang_code not in json_data:
continue
try:
return json_data[lang_code]["presets"]["presets"]
except KeyError:
pass
return {} return {}
@ -479,10 +526,10 @@ def osm_display_name(tags):
def street_address_in_tags(tags): def street_address_in_tags(tags):
return "addr:housenumber" in tags and "addr:street" in tags return "addr:housenumber" in tags and "addr:street" in tags
def find_osm_candidates(item, limit=60, max_distance=400, names=None): def find_osm_candidates(item, limit=80, max_distance=450, names=None):
item_id = item.item_id item_id = item.item_id
item_is_linear_feature = item.is_linear_feature()
item_is_street = item.is_street() item_is_street = item.is_street()
print("is street:", item_is_street)
check_is_street_number_first(item.locations[0].get_lat_lon()) check_is_street_number_first(item.locations[0].get_lat_lon())
@ -494,8 +541,10 @@ def find_osm_candidates(item, limit=60, max_distance=400, names=None):
tags = column('tags', postgresql.HSTORE) tags = column('tags', postgresql.HSTORE)
tag_list = get_item_tags(item) tag_list = get_item_tags(item)
# tag_filters = get_tag_filter(point.c.tags, tag_list)
# print(tag_filters)
s_point = (select([literal('point'), point.c.osm_id, point.c.tags.label('tags'), s_point = (select([literal('point').label('t'), point.c.osm_id, point.c.tags.label('tags'),
func.min(func.ST_DistanceSphere(model.ItemLocation.location, point.c.way)).label('dist'), func.min(func.ST_DistanceSphere(model.ItemLocation.location, point.c.way)).label('dist'),
func.ST_AsText(point.c.way), func.ST_AsText(point.c.way),
func.ST_AsGeoJSON(point.c.way), func.ST_AsGeoJSON(point.c.way),
@ -507,7 +556,7 @@ def find_osm_candidates(item, limit=60, max_distance=400, names=None):
or_(*get_tag_filter(point.c.tags, tag_list)))). or_(*get_tag_filter(point.c.tags, tag_list)))).
group_by(point.c.osm_id, point.c.tags, point.c.way)) group_by(point.c.osm_id, point.c.tags, point.c.way))
s_line = (select([literal('line'), line.c.osm_id, line.c.tags.label('tags'), s_line = (select([literal('line').label('t'), line.c.osm_id, line.c.tags.label('tags'),
func.min(func.ST_DistanceSphere(model.ItemLocation.location, line.c.way)).label('dist'), func.min(func.ST_DistanceSphere(model.ItemLocation.location, line.c.way)).label('dist'),
func.ST_AsText(func.ST_Centroid(func.ST_Collect(line.c.way))), func.ST_AsText(func.ST_Centroid(func.ST_Collect(line.c.way))),
func.ST_AsGeoJSON(func.ST_Collect(line.c.way)), func.ST_AsGeoJSON(func.ST_Collect(line.c.way)),
@ -518,7 +567,7 @@ def find_osm_candidates(item, limit=60, max_distance=400, names=None):
or_(*get_tag_filter(line.c.tags, tag_list)))). or_(*get_tag_filter(line.c.tags, tag_list)))).
group_by(line.c.osm_id, line.c.tags)) group_by(line.c.osm_id, line.c.tags))
s_polygon = (select([literal('polygon'), polygon.c.osm_id, polygon.c.tags.label('tags'), s_polygon = (select([literal('polygon').label('t'), polygon.c.osm_id, polygon.c.tags.label('tags'),
func.min(func.ST_DistanceSphere(model.ItemLocation.location, polygon.c.way)).label('dist'), func.min(func.ST_DistanceSphere(model.ItemLocation.location, polygon.c.way)).label('dist'),
func.ST_AsText(func.ST_Centroid(func.ST_Collect(polygon.c.way))), func.ST_AsText(func.ST_Centroid(func.ST_Collect(polygon.c.way))),
func.ST_AsGeoJSON(func.ST_Collect(polygon.c.way)), func.ST_AsGeoJSON(func.ST_Collect(polygon.c.way)),
@ -530,11 +579,11 @@ def find_osm_candidates(item, limit=60, max_distance=400, names=None):
group_by(polygon.c.osm_id, polygon.c.tags). group_by(polygon.c.osm_id, polygon.c.tags).
having(func.ST_Area(func.ST_Collect(polygon.c.way)) < 20 * func.ST_Area(bbox_list[0]))) having(func.ST_Area(func.ST_Collect(polygon.c.way)) < 20 * func.ST_Area(bbox_list[0])))
tables = ([] if item_is_street else [s_point]) + [s_line, s_polygon] tables = ([] if item_is_linear_feature else [s_point]) + [s_line, s_polygon]
s = select([union(*tables).alias()]).where(dist < max_distance).order_by(dist) s = select([union(*tables).alias()]).where(dist < max_distance).order_by(dist)
if names: if names:
s = s.where(tags["name"].in_(names)) s = s.where(or_(tags["name"].in_(names), tags["old_name"].in_(names)))
if item_is_street: if item_is_street:
s = s.where(tags["highway"] != "bus_stop") s = s.where(tags["highway"] != "bus_stop")
@ -550,7 +599,7 @@ def find_osm_candidates(item, limit=60, max_distance=400, names=None):
if limit: if limit:
s = s.limit(limit) s = s.limit(limit)
# print(s.compile(compile_kwargs={"literal_binds": True})) print(s.compile(compile_kwargs={"literal_binds": True}))
conn = database.session.connection() conn = database.session.connection()
nearby = [] nearby = []
@ -648,8 +697,16 @@ def item_detail(item):
street_address = get_item_street_addresses(item) street_address = get_item_street_addresses(item)
heritage_designation = [get_item(v["numeric-id"]).label() heritage_designation = []
for v in item.get_claim("P1435")] for v in item.get_claim("P1435"):
if not v:
print('heritage designation missing:', item.qid)
continue
heritage_designation_item = get_item(v["numeric-id"])
heritage_designation.append({
"qid": v["id"],
"label": heritage_designation_item.label(),
})
d = { d = {
"qid": item.qid, "qid": item.qid,
@ -658,7 +715,11 @@ def item_detail(item):
"markers": locations, "markers": locations,
"image_list": image_filenames, "image_list": image_filenames,
"street_address": street_address, "street_address": street_address,
"isa_list": item.get_isa_qids(), # "isa_list": item.get_isa_qids(),
"isa_list": [
{"qid": isa["id"], "label": get_item(isa["numeric-id"]).label()}
for isa in item.get_isa()
],
"closed": item.closed(), "closed": item.closed(),
"heritage_designation": heritage_designation, "heritage_designation": heritage_designation,
} }
@ -746,6 +807,8 @@ def isa_incremental_search(search_terms):
func.length(en_label) < 20, func.length(en_label) < 20,
) )
print(q.statement.compile(compile_kwargs={"literal_binds": True}))
ret = [] ret = []
for item in q: for item in q:
cur = { cur = {