aboutsummaryrefslogtreecommitdiffstats
path: root/app.py
blob: 2d6d8ecb18299c77f78da28f94c731101a7fe712 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
import database
import urllib.parse
import flask
import json
import os

app = flask.Flask(__name__)

if not os.path.exists(".docker"):
    import dotenv
    dotenv.load_dotenv(dotenv_path = "db.env")
    host = "srv.home"
else:
    host = "db" 

@app.route("/")
def serve_index():
    with open("charts.json", "r") as f:
        charts = json.load(f)

    return flask.render_template(
        "index.html.j2",
        title = "UK Gender Pay Gap",
        charts = charts["index"]
    )

@app.route("/search_click", methods = ["POST"])
def search_redirect():
    return flask.redirect("/search?s=%s" % urllib.parse.quote_plus(dict(flask.request.form)["search"]))

@app.route("/search")
def search():
    with database.PayGapDatabase(host = host) as db:
        search_text = flask.request.args.get("s")
        companies = db.search_company(search_text)
        if len(companies) == 1:
            return flask.redirect("/company/%s" % companies[0][1])

        return flask.render_template(
            "search.html.j2",
            title = "Search",
            companies = companies
        )

if __name__ == "__main__":
    app.run("0.0.0.0", port = 5005, debug = True)