Aggiornato il crawler in modo che si adatti a entrambe le modalita' di valutazione delle condizioni e salvati i tweet raccolti all'interno del database

nest_crawler/__main__.py

@@ -0,0 +1,104 @@
+from nest_backend.app import app
+from nest_backend.database import *
+import tweepy as tw
+import nltk
+from nltk.corpus import stopwords
+from datetime import datetime
+
+ext.init_app(app=app)
+
+
+def authenticate():
+    c_k = "GEhtSyP9e98mzFeiOCSW0lvQX"
+    c_s = "438cmYrl5xqaX2W7I2Bf5A9nF1pN5VtM9f77WYQnAXg1BwKJ27"
+    a_t = "1380217745732689921-8gCfr8Zx9YHKvo4OVP3HAr3kfMRkgz"
+    a_t_s = "jGOlgTs1i1itGMxDxAqFEDnv7QAui772n9hGxeSIKcwzS"
+    b_t = "AAAAAAAAAAAAAAAAAAAAANjpOQEAAAAAYFKrqByk6WTMEwIYUJLrVnPeQuo%3DJUJUW2mFq9n20A7BtIRMT24PPFg1NXVtetrfp0tH1NkDFFNGo5"
+
+    auth = tw.OAuthHandler(c_k, c_s)
+    auth.set_access_token(a_t, a_t_s)
+    api = tw.API(auth, wait_on_rate_limit=True)
+    #client = tw.Client(b_t, c_k, c_s, a_t, a_t_s, wait_on_rate_limit=True);
+    return api
+
+
+def search_repo_conditions(repository_id):
+    api = authenticate()
+    geocode="44.3591600,11.7132000,20km"
+
+    repo = Repository.query.filter_by(id=repository_id).first()
+    if(repo == None):
+        print("Non esiste una repository con questo id")
+        return False
+    conditions = [use for use in repo.conditions]
+    evaluation_mode = repo.evaluation_mode
+    conditions_type = dict()
+    #Dividing condition into condition types
+    for condition in conditions:
+        # print(condition.id)
+        if condition.type not in conditions_type.keys():
+            conditions_type[condition.type]=[condition.content]
+        else:
+            conditions_type[condition.type].append(condition.content)
+
+    queryString = ""
+
+    queryConjunction = "OR" if (evaluation_mode == ConditionMode.all_or) else ""
+
+    tweetsFound = []
+
+    for types in conditions_type.keys():
+        print(types, ":", conditions_type[types])
+    coordinates_string = ""
+    if ConditionType.hashtag in conditions_type.keys():
+        for condition_content in conditions_type[ConditionType.hashtag]:
+            queryString += ("#" + condition_content + " " + queryConjunction + " ")
+    if ConditionType.coordinates in conditions_type.keys():
+        if evaluation_mode == ConditionMode.all_and:
+            if len(conditions_type[ConditionType.coordinates])==1:
+                coordinates_tweet = conditions_type[ConditionType.coordinates][0].split()
+                coordinates_string = coordinates_tweet[2]+","+coordinates_tweet[3]+","+coordinates_tweet[1]+"km"
+            else:
+                return None
+        elif evaluation_mode == ConditionMode.all_or:
+            for condition_content in conditions_type[ConditionType.coordinates]:
+                coordinates_tweet = condition_content.split()
+                coordinates_string = coordinates_tweet[2]+","+coordinates_tweet[3]+","+coordinates_tweet[1]+"km"
+                print(coordinates_string)
+                for tweet in tw.Cursor(method=api.search, geocode=coordinates_string).items(10):
+                    tweetsFound.append(tweet)
+    if ConditionType.user in conditions_type.keys():
+        for condition_content in conditions_type[ConditionType.user]:
+            queryString += ("from:" + condition_content + " " + queryConjunction + " ")
+    if ConditionType.time in conditions_type.keys():
+        for condition_content in conditions_type[ConditionType.time]:
+            if condition_content[0]=='<':
+                queryString += ("until:" + condition_content + " " + queryConjunction + " ")
+            elif condition_content[0]=='>':
+                queryString += ("since:" + condition_content + " " + queryConjunction + " ")
+    queryString = queryString[:-len(queryConjunction)-1]
+    print(queryString)
+    if(evaluation_mode == ConditionMode.all_or):
+        for tweet in tw.Cursor(method=api.search, q=queryString).items(10):
+            tweetsFound.append(tweet)
+            print(tweet.user.name + ' : ' + tweet.text)
+    elif(evaluation_mode==ConditionMode.all_and):
+        for tweet in tw.Cursor(method=api.search, q=queryString, geocode=coordinates_string).items(10):
+            tweetsFound.append(tweet)
+            print(tweet.user.name + ' : ' + tweet.text)
+    for tweet in tweetsFound:
+        if not Tweet.query.filter_by(snowflake=str(tweet.id)).all():
+            tweetDB = Tweet(snowflake=tweet.id, content=tweet.text, location=tweet.geo.coordinate.coordinates if tweet.geo is not None else "", insert_time=str(datetime.now()))
+            ext.session.add(tweetDB)
+        if not Composed.query.filter_by(snowflake=str(tweet.id), rid=repository_id).all():
+            composed = Composed(rid=repository_id, snowflake=tweet.id)
+            ext.session.add(composed)
+            ext.session.commit()
+
+
+if __name__ == "__main__":
+    search_repo_conditions(16)
+    #print(stopwords.words('italian'))
+    with app.app_context():
+        ext.create_all(app=app)
+        #start_exploring()