from selenium import webdriver import time # 导入selenium4 from selenium.webdriver.common.by import By # 创建一个浏览器实例 driver = webdriver.Chrome() # 打开网站 driver.get('http://192.168.31.114:4399/') time.sleep(3) # 输入账号 driver.find_element(By.XPATH, "/html/body/div[1]/div/div/div[1]/div[2]/div[1]/div/v-card-texts/form/div[1]/div[1]/div/div[3]/input").send_keys("13732237096") # 输入密码 driver.find_element(By.XPATH, "/html/body/div[1]/div/div/div[1]/div[2]/div[1]/div/v-card-texts/form/div[2]/div[1]/div/div[3]/input").send_keys( "zxcvbnm123") # 点击空白处等待登陆按钮生效 driver.find_element(By.CLASS_NAME, "v-messages").click() # 点击登陆 driver.find_element(By.XPATH, "/html/body/div[1]/div/div/div[1]/div[2]/div[1]/div/v-card-texts/form/button").click() time.sleep(2) # 关闭浏览器 driver.quit() print("注册成功")这段代码需要怎么优化
时间: 2023-12-11 07:04:55 浏览: 174
这段代码可以进行如下优化:
1. 使用显式等待代替 time.sleep(),可以更加准确地等待页面元素加载完成。
2. 使用 Page Object 设计模式,将页面元素和操作封装到一个 Page 类中,提高代码的可维护性。
3. 使用配置文件或命令行参数来传递账号和密码,避免将敏感信息硬编码在代码中。
下面是一个可能的优化示例:
```
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
class LoginPage:
url = 'http://192.168.31.114:4399/'
account_input = (By.XPATH, "/html/body/div[1]/div/div/div[1]/div[2]/div[1]/div/v-card-texts/form/div[1]/div[1]/div/div[3]/input")
password_input = (By.XPATH, "/html/body/div[1]/div/div/div[1]/div[2]/div[1]/div/v-card-texts/form/div[2]/div[1]/div/div[3]/input")
login_button = (By.XPATH, "/html/body/div[1]/div/div/div[1]/div[2]/div[1]/div/v-card-texts/form/button")
def __init__(self, driver):
self.driver = driver
def open(self):
self.driver.get(LoginPage.url)
def input_account(self, account):
element = WebDriverWait(self.driver, 10).until(EC.presence_of_element_located(LoginPage.account_input))
element.send_keys(account)
def input_password(self, password):
element = WebDriverWait(self.driver, 10).until(EC.presence_of_element_located(LoginPage.password_input))
element.send_keys(password)
def click_login_button(self):
element = WebDriverWait(self.driver, 10).until(EC.element_to_be_clickable(LoginPage.login_button))
element.click()
if __name__ == '__main__':
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--account', required=True, help='Login account')
parser.add_argument('--password', required=True, help='Login password')
args = parser.parse_args()
driver = webdriver.Chrome()
login_page = LoginPage(driver)
login_page.open()
login_page.input_account(args.account)
login_page.input_password(args.password)
login_page.click_login_button()
driver.quit()
print("登陆成功")
```
在这个示例中,我们使用了 Page Object 设计模式,将页面元素和操作都封装到了 LoginPage 类中。在主程序中,我们使用了 argparse 模块来解析命令行参数,避免了将账号和密码硬编码在代码中。同时,我们使用了显式等待来等待页面元素加载完成。
阅读全文