forked from sfd226/vote
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathapp.py
51 lines (40 loc) · 1.2 KB
/
app.py
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
47
48
49
50
51
from flask import Flask, render_template, request, make_response, g
from redis import Redis
import os
import socket
import random
import json
option_a = os.getenv("OPTION_A", "Visual Studio")
option_b = os.getenv("OPTION_B", "JetBrains Rider")
hostname = socket.gethostname()
version = "v1"
app = Flask(__name__)
def get_redis():
if not hasattr(g, "redis"):
g.redis = Redis(host="redis", db=0, socket_timeout=5)
return g.redis
@app.route("/", methods=["POST", "GET"])
def hello():
voter_id = request.cookies.get("voter_id")
if not voter_id:
voter_id = hex(random.getrandbits(64))[2:-1]
vote = None
if request.method == "POST":
redis = get_redis()
vote = request.form["vote"]
data = json.dumps({"voter_id": voter_id, "vote": vote})
redis.rpush("votes", data)
resp = make_response(
render_template(
"index.html",
option_a=option_a,
option_b=option_b,
hostname=hostname,
vote=vote,
version=version,
)
)
resp.set_cookie("voter_id", voter_id)
return resp
if __name__ == "__main__":
app.run(host="0.0.0.0", port=80, debug=True, threaded=True)