[转载]Python自动切换wifi

前言

最近在打游戏的时候经常会遇到断网的情况，每次都需要切出去切换wifi，很烦人，所以就写了个用Python自动切换可用的wifi来减少困扰。

几个系统命令

这次我们是使用python中的os模块来模拟命令行来执行命令切换wifi，所以在写程序之前，可以先了解一下几个命令。

查看当前wifi：netsh wlan show interfaces查看所有wifi：netsh wlan show profiles连接wifi：netsh wlan connect name="wifi名称"

思路

这次我们写的程序的主要思路如下：

1.获取当前wifi
2.测试当前wifi能否ping通百度
3.如果能ping通则等待5s后继续测试
4.如果ping不通则在能够连接的wifi中随机选择一个来连接

代码

获取当前wifi

import os
import subprocessdef get_current_wifi():cmd = 'netsh wlan show interfaces'p = subprocess.Popen(cmd,stdin=subprocess.PIPE,stdout=subprocess.PIPE,stderr=subprocess.PIPE,shell=True)ret = p.stdout.read()index = ret.find("SSID")if index > 0:return ret[index:].split(':')[1].split('\r\n')[0].strip()else:return None

这里我们使用subprocess.Popen函数来模拟执行命令行命令，并通过read()方法得到命令行的结果，接着对结果进行分析可以得到当前的wifi。

测试能否ping通

def check_ping(ip, count=1, timeout=1000):cmd = 'ping -n %d -w %d %s > NUL' % (count, timeout, ip)res = os.system(cmd)return 'ok' if res == 0 else 'failed'

这里我们首先构建了一个cmd命令来ping我们自己传递过来的ip地址，然后使用os.system()函数执行该命令，如果返回值为0则ping通，否则失败。

自动切换wifi

import randomdef auto_switch_wifi(wifiList):wifi = random.choice(wifiList)cmd = 'netsh wlan connect name={}".format(wifi)res = os.system(cmd)return 'ok' if res == 0 else 'failed'

在auto_switch_wifi()函数中，我们接收一个可用的wifi列表，然后再列表中随机选择一个wifi进行切换，如果成功则返回ok。

到这里我们的几大基本模块已经写完了，下面上完整代码。

# _*_ coding:utf-8 _*_
import os
import time
import subprocess
import randomdef check_ping(ip, count=1, timeout=1000):cmd = 'ping -n %d -w %d %s > NUL' % (count, timeout, ip)# 通过os.system()方法执行命令response = os.system(cmd)return 'ok' if response == 0 else 'failed'def get_current_wifi():cmd = 'netsh wlan show interfaces'p = subprocess.Popen(cmd,stdin=subprocess.PIPE,stdout=subprocess.PIPE,stderr=subprocess.PIPE,shell=True)ret = p.stdout.read()index = ret.find('SSID')if index > 0:return ret[index:].split(':')[1].split('\r\n')[0].strip()def auto_switch_wifi(wifiList):wifi = random.choice(wifiList)cmd = 'netsh wlan connect name="%s"' % wifires = os.system(cmd)return 'ok' if res == 0 else 'failed'def main():# 百度ipipTest = '61.135.169.121'# 可以切换的wifiwifiList = ['HUAWEI-5DD8']while True:current_wifi = get_current_wifi()print "当前的wifi为：", current_wifiif check_ping(ipTest, 2) != 'ok':print "联网失败，正在切换wifi"if auto_switch_wifi(wifiList) == 'ok':print "切换成功"print "-" * 40else:continuetime.sleep(5)else:print "可以成功联网"print '-' * 40time.sleep(5)if __name__ == "__main__":main()

总结

人生苦短，我用python！代码还有可以完善的地方，如果想要扩展更多功能的童鞋可以自己探索哈！