unimore/pds-2021-g2-nest
unimore/pds-2021-g2-nest
1
Fork 0
mirror of https://github.com/pds-nest/nest.git synced 2024-11-22 04:54:18 +00:00
Code Activity

🐛 Fix email and tweet posting

Browse source
This commit is contained in:
Steffo 2021-05-30 17:07:19 +02:00
parent e1f4b360f5
commit 231b93e851
Signed by: steffo
GPG key ID: 6965406171929D01
1 changed files with 15 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
nest_crawler/alert_trigger.py
View file

@ -2,7 +2,12 @@ from datetime import datetime, timedelta
from nest_backend.database import * from nest_backend.database import *
from .authentication import authenticate from .authentication import authenticate
import smtplib import smtplib
import os
import tweepy as tw import tweepy as tw
import datetime
MESSAGE = "❗ {alert_name}: la soglia di allerta è stata superata alle {now}!"
def is_repo_alert_triggered(repository_id): def is_repo_alert_triggered(repository_id):
@ -37,20 +42,20 @@ def is_repo_alert_triggered(repository_id):
ext.session.commit() ext.session.commit()
print("alert triggered") print("alert triggered")
alerts_triggered.append(alert) alerts_triggered.append(alert)
#send_notification_email(alert) send_notification_email(alert)
send_notification_tweet(alert) send_notification_tweet(alert)
def send_notification_email(alert): def send_notification_email(alert):
owner_repo = alert.repository.owner
conditions_string = ''
for condition in alert.conditions:
conditions_string += condition.condition.content + ','
conditions_string = conditions_string[:-1]
smtpObj = None
try: try:
with smtplib.SMTP(host='localhost') as smtpObj: with smtplib.SMTP_SSL(host=os.environ["SMTP_HOST"], port=587) as smtpObj:
smtpObj.sendmail("alert@nest.com", owner_repo.email, "Alert triggered") smtpObj.ehlo()
smtpObj.starttls()
smtpObj.ehlo()
smtpObj.login(os.environ["SMTP_USERNAME"], os.environ["SMTP_PASSWORD"])
smtpObj.sendmail(os.environ["SMTP_FROM_EMAIL"],
alert.repository.owner.email,
MESSAGE.format(alert_name=alert.name, now=datetime.datetime.now().isoformat()))
print("Successfully sent email") print("Successfully sent email")
except smtplib.SMTPException: except smtplib.SMTPException:
print("Error: unable to send email") print("Error: unable to send email")
@ -64,7 +69,7 @@ def send_notification_tweet(alert):
conditions_string = conditions_string[:-1] conditions_string = conditions_string[:-1]
print(conditions_string) print(conditions_string)
try: try:
api.update_status(f"L'alert {alert.name} è stato attivato! C'è stato un incremento di popolarità negli argomenti di ricerca {conditions_string}") api.update_status(MESSAGE.format(alert_name=alert.name, now=datetime.datetime.now().isoformat()))
except tw.errors.Forbidden: except tw.errors.Forbidden:
print("Il tweet e' gia' stato pubblicato") print("Il tweet e' gia' stato pubblicato")