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 12:04:55 浏览: 26
可以将这段代码封装成一个函数,函数名可以为`login_website`,函数的参数可以包括账号和密码,这样可以方便地调用函数来实现登录网站的功能。同时,可以使用`try except`语句来捕获异常,以便更好地处理错误。以下是示例代码:
```python
from selenium import webdriver
import time
from selenium.webdriver.common.by import By
def login_website(username, password):
try:
# 创建一个浏览器实例
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(username)
# 输入密码
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(password)
# 点击空白处等待登陆按钮生效
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("登录成功")
except Exception as e:
print("登录失败:", e)
# 调用函数
login_website("13732237096", "zxcvbnm123")
```
这样,当需要登录网站时,只需要调用`login_website`函数并传入正确的账号和密码即可。这种方式可以提高代码的可维护性,使代码更加简洁易读。