mirror of
https://github.com/task-17-lct/backend.git
synced 2024-11-30 20:23:49 +03:00
65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
|
import json
|
||
|
|
|
||
|
|
with open("data/osm/theme_park.json") as f:
|
||
|
|
data = json.load(f)
|
||
|
|
|
||
|
|
nodes_loc = {}
|
||
|
|
|
||
|
|
for el in data["elements"]:
|
||
|
|
if "lat" in el and "lon" in el:
|
||
|
|
nodes_loc[el["id"]] = [el["lat"], el["lon"]]
|
||
|
|
|
||
|
|
|
||
|
|
ress = []
|
||
|
|
|
||
|
|
for el in data["elements"]:
|
||
|
|
if "tags" in el and "name" in el["tags"] and "lat" in el and "lon" in el:
|
||
|
|
nodes = []
|
||
|
|
|
||
|
|
if "nodes" in el:
|
||
|
|
nodes = [nodes_loc[x] for x in el["nodes"]]
|
||
|
|
|
||
|
|
info = el["tags"]
|
||
|
|
if "tourism" in info:
|
||
|
|
del info["tourism"]
|
||
|
|
res = {
|
||
|
|
"title": info["name:ru"] if "name:ru" in info else info["name"],
|
||
|
|
"type": "theme_park",
|
||
|
|
"parser_source": "openstreetmap.org",
|
||
|
|
"lat": el["lat"],
|
||
|
|
"lon": el["lon"],
|
||
|
|
}
|
||
|
|
if "addr:full" in info:
|
||
|
|
res["address"] = info["addr:full"]
|
||
|
|
else:
|
||
|
|
addr_res = ""
|
||
|
|
names = [x.split(":")[1] for x in info.keys() if "addr:" in x]
|
||
|
|
|
||
|
|
for nnn in ["postcode", "city", "street", "housenumber"]:
|
||
|
|
if nnn in names:
|
||
|
|
val = info["addr:" + nnn]
|
||
|
|
if addr_res:
|
||
|
|
if nnn in ["city", "street"]:
|
||
|
|
addr_res += f", {val}"
|
||
|
|
else:
|
||
|
|
addr_res += f" {val}"
|
||
|
|
else:
|
||
|
|
addr_res = val
|
||
|
|
|
||
|
|
if addr_res:
|
||
|
|
res["address"] = addr_res
|
||
|
|
|
||
|
|
if "description" in info:
|
||
|
|
res["description"] = info["description"]
|
||
|
|
|
||
|
|
res["extra_kwargs"] = info
|
||
|
|
|
||
|
|
if nodes:
|
||
|
|
res["extra_kwargs"]["nodes"] = nodes
|
||
|
|
|
||
|
|
ress.append(res)
|
||
|
|
|
||
|
|
|
||
|
|
def get_theme_park():
|
||
|
|
return ress
|