import json

with open("data/osm/shops.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": "shop",
            "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_shops():
    return ress