-
Notifications
You must be signed in to change notification settings - Fork 68
/
app.py
44 lines (36 loc) · 1.25 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
from flask import Flask, render_template, request
from flask.ext.sqlalchemy import SQLAlchemy
from flask.ext.heroku import Heroku
app = Flask(__name__)
#app.config['SQLALCHEMY_DATABASE_URI'] = 'postgresql://localhost/pre-registration'
heroku = Heroku(app)
db = SQLAlchemy(app)
# Create our database model
class User(db.Model):
__tablename__ = "users"
id = db.Column(db.Integer, primary_key=True)
email = db.Column(db.String(120), unique=True)
def __init__(self, email):
self.email = email
def __repr__(self):
return '<E-mail %r>' % self.email
# Set "homepage" to index.html
@app.route('/')
def index():
return render_template('index.html')
# Save e-mail to database and send to success page
@app.route('/prereg', methods=['POST'])
def prereg():
email = None
if request.method == 'POST':
email = request.form['email']
# Check that email does not already exist (not a great query, but works)
if not db.session.query(User).filter(User.email == email).count():
reg = User(email)
db.session.add(reg)
db.session.commit()
return render_template('success.html')
return render_template('index.html')
if __name__ == '__main__':
#app.debug = True
app.run()