python+selenium线性脚本模块化

"""

将线性脚本转化为模块脚本
脚本重复代码抽离成一个模块
1、
登陆电商系统
2、
退出电商系统
"""
# 设计一个方法，这个方法提供注册的基本步骤，通过参数传入数据
from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
import time
class very_login():
    # 实现浏览器对象创建
    def __init__(self,url):
        # 创建浏览器对象
        self.driver =webdriver.Chrome()
        # 隐性等待
        self.driver.implicitly_wait(10)
        self.driver.get(url)

    def login(self):
        # 点击登陆按钮
        self.driver.find_element_by_link_text().click()
        # 输入用户名
        self.driver.find_element_by_xpath().send_keys()
        # 输入密码
        self.driver.find_element_by_xpath().send_keys()
        # 点击登陆
        self.driver.find_element_by_xpath().click()

        time.sleep(5)

    # 退出登录
    def quita(self):
        ele = self.driver.find_element_by_xpath()
        # 鼠标悬浮
        ActionChains(self.driver).move_to_element(ele).perform()
        self.driver.find_element_by_link_text("退出").click()



    # 退出浏览器对象
    def quitb(self):
        self.driver.quit()

if __name__ == '__main__':
    aa =very_login()
    aa.login()
    aa.quita()
    aa.quitb()