首先看你的代码，核心问题出在无限循环的逻辑漏洞和不完善的登录成功/失败处理上，导致程序卡在登录环节无法进入后续的main函数。我们一步步拆解问题：

主要问题点 #

1. 无限循环无法终止：当用密码B登录成功后，代码没有任何逻辑跳出while True循环，会再次回到循环开头重新输入密码A，陷入死循环，永远执行不到后面的main()调用。
2. 密码B登录后无验证：输入密码B点击登录后，你没有检查是否登录成功，也没有处理登录成功后的退出逻辑。
3. 过时的Selenium API：find_element_by_xpath这类方法已经被Selenium 4.x弃用，建议使用新的find_element(By.XPATH, 'xxx')语法。
4. 硬编码等待不可靠：time.sleep是固定等待，页面加载快慢会影响元素定位，建议用显式等待替代。

修正后的代码 #

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import time

driver = webdriver.Chrome()
wait = WebDriverWait(driver, 10)  # 显式等待，最长等待10秒

def login():
    driver.get('website_url')
    # 先定位账号和密码输入框，避免重复查找
    account_input = wait.until(EC.presence_of_element_located((By.XPATH, '//*[@id="Account"]')))
    password_input = wait.until(EC.presence_of_element_located((By.XPATH, '//*[@id="Password"]')))
    login_btn = wait.until(EC.element_to_be_clickable((By.XPATH, '//*[@class="loginBtn"]')))
    
    # 先尝试密码A登录
    account_input.send_keys('Username')
    password_input.send_keys('Password A')
    login_btn.click()
    time.sleep(1)  # 给页面一点跳转时间
    
    # 检查是否登录失败（出现密码错误提示）
    try:
        error_msg = wait.until(EC.presence_of_element_located((By.XPATH, '//*[@class="field-validation-error"]'))).text
        if error_msg == "Password Error":
            print("密码A错误，尝试密码B...")
            # 清空输入框，用密码B登录
            account_input.clear()
            password_input.clear()
            account_input.send_keys('Username')
            password_input.send_keys('Password B')
            login_btn.click()
            time.sleep(1)
            # 再次检查是否登录成功（这里可以根据登录后的页面特征判断，比如某个只有登录后才有的元素）
            wait.until(EC.presence_of_element_located((By.XPATH, '//*[某个登录成功后的元素XPath]')))
            print("密码B登录成功！")
    except:
        # 没有捕获到错误提示，说明密码A登录成功
        print("密码A登录成功！")
    
    # 登录成功后退出函数，继续执行后续代码
    return driver

login()
# 现在可以正常执行main函数了
def main():
    driver.get('webpage_url')
    # 这里可以添加后续的业务逻辑

main()

关键改进说明 #

• 移除无限循环：改为先尝试密码A，失败则尝试密码B，两次尝试后确认登录成功，直接退出函数，避免死循环。
• 显式等待替代sleep：用WebDriverWait等待元素出现/可点击，比固定sleep更稳定，能适应不同的页面加载速度。
• 登录成功验证：密码B登录后，通过等待登录后才会出现的元素（比如用户头像、首页特定元素）来确认登录成功，确保程序进入后续流程。
• 优化元素定位：提前定位常用元素，避免重复查找，提升代码效率。

另外，建议你根据实际登录后的页面，替换代码中//*[某个登录成功后的元素XPath]为真实的元素定位，这样能更准确地判断登录状态。

内容的提问来源于stack exchange，提问作者WCJ