LISHUZUOXUN_yangjiang/LSZXNetWork/NetworkDriverWin32/wifi_ap.py

import os
import re
import subprocess
import time
from subprocess import Popen, PIPE
from ping3 import ping

from AcrossPlatform.get_platform import GLOBAL_DIR
from LogRecord.log_recorder import GLOBAL_LOG

WIFI_AP_DRIVER = os.path.join(GLOBAL_DIR, 'LSZXNetWork', 'NetworkDriverWin32', 'WiFiDirectLegacyAPDemo.exe')
WIFI_AP_EXE = 'WiFiDirectLegacyAPDemo.exe'
DEFAULT_IP_SEGMENT = r"192.168.[0-9]{1,3}.[0-9]{1,3}"


# 获得自身的ip地址
def get_self_ip_address():
    ip_list = get_machine_ip_address()
    if ip_list:
        return ip_list[0]
    else:
        return "0.0.0.0"


# 获得同个网段下的设备
def get_machine_ip_address():
    echo = os.popen("arp -a").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


def kill_port(port):
    response = os.popen(f"netstat -nao | findstr \"{port}\"").read()
    pid_mes = re.search(r"(?<= )[0-9]+(?![.:])", response)
    if pid_mes:
        pid = pid_mes.group()
        # 运行一个命令，并将标准输出和标准错误重定向到 DEVNULL
        process = subprocess.Popen(
            ["taskkill", "-pid", f"{pid}", "-f"],
            stdout=subprocess.DEVNULL,
            stderr=subprocess.DEVNULL
        )
        # 等待命令执行完成
        process.wait()


# 结束wifi
def stop_wifi_ap():
    # 运行一个命令，并将标准输出和标准错误重定向到 DEVNULL
    process = subprocess.Popen(
        ["taskkill", "/f", "/im", WIFI_AP_EXE],
        stdout=subprocess.DEVNULL,
        stderr=subprocess.DEVNULL
    )
    # 等待命令执行完成
    process.wait()


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()
        time.sleep(0.1)
        self.proc = Popen(
            args=WIFI_AP_DRIVER, shell=True, encoding='utf-8', stdin=PIPE,
            stdout=PIPE
        )
        self.ssid = None
        self.password = None
        self.enable = False

    def reboot(self):
        try:
            stop_wifi_ap()
            self.proc.kill()
            stop_wifi_ap()
            time.sleep(1)
            self.proc = Popen(
                args=WIFI_AP_DRIVER, shell=True, encoding='utf-8', stdin=PIPE,
                stdout=PIPE
            )
            if self.enable:
                self.set_ssid(ssid=self.ssid)
                self.set_password(password=self.password)
                self.start()
            GLOBAL_LOG.write(f"AP重启成功")
        except Exception as e:
            GLOBAL_LOG.write(f"AP重启出错，原因是：{e.args}。")
            return

    def send_cmd(self, cmd, timeout=1):
        start_time = time.time()
        while time.time() - start_time < timeout:
            try:
                self.proc.stdin.write(f"{cmd}\n")
                self.proc.stdin.flush()
                break
            except Exception as e:
                GLOBAL_LOG.write(f"AP启动出错，原因是：{e.args}。")
                self.reboot()

    def set_ssid(self, ssid):
        self.ssid = ssid
        self.send_cmd(f"ssid {ssid}")

    def set_password(self, password):
        self.password = password
        self.send_cmd(f"pass {password}")

    def start(self):
        self.send_cmd("start")
        self.enable = True

    def stop(self):
        self.send_cmd("stop")
        self.enable = False

    def set_auto_accept(self, mode=True):
        if mode:
            self.send_cmd("autoaccept 1")
        else:
            self.send_cmd("autoaccept 0")

    def kill_wifi(self):
        self.send_cmd("quit")