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/&", 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 %} +

Password

+
+ {{ form.hidden_tag() }} +

+ {{ form.password.label }}
+ {{ form.password(size=32) }}
+ {% for error in form.password.errors %} + [{{ error }}] + {% endfor %} +

+

{{ form.submit() }}

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