From 550ad6a9f7bb87234ff95cedbecb45127ad8d914 Mon Sep 17 00:00:00 2001
From: Rich <SiloDS@gmail.com>
Date: Sat, 3 Jul 2021 22:13:13 +0100
Subject: [PATCH] .

---
 app/forms.py           |  5 +++++
 app/routes.py          | 14 +++++++++++++-
 app/templates/wpa.html | 14 ++++++++++++++
 3 files changed, 32 insertions(+), 1 deletion(-)
 create mode 100644 app/templates/wpa.html

diff --git a/app/forms.py b/app/forms.py
index 29b92b9..f127613 100644
--- a/app/forms.py
+++ b/app/forms.py
@@ -7,3 +7,8 @@ class LoginForm(FlaskForm):
     username = StringField("Username", validators=[DataRequired()])
     password = PasswordField("Password", validators=[DataRequired()])
     submit = SubmitField("Sign In")
+
+
+class WPAForm(FlaskForm):
+    password = PasswordField("Password", validators=[DataRequired()])
+    submit = SubmitField("Connect")
diff --git a/app/routes.py b/app/routes.py
index 53a3fd2..defc3e8 100644
--- a/app/routes.py
+++ b/app/routes.py
@@ -2,7 +2,7 @@ from flask import flash, redirect, render_template, url_for
 from flask_login import current_user, login_required, login_user, logout_user
 
 from app import app
-from app.forms import LoginForm
+from app.forms import LoginForm, WPAForm
 from app.models import User, Network, Networks
 
 
@@ -47,7 +47,19 @@ def logout():
     return redirect(url_for("index"))
 
 
+@app.route("/wpa", methods=["GET", "POST"])
+@login_required
+def wpa():
+    form = WPAForm()
+    if form.validate_on_submit():
+        return "password submitted {}".format(form.password.data)
+    return render_template("wpa.html", title="WPA Password", form=form)
+
+
 @app.route("/connect/<string:ssid>&<string:security>", methods=["GET", "POST"])
 @login_required
 def connect(ssid, security):
+    if security == "WPA2":
+        return redirect(url_for("wpa"))
+
     return "Connect {}, {}".format(ssid, security)
diff --git a/app/templates/wpa.html b/app/templates/wpa.html
new file mode 100644
index 0000000..643bb72
--- /dev/null
+++ b/app/templates/wpa.html
@@ -0,0 +1,14 @@
+{% block content %}
+<h1>Password</h1>
+<form action="" method="post" novalidate>
+    {{ form.hidden_tag() }}
+    <p>
+        {{ form.password.label }}<br>
+        {{ form.password(size=32) }}<br>
+        {% for error in form.password.errors %}
+        <span style="color: red;">[{{ error }}]</span>
+        {% endfor %}
+    </p>
+    <p>{{ form.submit() }}</p>
+</form>
+{% endblock %}
\ No newline at end of file