Python 在脚本中调用函数

我正在写一个从这些开始的脚本

import time

from selenium import webdriver

from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By

driver = webdriver.Chrome(....)

然后我想在这个文件内部或外部定义一个函数，它使用driver的值。。。但无论我做什么，我都不会保留driver的值，即使我将这些行复制并粘贴到函数中。。。我的函数输入变量不包括驱动程序。。。但在它里面，它需要司机。。。就像：

def test(i,j)

---

我是Python新手。我知道在一些语言中，你可以调用驱动程序一个全局变量，然后它的值将被带入函数中。。。但是我应该如何在Python中设置所有代码的格式，方法是高亮显示并按ctrl+k。请向我们显示要复制粘贴这些行的文件的完整代码。