pywifi 网卡反复断开连接

1、STA 反复断开连接



# _*_coding:utf-8 _*_
__author__ = 'juzi_juzi'

"""
环境准备，windows 平台
1、安装python3 环境，对具体版本无要求,具体安装这里不介绍，https://www.python.org/ 下载即可；
2、安装pywifi 模块；如果不是用第三方IDE工具，直接cmd 安装：pip3 install pywifi;
3、安装comtypes；如果不是用第三方IDE工具，直接cmd 安装：pip3 install comtypes;
4、不使用第三方IDE 工具，直接cmd（管理员权限）运行：python3 wifi.py
"""
import pywifi
import time,subprocess
import logging, os
from datetime import date


GCOUNT = 0
GSUCCESSCOUNT = 0
GFAILEDCOUNT = 0
PINGFAILCOUNT = 0

def log():
    """
    定义日志相关内容；
    :return:None；
    """
    logger = logging.getLogger()
    logger.setLevel(logging.INFO)  # Log等级总开关
    timestruct =date.today().strftime('%Y-%m-%d')
    lastname = '{}-{}'.format(timestruct,'wifi.log')
    filename = os.path.join(os.getcwd(), lastname)
    fh = logging.FileHandler(filename)
    fh.setLevel(logging.DEBUG)  # 输出到file的log等级的开关
    formatter = logging.Formatter("%(asctime)s==%(levelname)s :%(message)s")
    fh.setFormatter(formatter)
    logger.addHandler(fh)
    ch = logging.StreamHandler()
    ch.setLevel(logging.DEBUG)  # 输出到file的log等级的开关
    ch.setFormatter(logging.Formatter("%(asctime)s==%(levelname)s :%(message)s"))
    logger.addHandler(ch)


def loginAP(ssid, password, waittimes=10):
    """
    STA connect AP
    :param ssid: SSID;
    :param password: login password;
    :return: connect result;
    """
    status = False
    pywifi.set_loglevel(logging.INFO)  # 输出日志
    wifi = pywifi.PyWiFi()  # 抓取WiFi接口
    ifaceList = wifi.interfaces()  # 抓取无线网卡列表
    global iface
    iface = ifaceList[0]  # 如果有无线网卡第一个一般就是你要的
    logging.info('get wireless adapter name:{}'.format(iface.name()))

    # iface.disconnect()
    profile = pywifi.Profile()  # 创建wifi链接文件
    profile.ssid = ssid  # wifi名称，不加会报错
    profile.auth = pywifi.const.AUTH_ALG_OPEN  # 网卡的开放
    profile.akm.append(pywifi.const.AKM_TYPE_WPA2PSK)  # wifi加密算法
    profile.cipher = pywifi.const.CIPHER_TYPE_CCMP # 加密单元
    profile.key = password  # 密码
    time.sleep(2)
    tmp_profile = iface.add_network_profile(profile)  # 设定新的链接文件
    iface.connect(tmp_profile)  # 连接AP'
    logging.info('connectinig AP......')
    time.sleep(5)
    logging.info('wait connection 5s.....')
    for i in range(waittimes):
        if iface.status() != pywifi.const.IFACE_CONNECTED:
            logging.info('connection status is :{}'.format(iface.status()))
            logging.info('retry check wirless status {}... wait 1s'.format(i))
            time.sleep(1)
            if iface.status() == pywifi.const.IFACE_CONNECTED:
                logging.info('connection status is :{}'.format(iface.status()))
                status = True
                break
        else:
            logging.info('connection AP succeed!')
            status = True
            break
    return status

def ping(host,count=5):
    """
    执行ping ,并且将结果进行按行保存；
    :param host: ping 的目标机器；
    :param count: ping 次数；
    :return: 结果返回并按行读取放到list 进行返回；
    """
    cmd = 'ping {} -n {}'.format(host, count)
    res = subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE)
    logging.info('execute ping {}:,count:{}'.format(host,count))
    out = res.stdout.readlines()
    rl =[]
    for o in out:
        print(o.decode(encoding='gb18030'))
        logging.info(o.decode(encoding='gb18030'))
        rl.append(o.decode(encoding='gb18030'))
    return rl


def check_ping(data, errorrange=25):
    """
    检查Ping是否通过，如果在允许的范围内，那么认为通过，否则失败；
    :param data: ping 获取到的执行结果的数组；
    :param errorrange:
    :return: list:True：ping 结果检查通过，False:结果检查失败；同时返回失败的次数；
    """
    result = False
    s = (data[-4:])
    logging.info('recevied ping result from ping:{}'.format(s))
    for sl in s:
        begin = ''.join(sl).find('(')
        end = ''.join(sl).find('%')
        if begin != -1 and end != -1:
            res =sl[begin + 1 : end ]
            logging.info(res)
            if int(res.strip()) <= errorrange:
                result = True
                break
            else:
                global PINGFAILCOUNT
                PINGFAILCOUNT += 1
                logging.info('ping failed .... pingfailecount:'.format(PINGFAILCOUNT))
                break
    logging.info('check ping result:{} '.format(result))
    return [result,PINGFAILCOUNT]

def apper_ping(ssid, passd, host, waittimes):
    """
    逻辑调用方法，登录AP并且执行ping 操作；
    :return:None；
    """
    pingr = False
    failedcount = PINGFAILCOUNT   #这里可能有点问题
    res = loginAP(ssid, passd, waittimes=waittimes)
    time.sleep(3)
    if res:
        pingres = ping(host, count=10)
        chkp = check_ping(pingres,errorrange=20)
        if chkp[0]:
            logging.info('check ping succeed!')
            pingr = True
        else:
            failedcount = chkp[1]
            logging.error('check ping FAIL!.... and failed count: '.format(failedcount))
    else:
        logging.info('connect AP FAIL')
        logging.info(res)
    return [pingr, failedcount]

if __name__ == '__main__':
    log()
    while True:
        GCOUNT += 1
        logging.info('
' * 2)
        logging.info('*' * 80)
        logging.info('test {} times.........'.format(GCOUNT))
        logging.info('*' * 80)
        if apper_ping('testpsk','12345678', host='192.168.204.254', waittimes=10)[0]:
            # iface.disconnect()
            GSUCCESSCOUNT += 1
        else:
            GFAILEDCOUNT += 1
        logging.info("=" * 40)
        logging.info('login all times:{}'.format(GCOUNT))
        logging.info('login all succeed times:{}'.format(GSUCCESSCOUNT))
        logging.info('login all failed times:{}'.format(GFAILEDCOUNT))
        logging.info('login ping failed times:{}'.format(PINGFAILCOUNT))
        logging.info('login ping failed rate:{}%'.format(PINGFAILCOUNT / GCOUNT * 100))
        logging.info('logdin succeed rate:{}%'.format(GSUCCESSCOUNT / GCOUNT * 100))
        logging.info("=" * 40)
        try:
            iface.remove_all_network_profiles() # 删除所有的wifi文件
            logging.info('delete all connection......')
            time.sleep(60)
        except Exception as e:
            logging.info('delete connection error,msg:{}'.format(e))