add store heatmap

1 files changed, 13 insertions, 0 deletions

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: