以下是一个使用Python中的pickle模块保存查询结果到内存的示例代码:
import pickle
def save_query_results(query, results):
# 创建一个保存查询结果的字典
query_results = {}
# 将查询语句作为字典的键,查询结果作为字典的值
query_results[query] = results
# 使用pickle模块将字典保存到内存中
with open('query_results.pickle', 'wb') as f:
pickle.dump(query_results, f)
def load_query_results():
# 使用pickle模块从内存中加载保存的查询结果
with open('query_results.pickle', 'rb') as f:
query_results = pickle.load(f)
# 返回查询结果字典
return query_results
# 示例使用
# 假设有一个查询语句和查询结果
query = "SELECT * FROM table"
results = ["result1", "result2", "result3"]
# 保存查询结果到内存中
save_query_results(query, results)
# 加载保存的查询结果
loaded_results = load_query_results()
# 打印查询结果
print(loaded_results[query])
这个示例中,save_query_results
函数将查询语句和查询结果保存到一个字典中,然后使用pickle模块将字典保存到文件中。load_query_results
函数从文件中加载保存的查询结果字典,并返回该字典。在示例的最后,我们通过打印查询结果来验证保存和加载的正确性。
下一篇:保存查询多个数据库的SQL文件