import shelve
import os
# 创建/连接数据库
db = shelve.open('files_shelf')
# 添加文件
def add_file(file_path):
name = os.path.basename(file_path)
with open(file_path, 'rb') as f:
db[name] = f.read()
# 获取文件
def get_file(file_name):
return db.get(file_name)
# 使用完后关闭
# db.close()