Anri/feurBot
Archived
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases Wiki Activity

* fix crash

Browse source 
* complete migration
This commit is contained in:
Mylloon 2022-08-08 01:20:35 +02:00
parent a40866e736
commit 2513236f6c
Signed by: Anri
GPG key ID: A82D63DFF8D1317F
1 changed files with 33 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

43
main.py
View file

@ -83,6 +83,7 @@ class Listener(StreamingClient):
): ):
super(Listener, self).__init__(bearer_token, wait_on_rate_limit=True) super(Listener, self).__init__(bearer_token, wait_on_rate_limit=True)
self.client = client self.client = client
self.cache = {}
def on_connect(self): def on_connect(self):
print(f"Début du scroll sur Twitter...") print(f"Début du scroll sur Twitter...")
@ -93,18 +94,40 @@ class Listener(StreamingClient):
if len(notice["reason"]) > 0: if len(notice["reason"]) > 0:
print(f"Raison : {notice['reason']}") print(f"Raison : {notice['reason']}")
def get_user(self, uid: int) -> str:
"""Return username by ID, with cache support"""
# If not cached
if not uid in self.cache:
# Fetch from Twitter
self.cache[uid] = self.client.get_user(
id=uid, user_auth=True).data.username
# Return the username
return self.cache[uid]
def on_tweet(self, tweet: Tweet): def on_tweet(self, tweet: Tweet):
# Check if the tweet is not a retweet # Check if the tweet is not a retweet
if not tweet.text.startswith("RT @"): if not tweet.text.startswith("RT @"):
# Fetch the last word of the tweet username = self.get_user(tweet.author_id)
tweet.username: str = self.client.get_user( # Clean the tweet
id=tweet.author_id, user_auth=True).data.username lastWord = cleanTweet(tweet.text)
lastWord = cleanTweet(tweet.text).split()[-1:][0]
# Log
if keys["VERBOSE"]: if keys["VERBOSE"]:
infoLastWord = f"dernier mot : \"{lastWord}\"" if len( infoLastWord = "dernier mot : "
lastWord) > 0 else "tweet ignoré car trop de hashtags" if len(lastWord) > 0:
infoLastWord += f"dernier mot : {lastWord}"
else:
infoLastWord += "tweet ignoré car trop de hashtags"
print( print(
f"Tweet trouvé de {tweet.username} ({infoLastWord})...", end=" ") f"Tweet trouvé de {username} ({infoLastWord})...", end=" ")
if len(lastWord) > 0:
return
# Fetch the last word of the tweet
lastWord = lastWord.split()[-1:][0]
# Check if the last word found is a supported word # Check if the last word found is a supported word
if lastWord in universalBase: if lastWord in universalBase:
answer = None answer = None
@ -142,13 +165,13 @@ class Listener(StreamingClient):
# Send the tweet with the answer # Send the tweet with the answer
self.client.create_tweet( self.client.create_tweet(
in_reply_to_tweet_id=tweet.id, text=choice(answer)) in_reply_to_tweet_id=tweet.id, text=choice(answer))
print(f"{tweet.username} s'est fait {answer[0]} !") print(f"{username} s'est fait {answer[0]} !")
except Exception as error: except Exception as error:
error = loads(error.response.text)["errors"][0] error = loads(error.response.text)["errors"][0]
# https://developer.twitter.com/en/support/twitter-api/error-troubleshooting # https://developer.twitter.com/en/support/twitter-api/error-troubleshooting
show_error = True show_error = True
if error["code"] == 385: if error["code"] == 385:
error["message"] = f"Tweet supprimé ou auteur ({tweet.username}) en privé/bloqué." error["message"] = f"Tweet supprimé ou auteur ({username}) en privé/bloqué."
show_error = False show_error = False
# Show error only if relevant, always in verbose # Show error only if relevant, always in verbose
@ -311,7 +334,7 @@ def start():
for rule in create_rules(tracked_users): for rule in create_rules(tracked_users):
stream.add_rules(StreamRule(rule)) stream.add_rules(StreamRule(rule))
stream.filter(threaded=True) stream.filter(threaded=True, tweet_fields=['author_id'])
if __name__ == "__main__": if __name__ == "__main__":