diff --git a/main.py b/main.py index a7b6794..a386bb5 100644 --- a/main.py +++ b/main.py @@ -3,7 +3,7 @@ from flask import Flask, render_template, request, redirect, url_for, session -from model import Donation +from model import Donation, Donor app = Flask(__name__) @@ -15,7 +15,17 @@ def home(): def all(): donations = Donation.select() return render_template('donations.jinja2', donations=donations) - + +@app.route('/new_donation', methods=['POST', 'GET']) +def create(): + if request.method == 'POST': + donor = request.form['donor'] + amount = int(request.form['amount']) + saved_donor = Donor.select().where(Donor.name == donor).get() + Donation(donor=saved_donor, value=amount).save() + return redirect(url_for('home')) + return render_template('new_donation.jinja2') + if __name__ == "__main__": port = int(os.environ.get("PORT", 6738)) diff --git a/requirements.txt b/requirements.txt index 41a4f05..523880a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,5 +4,5 @@ itsdangerous==0.24 Jinja2==2.10 MarkupSafe==1.0 peewee==3.2.2 -psycopg2-binary==2.7.4 +psycopg2-binary==2.8.1 Werkzeug==0.14.1 diff --git a/templates/base.jinja2 b/templates/base.jinja2 index f10648b..e207d1a 100644 --- a/templates/base.jinja2 +++ b/templates/base.jinja2 @@ -3,6 +3,10 @@