100 lines
3.4 KiB
Python
100 lines
3.4 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import sendmail
|
||
|
|
import datetime
|
||
|
|
import email
|
||
|
|
import logger
|
||
|
|
import pytoml as toml
|
||
|
|
import imaplib
|
||
|
|
import sys
|
||
|
|
|
||
|
|
class Mailbot(object):
|
||
|
|
"""
|
||
|
|
Bot which sends Mails if mentioned via twitter/mastodon, and tells
|
||
|
|
other bots that it received mails.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self, config, logger):
|
||
|
|
"""
|
||
|
|
Creates a Bot who listens to mails and forwards them to other
|
||
|
|
bots.
|
||
|
|
|
||
|
|
:param config: (dictionary) config.toml as a dictionary of dictionaries
|
||
|
|
"""
|
||
|
|
self.config = config
|
||
|
|
self.logger = logger
|
||
|
|
|
||
|
|
try:
|
||
|
|
self.mailinglist = self.config["mail"]["list"]
|
||
|
|
except KeyError:
|
||
|
|
self.mailinglist = None
|
||
|
|
|
||
|
|
self.mailbox = imaplib.IMAP4_SSL(self.config["mail"]["imapserver"])
|
||
|
|
# context = ssl.create_default_context()
|
||
|
|
# print(self.mailbox.starttls(ssl_context=context)) # print is a debug
|
||
|
|
try:
|
||
|
|
rv, data = self.mailbox.login(self.config["mail"]["user"],
|
||
|
|
self.config["mail"]["passphrase"])
|
||
|
|
except imaplib.IMAP4.error:
|
||
|
|
logmsg = "Login to mail server failed."
|
||
|
|
logmsg = logmsg + logger.generate_tb(sys.exc_info())
|
||
|
|
logger.log(logmsg)
|
||
|
|
logger.shutdown(logmsg)
|
||
|
|
|
||
|
|
def listen(self):
|
||
|
|
"""
|
||
|
|
listen for mails which contain goodwords but no badwords.
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
rv, data = self.mailbox.select("Inbox")
|
||
|
|
if rv == 'OK':
|
||
|
|
rv, data = self.mailbox.search(None, "ALL")
|
||
|
|
print(data)
|
||
|
|
rv, data = self.mailbox.search(None, "ALL")
|
||
|
|
if rv != 'OK':
|
||
|
|
print("No messages found!")
|
||
|
|
return
|
||
|
|
|
||
|
|
for num in data[0].split():
|
||
|
|
rv, data = self.mailbox.fetch(num, '(RFC822)')
|
||
|
|
if rv != 'OK':
|
||
|
|
print("ERROR getting message", num)
|
||
|
|
return
|
||
|
|
msg = email.message_from_bytes(data[0][1])
|
||
|
|
hdr = email.header.make_header(email.header.decode_header(msg['Subject']))
|
||
|
|
subject = str(hdr)
|
||
|
|
print('Message %s: %s' % (num, subject))
|
||
|
|
print('Raw Date:', msg['Date'])
|
||
|
|
# Now convert to local date-time
|
||
|
|
date_tuple = email.utils.parsedate_tz(msg['Date'])
|
||
|
|
if date_tuple:
|
||
|
|
local_date = datetime.datetime.fromtimestamp(
|
||
|
|
email.utils.mktime_tz(date_tuple))
|
||
|
|
print ("Local Date:", local_date.strftime("%a, %d %b %Y %H:%M:%S"))
|
||
|
|
#print(msg.as_string())
|
||
|
|
author = msg.get("From") # get mail author from email header
|
||
|
|
text = msg.get_payload()
|
||
|
|
print(author)
|
||
|
|
print(text)
|
||
|
|
# :todo check if they match trigger
|
||
|
|
# :todo return a nice list of warning messages
|
||
|
|
|
||
|
|
def send_warning(self, statuses):
|
||
|
|
"""
|
||
|
|
sends warnings by twitter & mastodon to a mailing list.
|
||
|
|
"""
|
||
|
|
for status in statuses:
|
||
|
|
mailer = sendmail.Mailer(self.config)
|
||
|
|
mailer.send(status, self.mailinglist, "Warnung: Kontrolleure gesehen")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
# read config in TOML format (https://github.com/toml-lang/toml#toml)
|
||
|
|
with open('config.toml') as configfile:
|
||
|
|
config = toml.load(configfile)
|
||
|
|
|
||
|
|
logger = logger.Logger(config)
|
||
|
|
|
||
|
|
m = Mailbot(config, logger)
|
||
|
|
m.listen()
|
||
|
|
|