时间戳

首页 > 解决方案 > Python不从字典中发送键

python - Python不从字典中发送键

问题描述

我正在尝试使用 selenium 自动执行一项任务。我的代码将分解为我定义的函数。

我首先阅读了一张 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

最后，我将所有内容都放入一个 main 函数中：

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没有数据为止。我认为我将数据传递给我的函数的方式有问题。任何帮助表示赞赏

标签： pythonseleniumdictionaryselenium-webdriver

解决方案

我已经解决了这个错误的所有问题，自动化的代码/过程完美运行！解析后的代码如下：

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()

推荐阅读