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():