Kywo/perma-alert
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: Kywo:fcfde85c31eb9de0a30e8dbcbcf9d8010bec3a61
Branches Tags
Kywo:master
Kywo:main
..
compare: Kywo:35217c6b4d6c9aa711262f7021d5f0d94b15426f
Branches Tags
Kywo:master
Kywo:main

No commits in common. "fcfde85c31eb9de0a30e8dbcbcf9d8010bec3a61" and "35217c6b4d6c9aa711262f7021d5f0d94b15426f" have entirely different histories.
fcfde85c31 ... 35217c6b4d

1 changed files with 17 additions and 42 deletions
Show Stats Expand all files Collapse all files

57
main.py
View File

@ -1,5 +1,4 @@
import datetime, imaplib, email, time, utm, requests, json, os, smtplib, logging, sys import datetime, imaplib, email, time, utm, requests, json, os, smtplib, logging, sys
from urllib.parse import unquote
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
if os.environ.get("DEBUG", False): if os.environ.get("DEBUG", False):
@ -92,7 +91,6 @@ class FetchEmail():
logger.debug(f"Using search term '{searchTerm}'") logger.debug(f"Using search term '{searchTerm}'")
(result, messages) = self.connection.search(None, searchTerm) (result, messages) = self.connection.search(None, searchTerm)
body = None body = None
attachment = []
if result == "OK": if result == "OK":
mail_ids = messages[0] mail_ids = messages[0]
@ -101,7 +99,7 @@ class FetchEmail():
if mail_ids == b"": if mail_ids == b"":
logger.debug(f"No Email found") logger.debug(f"No Email found")
self.close_connection() self.close_connection()
return None, attachment return None
try: try:
#ret, data = self.connection.fetch(id_list[-1], '(UID BODY[TEXT])') #ret, data = self.connection.fetch(id_list[-1], '(UID BODY[TEXT])')
@ -117,29 +115,19 @@ class FetchEmail():
if email_msg.is_multipart(): if email_msg.is_multipart():
for part in email_msg.walk(): for part in email_msg.walk():
if part.get_content_type() == "text/html": if part.get_content_type() == "text/html":
logger.debug("Found HTML File in Email")
body = part.get_payload(decode=True) body = part.get_payload(decode=True)
body = body.decode() body = body.decode()
elif part.get_content_type() == "text/plain": elif part.get_content_type() == "text/plain":
logger.debug("Found Text in Email")
body = part.get_payload(decode=True) body = part.get_payload(decode=True)
body = body.decode() body = body.decode()
if part.get_content_type() == "application/pdf":
logger.debug("Found PDF File in Email")
attachment.append({
"filename": part.get_filename(),
"type": "application/pdf",
"content": part.get_payload(decode=True)
})
except Exception as err: except Exception as err:
logger.error(f"Could not retrieve email. Got: {err}", exc_info=True) logger.error(f"Could not retrieve email. Got: {err}", exc_info=True)
self.close_connection() self.close_connection()
return None, attachment return None
logger.debug(f"Found an email") logger.debug(f"Found an email")
return str(body), attachment return str(body)
def parse_email_address(self, email_address): def parse_email_address(self, email_address):
""" """
@ -205,23 +193,23 @@ class Alarmdepesche():
for td in tds: for td in tds:
key = str(str(td.string).strip()) key = str(str(td.string).strip())
if "Druckdatum:" in key: if key == "Druckdatum:":
self.timestamp = datetime.datetime.strptime(self.get_string(td), "%d.%m.%Y %H:%M:%S") self.timestamp = datetime.datetime.strptime(self.get_string(td), "%d.%m.%Y %H:%M:%S")
if "Einsatzstichwort:" in key: if key == "Einsatzstichwort:":
self.intervention_code = self.get_string(td) self.intervention_code = self.get_string(td)
if "Sachverhalt:" in key: if key == "Sachverhalt:":
self.intervention_message = self.get_string(td) self.intervention_message = self.get_string(td)
if "Alarmiertes Einsatzmittel:" in key: if key == "Alarmiertes Einsatzmittel:":
engine = self.get_string(td) engine = self.get_string(td)
if "," in engine: if "," in engine:
logger.warning(f"Unusual engine format, got {engine} only using last entry") logger.warning(f"Unusual engine format, got {engine} only using last entry")
engine = engine.split(",")[-1].strip() engine = engine.split(",")[-1].strip()
self.engine = engine self.engine = engine
if "Einsatznummer:" in key: if key == "Einsatznummer:":
self.intervention_number = self.get_string(td) self.intervention_number = self.get_string(td)
if "Stadt:" in key: if key == "Stadt:":
self.city = self.get_string(td) self.city = self.get_string(td)
if "Info:" in key: if key == "Info:":
self.info = self.get_string(td) self.info = self.get_string(td)
if "UTM - Koordinaten:" in key: if "UTM - Koordinaten:" in key:
utm_string = self.get_string(td).split(" ") utm_string = self.get_string(td).split(" ")
@ -293,18 +281,12 @@ class Email():
smtp_client : smtplib.SMTP smtp_client : smtplib.SMTP
def __init__(self, smtp_server, smtp_port, smtp_user, smtp_password): def __init__(self, smtp_server, smtp_port, smtp_user, smtp_password):
if os.environ.get("SMTP_SERVER", None):
smtp_server = os.environ.get("SMTP_SERVER")
if os.environ.get("SMTP_PORT", None):
smtp_port = os.environ.get("SMTP_PORT")
logger.debug(f"Connecting to SMTP Server {smtp_server}") logger.debug(f"Connecting to SMTP Server {smtp_server}")
try: try:
self.smtp_client = smtplib.SMTP(smtp_server, smtp_port) self.smtp_client = smtplib.SMTP(smtp_server, smtp_port)
self.smtp_client.ehlo() self.smtp_client.ehlo()
if os.environ.get("MAIL_SMTP_STARTTLS", True): if os.environ.get("MAIL_SMTP_STARTTLS", True):
self.smtp_client.starttls() self.smtp_client.starttls()
if os.environ.get("SMTP_AUTH", True):
self.smtp_client.login(smtp_user, smtp_password) self.smtp_client.login(smtp_user, smtp_password)
logger.debug("Connection to SMTP Sever was successfull") logger.debug("Connection to SMTP Sever was successfull")
except BaseException as err: except BaseException as err:
@ -315,10 +297,9 @@ class Email():
self.smtp_client.close() self.smtp_client.close()
def sendmail(self, alarmdepesche, msg_from, msg_to, attachment=None): def sendmail(self, alarmdepesche, msg_from, msg_to):
from email.mime.multipart import MIMEMultipart from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText from email.mime.text import MIMEText
from email.mime.application import MIMEApplication
logger.debug(f"Preparing email for {msg_to}") logger.debug(f"Preparing email for {msg_to}")
@ -340,11 +321,6 @@ class Email():
msg.attach(part1) msg.attach(part1)
msg.attach(part2) msg.attach(part2)
if attachment:
part3 = MIMEApplication(attachment[0]["content"], Name=unquote(attachment[0]["filename"]), _subtype="pdf")
part3['Content-Disposition'] = f'attachment; filename="{unquote(attachment[0]["filename"])}"'
msg.attach(part3)
logger.info(f"Sending email to {msg_to}") logger.info(f"Sending email to {msg_to}")
# sendmail function takes 3 arguments: sender's address, recipient's address # sendmail function takes 3 arguments: sender's address, recipient's address
@ -365,7 +341,6 @@ if __name__ == '__main__':
mail_user = os.environ.get('MAIL_USER', None) mail_user = os.environ.get('MAIL_USER', None)
mail_from = os.environ.get('MAIL_FROM', mail_user) mail_from = os.environ.get('MAIL_FROM', mail_user)
mail_pass = os.environ.get('MAIL_PASS', None) mail_pass = os.environ.get('MAIL_PASS', None)
mail_port = os.environ.get('MAIL_PORT', 587)
if not mail_server or not mail_user or not mail_pass: if not mail_server or not mail_user or not mail_pass:
logger.error("Environment variables not set!, exiting...") logger.error("Environment variables not set!, exiting...")
@ -388,18 +363,18 @@ if __name__ == '__main__':
fetchmail = FetchEmail(mail_server, mail_user, mail_pass) fetchmail = FetchEmail(mail_server, mail_user, mail_pass)
fetchmail.select_mailbox() fetchmail.select_mailbox()
html_message, attachment = fetchmail.fetch_messages_content("FROM els.112@cgdis.lu (UNSEEN)") html_message = fetchmail.fetch_messages_content("FROM els.112@cgdis.lu (UNSEEN)")
if html_message: if html_message:
current_alarmdepsche = Alarmdepesche() current_alarmdepsche = Alarmdepesche()
current_alarmdepsche.from_html(html_message) current_alarmdepsche.from_html(html_message)
current_alarmdepsche.saveToDisk() current_alarmdepsche.saveToDisk()
metric_api_up = current_alarmdepsche.metric_api_up metric_api_up = current_alarmdepsche.metric_api_up
members = current_alarmdepsche.get_members_emails() members = current_alarmdepsche.get_members_emails()
sendmail = Email(mail_server, mail_port, mail_user, mail_pass) sendmail = Email(mail_server, 587, mail_user, mail_pass)
for email_addr in members: for email_addr in members:
sendmail.sendmail(current_alarmdepsche, mail_from, email_addr, attachment) sendmail.sendmail(current_alarmdepsche, mail_from, email_addr)
else: else:
logger.info("No message found, exiting...") logger.info("Message could not be loaded, exiting...")
timer_global.end() timer_global.end()
prometheus.append_gauge_metric("permaalert_execution_time", timer_global.took, prometheus.append_gauge_metric("permaalert_execution_time", timer_global.took,