Made get_all_mentioned to take into account sometimes tweets might have no entities defined

This commit is contained in:
Manuel Cortez 2021-06-27 18:04:26 -05:00
parent 002e1ccb55
commit 4b60a79e49

View File

@ -100,10 +100,11 @@ def is_media(tweet):
def get_all_mentioned(tweet, conf, field="screen_name"): def get_all_mentioned(tweet, conf, field="screen_name"):
""" Gets all users that have been mentioned.""" """ Gets all users that have been mentioned."""
results = [] results = []
for i in tweet.entities["user_mentions"]: if hasattr(tweet, "entities") and tweet.entities.get("user_mentions"):
if i["screen_name"] != conf["user_name"] and i["id_str"] != tweet.user: for i in tweet.entities["user_mentions"]:
if i.get(field) not in results: if i["screen_name"] != conf["user_name"] and i["id_str"] != tweet.user:
results.append(i.get(field)) if i.get(field) not in results:
results.append(i.get(field))
return results return results
def get_all_users(tweet, session): def get_all_users(tweet, session):