diff --git a/README.md b/README.md index 86dc732..96f25a2 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,7 @@ # Flask Mailroom Application -Running at [http://afternoon-reef-51666.herokuapp.com/donations/](http://afternoon-reef-51666.herokuapp.com/donations/). +Running at [http://stormy-hollows-70309.herokuapp.com/donations/](http://stormy-hollows-70309.herokuapp.com/donations/). +http://stormy-hollows-70309.herokuapp.com/add ## Your Task diff --git a/main.py b/main.py index a7b6794..b48adfb 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__) @@ -16,6 +16,33 @@ def all(): donations = Donation.select() return render_template('donations.jinja2', donations=donations) +@app.route('/add', methods=['GET', 'POST']) +def add(): + + if request.method == 'POST': + # Handle the form submission + # f_name = request.form['name'] + # amount = request.form['amount'] + query = Donor.select().where(Donor.name == request.form['name']) + if query.exists(): + Donation(donor=request.form['name'], value=request.form['amount']).save() + else: + f_name = Donor(name=request.form['name']) + f_name.save() + Donation(donor=f_name, value=request.form['amount']).save() + + + + # if f_name == d_name: + # # existing_donation = Donation.select().where(Donor.name == request.form['name']).get() + # # updated_donation = existing_donation.value + int(amount) + # # Donation.update(value=updated_donation).where(Donor.name == request.form['name']).execute() + # Donor.create(name=f_name) + # Donation. + return redirect(url_for('all')) + + else: + return render_template('new_donations.jinja2') if __name__ == "__main__": port = int(os.environ.get("PORT", 6738)) diff --git a/model.py b/model.py index baa7e50..0ffc3a0 100644 --- a/model.py +++ b/model.py @@ -13,7 +13,7 @@ class Meta: class Donation(Model): value = IntegerField() - donor = ForeignKeyField(Donor, backref='donations') + donor = ForeignKeyField(model=Donor, backref='donations') class Meta: database = db diff --git a/templates/new_donations.jinja2 b/templates/new_donations.jinja2 new file mode 100644 index 0000000..aef2a93 --- /dev/null +++ b/templates/new_donations.jinja2 @@ -0,0 +1,15 @@ +{% extends 'base.jinja2' %} + +{% block subtitle %}New Donations {% endblock subtitle %} + +{% block content %} +{#

Total: {{ session.total }}

#} +
+ + + +
+{#
#} +{# #} +{#
#} +{% endblock content %} \ No newline at end of file