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


@app.route("/all.json")
def all():
    rows = db_handler.get_all()
    for row in rows:
        print(row['polygon'])
        coordinates = []
        for coordinate in json.loads(row['polygon'])['coordinates'][0]:
            coordinates.append([coordinate[1], coordinate[0]])
        row['polygon'] = coordinates
        print(row['polygon'])
    return json.dumps(rows)