如何使用pickle（python）加载文件夹中的每个文件

Question

我正在编写一个脚本，用pickle将每个文件加载到文件夹中！但是我不知道如何加载每个文件，然后执行操作，完成操作，然后对下一个文件执行操作！

使用网络驱动程序加载Cookie的示例：

import pickle
import selenium.webdriver

driver = selenium.webdriver.Firefox()
driver.get("http://www.facebook.com")
cookies = pickle.load(open("cookies.txt", "rb"))
for cookie in cookies:
    driver.add_cookie(cookie)
driver.get("http://www.facebook.com")
#then perform an action 

文件夹中有多个文件cookies1.txt，cookies2.txt，cookies3.txt ...，但是我不知道如何一个个地运行每个文件，有人可以帮我吗？谢谢

Answer 1

首先获取文件夹中所有cookie文件的名称：

import os
files = os.listdir(path)

从不同文件加载所有cookie：

for f in files:
    cookies = pickle.load(open(f, "rb"))
    for cookie in cookies:
        driver.add_cookie(cookie)

然后做你的东西