83 lines
2.1 KiB
Python
83 lines
2.1 KiB
Python
import email
|
|
import imaplib
|
|
import time
|
|
from SubjectHandler import SubjectHandler
|
|
|
|
# custom imports for nodes
|
|
# import yourLib
|
|
|
|
SubHandler = SubjectHandler()
|
|
|
|
mail_addr = None
|
|
mail_pwd = None
|
|
|
|
"""
|
|
CONFIG SECTION
|
|
"""
|
|
imap_port = 993
|
|
imap_server = "imap.gmail.com"
|
|
mail_domain = "gmail.com"
|
|
mail_time_delta = 30 # in seconds
|
|
|
|
secrets_file = "./secrets.txt"
|
|
|
|
# add all nodes here in this(↓) pattern
|
|
# SubHandler += SubjectHandler.Node(regex, function)
|
|
SubHandler += SubjectHandler.Node("Test[0-9]*", lambda x: print(str(x)))
|
|
|
|
"""
|
|
END OF CONFIG SECTION
|
|
"""
|
|
|
|
|
|
def import_secrets(file):
|
|
"""
|
|
will save the secrets like mail address and password to variables, so they aren't hardcoded
|
|
:param file: path to file: content is '[MAIL]\n[PWD]'
|
|
:return: secrets, that can be saved to global variables
|
|
"""
|
|
global mail_addr
|
|
global mail_pwd
|
|
with open(file) as f:
|
|
mail_addr = f.readline()
|
|
mail_pwd = f.readline()
|
|
if '@' not in mail_addr:
|
|
mail_addr += '@' + mail_domain
|
|
|
|
|
|
def mail_handling():
|
|
"""
|
|
Opens inbox, checks for new email fetches then and deletes them from the inbox
|
|
:return: array of emails: subject, sender, body
|
|
"""
|
|
messages = []
|
|
mail_connection = imaplib.IMAP4_SSL(imap_server, imap_port)
|
|
mail_connection.login(mail_addr, mail_pwd)
|
|
mail_connection.select('inbox')
|
|
|
|
typ, data = mail_connection.search(None, 'ALL')
|
|
mail_ids = data[0].split()
|
|
for i in reversed(mail_ids):
|
|
typ, data = mail_connection.fetch(i, '(RFC822)')
|
|
for response_part in data:
|
|
if isinstance(response_part, tuple):
|
|
msg = email.message_from_string(response_part[1].decode('utf-8'))
|
|
email_subject = msg['subject']
|
|
email_from = msg['from']
|
|
email_content = str(msg)
|
|
messages.append([email_subject, email_from, email_content])
|
|
mail_connection.store(i, '+FLAGS', '\\Deleted')
|
|
mail_connection.logout()
|
|
return messages
|
|
|
|
|
|
import_secrets(secrets_file)
|
|
while True:
|
|
SubHandler.run(mail_handling())
|
|
time.sleep(mail_time_delta)
|
|
|
|
|
|
|
|
|
|
|