import subprocess from config import RPI, debug 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, WPAForm from app.models import Network, Networks, User CMD_SCAN = "sudo nmcli -t -f SSID,SIGNAL,IN-USE,SECURITY -e yes -m tab device wifi list ifname wlxf81a6719febb --rescan yes" CMD_JOIN = "sudo nmcli device wifi connect *SSID* ifname wlxf81a6719febb" CMD_JOINPW = "sudo nmcli device wifi connect *SSID* ifname wlxf81a6719febb password *PASSWORD*" CMD_DISCONNECT = "sudo nmcli device disconnect wlxf81a6719febb" # Error: Connection activation failed: (7) Secrets were required, but not provided. # Device 'wlxf81a6719febb' successfully activated with '11111-1111-11111-111111-11111111' def scan_networks(): scan = [] if not RPI: output = subprocess.run(CMD_SCAN.split(" "), stdout=subprocess.PIPE).stdout.decode("utf-8") for line in output.splitlines(): t = line.split(":") if t[0] != "": scan.append(line) else: output = subprocess.run(["sudo", "iwgetid"], stdout=subprocess.PIPE).stdout.decode("utf-8") current = output.partition("ESSID:")[2].strip().strip('"') output = subprocess.run(["sudo", "iwlist", "wlan0", "scan"], stdout=subprocess.PIPE).stdout.decode("utf-8") a = 0 scan = [] for line in output.splitlines(): if a == 0: connected = " " if line.partition("ESSID:")[2].strip('"') != "": a = 1 ssid = line.partition("ESSID:")[2].strip('"') if ssid == current: connected = "*" if a == 1: if line.find("Encryption key:off") != -1: password = " " a = 2 else: password = "WPA2" a = 2 if a == 2: if line.partition("Signal level=")[2].split("/")[0] != "": signal = line.partition("Signal level=")[2].split("/")[0] a = 0 scan.append(ssid + ":" + signal + ":" + connected + ":" + password) return scan @app.route("/") @app.route("/index") @login_required def index(): results = [] scan = ["rpi:100: :WPA2", "Home:94:*:WPA2", "HOME2:48: :WPA2", "BT:23: :"] if not debug: scan = scan_networks() for network in scan: item = [ network.split(":")[0], network.split(":")[1], "Yes" if network.split(":")[2] == "*" else "", "None" if network.split(":")[3].strip() == "" else "WPA2", ] results.append(item) # table = Networks(results) # table.border = True t = {} for a in results: t["ssid"] = a[0] return render_template("index.html", index_table=results) @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("/wpa/", methods=["GET", "POST"]) @login_required def wpa(ssid): # wlan0: flags=4098 mtu 1500 # wlan1: flags=4163 mtu 1500 if not RPI: form = WPAForm() if form.validate_on_submit(): cmd = CMD_JOINPW.replace("*SSID*", ssid) cmd = cmd.replace("*PASSWORD*", form.password.data) output = subprocess.run(cmd.split(" "), stdout=subprocess.PIPE, stderr=subprocess.PIPE).stdout.decode("utf-8") if output.find("Error") == -1 and output != "": return render_template("message.html", message="Successfully connected to {}".format(ssid)) return render_template("message.html", message="Failed to connect to {}".format(ssid)) return render_template("wpa.html", title="WPA Password", form=form) else: form = WPAForm() if form.validate_on_submit(): wpa = '''country=GB # Your 2-digit country code ctrl_interface=DIR=/var/run/wpa_supplicant GROUP=netdev network={{ ssid="{}" psk="{}" key_mgmt=WPA-PSK }} '''.format(ssid,form.password.data) with open('/tmp/arg.txt','wt') as f: f.write(wpa) output = subprocess.run(['sudo', '/usr/sbin/ifdown','wlan0'], stdout=subprocess.PIPE, stderr=subprocess.PIPE).stdout.decode("utf-8") output = subprocess.run(['sudo', '/usr/bin/systemctl','stop','wpa_supplicant'], stdout=subprocess.PIPE, stderr=subprocess.PIPE).stdout.decode("utf-8") output = subprocess.run(['sudo', 'mv','/tmp/arg.txt','/etc/wpa_supplicant/wpa_supplicant.conf'], shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE).stdout.decode("utf-8") try: output = subprocess.run(['sudo', '/usr/bin/systemctl','start','wpa_supplicant'], stdout=subprocess.PIPE, stderr=subprocess.PIPE).stdout.decode("utf-8") output = subprocess.run(['sudo', '/usr/sbin/ifup','wlan0'], stdout=subprocess.PIPE, stderr=subprocess.PIPE,timeout=10).stdout.decode("utf-8") except: return render_template("message.html", message="Failt to connected to {}".format(ssid)) return render_template("message.html", message="Successfully connected to {}".format(ssid)) 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", ssid=ssid)) cmd = CMD_JOIN.replace("*SSID*", ssid) output = subprocess.run(cmd.split(" "), stdout=subprocess.PIPE, stderr=subprocess.PIPE).stdout.decode("utf-8") if output.find("Error") == -1: return render_template("message.html", message="Successfully connected to {}".format(ssid)) return render_template("message.html", message="Failed to connect to {}".format(ssid)) @app.route("/disconnect/", methods=["GET", "POST"]) @login_required def disconnect(ssid): if not RPI: output = subprocess.run(CMD_DISCONNECT.split(" "), stdout=subprocess.PIPE, stderr=subprocess.PIPE).stdout.decode("utf-8") if output.find("successfully disconnected") != -1: return render_template("message.html", message="Sucessfully disconnected from {}".format(ssid)) return render_template("message.html", message="Failed to Disconnect from {}".format(ssid)) else: output = subprocess.run(['sudo','/usr/sbin/ifdown','wlan0'], stdout=subprocess.PIPE, stderr=subprocess.PIPE).stdout.decode("utf-8") return render_template("message.html", message="Sucessfully disconnected from {}".format(ssid))