上个版本不生成excel文件,这个会自动保存到excel[Python] 纯文本查看 复制代码 # -*- coding: UTF-8 -*-
import os
import sys
import xlwt
import time
from datetime import datetime
# 定义一个函数checkWIFI,获取电脑连接过的所有wifi名称和密码,结果以列表形式返回
def checkWIFI():
# 定义一个空列表,用来存放查询结果
list = []
# 查询所有的wifi名称
message = os.popen('netsh wlan show profiles').readlines()
# print(message)
# 不是笔记本,或者wifi模块未启动
if(message[0].find(u"没有运行")) != -1:
return message[0]
# 获取的结果是一个列表list,需要进行遍历
for i in message:
# 遍历结果含有中文会乱码,需要进行编码
result = i.strip().encode('gb2312').decode('gb2312','strict')
# print(result)
# print(result+"的长度为"+str(len(result))+",数据类型为"+str(type(result)))
# 检查每一个结果中是否含有指定关键字
if result.find(u"所有用户配置文件 : ") != -1:
# print("位置:"+str(result.find(u": ")))
# 从位置11开始截取
# print("wifi名称:"+result[11:])
# netsh wlan show profiles name="Xiaomi_216E" key=clear
# 如果找到关键字,就截取指定位置的字符串,即wifi名称,再拼接成cmd命令
command = 'netsh wlan show profiles name="' + result[11:] + '" key=clear'
# print(command)
# 执行拼接好的命令,获取含有密码的结果
per_wifi = os.popen(command.encode('gb2312').decode('gb2312')).readlines()
# print(per_wifi)
# 获取的结果是一个列表list,需要进行遍历
for j in per_wifi:
# 遍历结果含有中文会乱码,需要进行编码
passwd = j.strip().encode('gb2312').decode('gb2312','strict')
# print(passwd)
# 检查每一个结果中是否含有指定关键字
if passwd.find(u"关键内容 :") != -1:
# 获取字符串指定位置的内容并判断是否为空
if passwd[18:] != '':
# 定义一个临时列表list_temp存放每一个wifi信息,即wifi名称和密码
list_temp = []
# 将wifi名称追加到列表list_temp
list_temp.append(result[11:])
# 将密码追加到列表list_temp
list_temp.append(passwd[18:])
# 将每个wifi信息作为一个整体追加到列表list
list.append(list_temp)
return list
# 生成excel文件 (数据,名称)
def write_data_to_excel(result,name):
# 将sql作为参数传递调用get_data并将结果赋值给result,(result为一个嵌套元组)
#[["wifi","password"],["jz@cfhec","jzusers88"]]
# 实例化一个Workbook()对象(即excel文件) 设置编码
wbk = xlwt.Workbook(encoding = 'utf-8')
# 创建一个worksheet
sheet = wbk.add_sheet('Sheet1',cell_overwrite_ok=True)
# 获取当前日期,得到一个datetime对象如:(2016, 8, 9, 23, 12, 23, 424000)
today = datetime.today()
# 将获取到的datetime对象仅取日期如:2016-8-9
today_date = datetime.date(today)
# 遍历result中的没个元素。
for i in range(len(result)):
#对result的每个子元素作遍历,
for j in range(len(result[i])):
#将每一行的每个元素按行号i,列号j,写入到excel中。
# 参数对应 行, 列, 值
sheet.write(i,j,result[i][j])
# 要保存的excel名称
excel_name = name+"_"+str(today_date)+'.xls'
# 保存
wbk.save(excel_name)
# 如果该文件不是被import,则执行下面代码。
if __name__ == '__main__':
print("正在查询......")
# 定义一个变量,存放调用checkWIFI的执行结果
password_list = checkWIFI()
print("\n返回结果如下:\n")
# 判断返回值是否是list
if isinstance(password_list,list):
# 将查询结果遍历输出
for n in password_list:
print("[+]" + "\t Wifi:\t" + n[0] + "\n\tPassword:\t" + n[1]+"\n")
# 执行写入文件
write_data_to_excel(password_list,"WifiPassword")
elif isinstance(password_list,str):
print ("[-]\t"+password_list)
print("10秒后退出")
time.sleep(10)
|