100 lines
2.9 KiB
Python
100 lines
2.9 KiB
Python
|
import os
|
|||
|
|
import re
|
|||
|
|
|
|||
|
|
from ping3 import ping
|
|||
|
|
|
|||
|
|
from AcrossPlatform.get_platform import GLOBAL_DIR
|
|||
|
|
from LSZXNetWork.NetworkDriverLinux.wifi import DEFAULT_DEVICE, LAN_DEVICE
|
|||
|
|
from LogRecord.log_recorder import GLOBAL_LOG
|
|||
|
|
|
|||
|
|
DEFAULT_IP_SEGMENT = r"192.168.[0-9]{1,3}.[0-9]{1,3}"
|
|||
|
|
AP_CONF_DIR = os.path.join(GLOBAL_DIR, "LSZXNetWork/NetworkDriverLinux/start_wifi_ap.sh")
|
|||
|
|
PATTERN_AP_CONF_DIR = os.path.join(GLOBAL_DIR, "LSZXNetWork/NetworkDriverLinux/pattern_start_wifi_ap.sh")
|
|||
|
|
STOP_AP_CONF_DIR = os.path.join(GLOBAL_DIR, "LSZXNetWork/NetworkDriverLinux/stop_wifi_ap.sh")
|
|||
|
|
PATTERN_STOP_AP_CONF_DIR = os.path.join(GLOBAL_DIR, "LSZXNetWork/NetworkDriverLinux/pattern_stop_wifi_ap.sh")
|
|||
|
|
MASTER_IP = "192.168.137.1"
|
|||
|
|
WLAN_DEVICE_TAG = "<wlan>"
|
|||
|
|
LAN_DEVICE_TAG = "<lan>"
|
|||
|
|
SSID_DEVICE_TAG = "<ssid>"
|
|||
|
|
PASS_DEVICE_TAG = "<passphrase>"
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 获得同个网段下的设备
|
|||
|
|
def get_machine_ip_address():
|
|||
|
|
echo = os.popen("arp -a | awk '{print $2}' | tr -d '()'").read()
|
|||
|
|
ip_list = re.findall(DEFAULT_IP_SEGMENT, echo)
|
|||
|
|
ip_list = list(filter(lambda x: not x.endswith("255") and not x.endswith("138.1"), ip_list))
|
|||
|
|
return ip_list
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 结束wifi
|
|||
|
|
def stop_wifi_ap():
|
|||
|
|
with open(PATTERN_STOP_AP_CONF_DIR, "r") as file:
|
|||
|
|
conf = file.read()
|
|||
|
|
conf = conf.replace(WLAN_DEVICE_TAG, DEFAULT_DEVICE)
|
|||
|
|
conf = conf.replace(LAN_DEVICE_TAG, LAN_DEVICE)
|
|||
|
|
with open(STOP_AP_CONF_DIR, "w") as file:
|
|||
|
|
file.write(conf)
|
|||
|
|
os.popen(f"sudo bash {STOP_AP_CONF_DIR}").read()
|
|||
|
|
|
|||
|
|
def kill_port(port):
|
|||
|
|
os.popen(f"sudo fuser -k {port}/tcp").read()
|
|||
|
|
|
|||
|
|
def get_wifi_ap_status():
|
|||
|
|
try:
|
|||
|
|
if isinstance(ping("192.168.137.1", timeout=500, unit="ms"), float):
|
|||
|
|
return True
|
|||
|
|
else:
|
|||
|
|
return False
|
|||
|
|
except Exception as e:
|
|||
|
|
GLOBAL_LOG.write(f"AP状态获取出错，原因是：{e.args}。")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
|
|||
|
|
class WiFiAP:
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
stop_wifi_ap()
|
|||
|
|
self.ssid = None
|
|||
|
|
self.password = None
|
|||
|
|
|
|||
|
|
def reboot(self):
|
|||
|
|
stop_wifi_ap()
|
|||
|
|
if self.ssid:
|
|||
|
|
self.set_ssid(ssid=self.ssid)
|
|||
|
|
self.set_password(password=self.password)
|
|||
|
|
self.start()
|
|||
|
|
|
|||
|
|
def set_ssid(self, ssid):
|
|||
|
|
self.ssid = ssid
|
|||
|
|
|
|||
|
|
def set_password(self, password):
|
|||
|
|
self.password = password
|
|||
|
|
|
|||
|
|
def edit_ap_conf(self):
|
|||
|
|
with open(PATTERN_AP_CONF_DIR, "r") as file:
|
|||
|
|
conf = file.read()
|
|||
|
|
conf = conf.replace(WLAN_DEVICE_TAG, DEFAULT_DEVICE)
|
|||
|
|
conf = conf.replace(LAN_DEVICE_TAG, LAN_DEVICE)
|
|||
|
|
conf = conf.replace(SSID_DEVICE_TAG, self.ssid)
|
|||
|
|
conf = conf.replace(PASS_DEVICE_TAG, self.password)
|
|||
|
|
with open(AP_CONF_DIR, "w") as file:
|
|||
|
|
file.write(conf)
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def start_cmd():
|
|||
|
|
os.popen(f"sudo bash {AP_CONF_DIR}").read()
|
|||
|
|
|
|||
|
|
def start(self):
|
|||
|
|
if self.ssid:
|
|||
|
|
self.edit_ap_conf()
|
|||
|
|
self.start_cmd()
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def stop():
|
|||
|
|
stop_wifi_ap()
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def kill_wifi():
|
|||
|
|
stop_wifi_ap()
|