owl-map/matcher/nominatim.py

from collections import OrderedDict

import json
import requests


class SearchError(Exception):
    pass


def lookup_with_params(**kwargs):
    url = "http://nominatim.openstreetmap.org/search"

    params = {
        "format": "jsonv2",
        "addressdetails": 1,
        "extratags": 1,
        "limit": 30,
        "namedetails": 1,
        "accept-language": "en",
        "polygon_text": 0,
    }
    params.update(kwargs)
    r = requests.get(url, params=params)
    if r.status_code == 500:
        raise SearchError

    try:
        return json.loads(r.text, object_pairs_hook=OrderedDict)
    except json.decoder.JSONDecodeError:
        raise SearchError(r)


def lookup(q):
    return lookup_with_params(q=q)


def get_us_county(county, state):
    if " " not in county and "county" not in county:
        county += " county"
    results = lookup(q="{}, {}".format(county, state))

    def pred(hit):
        return (
            "osm_type" in hit
            and hit["osm_type"] != "node"
            and county in hit["display_name"].lower()
        )

    return next(filter(pred, results), None)


def get_us_city(name, state):
    results = lookup_with_params(city=name, state=state)
    if len(results) != 1:
        results = [
            hit for hit in results if hit["type"] == "city" or hit["osm_type"] == "node"
        ]
        if len(results) != 1:
            print("more than one")
            return
    hit = results[0]
    if hit["type"] not in ("administrative", "city"):
        print("not a city")
        return
    if hit["osm_type"] == "node":
        print("node")
        return
    if not hit["display_name"].startswith(name):
        print("wrong name")
        return
    assert "osm_type" in hit and "osm_id" in hit and "geotext" in hit
    return hit


def get_hit_name(hit):
    address = hit.get("address")
    if not address:
        return hit["display_name"]

    address_values = list(address.values())
    n1 = address_values[0]
    if len(address) == 1:
        return n1

    country = address.pop("country")
    country_code = address.pop("country_code")
    if country_code:
        country_code == country_code.lower()

    if country_code == "us" and "state" in address:
        state = address["state"]
        return f"{n1}, {state}, USA"

    if country_code == "gb":
        country = "UK"

    if len(address) == 1:
        return f"{n1}, {country}"
    else:
        n2 = address_values[1]
        return f"{n1}, {n2}, {country}"

def get_hit_label(hit):
    tags = hit["extratags"]
    designation = tags.get("designation")
    category = hit["category"]
    hit_type = hit["type"]

    if designation:
        return designation.replace("_", " ")

    if category == "boundary" and hit_type == "administrative":
        place = tags.get("place") or tags.get("linked_place")

        if place:
            return f"{place} {category}"

    if category == "place":
        return f"{hit_type}"

    return f"{hit_type} {category}"