From dd60df3f18af9eb6680e9e4aabf64e0d528e92ed Mon Sep 17 00:00:00 2001 From: cellarspoon Date: Mon, 6 Dec 2021 11:10:58 +0100 Subject: [PATCH] maybe radio fields --- app.py | 40 +++++++++++++++++++++++++++++++++------- templates/index.html | 11 ++++++++--- 2 files changed, 41 insertions(+), 10 deletions(-) diff --git a/app.py b/app.py index 59157b8..84a4f26 100644 --- a/app.py +++ b/app.py @@ -6,7 +6,7 @@ from flask_wtf import FlaskForm from flask_wtf.csrf import CSRFProtect from flask_wtf.file import FileAllowed, FileField, FileRequired from werkzeug.utils import secure_filename -from wtforms import StringField +from wtforms import RadioField, StringField from wtforms.validators import DataRequired app = Flask(__name__) @@ -21,18 +21,44 @@ class UploadForm(FlaskForm): pdf = FileField(validators=[FileAllowed(["pdf"], "PDFs only!"), FileRequired()]) +class CardSidesForm(FlaskForm): + title = RadioField("title", choices=[("side a", "side a"), ("side b", "side b")]) + + @app.route("/") def home(): - form = UploadForm() - return render_template("index.html", form=form) + upload_form = UploadForm() + card_sides_form = CardSidesForm() + return render_template( + "index.html", upload_form=upload_form, card_sides_form=card_sides_form + ) @app.route("/upload", methods=["POST"]) def upload(): - form = UploadForm() - if form.validate_on_submit(): - f = form.pdf.data + upload_form = UploadForm() + card_sides_form = CardSidesForm() + + if upload_form.validate_on_submit(): + f = upload_form.pdf.data filename = secure_filename(f.filename) f.save(os.path.join(CWD, "pdfs", filename)) return redirect(url_for("home")) - return render_template("index.html", form=form) + return render_template( + "index.html", upload_form=upload_form, card_sides_form=card_sides_form + ) + + +@app.route("/generate", methods=["POST"]) +def generate(): + upload_form = UploadForm() + card_sides_form = CardSidesForm() + + if card_sides_form.validate_on_submit(): + print(card_sides_form.title.data) + else: + print("didnt work") + + return render_template( + "index.html", upload_form=upload_form, card_sides_form=card_sides_form + ) diff --git a/templates/index.html b/templates/index.html index 5617433..67f0200 100644 --- a/templates/index.html +++ b/templates/index.html @@ -7,16 +7,21 @@
- {{ form.csrf_token }} {{ form.pdf }} + {{ upload_form.csrf_token }} {{ upload_form.pdf }}
- {% if form.pdf.errors %} + {% if upload_form.pdf.errors %} {% endif %} + +
+ {{ card_sides_form.csrf_token }} {{ card_sides_form.title }} + +