edward
/
owl-map
1
1
Fork 1
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

Add types.

This commit is contained in:
Edward Betts 2023-05-14 18:17:16 +00:00
parent 2424fe0fdf
commit b1ae7aa8dd
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
matcher/api.py
View File

@ -40,21 +40,20 @@ skip_tags = {
} }
def get_country_iso3166_1(lat, lon): def get_country_iso3166_1(lat: float, lon: float) -> set[str]:
"""For a given lat/lon return a set of ISO country codes. """For a given lat/lon return a set of ISO country codes.
Also cache the country code in the global object. Also cache the country code in the global object.
Normally there should be only one country. Normally there should be only one country.
""" """
point = func.ST_SetSRID(func.ST_MakePoint(lon, lat), srid) point = func.ST_SetSRID(func.ST_MakePoint(lon, lat), srid)
alpha2_codes = set() alpha2_codes: set[str] = set()
q = model.Polygon.query.filter( q = model.Polygon.query.filter(
func.ST_Covers(model.Polygon.way, point), model.Polygon.admin_level == "2" func.ST_Covers(model.Polygon.way, point), model.Polygon.admin_level == "2"
) )
for country in q: for country in q:
alpha2 = country.tags.get("ISO3166-1") alpha2: str = country.tags.get("ISO3166-1")
if not alpha2: if not alpha2:
continue continue
alpha2_codes.add(alpha2) alpha2_codes.add(alpha2)