Changes admin_insert to use WTForms

app/admin_insert.py

@@ -4,22 +4,24 @@ from flask import (
 from werkzeug.exceptions import abort
 
 from app.db import get_db
+from app.forms import StationForm
 
 bp = Blueprint('admin_insert', __name__)
 
 @bp.route('/admin_insert', methods=('GET', 'POST'))
 def admin():
-    if request.method == 'POST':
-        eName = request.form['eName']
-        office = request.form['oficina']
-
+    form = StationForm()
+    if form.validate_on_submit():
         db = get_db()
         db.execute(
             'INSERT INTO station (nombre_empleado, oficina, last_turn)'
             ' VALUES (?, ?, ?)',
-            (eName, office, 0)
+            (form.eName.data, form.office.data, 0)
         )
         db.commit()
+
+        flash('Added station for  {} ({})'.format(form.eName.data, form.office.data))
+
         return redirect(url_for('admin_insert'))
 
-    return render_template('admin_insert.html')
+    return render_template('admin_insert.html', form=form)

app/forms.py

@@ -5,4 +5,9 @@ from wtforms.validators import DataRequired
 class TurnForm(FlaskForm):
     cName = StringField('Name', validators=[DataRequired()])
     cEmail = StringField('Email', validators=[DataRequired()])
-    submit = SubmitField('Get turn')
+    submit = SubmitField('Get turn')
+
+class StationForm(FlaskForm):
+    eName = StringField('Name', validators=[DataRequired()])
+    office = StringField('Office', validators=[DataRequired()])
+    submit = SubmitField('Insert Station')

app/templates/admin_insert.html

@@ -5,11 +5,12 @@
 {% endblock %}
 
 {% block content %}
-    <form method="post">
-        <label for="eName">Name</label>
-        <input name="eName" id="name" required></input><br>
-        <label for="eName">Oficina</label>
-        <input name="oficina" id="oficina" required></input><br>
-        <input type="submit" value="Insert Station">
+    <form action="" method="post">
+        {{ form.hidden_tag() }}
+        {{ form.eName.label }}<br>
+        {{ form.eName }}<br>
+        {{ form.office.label }}<br>
+        {{ form.office }}<br>
+        {{ form.submit }}
     </form>
 {% endblock %}

app/templates/index.html

@@ -16,6 +16,6 @@
             <input type="radio" name="station" id={{ "station{}".format(station['s_id']) }} value={{ station['s_id'] }} required> {{ "{} ({})".format(station['nombre_empleado'],station['oficina']) }} </input><br>
         {% endfor %}
 
-        <input type="submit" value="Get ticket">
+        {{ form.submit }}
     </form>
 {% endblock %}