From 3a8225c6816d941c6c079af256cc57b143296efc Mon Sep 17 00:00:00 2001 From: Guillaume Horel Date: Mon, 24 Jul 2017 11:26:39 -0400 Subject: fix --- email_helpers.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'email_helpers.py') diff --git a/email_helpers.py b/email_helpers.py index eeb2168..a3e2b29 100644 --- a/email_helpers.py +++ b/email_helpers.py @@ -134,13 +134,13 @@ if __name__ == "__main__": "(date, news_id, user_id, content, content_cache) " \ "VALUES(?, ?, ?, ?, ?)" try: - with open('.pickle') as fh: + with open('.pickle', 'rb') as fh: already_seen = pickle.load(fh) except FileNotFoundError: already_seen = set() for msg_id in ListMessagesWithLabels(GmailMessage._service, 'me', 'INBOX'): - if msg_id in already_seen: + if msg_id['id'] in already_seen: continue mail = GmailMessage.from_id(msg_id['id']) m = regex.match(mail['To']) @@ -169,4 +169,5 @@ if __name__ == "__main__": db.commit() print(msg_id['id'], news_id, email_addr, date, comment) already_seen.add(msg_id['id']) - pickle.dump(already_seen, '.pickle') + with open('.pickle', 'wb') as fh: + pickle.dump(already_seen, fh) -- cgit v1.2.3-70-g09d2