mirror of
https://github.com/pds-nest/nest.git
synced 2024-11-22 04:54:18 +00:00
76 lines
3.1 KiB
Python
76 lines
3.1 KiB
Python
from datetime import datetime, timedelta
|
|
from nest_backend.database import *
|
|
from .authentication import authenticate
|
|
import smtplib
|
|
import os
|
|
import tweepy as tw
|
|
|
|
|
|
MESSAGE = "❗ {alert_name}: la soglia di allerta è stata superata alle {now}!"
|
|
|
|
|
|
def is_repo_alert_triggered(repository_id):
|
|
repo = Repository.query.filter_by(id=repository_id).first()
|
|
if repo is None:
|
|
print("Non esiste una repository con questo id")
|
|
return False
|
|
alerts = [alert for alert in repo.alerts]
|
|
alerts_triggered = []
|
|
for alert in alerts:
|
|
evaluation_mode = alert.evaluation_mode
|
|
conditions = [condition.condition for condition in alert.conditions]
|
|
repo_tweets = [tweet.tweet for tweet in repo.tweets]
|
|
alert_tweets = set(repo_tweets)
|
|
if evaluation_mode == ConditionMode.all_and:
|
|
for condition in conditions:
|
|
alert_tweets.intersection(set([tweet.tweet for tweet in condition.tweets]))
|
|
elif evaluation_mode == ConditionMode.all_or:
|
|
conditions_tweet = set()
|
|
for condition in conditions:
|
|
conditions_tweet.update([tweet.tweet for tweet in condition.tweets])
|
|
alert_tweets = alert_tweets.intersection(conditions_tweet)
|
|
end_time = datetime.now()
|
|
window_size_hours = timedelta(hours=alert.window_size)
|
|
last_notification_time = min([notification.ora for notification in alert.notifications] if len(alert.notifications)>0 else [end_time - window_size_hours])
|
|
start_time = max(end_time - window_size_hours, last_notification_time)
|
|
alert_tweets = [tweet for tweet in alert_tweets if (end_time > tweet.insert_time > start_time)]
|
|
print(f"I tweet corrispondenti sono: {len(alert_tweets)}")
|
|
if len(alert_tweets) >= alert.limit:
|
|
alert_notification = Notification(ora=str(datetime.now()), alert_id=alert.id)
|
|
ext.session.add(alert_notification)
|
|
ext.session.commit()
|
|
print("Alert triggered!")
|
|
alerts_triggered.append(alert)
|
|
send_notification_email(alert)
|
|
send_notification_tweet(alert)
|
|
|
|
|
|
def send_notification_email(alert):
|
|
try:
|
|
with smtplib.SMTP(host=os.environ["SMTP_HOST"], port=587) as smtpObj:
|
|
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.now().isoformat()).encode("utf8"))
|
|
print("Successfully sent email")
|
|
except smtplib.SMTPException:
|
|
print("Error: unable to send email")
|
|
|
|
|
|
def send_notification_tweet(alert):
|
|
api = authenticate()
|
|
try:
|
|
api.update_status(MESSAGE.format(alert_name=alert.name, now=datetime.now().isoformat()))
|
|
except tw.errors.Forbidden:
|
|
print("Il tweet e' gia' stato pubblicato")
|
|
|
|
|
|
__all__ = (
|
|
"MESSAGE",
|
|
"is_repo_alert_triggered",
|
|
"send_notification_email",
|
|
"send_notification_tweet",
|
|
)
|