Pickle 加载目录中的所有 .pkl 文件

Question

我正在为 Windows 7 使用 Python 3.3 制作通讯录应用程序。我将联系人信息存储在 pickle 文件 (.pkl) 中。我想加载文件夹中的所有 pkl 文件并用 pickle 加载它们，并使用我的 GUI 显示所有联系人的目录。到目前为止，这是我加载文件夹中所有泡菜文件的代码：

for root, dirs, files, in os.walk("LIP Source Files/Contacts/Contact Book"):
    for file in files:
        if file.endswith(".pkl"):
           contacts = file
           print(contacts)
           opencontacts = open(os.getcwd() + "/LIP Source Files/Contacts/Contact Book/" + contacts, 'rb')
           loadedcontacts = pickle.load(contacts)
           print(loadedcontacts)
       else:
          lipgui.msgbox("No contacts found!")

这是 lipgui.choicebox() 的代码：

     def choicebox(msg="Pick something."
, title=" "
, choices=()
):
"""
Present the user with a list of choices.
return the choice that he selects.
return None if he cancels the selection selection.

@arg msg: the msg to be displayed.
@arg title: the window title
@arg choices: a list or tuple of the choices to be displayed
"""
if len(choices) == 0: choices = ["Program logic error - no choices were specified."]

global __choiceboxMultipleSelect
__choiceboxMultipleSelect = 0
return __choicebox(msg,title,choices)

Answer 1

您的问题已经做了一些加载联系人的工作。 loadedcontacts = pickle.load(contacts) 行是一个好方法。但是pickle.load 需要打开的文件而不是文件名。所以不要传递contacts，而是传递opencontacts。

您可以通过在外循环之前创建一个列表来将联系人保存在一个列表中：

allcontacts = [] # Creates an empty list
for root, dirs, files in os.walk("LIP Source Files/Contacts/Contact Book"):
    # Omitted

然后，您将取消选中的每个联系人附加到该列表中：

            loadedcontacts = pickle.load(opencontacts)
            allcontacts.append(loadedcontacts)

顺便说一句：当您不再需要打开的文件时，您应该关闭它。在此示例中，这意味着您在调用 loadedcontacts = pickle.load(opencontacts) 之后调用 opencontacts.close()。