from selenium import webdriver
driver=webdriver.Firefox()
driver.implicitly_wait(10)
driver.get("http://www.126.com")

def login():
    driver.find_element_by_xpath("//input[@name='email']").clear()
    driver.find_element_by_xpath("//input[@name='email']").send_keys("username")
    driver.find_element_by_name("password").clear()
    driver.find_element_by_name("password").send_keys("password")
    driver.find_element_by_id("dologin").click()
def logout():
    driver.find_element_by_link_text("退出").click()
    driver.quit()
login()
logout()
 

將登入的操作封裝到login()函式中，將退出的操作封裝到logout()函式中，用到時呼叫即可

public.py

class Login():
    def user_login(self,driver):
        driver.find_element_by_xpath("//input[@name='email']").clear()
        driver.find_element_by_xpath("//input[@name='email']").send_keys("username")
        driver.find_element_by_name("password").clear()
        driver.find_element_by_name("password").send_keys("password")
        driver.find_element_by_id("dologin").click()
    def user_logout(self,driver):
        driver.find_element_by_link_text("退出").click()
        driver.quit()
 

from selenium import webdriver
from public import Login
driver=webdriver.Firefox()
driver.implicitly_wait(10)
driver.get("http://www.126.com")
Login().user_login(driver)
Login().user_logout(driver)