amorfo77/parsedmarc
1
0
Fork 0
mirror of https://github.com/domainaware/parsedmarc.git synced 2026-07-09 18:25:09 +00:00
Code Issues Projects Releases Wiki Activity

PEP8 fixes

Browse Source
This commit is contained in:
Sean Whalen
2022-04-21 17:46:27 -04:00
parent 4faf8ad651
commit 90a53f2217
1 changed files with 21 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
parsedmarc/__init__.py
+21 -11
View File
@@ -1200,6 +1200,7 @@ def get_dmarc_reports_from_mailbox(connection: MailboxConnection,
return results
def get_gmail_api_creds(token_file="token.json",credentials_file="credentials.json",scopes=['https://www.googleapis.com/auth/gmail.modify']):
creds = None
@@ -1218,10 +1219,11 @@ def get_gmail_api_creds(token_file="token.json",credentials_file="credentials.js
token.write(creds.to_json())
return creds
def get_dmarc_reports_from_gmail_api(credentials_file=".credentials",token_file=".token",
reports_label="INBOX", archive_label = "DMARC Archive",
offline=False, ip_db_path=None,
scopes = ['https://mail.google.com/'], include_spam_trash=False,
scopes=['https://mail.google.com/'], include_spam_trash=False,
nameservers=None, dns_timeout=2.0,
strip_attachment_payloads=False,delete=False,
test=False,parallel=False):
@@ -1233,7 +1235,6 @@ def get_dmarc_reports_from_gmail_api(credentials_file=".credentials",token_file=
aggregate_report_msg_uids = []
forensic_report_msg_uids = []
creds = get_gmail_api_creds(token_file,credentials_file,scopes)
service = build('gmail', 'v1', credentials=creds)
@@ -1271,32 +1272,42 @@ def get_dmarc_reports_from_gmail_api(credentials_file=".credentials",token_file=
aggregate_label_id = label['id']
if reports_label_id is None:
logger.debug("Creating label {0} for reports".format(reports_label))
label = service.users().labels().create(userId='me',body={'name': reports_label, 'messageListVisibility': 'show'}).execute()
label = service.users().labels().create(userId='me',
body={'name': reports_label,
'messageListVisibility': 'show'}).execute()
reports_label_id = label['id']
if archive_label_id is None:
logger.debug("Creating label {0} for archive".format(archive_label))
label = service.users().labels().create(userId='me',body={'name': archive_label, 'messageListVisibility': 'show'}).execute()
label = service.users().labels().create(userId='me',
body={'name': archive_label,
'messageListVisibility': 'show'}).execute()
archive_label_id = label['id']
if forensic_label_id is None:
logger.debug("Creating label {0} for forensic reports".format(forensic_label))
label = service.users().labels().create(userId='me',body={'name': forensic_label, 'messageListVisibility': 'show'}).execute()
label = service.users().labels().create(userId='me',
body={'name': forensic_label,
'messageListVisibility': 'show'}).execute()
forensic_label_id = label['id']
if aggregate_label_id is None:
logger.debug("Creating label {0} for aggregate reports".format(aggregate_label))
label = service.users().labels().create(userId='me',body={'name': aggregate_label, 'messageListVisibility': 'show'}).execute()
label = service.users().labels().create(userId='me',
body={'name': aggregate_label,
'messageListVisibility': 'show'}).execute()
aggregate_label_id = label['id']
if invalid_label_id is None:
logger.debug("Creating label {0} for invalid reports".format(invalid_label))
label = service.users().labels().create(userId='me',body={'name': invalid_label, 'messageListVisibility': 'show'}).execute()
label = service.users().labels().create(userId='me',
body={'name': invalid_label,
'messageListVisibility': 'show'}).execute()
invalid_label_id = label['id']
results = service.users().messages().list(userId='me',includeSpamTrash=include_spam_trash,labelIds=[reports_label_id]).execute()
messages = results.get('messages',[])
results = service.users().messages().list(userId='me', includeSpamTrash=include_spam_trash,
labelIds=[reports_label_id]).execute()
messages = results.get('messages', [])
total_messages = results['resultSizeEstimate']
while(messages):
@@ -1399,7 +1410,6 @@ def get_dmarc_reports_from_gmail_api(credentials_file=".credentials",token_file=
msg_uid, e)
logger.error("GMail error: {0}".format(e))
results = OrderedDict([("aggregate_reports", aggregate_reports),
("forensic_reports", forensic_reports)])