pds-2021-g2-nest/nest_crawler/alert_trigger.py

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",
)