-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathapp.py
55 lines (48 loc) · 1.66 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
52
53
54
55
import os
from flask import *
from flaskext.sqlalchemy import SQLAlchemy
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = os.environ['DATABASE_URL']
db = SQLAlchemy(app)
db.create_all()
@app.route("/", methods = ["GET", "POST"])
def index():
if request.method == "GET":
#log_access(request)
return render_template('form.html')
elif request.method == "POST":
log_form(request)
return render_template('chart.html', income=request.form['income'])
def log_access(request):
entry = Entry(request.environ['REMOTE_ADDR'],
request.environ['HTTP_USER_AGENT'],
time.gmtime())
db.session.add(entry)
db.session.commit()
def log_form(request):
entry = Entry(request.environ['REMOTE_ADDR'],
request.environ['HTTP_USER_AGENT'],
time.gmtime(),
request.form['zip'],
request.form['income'],
request.form['pct'])
db.session.add(entry)
db.session.commit()
if __name__ == "__main__":
port = int(os.environ.get("PORT", 5000))
app.run(host='0.0.0.0', port=port)
class Entry(db.Model):
id = db.Column(db.Integer, db.Sequence('id_seq'), primary_key=True)
ip = db.Column(db.String(80))
user_agent = db.Column(db.Text)
zip = db.Column(db.Integer)
income = db.Column(db.Integer)
guess = db.Column(db.Integer)
time = db.Column(db.DateTime)
def __init__(self, ip, user_agent, time, zip=Null, income=Null, guess=Null):
self.ip = ip
self.user_agent = user_agent
self.zip = zip
self.income = income
self.guess = guess
self.time = time