【Python】檔案中csv讀取在寫入csv讀取的資料和執行是否成功。

標籤：username   port   btn   組成   data   res   ==   讀取檔案   div   

 

 

 1 # coding=utf-8 2 from selenium import webdriver 3 from time import sleep 4 import keyword 5 from selenium.webdriver.common.keys import Keys 6 from selenium.webdriver.support.wait import WebDriverWait 7 from selenium.webdriver.support.ui import Select 8 import csv 9 import random10 import io11 12 #檔案路徑13 inputFilePath = "Documents\LoginInputData.csv"14 outputFilePath = "Documents\LoginOutputData.csv"15 16 def ReadCsvFileToList(filePath):17 18     with  io.open(filePath, "r", encoding=‘utf-8‘) as csvfile:19         list = csvfile.readlines()20     return  list21 22 def SaveListToFileToList(filePath):23 24     csvfile = file(filePath, ‘wb‘)25     writer = csv.writer(csvfile)26     #表頭27     writer.writerow([‘UserName‘, ‘Pwd‘, ‘‘])28     writer.writerows(outputString)29     csvfile.close()30 31 #通過Id判斷該元素是否存在32 def IsElementExist(driver,id):33     try:34         driver.find_element_by_id(id)35         return True36     except:37         return False38 39 if __name__=="__main__":40 41     #讀取檔案到list數組裡面42     lists = ReadCsvFileToList(inputFilePath)43 #lists的組成：["grace,1","grace2,2"]44     #聲明一個輸出數組 來儲存結果集45     outputString=[]46 47     #迴圈數組48     for list in lists:49 50         #擷取每一行資料 並以逗號拆封為使用者名稱和密碼51         #list ="grace,1"52         #list.split(",")=[grace1,1]#csv是用逗號分隔，下面的list.split(",")[0]中的第1個元素即grace1,list.split(",")[1]中的第2個元素即153         userName=list.split(",")[0].strip()54         pwd= list.split(",")[1].strip()55 56         print(userName)57         print(pwd)58 59         #開始瀏覽登陸60         driver = webdriver.Firefox()61         driver.get("http://demo.pingnanlearning.com/test/login/index.php")62 63         driver.find_element_by_id("username").send_keys(userName)64         sleep(1)65         driver.find_element_by_id("password").send_keys(pwd)66         sleep(1)67         driver.find_element_by_id("loginbtn").click()68 69         sleep(2)70 71         result="Unsuccessful"72         #il判斷是否登陸成功73         if IsElementExist(driver, "maincontent"):74             #成功75             result ="Successful"76         #將結果追加到儲存結果集77         outputString.append([userName,pwd,result])78         driver.quit()79 80     #81     # 迴圈結束82     print(outputString)83 84     #將結果集寫入檔案85     SaveListToFileToList(outputFilePath)

 

【Python】檔案中csv讀取在寫入csv讀取的資料和執行是否成功。