From 5db3cbb0b2ae8000f338f1936401145adf5d058e Mon Sep 17 00:00:00 2001
From: "g.minoccari" <g.minoccari@gmail.com>
Date: Wed, 26 May 2021 15:39:21 +0200
Subject: [PATCH] Aggiunte funzioni per il controllo degli alert di un
 repository dato come parametro e funzioni di supporto per la notifica tramite
 mail e tramite tweet

---
 nest_crawler/__main__.py | 75 ++++++++++++++++++++++++++++++++++++++--
 1 file changed, 72 insertions(+), 3 deletions(-)

diff --git a/nest_crawler/__main__.py b/nest_crawler/__main__.py
index 468a3ce..a5fd602 100644
--- a/nest_crawler/__main__.py
+++ b/nest_crawler/__main__.py
@@ -1,7 +1,9 @@
+import random
 from nest_backend.app import app
 from nest_backend.database import *
 import tweepy as tw
-from datetime import datetime
+from datetime import datetime, timedelta
+import smtplib
 
 ext.init_app(app=app)
 
@@ -9,8 +11,8 @@ ext.init_app(app=app)
 def authenticate():
     c_k = "GEhtSyP9e98mzFeiOCSW0lvQX"
     c_s = "438cmYrl5xqaX2W7I2Bf5A9nF1pN5VtM9f77WYQnAXg1BwKJ27"
-    a_t = "1380217745732689921-8gCfr8Zx9YHKvo4OVP3HAr3kfMRkgz"
-    a_t_s = "jGOlgTs1i1itGMxDxAqFEDnv7QAui772n9hGxeSIKcwzS"
+    a_t = "1380217745732689921-IW3U1JlxhnQeGBUrnHZ2nxbxhksXUZ"
+    a_t_s = "EUoYNoj72rb2q00tUIW8eTcLJAhUAYPstZlV78W9cPpEJ"
 
     auth = tw.OAuthHandler(c_k, c_s)
     auth.set_access_token(a_t, a_t_s)
@@ -165,6 +167,73 @@ def search_repo_conditions(repository_id):
             ext.session.add(composed)
             ext.session.commit()
 
+
+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):
+    owner_repo = alert.repository.owner
+    conditions_string = ''
+    for condition in alert.conditions:
+        conditions_string += condition.condition.content + ','
+    conditions_string = conditions_string[:-1]
+    try:
+        smtpObj = smtplib.SMTP('localhost')
+        smtpObj.sendmail("alert@nest.com", owner_repo.email, "Alert triggered")
+        print("Successfully sent email")
+    except smtplib.SMTPException:
+        print("Error: unable to send email")
+    finally:
+        if smtpObj is not None:
+            smtpObj.close()
+
+def send_notification_tweet(alert):
+    api = authenticate()
+    conditions_string = ''
+    for condition in alert.conditions:
+        conditions_string += condition.condition.content + ','
+    conditions_string = conditions_string[:-1]
+    print(conditions_string)
+    api.update_status(f"L'alert {alert.name} è stato attivato! C'è stato un incremento di popolarità negli argomenti di ricerca {conditions_string}")
+    
+
+def send_test_tweet():
+    api = authenticate()
+    api.update_status("Test")
+
 if __name__ == "__main__":
     search_repo_conditions(16)
     with app.app_context():