54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
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.models import User, Network, Networks
|
|
|
|
|
|
@app.route("/")
|
|
@app.route("/index")
|
|
@login_required
|
|
def search_results():
|
|
results = []
|
|
scan = ["rpi:100: :WPA2", "Home:94:*:WPA2", "HOME2:48: :WPA2", "BT:23: :"]
|
|
for item in scan:
|
|
erm = Network(
|
|
item.split(":")[0],
|
|
item.split(":")[1],
|
|
"Yes" if item.split(":")[2] == "*" else "",
|
|
"None" if item.split(":")[3].strip() == "" else "WPA2",
|
|
)
|
|
results.append(erm)
|
|
|
|
table = Networks(results)
|
|
table.border = True
|
|
return render_template("index.html", table=table)
|
|
|
|
|
|
@app.route("/login", methods=["GET", "POST"])
|
|
def login():
|
|
if current_user.is_authenticated:
|
|
return redirect(url_for("index"))
|
|
form = LoginForm()
|
|
if form.validate_on_submit():
|
|
user = User.query.filter_by(username=form.username.data).first()
|
|
if user is None or not user.check_password(form.password.data):
|
|
flash("Invalid username or password")
|
|
return redirect(url_for("login"))
|
|
login_user(user)
|
|
return redirect(url_for("index"))
|
|
return render_template("login.html", title="Sign In", form=form)
|
|
|
|
|
|
@app.route("/logout")
|
|
def logout():
|
|
logout_user()
|
|
return redirect(url_for("index"))
|
|
|
|
|
|
@app.route("/connect/<string:ssid>", methods=["GET", "POST"])
|
|
@login_required
|
|
def connect(ssid):
|
|
return "Connect {}".format(ssid)
|