import json
import os

import flask

import db_handler

app = flask.Flask(__name__, template_folder='templates', static_folder='static', static_url_path='')


@app.route("/")
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()


@app.route("/voronoi_polygons.json")
def voronoi_polygons():
    d = db_handler.voronoi_polygons()
    b = []
    for a in json.loads(d)['geometries']:
        c = []
        for g in a['coordinates'][0]:
            c.append([g[1], g[0]])
        b.append(c)
    return json.dumps(b)