Python不從字典發送密鑰

Question

我正在嘗試使用硒來自動化任務。 我的代碼將分解為我定義的函數。

我首先閱讀一張Excel工作表，然后將數據作為字符串存儲在字典中。

def get_excel():
    d = defaultdict(list)

    workbook = openpyxl.load_workbook(sys.argv[1])
    sheet = workbook.get_sheet_by_name('Sheet1')

    row_count = sheet.max_row
    for r in range(2, row_count + 1):
        d[str(sheet.cell(r, 4).value)].append((str(sheet.cell(r, 1).value), str(sheet.cell(r, 2).value), str(sheet.cell(r, 3).value)))
    return d

現在，我有一個單獨的函數，可從excel獲取數據：

def get_emp_data(FirstName, LastName, Email1, EmployeeID, driver):
    f_name = driver.find_element_by_name('f_name')
    l_name = driver.find_element_by_name('l_name')
    email = driver.find_element_by_name('contact_id.email')
    employeeID = driver.find_element_by_name('contact_id.custom')

    time.sleep(5)
    #writing in the fields
    f_name.send_keys(FirstName)
    l_name.send_keys(LastName)
    email.send_keys(Email1)
    employeeID.send_keys(EmployeeID)

    #clicking save button
    save = driver.find_element_by_id('saveButton').click()

    driver.get('link to web address')

接下來，我有一個執行自動化的功能。 它傳入來自Excel工作表的數據。

def log_in():
    chrome_path = ("C:\\chromedriver.exe")

    driver = webdriver.Chrome(chrome_path)
    driver.get("link to web address")

    #setting variables for username and password fields
    username = driver.find_element_by_name('login')
    password = driver.find_element_by_name('password')

    #input email and password
    username.send_keys("my email")
    password.send_keys("my password")

    #click submit
    driver.find_element_by_id('submitButton').click()

    #click New Customer Button
    time.sleep(5)
    driver.find_element_by_id('newCustomerButton').click()
    return driver

最后，我將所有內容放入主函數中：

def main():
    excel_data = get_excel()
    driver = log_in()
    time.sleep(2)
    for EmployeeID in excel_data:
        try:
            get_emp_data(excel_data[EmployeeID][0], excel_data[EmployeeID][1], excel_data[EmployeeID][2], EmployeeID, driver)
        continue
    except:
        print(FirstName)

然后我打電話給main（）

main()

除了編寫excel的值外，代碼還執行所有應做的工作。 它應該為log_in（），使用excel中的第一行數據，單擊“保存”，轉到下一個編輯頁面，並使用excel中的下一行數據，然后再次保存，直到excel中沒有更多數據為止。 我認為將數據傳遞到函數中的方式存在問題。 任何幫助表示贊賞

Answer 1

我已解決此錯誤的所有問題，並且自動化的代碼/過程運行完美！ 解析的代碼如下：

from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from collections import defaultdict
import openpyxl
import time
import os
import sys
import traceback

def get_excel():
    d = defaultdict(list)

    workbook = openpyxl.load_workbook(sys.argv[1])
    sheet = workbook.get_sheet_by_name('Sheet1')

    row_count = sheet.max_row
    for r in range(2, row_count + 1):
        d[(sheet.cell(r, 1).value), (sheet.cell(r, 2).value), (sheet.cell(r, 3).value), (sheet.cell(r, 4).value)]
    return d

def get_emp_data(FirstName, LastName, Email1, EmployeeID, driver):
    emp_type = driver.find_element_by_xpath("//select[@id='name_of_id']/option[text()='option_name']").click()
    f_name = driver.find_element_by_name('f_name')
    l_name = driver.find_element_by_name('l_name')
    email = driver.find_element_by_name('contact_id.email')
    employeeID = driver.find_element_by_name('contact_id.custom')

    f_name.send_keys(FirstName)
    l_name.send_keys(LastName)
    email.send_keys(Email1)
    employeeID.send_keys(EmployeeID)

    save = driver.find_element_by_id('saveButton').click()

def log_in():
    chrome_path = ("C:\chromedriver.exe")

    driver = webdriver.Chrome(chrome_path)
    driver.get("link to webpage")

    username = driver.find_element_by_name('login')
    password = driver.find_element_by_name('password')

    username.send_keys("username")
    password.send_keys("password")

    driver.find_element_by_id('submitButton').click()

    time.sleep(5)
    driver.find_element_by_id('newCustomerButton').click()

    return driver

def main():
    excel_data = get_excel()
    driver = log_in()
    time.sleep(3)
    for EmployeeID in excel_data:
        try:
            get_emp_data(EmployeeID[0], EmployeeID[1], EmployeeID[2], EmployeeID[3], driver)
            time.sleep(3)
            driver.get("link to webpage")
            time.sleep(5)
            driver.find_element_by_id('newCustomerButton').click()
            time.sleep(2)
            continue
        except:
            driver.quit()
    driver.get("link to webpage")

main()