coffeeordeath/python-tools
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add influx data

Browse Source
This commit is contained in:
Alexander Hosking 2023-01-08 23:10:07 -05:00
parent 7451cf0685
commit 9c08687f25
1 changed files with 23 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
gmail/email_count.py
View File

@ -1,9 +1,13 @@
import os import os
import imaplib import imaplib
import json import json
from dotenv import load_dotenv from time import strftime
import influxdb_client
from influxdb_client.client.write_api import SYNCHRONOUS
from dotenv import load_dotenv, dotenv_values
load_dotenv() load_dotenv()
config = dotenv_values(".env")
EMAIL_ACCOUNTS = json.loads(os.environ.get("MAIL_ACCOUNTS")) EMAIL_ACCOUNTS = json.loads(os.environ.get("MAIL_ACCOUNTS"))
EMAIL_PASSWORDS = json.loads(os.environ.get("MAIL_PASSWORDS")) EMAIL_PASSWORDS = json.loads(os.environ.get("MAIL_PASSWORDS"))
@ -11,6 +15,14 @@ EMAIL_PASSWORDS = json.loads(os.environ.get("MAIL_PASSWORDS"))
# to convert lists to dictionary # to convert lists to dictionary
accounts = dict(zip(EMAIL_ACCOUNTS, EMAIL_PASSWORDS)) accounts = dict(zip(EMAIL_ACCOUNTS, EMAIL_PASSWORDS))
print(accounts) print(accounts)
# Influx DB
client = influxdb_client.InfluxDBClient(
url=config['DB_URL'],
token=config['DB_TOKEN'],
org=config['DB_ORG']
)
################ IMAP SSL ############################## ################ IMAP SSL ##############################
for account in accounts: for account in accounts:
@ -30,8 +42,8 @@ for account in accounts:
resp_code, messages = imap_ssl.search(None, 'UnSeen') resp_code, messages = imap_ssl.search(None, 'UnSeen')
if resp_code == 'OK': if resp_code == 'OK':
if len(messages[0].split()) > 0: if len(messages[0].split()) > 0:
#print('True') # print('True')
#print(messages[0].split()) # print(messages[0].split())
unread_messages = len(messages[0].split()) unread_messages = len(messages[0].split())
else: else:
print('False') print('False')
@ -42,4 +54,12 @@ for account in accounts:
imap_ssl.close() imap_ssl.close()
if unread_messages: if unread_messages:
message = f"{account} has {unread_messages} messages." message = f"{account} has {unread_messages} messages."
date = strftime("%Y-%m-%d")
time = strftime("%H:%M:%S")
timestamp = date + "T" + time + "Z"
write_api = client.write_api(write_options=SYNCHRONOUS)
point = influxdb_client.Point("unread_emails").time(timestamp).tag("account", account)\
.field("count", unread_messages)
write_api.write(bucket="personal_data",
org=config['DB_ORG'], record=point)
print(message) print(message)