selenium中把代码封装成类调用

可以将Selenium代码封装成一个类，然后在其他文件或程序中调用该类的方法。以下是一个简单的示例：

from selenium import webdriver

class MySelenium:
    def __init__(self):
        self.driver = webdriver.Chrome() # 初始化浏览器驱动

    def open_url(self, url):
        self.driver.get(url) # 打开指定链接

    def click_element(self, element):
        el = self.driver.find_element_by_xpath(element)
        el.click() # 点击指定元素

    def close_browser(self):
        self.driver.quit() # 关闭浏览器

if __name__ == '__main__':
    my_selenium = MySelenium()
    my_selenium.open_url('https://www.baidu.com')
    my_selenium.click_element('//*[@id="su"]')
    my_selenium.close_browser()

在上面的示例代码中，我们定义了一个名为`MySelenium`的类，该类包含了打开链接、点击元素和关闭浏览器等方法。在`__init__`方法中初始化了一个Chrome浏览器驱动。

在主程序中，我们实例化了`MySelenium`对象，并调用了`open_url`和`click_element`方法来打开百度搜索页面并点击搜索按钮。最后，我们调用了`close_browser`方法关闭浏览器。

通过将Selenium代码封装成一个类，我们可以更方便地在其他程序中调用它，提高代码复用性和可维护性。