show all stores, and their respective polygons

1 files changed, 9 insertions, 1 deletions

diff --git a/src/shop_map_node5_net.py b/src/shop_map_node5_net.py
index 5154a0d..eafad5c 100644
--- a/src/shop_map_node5_net.py
+++ b/src/shop_map_node5_net.py
@@ -1,10 +1,10 @@
 import json
+import os
 
 import flask
 
 import db_handler
 
-
 app = flask.Flask(__name__, template_folder='templates', static_folder='static', static_url_path='')
 
 
@@ -13,6 +13,14 @@ def index():
     return flask.redirect('/index.html')
 
 
+@app.route("/chains")
+def chains():
+    icons = os.listdir(os.path.join(os.getcwd(), 'src', 'static', 'icons'))
+    icons.pop(icons.index('Unknown.png'))
+    chains = [os.path.splitext(a)[0] for a in icons]
+    return json.dumps(chains)
+
+
 @app.route("/supermarkets.json")
 def supermarkets():
     return db_handler.get_supermarkets()