|
@@ -18,7 +18,7 @@ app = Flask(__name__)
|
|
app.config.from_object(__name__)
|
|
app.config.from_object(__name__)
|
|
|
|
|
|
def connect_db():
|
|
def connect_db():
|
|
- return sqlite3.connect(app.config['DATABASE'])
|
|
|
|
|
|
+ return sqlite3.connect(app.config['DATABASE'])
|
|
|
|
|
|
@app.before_request
|
|
@app.before_request
|
|
def before_request():
|
|
def before_request():
|
|
@@ -28,6 +28,10 @@ def before_request():
|
|
def teardown_request(exception):
|
|
def teardown_request(exception):
|
|
g.db.close()
|
|
g.db.close()
|
|
|
|
|
|
|
|
+@app.route('/')
|
|
|
|
+def home():
|
|
|
|
+ return render_template('index.html')
|
|
|
|
+
|
|
@app.route('/admin/votes')
|
|
@app.route('/admin/votes')
|
|
def show_votes():
|
|
def show_votes():
|
|
cur = g.db.execute('select title, description, date from votes order by id desc')
|
|
cur = g.db.execute('select title, description, date from votes order by id desc')
|
|
@@ -44,7 +48,7 @@ def add_vote():
|
|
[request.form['title'], request.form['description'], ndate])
|
|
[request.form['title'], request.form['description'], ndate])
|
|
g.db.commit()
|
|
g.db.commit()
|
|
flash('New entry was successfully posted')
|
|
flash('New entry was successfully posted')
|
|
- return redirect(url_for('show_votes'))
|
|
|
|
|
|
+ return redirect(url_for('home'))
|
|
|
|
|
|
@app.route('/login', methods=['GET', 'POST'])
|
|
@app.route('/login', methods=['GET', 'POST'])
|
|
def login():
|
|
def login():
|
|
@@ -56,15 +60,16 @@ def login():
|
|
error = 'Invalid password'
|
|
error = 'Invalid password'
|
|
else:
|
|
else:
|
|
session['logged_in'] = True
|
|
session['logged_in'] = True
|
|
|
|
+ session['nickname'] = request.form['username']
|
|
flash('You were logged in')
|
|
flash('You were logged in')
|
|
- return redirect(url_for('show_votes'))
|
|
|
|
|
|
+ return redirect(url_for('home'))
|
|
return render_template('login.html', error=error)
|
|
return render_template('login.html', error=error)
|
|
|
|
|
|
@app.route('/logout')
|
|
@app.route('/logout')
|
|
def logout():
|
|
def logout():
|
|
session.pop('logged_in', None)
|
|
session.pop('logged_in', None)
|
|
flash('You were logged out')
|
|
flash('You were logged out')
|
|
- return redirect(url_for('show_votes'))
|
|
|
|
|
|
+ return redirect(url_for('home'))
|
|
|
|
|
|
|
|
|
|
|
|
|