From 0d1ba0fc1bc2c64c7aaaf71f98fd96f35b8b1bab Mon Sep 17 00:00:00 2001 From: user Date: Sun, 7 Apr 2024 23:10:05 +0200 Subject: add store heatmap --- src/db_handler.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) (limited to 'src/db_handler.py') diff --git a/src/db_handler.py b/src/db_handler.py index 0fd3ec1..5a2ab73 100644 --- a/src/db_handler.py +++ b/src/db_handler.py @@ -56,6 +56,19 @@ def get_supermarkets() -> (list[dict], list[str]): return supermarkets +def get_stores() -> (list[dict], list[str]): + with psycopg.connect(**db_con_params) as conn: + with conn.cursor() as cur: + cur.execute(""" + SELECT + ST_Y(ST_Transform(geom, 4326)) AS lat, ST_X(ST_Transform(geom, 4326)) AS long + FROM shop; + """) + + stores = cur.fetchall() + return stores + + def voronoi_polygons() -> (list[dict], list[str]): with psycopg.connect(**db_con_params, row_factory=psycopg.rows.dict_row) as conn: with conn.cursor() as cur: -- cgit v1.2.3