app/__init__.py

@@ -25,4 +25,8 @@ def create_app(test_config=None):
     app.register_blueprint(index.bp)
     app.add_url_rule('/', endpoint='index')
 
+    from . import admin_insert
+    app.register_blueprint(admin_insert.bp)
+    app.add_url_rule('/admin_insert', endpoint='admin_insert')
+
     return app

app/admin_insert.py

@@ -0,0 +1,27 @@
+from flask import (
+    Blueprint, flash, g, redirect, render_template, request, url_for
+)
+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():
+    form = StationForm()
+    if form.validate_on_submit():
+        db = get_db()
+        db.execute(
+            'INSERT INTO station (nombre_empleado, oficina, last_turn)'
+            ' VALUES (?, ?, ?)',
+            (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', form=form)

app/forms.py

@@ -0,0 +1,13 @@
+from flask_wtf import FlaskForm
+from wtforms import StringField, PasswordField, BooleanField, SubmitField
+from wtforms.validators import DataRequired
+
+class TurnForm(FlaskForm):
+    cName = StringField('Name', validators=[DataRequired()])
+    cEmail = StringField('Email', validators=[DataRequired()])
+    submit = SubmitField('Get turn')
+
+class StationForm(FlaskForm):
+    eName = StringField('Name', validators=[DataRequired()])
+    office = StringField('Office', validators=[DataRequired()])
+    submit = SubmitField('Insert Station')

app/index.py

@@ -6,39 +6,38 @@ from flask import (
 from werkzeug.exceptions import abort
 
 from app.db import get_db
+from app.forms import TurnForm
 
 bp = Blueprint('index', __name__)
 
 @bp.route('/', methods=('GET', 'POST'))
 def index():
-    if request.method == 'POST':
-        cName = request.form['cName']
-        cEmail = request.form['cEmail']
-        station_name = request.form['station']
+    form = TurnForm()
+    if form.validate_on_submit():
+        station_id = request.form['station']
 
         db = get_db()
-        station_query_res = db.execute(
-            'SELECT s_id FROM station WHERE nombre_empleado=?',
-            (station_name,)
-        ).fetchone()
-        if station_query_res is not None:
-            station_id = station_query_res[0]
         db.execute(
             'UPDATE station SET last_turn = last_turn + 1 WHERE s_id=?',
             (station_id,)
         )
-        turn_query_res = db.execute(
+        turn = db.execute(
             'SELECT last_turn FROM station WHERE s_id=?',
             (station_id,)
-        ).fetchone()
-        if turn_query_res is not None:
-            turn = turn_query_res[0]
+        ).fetchone()[0]
         db.execute(
             'INSERT INTO turno (cName, cEmail, turn, timeArrival, station)'
             ' VALUES (?, ?, ?, ?, ?)',
-            (cName, cEmail, turn, '{0:%H:%M}'.format(datetime.now()), station_id)
+            (form.cName.data, form.cEmail.data, turn, '{0:%H:%M}'.format(datetime.now()), station_id)
         )
+        station_info = db.execute(
+            'SELECT nombre_empleado, oficina FROM station WHERE s_id=?',
+            (station_id,)
+        ).fetchone()
         db.commit()
+
+        flash('Your turn for {} ({}) is number {}'.format(station_info['nombre_empleado'], station_info['oficina'],  turn))
+
         return redirect(url_for('index'))
 
     db = get_db()
@@ -46,4 +45,4 @@ def index():
         'SELECT s_id, nombre_empleado, oficina FROM station'
         ' ORDER BY oficina ASC'
     ).fetchall()
-    return render_template('index.html', stations=stations)
+    return render_template('index.html', stations=stations, form=form)

app/templates/admin_insert.html

@@ -0,0 +1,16 @@
+{% extends 'base.html' %}
+
+{% block header %}
+    <h1>{% block title %}UPR Queue: Insert Station{% endblock %}</h1>
+{% endblock %}
+
+{% block content %}
+    <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/base.html

@@ -6,7 +6,7 @@
         {% block header %}{% endblock %}
     </header>
     {% for message in get_flashed_messages() %}
-        <div class="flash">{{ message }}</div>
+        <div class="flash">{{ message }}</div><br>
     {% endfor %}
     {% block content %}{% endblock %}
 </section>

app/templates/index.html

@@ -5,16 +5,17 @@
 {% endblock %}
 
 {% block content %}
-    <form method="post">
-        <label for="cName">Name</label>
-        <input name="cName" id="name" required></input><br>
-		<label for="cEmail">email</label>
-        <input name="cEmail" id="email" required></input><br>
+    <form action="" method="post">
+        {{ form.hidden_tag() }}
+        {{ form.cName.label }}<br>
+        {{ form.cName }}<br>
+        {{ form.cEmail.label }}<br>
+        {{ form.cEmail }}<br>
 
         {% for station in stations %}
-            <input type="radio" name="station" id={{ "station{}".format(station['s_id']) }} value={{ station['nombre_empleado'] }}> {{ "{} ({})".format(station['nombre_empleado'],station['oficina']) }} </input><br>
+            <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 %}