如何将selenium python代码拆分成多个函数

from selenium import webdriver

def open_browser():
    # 打开浏览器
    driver = webdriver.Chrome()
    return driver

def login(driver, username, password):
    # 登录操作
    driver.get("https://example.com/login")
    driver.find_element_by_id("username").send_keys(username)
    driver.find_element_by_id("password").send_keys(password)
    driver.find_element_by_id("login-button").click()

def search(driver, keyword):
    # 搜索操作
    driver.find_element_by_id("search-box").send_keys(keyword)
    driver.find_element_by_id("search-button").click()

def get_search_results(driver):
    # 获取搜索结果
    results = driver.find_elements_by_class_name("result-item")
    return results

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

# 主函数
def main():
    driver = open_browser()
    login(driver, "username", "password")
    search(driver, "keyword")
    results = get_search_results(driver)
    for result in results:
        print(result.text)
    close_browser(driver)

if __name__ == "__main__":
    main()