有没有更好的方法可以使这段代码更短且不重复？

Question

我正在改进我的代码，使其更易于阅读、更短且更具动态性。

我使用 dotenv 隐藏了用户名和密码。

如您所见，它首先打开一个驱动程序，然后在函数中，我要求驱动程序进入 URL，然后取决于我在参数中输入的帐户，驱动程序将访问该特定电子邮件。

我已经考虑在多个变量中分配驱动程序操作，但由于 Web 登录有 2 个步骤；

1. 帐号和密码
2. 加密密码

不能那样做。

有什么想法吗？

load_dotenv()


options = webdriver.ChromeOptions()
executable_path = {"executable_path": "/usr/local/bin/chromedriver"}
browser = Browser("chrome", **executable_path, headless=False)
bw = browser.driver

html = browser.html
soup = BeautifulSoup(html, "html.parser")


def open_web_email(account):
    """[This function will open web email]

    Arguments:
        account {[account]} -- [returns what email will open]
    """
    bw.get("https://webemail.com/login")
    bw.set_window_size(1080, 820)
    # username = bw.find_element_by_xpath('//*[@id="username"]')
    # password = bw.find_element_by_xpath('//*[@id="password"]')
    # login_butt = bw.find_element_by_xpath('//*[@id="login_btn"]')
    if account == "1":
        time.sleep(2)
        bw.find_element_by_xpath('//*[@id="username"]').send_keys(os.getenv("P_USERNAME_ONE"))
        bw.find_element_by_xpath('//*[@id="password"]').send_keys(os.getenv("P_PSW_ONE"))
        bw.find_element_by_xpath('//*[@id="login_btn"]').click()


        bw.find_element_by_xpath('//*[@id="mailboxPassword"]').send_keys(os.getenv("P_MAILBOX_ONE"))
        bw.find_element_by_xpath('//*[@id="unlock_btn"]').click()
    elif account == "2":
        bw.find_element_by_xpath('//*[@id="username"]').send_keys(os.getenv("P_USERNAME_TWO"))
        bw.find_element_by_xpath('//*[@id="password"]').send_keys(os.getenv("P_PSW_TWO"))
        bw.find_element_by_xpath('//*[@id="login_btn"]').click()


        bw.find_element_by_xpath('//*[@id="mailboxPassword"]').send_keys(os.getenv("P_MAILBOX_TWO"))
        bw.find_element_by_xpath('//*[@id="unlock_btn"]').click()

也许更好的方法是使用类？

Answer 1

您可以创建另一个方法并将参数传递给该方法以使其更短。根据您的代码，只有参数不同

load_dotenv()

options = webdriver.ChromeOptions()
executable_path = {"executable_path": "/usr/local/bin/chromedriver"}
browser = Browser("chrome", **executable_path, headless=False)
bw = browser.driver

html = browser.html
soup = BeautifulSoup(html, "html.parser")


def open_web_email(account):
    """[This function will open web email]

    Arguments:
        account {[account]} -- [returns what email will open]
    """
    bw.get("https://webemail.com/login")
    bw.set_window_size(1080, 820)
    if account == "1":
        time.sleep(2)
        createAccount(P_USERNAME_ONE,P_PSW_ONE,P_MAILBOX_ONE)
    elseif account == "2":
        createAccount(P_USERNAME_TWO,P_PSW_TWO,P_MAILBOX_TWO)


def createAccount(username,password,mailBox)
{

        bw.find_element_by_xpath('//*[@id="username"]').send_keys(os.getenv("username"))
        bw.find_element_by_xpath('//*[@id="password"]').send_keys(os.getenv("password"))
        bw.find_element_by_xpath('//*[@id="login_btn"]').click()


        bw.find_element_by_xpath('//*[@id="mailboxPassword"]').send_keys(os.getenv("mailBox"))
        bw.find_element_by_xpath('//*[@id="unlock_btn"]').click()

}