From bc99f15219437569bb6f59136af1d78926b04a8e Mon Sep 17 00:00:00 2001 From: IrvingGao <1729854488@qq.com> Date: Mon, 27 May 2024 21:20:54 +0800 Subject: [PATCH] wifi json --- README.md | 10 +++++- wifi_manager.py | 82 ++++++++++++++++++++++++++++++------------------- 2 files changed, 59 insertions(+), 33 deletions(-) diff --git a/README.md b/README.md index cdfc7da..99e7c2e 100644 --- a/README.md +++ b/README.md @@ -62,6 +62,7 @@ Description=Hotspot Service After=network.target [Service] +ExecPre=/bin/sleep 5 ExecStart=create_ap wlan0 eth0 Takway-Toys --no-virt User=root @@ -174,7 +175,7 @@ nmcli dev wifi connect Innoxsz-Public password innox2023 - 断开Wi-Fi连接: ``` -nmcli dev disconnect iface wlan0 +nmcli dev disconnect wlan0 ``` - 扫描Wi-Fi: @@ -187,3 +188,10 @@ nmcli dev wifi ``` nmcli dev status ``` + + +### create_ap + +``` +sudo create_ap wlan0 eth0 Takway-Toys --no-virt ++``` \ No newline at end of file diff --git a/wifi_manager.py b/wifi_manager.py index 4bc8c00..4a68cec 100644 --- a/wifi_manager.py +++ b/wifi_manager.py @@ -5,6 +5,7 @@ import psutil import signal import time import datetime +import json from flask import Flask, render_template, request, redirect, url_for, make_response try: @@ -23,16 +24,16 @@ app = Flask(__name__) def close_app(): if led_enabled: orangepi.set_led_off('blue') - + # 获取当前Flask应用程序的进程ID flask_pid = os.getpid() - + # 获取所有名为'python3'的进程ID python_pids = [p.info['pid'] for p in psutil.process_iter(attrs=['pid', 'name']) if p.info['name'] == 'python3'] - + # 关闭Flask应用程序进程 os.kill(flask_pid, signal.SIGTERM) - + # 关闭Python进程 for pid in python_pids: os.kill(pid, signal.SIGTERM) @@ -54,8 +55,8 @@ def close_hotspot(): orangepi.set_led_off('red') except subprocess.CalledProcessError as e: print(f"{datetime.datetime.now()}: Error stopping hotspot: {e}") - - + + # 检测 Wi-Fi 连接状态 def check_wifi_connection(): cmd = "nmcli dev status" @@ -71,13 +72,13 @@ def check_wifi_connection(): if led_enabled: orangepi.set_led_off('blue') return True, wifi_ssid - return False + return False, None def scan_wifi(): subprocess.run(['nmcli', 'dev', 'wifi', 'rescan'], check=True) cmd = "nmcli dev wifi" result = subprocess.run(cmd, shell=True, capture_output=True, text=True) - + ssid_list = [] wifi_list = [] output = result.stdout.strip() @@ -108,11 +109,11 @@ def scan_wifi(): # 去除mac地址 ssid = ssid.split(' ')[1:] ssid = ''.join(ssid) - + if ssid in ssid_list: continue ssid_list.append(ssid) - + # 提取强度 strength = None for i in range(len(columns)): @@ -122,18 +123,30 @@ def scan_wifi(): # print("MAC地址:", mac_address) # print("Wi-Fi名称:", ssid) # print("强度:", strength) - + wifi_list.append({'ssid': ssid, 'signal': strength, 'mac': mac_address}) if len(wifi_list) == 15: break # save wifi_list to file - with open('wifi_list.txt', 'w') as f: - for wifi in wifi_list: - print(f"{wifi}") - f.write(f"{wifi}\n") - + with open('wifi_list.json', 'w') as f: + json.dump(wifi_list, f) + return wifi_list +def load_saved_wifi(): + try: + with open('wifi_list.json', 'r') as f: + wifi_list = json.load(f) + return wifi_list + except FileNotFoundError: + return [] + +def save_wifi(ssid, password): + wifi_list = load_saved_wifi() + wifi_list.append({'ssid': ssid, 'password': password}) + with open('wifi_list.json', 'w') as f: + json.dump(wifi_list, f) + # 连接 Wi-Fi def connect_wifi(ssid, password): # 连接到用户选择的 Wi-Fi 网络 @@ -142,6 +155,7 @@ def connect_wifi(ssid, password): output_str = output.decode('utf-8') # 将输出转换为字符串 if "successfully" in output_str: print(f"{datetime.datetime.now()}: Successfully connected to Wi-Fi: {ssid}") + save_wifi(ssid, password) # 保存连接成功的Wi-Fi信息 return True else: print(f"{datetime.datetime.now()}: Error connecting to Wi-Fi: {output_str}") @@ -169,13 +183,13 @@ def disconnect_wifi(): @app.route('/') def index(): global wifi_list - + response = make_response(render_template('index.html', wifi_list=wifi_list)) response.headers.set('Content-Type', 'text/html') response.headers.set('Apple-Web-App-Capable', 'yes') response.headers.set('Apple-Mobile-Web-App-Status-Bar-Style', 'black-translucent') return response - + # 提交 Wi-Fi 信息 @app.route('/submit', methods=['POST']) def submit(): @@ -184,20 +198,23 @@ def submit(): ssid = request.form['ssid'] password = request.form['password'] print(f"{datetime.datetime.now()}: Connecting to Wi-Fi: {ssid} with password {password}") - + # 关闭热点 close_hotspot() - + time.sleep(5) # 连接到用户选择的 Wi-Fi 网络 if connect_wifi(ssid, password): close_app() - - if not check_wifi_connection(): + + connected, wifi_ssid = check_wifi_connection() + if not connected: print(f"{datetime.datetime.now()}: Wi-Fi连接失败。") wifi_list = scan_wifi() start_hotspot() - + else: + save_wifi(wifi_ssid, password) # 保存连接成功的Wi-Fi信息 + return redirect(url_for('index')) @@ -207,24 +224,25 @@ if __name__ == '__main__': disconnect_wifi() wifi_list = scan_wifi() print(wifi_list) - + start_hotspot() - + # app.run(host='0.0.0.0', port=80) - + if connect_wifi("Innoxsz-Public", "innox2023"): close_app() if led_enabled: orangepi.set_led_on('blue') - - if check_wifi_connection(): + + connected, wifi_ssid = check_wifi_connection() + if connected: print(f"{datetime.datetime.now()}: 系统已自动连接到 Wi-Fi 网络，退出程序") close_app() else: - wifi_list = scan_wifi() + wifi_list = load_saved_wifi() + if not wifi_list: + wifi_list = scan_wifi() print(f"{datetime.datetime.now()}: 未连接到 Wi-Fi 网络") start_hotspot() - app.run(host='0.0.0.0', port=80) - - \ No newline at end of file + app.run(host='0.0.0.0', port=80) \ No newline at end of file