121 lines
3.8 KiB
Python
121 lines
3.8 KiB
Python
|
import os
|
|||
|
|
import re
|
|||
|
|
import time
|
|||
|
|
|
|||
|
|
import psutil
|
|||
|
|
import pywifi
|
|||
|
|
from pywifi import const
|
|||
|
|
|
|||
|
|
from LogRecord.log_recorder import GLOBAL_LOG
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_wifi_name():
|
|||
|
|
echo = os.popen("netsh wlan show interfaces | findstr /C:\"SSID\"").read()
|
|||
|
|
wifi_name_re = re.search(r"(?<=:\s).*", echo)
|
|||
|
|
if wifi_name_re:
|
|||
|
|
return wifi_name_re.group()
|
|||
|
|
else:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
|
|||
|
|
class WIFI:
|
|||
|
|
|
|||
|
|
def __init__(self) -> None:
|
|||
|
|
super().__init__()
|
|||
|
|
self.iface = None
|
|||
|
|
self.wifi_device_initial()
|
|||
|
|
self.log = GLOBAL_LOG
|
|||
|
|
|
|||
|
|
# WiFi驱动初始化
|
|||
|
|
def wifi_device_initial(self):
|
|||
|
|
while True:
|
|||
|
|
try:
|
|||
|
|
wifi = pywifi.PyWiFi()
|
|||
|
|
self.iface = wifi.interfaces()[0] # acquire the first Wlan card,maybe not
|
|||
|
|
break
|
|||
|
|
except Exception as e:
|
|||
|
|
self.log.write(f"WiFi驱动初始化失败，原因是：{e.args}。")
|
|||
|
|
time.sleep(1)
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
def get_connected_wifi_name(self):
|
|||
|
|
try:
|
|||
|
|
if self.wifi_connect_status():
|
|||
|
|
wifi_name = get_wifi_name()
|
|||
|
|
return wifi_name
|
|||
|
|
except Exception as e:
|
|||
|
|
self.log.write(f"获取连接的WiFi出错，原因是：{e.args}。")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def wifi_connect_status(self):
|
|||
|
|
try:
|
|||
|
|
if self.iface.status() in [const.IFACE_CONNECTED, const.IFACE_INACTIVE]:
|
|||
|
|
return True
|
|||
|
|
else:
|
|||
|
|
return False
|
|||
|
|
except Exception as e:
|
|||
|
|
self.log.write(f"获取WiFi状态出错，原因是：{e.args}。")
|
|||
|
|
self.wifi_device_initial()
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def disconnect(self):
|
|||
|
|
try:
|
|||
|
|
while self.wifi_connect_status():
|
|||
|
|
self.iface.disconnect()
|
|||
|
|
except Exception as e:
|
|||
|
|
self.log.write(f"断开WiFi出错，原因是：{e.args}。")
|
|||
|
|
self.wifi_device_initial()
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def get_ip():
|
|||
|
|
dic = psutil.net_if_addrs()
|
|||
|
|
container = {}
|
|||
|
|
for adapter in dic:
|
|||
|
|
snic_list = dic[adapter]
|
|||
|
|
ipv4 = '无 ipv4 地址'
|
|||
|
|
for snic in snic_list:
|
|||
|
|
if snic.family.name == 'AF_INET':
|
|||
|
|
ipv4 = snic.address
|
|||
|
|
container[str(adapter)] = str(ipv4)
|
|||
|
|
return container
|
|||
|
|
|
|||
|
|
def scan_wifi(self):
|
|||
|
|
try:
|
|||
|
|
self.iface.scan()
|
|||
|
|
time.sleep(5)
|
|||
|
|
base_wifi = self.iface.scan_results()
|
|||
|
|
wifi_list = [str(wifi.ssid) for wifi in base_wifi]
|
|||
|
|
return wifi_list
|
|||
|
|
except Exception as e:
|
|||
|
|
self.log.write(f"扫描WiFi出错，原因是：{e.args}。")
|
|||
|
|
self.wifi_device_initial()
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
def connect_wifi(self, ssid, password):
|
|||
|
|
try:
|
|||
|
|
self.disconnect()
|
|||
|
|
for i in range(10):
|
|||
|
|
time.sleep(1)
|
|||
|
|
if not self.wifi_connect_status():
|
|||
|
|
break
|
|||
|
|
profile = pywifi.Profile() # 配置文件
|
|||
|
|
profile.ssid = ssid # wifi名称
|
|||
|
|
profile.auth = const.AUTH_ALG_OPEN # 需要密码
|
|||
|
|
profile.akm.append(const.AKM_TYPE_WPA2PSK) # 加密类型
|
|||
|
|
profile.cipher = const.CIPHER_TYPE_CCMP # 加密单元
|
|||
|
|
profile.key = password # wifi密码
|
|||
|
|
|
|||
|
|
# self.iface.remove_all_network_profiles() # 删除其它配置文件
|
|||
|
|
tmp_profile = self.iface.add_network_profile(profile) # 加载配置文件
|
|||
|
|
self.iface.connect(tmp_profile)
|
|||
|
|
for i in range(20):
|
|||
|
|
time.sleep(1)
|
|||
|
|
if self.iface.status() == const.IFACE_CONNECTED:
|
|||
|
|
return True
|
|||
|
|
return False
|
|||
|
|
except Exception as e:
|
|||
|
|
self.log.write(f"链接WiFi出错，原因是：{e.args}。")
|
|||
|
|
self.wifi_device_initial()
|
|||
|
|
return False
|