【Python+Selenium】猪猪练习成功版：csv文件的输入和输出（运行环境：python3.5版本）

自己实践成功的从csv文件中读取用户名密码并实际登录系统后判断是否登录成功，并将已经运行的用户名密码及运行结果输出到一个新的csv文件中~

 # coding=utf-8

 from selenium import webdriver

 from time import sleep

 import keyword

 from selenium.webdriver.common.keys import Keys

 from selenium.webdriver.support.wait import WebDriverWait

 from selenium.webdriver.support.ui import Select

 import csv

 import random

 import io

 #文件路径

 inputFilePath = "E:\script\python-script\csvfile_input.csv"

 outputFilePath = "E:\script\python-script\csvfile_output.csv"

 #判断成功登录后是否存在对应元素

 def IsElementExist(driver,classname):

     try:

         driver.find_element_by_class_name(classname)

         return True

     except:

         return False

 #读取csv文件

 def Readcsvfile(filePath):

     csvFile = open(filePath, "r")

     reader = csv.reader(csvFile)  # 返回的是迭代类型

     return reader

 #将读取的内容写入一个新的csv文档

 def Savecsvfile(filePath):

     csvFile2 = open(filePath, 'w', newline='')  # 设置newline，否则两行之间会空一行

     writer = csv.writer(csvFile2)

     writer.writerow(['UserName', 'Pwd','Result'])

     writer.writerows(outputString)

     #csvfile.close()

 #主函数

 if __name__=="__main__":

     outputString = []

     lists=Readcsvfile(inputFilePath)

     for list in lists:

         userName=list[0]

         pwd=list[1]

         print(userName)

         print(pwd)

         driver = webdriver.Chrome()

         driver.get("http://10.10.1.9/super/login/index.php")

         driver.find_element_by_id("username").send_keys(list[0])

         sleep(1)

         driver.find_element_by_id("password").send_keys(list[1])

         driver.find_element_by_id("loginbtn").click()

         result = "Unsuccessful"

         if IsElementExist(driver, "new_index_bottom_left_news"):

             result = "Successful"

         print(result)

         outputString.append([userName, pwd,result])

         driver.quit()

     Savecsvfile(outputFilePath)