首页 > 解决方案 > Pickle (Python 3.6) 写入空文件

问题描述

我正在尝试自学 Python,因此创建了一个愚蠢的脚本来检查博客网站以检查新更新,然后保存与更新相关的元数据。如果有新帖子,我打开以前的元数据,附加新元数据,然后保存。但是,我发现很多时候(无法弄清楚它何时起作用和不起作用),这些更新会产生一个空文件,并且我会丢失所有元数据。

if new_post_count > 0:
        file_name = 'all_posts' + user

        previous_posts = pickle.load(open(file_name, 'rb'))

        current_posts = get_posts(client, user, start_from_posts=0, total_posts=new_post_count)

        all_posts = previous_posts.extend(current_posts)

        f = open(file_name, 'wb')
        pickle.dump(all_posts, f)
        f.close() 

看看论坛,除了 pickle 之外,使用其他东西来保存我的数据可能更有意义(有什么建议吗?)。即使是这样,我仍然想知道我在这里做错了什么,所以我以后不会犯同样的错误。

标签: pythonpython-3.xsavepickle

解决方案


问题不在于 pickle 模块,而在于以下内容:

 all_posts = previous_posts.extend(current_posts)

实际发生的是调用了extend方法,扩展了 previous_posts ,一旦成功完成返回关键字None

然后将此关键字分配给 all_posts 而不是 previous_posts 的内容,然后将其写入文件。

尝试如下修改:

if new_post_count > 0:
    file_name = 'all_posts' + user

    previous_posts = pickle.load(open(file_name, 'rb'))

    current_posts = get_posts(client, user, start_from_posts=0, total_posts=new_post_count)

    previous_posts.extend(current_posts)

    f = open(file_name, 'wb')
    pickle.dump(previous_posts, f)
    f.close() 

更好的是包括让-弗朗索瓦的建议:

if new_post_count > 0:
    file_name = 'all_posts' + user
    with open(file_name, 'rb') as f:
        previous_posts = pickle.load(f)

    current_posts = get_posts(client, user, start_from_posts=0, total_posts=new_post_count)

    previous_posts.extend(current_posts)

    with open(file_name, 'wb') as f:
        pickle.dump(previous_posts, f) 

推荐阅读