diff options
Diffstat (limited to 'python/download_emails.py')
| -rw-r--r-- | python/download_emails.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/python/download_emails.py b/python/download_emails.py index be18b1f7..7bf41561 100644 --- a/python/download_emails.py +++ b/python/download_emails.py @@ -13,7 +13,7 @@ import base64 import binascii from send_email import get_gmail_service import argparse -flags = argparse.ArgumentParser(parents=[tools.argparser]).parse_args() +#flags = argparse.ArgumentParser(parents=[tools.argparser]).parse_args() SCOPES = 'https://www.googleapis.com/auth/gmail.readonly' CLIENT_SECRET_FILE = 'secret.json' @@ -78,7 +78,7 @@ def get_msg(service, user_id, msg_id): def msg_content(msg): """Extract subject and body from a gmail message""" subject = [x['value'] for x in msg['payload']['headers'] if x['name']=='Subject'][0] - content = base64.b64decode(msg['payload']['body']['data']).decode('utf-8') + content = base64.urlsafe_b64decode(msg['payload']['body']['data']).decode('utf-8') date = msg['internalDate'] ## date /1000 to get timestamp return subject, content, date @@ -93,7 +93,7 @@ def update_emails(): if msg['id'] not in current_msgs: try: subject, content, date = msg_content(get_msg(service, 'me', msg['id'])) - except (binascii.Error, KeyError): + except (binascii.Error, KeyError, UnicodeDecodeError) as e: print("error decoding {0}".format(msg['id'])) continue else: |