Format code with black

This commit is contained in:
Edward Betts 2023-10-10 10:08:59 +01:00
parent 6e9b2eb50e
commit 5c01d9aebf
4 changed files with 11 additions and 13 deletions

View file

@ -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))

View file

@ -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()

View file

@ -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}"