From 5c01d9aebf4310fce34c059456f8c219b22f055c Mon Sep 17 00:00:00 2001 From: Edward Betts Date: Tue, 10 Oct 2023 10:08:59 +0100 Subject: [PATCH] Format code with black --- geocode/model.py | 12 +++++++----- geocode/scotland.py | 1 + geocode/wikidata.py | 1 + lookup.py | 10 ++-------- 4 files changed, 11 insertions(+), 13 deletions(-) diff --git a/geocode/model.py b/geocode/model.py index 8230b0b..a4b1802 100644 --- a/geocode/model.py +++ b/geocode/model.py @@ -12,6 +12,7 @@ from .database import session Base = declarative_base() Base.query = session.query_property() + class Polygon(Base): __tablename__ = "planet_osm_polygon" @@ -35,10 +36,12 @@ class Polygon(Base): @classmethod def coords_within(cls, lat, lon): point = func.ST_SetSRID(func.ST_MakePoint(lon, lat), 4326) - return (cls.query.filter(cls.admin_level.isnot(None), - cls.admin_level.regexp_match("^\d+$"), - func.ST_Within(point, cls.way)) - .order_by(cls.area, cast(cls.admin_level, Integer).desc())) + return cls.query.filter( + cls.admin_level.isnot(None), + cls.admin_level.regexp_match("^\d+$"), + func.ST_Within(point, cls.way), + ).order_by(cls.area, cast(cls.admin_level, Integer).desc()) + class Scotland(Base): __tablename__ = "scotland" @@ -54,4 +57,3 @@ class Scotland(Base): name = Column(String(50)) geom = Column(Geometry("MULTIPOLYGON", srid=27700)) - diff --git a/geocode/scotland.py b/geocode/scotland.py index 60c9b43..d87dd2b 100644 --- a/geocode/scotland.py +++ b/geocode/scotland.py @@ -1,6 +1,7 @@ from flask import current_app import psycopg2 + def get_scotland_code(lat, lon): conn = psycopg2.connect(**current_app.config["DB_PARAMS"]) cur = conn.cursor() diff --git a/geocode/wikidata.py b/geocode/wikidata.py index 7588816..7e17c0c 100644 --- a/geocode/wikidata.py +++ b/geocode/wikidata.py @@ -65,6 +65,7 @@ def wd_uri_to_qid(value): assert value.startswith(wd_entity) return value[len(wd_entity) - 1 :] + def geosearch_query(lat, lon): if isinstance(lat, float): lat = f"{lat:f}" diff --git a/lookup.py b/lookup.py index c75d1d0..80efb8b 100755 --- a/lookup.py +++ b/lookup.py @@ -12,7 +12,7 @@ database.init_app(app) def get_random_lat_lon(): - """ Select random lat/lon within the UK """ + """Select random lat/lon within the UK""" south, east = 50.8520, 0.3536 north, west = 53.7984, -2.7296 @@ -179,13 +179,7 @@ def detail_page(): reply = lat_lon_to_wikidata(lat, lon) except wikidata.QueryError as e: query, r = e.args - return render_template( - "query_error.html", - lat=lat, - lon=lon, - query=query, - r=r - ) + return render_template("query_error.html", lat=lat, lon=lon, query=query, r=r) return render_template("detail.html", lat=lat, lon=lon, **reply)