Compare commits

..

No commits in common. "master" and "master" have entirely different histories.

74
main.py
View File

@ -13,18 +13,11 @@ from email.mime.text import MIMEText
import cherrypy import cherrypy
import logging import logging
import threading
logging.basicConfig(stream=sys.stdout, logging.basicConfig(stream=sys.stdout,
format='%(name)s - %(levelname)s - %(message)s') format='%(name)s - %(levelname)s - %(message)s')
def soft_return(msg):
time.sleep(0.2)
return(msg)
class SMTPWebhookApp: class SMTPWebhookApp:
""" """
SMTP webhook server SMTP webhook server
@ -56,49 +49,33 @@ class SMTPWebhookApp:
body = req_body['body'] body = req_body['body']
to_mail = req_body['to_mail'] to_mail = req_body['to_mail']
try:
client = smtplib.SMTP(host=SMTP_SERVER_HOST, port=SMTP_SERVER_PORT)
except (smtplib.SMTPConnectionError, TimeoutError) as e:
time.sleep(0.2)
logging.error("To: {}, error: {}".format(to_mail, e))
return {"status": 400, "message": "SMTP client error: {}.".format(
e)}
msg = MIMEText(body, 'plain', 'utf-8') msg = MIMEText(body, 'plain', 'utf-8')
msg['Subject'] = Header(subject, 'utf-8') msg['Subject'] = Header(subject, 'utf-8')
msg['From'] = FROM_MAIL msg['From'] = FROM_MAIL
msg['To'] = to_mail msg['To'] = to_mail
try:
c = 0 client.ehlo()
while (c < 2): client.starttls()
try: client.ehlo()
mythread.client.sendmail(msg['From'], client.login(msg["From"], FROM_MAIL_PASSWORD)
msg['To'], msg.as_string()) client.sendmail(msg['From'], msg['To'], msg.as_string())
response = {"status": 200, "message": "mail sent successfully"} response = {"status": 200, "message": "mail sent successfully"}
except (AttributeError, ValueError, logging.info("To: {}, mail sent successfully".format(to_mail))
smtplib.SMTPServerDisconnected): except smtplib.SMTPException as e:
if (c == 1): time.sleep(0.2)
soft_return("can't connect") logging.error("To: {}, error: {}".format(to_mail, e))
try: response = {"status": 500, "message": "some error: {}".format(e)}
print("- - - - - CREATING CLIENT - - - - -") finally:
mythread.client = smtplib.SMTP(host=SMTP_SERVER_HOST, client.quit()
port=SMTP_SERVER_PORT) return response
mythread.client.starttls()
except (smtplib.SMTPException, TimeoutError) as e:
soft_return("can't connect")
response = {"status": 500,
"message": "some error: {}".format(e)}
print("response1: ", response)
try:
mythread.client.login(msg["From"], FROM_MAIL_PASSWORD)
except smtplib.SMTPException as e:
soft_return(e)
response = {"status": 500,
"message": "some error: {}".format(e)}
print("response2: ", response)
except smtplib.SMTPException as e:
soft_return("can't send for some reason")
response = {"status": 500,
"message": "some error: {}".format(e)}
print("response3: ", response)
finally:
c += 1
return response
@cherrypy.expose @cherrypy.expose
def send_mail(self): def send_mail(self):
@ -112,6 +89,7 @@ class SMTPWebhookApp:
ROOT = SMTPWebhookApp() ROOT = SMTPWebhookApp()
SMTP_SERVER_PORT = 587 SMTP_SERVER_PORT = 587
CHERRYPY_SERVER_HOST = "0.0.0.0" CHERRYPY_SERVER_HOST = "0.0.0.0"
CHERRYPY_SERVER_PORT = 8081 CHERRYPY_SERVER_PORT = 8081
@ -131,10 +109,6 @@ if __name__ == "__main__":
ENGINE = cherrypy.engine ENGINE = cherrypy.engine
cherrypy.tree.mount(ROOT, config={}) cherrypy.tree.mount(ROOT, config={})
mythread = threading.local()
mythread.client = smtplib.SMTP(host=SMTP_SERVER_HOST,
port=SMTP_SERVER_PORT)
mythread.client.starttls()
if hasattr(ENGINE, "signal_handler"): if hasattr(ENGINE, "signal_handler"):
ENGINE.signal_handler.subscribe() ENGINE.signal_handler.subscribe()