* rename and little bit of cleanup
* add forcelist
This commit is contained in:
parent
4a26525905
commit
556b6fb93b
2 changed files with 13 additions and 7 deletions
|
@ -28,6 +28,7 @@ CONSUMER_SECRET | Clé secrète API disponible dans la section `Consumer Keys`
|
||||||
PSEUDOS | Pseudos du ou des compte.s que vous voulez écouter pour le snipe (a séparer avec une virgule **sans** espaces)
|
PSEUDOS | Pseudos du ou des compte.s que vous voulez écouter pour le snipe (a séparer avec une virgule **sans** espaces)
|
||||||
WHITELIST | Pseudos des comptes qui ne seront pas touché par le Bot (facultatif, a séparer avec une virgule **sans** espaces, par défaut la liste est vide)
|
WHITELIST | Pseudos des comptes qui ne seront pas touché par le Bot (facultatif, a séparer avec une virgule **sans** espaces, par défaut la liste est vide)
|
||||||
VERBOSE | Affiche plus de messages dans la console [False\|True] (facultatif, par défaut sur False)
|
VERBOSE | Affiche plus de messages dans la console [False\|True] (facultatif, par défaut sur False)
|
||||||
|
FORCELIST | Force le bot à écouter certains comptes (séparer les comptes avec une virgule **sans** espaces, par défaut la liste est vide)
|
||||||
|
|
||||||
### En local
|
### En local
|
||||||
|
|
||||||
|
|
19
main.py
19
main.py
|
@ -24,6 +24,11 @@ def load(variables) -> dict:
|
||||||
res = list(set(environ[var].split(",")) - {""})
|
res = list(set(environ[var].split(",")) - {""})
|
||||||
except:
|
except:
|
||||||
res = [] # if not its an empty list
|
res = [] # if not its an empty list
|
||||||
|
elif var == "FORCELIST": # check if FORCELIST is set
|
||||||
|
try:
|
||||||
|
res = list(set(environ[var].split(",")) - {""})
|
||||||
|
except:
|
||||||
|
res = [] # if not its an empty list
|
||||||
else:
|
else:
|
||||||
res = environ[var]
|
res = environ[var]
|
||||||
if var == "PSEUDOS":
|
if var == "PSEUDOS":
|
||||||
|
@ -164,10 +169,10 @@ def createBaseAnswers(word) -> list:
|
||||||
"""Generates default answers for a given word."""
|
"""Generates default answers for a given word."""
|
||||||
return [word, f"({word})", word.upper(), f"{word} lol"]
|
return [word, f"({word})", word.upper(), f"{word} lol"]
|
||||||
|
|
||||||
def main(accessToken: str, accessTokenSecret: str, consumerKey: str, consumerSecret: str, users: list):
|
def start():
|
||||||
"""Main method."""
|
"""Start the bot."""
|
||||||
auth = OAuthHandler(consumerKey, consumerSecret)
|
auth = OAuthHandler(keys["CONSUMER_KEY"], keys["CONSUMER_SECRET"])
|
||||||
auth.set_access_token(accessToken, accessTokenSecret)
|
auth.set_access_token(keys["TOKEN"], keys["TOKEN_SECRET"])
|
||||||
|
|
||||||
api = API(auth_handler = auth, wait_on_rate_limit = True)
|
api = API(auth_handler = auth, wait_on_rate_limit = True)
|
||||||
|
|
||||||
|
@ -186,7 +191,7 @@ def main(accessToken: str, accessTokenSecret: str, consumerKey: str, consumerSec
|
||||||
whitelist = f"@{', @'.join(keys['WHITELIST'])}"
|
whitelist = f"@{', @'.join(keys['WHITELIST'])}"
|
||||||
print(f"Liste des comptes ignorés : {whitelist}.")
|
print(f"Liste des comptes ignorés : {whitelist}.")
|
||||||
|
|
||||||
listener = Listener(api, users)
|
listener = Listener(api, keys["PSEUDOS"] + keys["FORCELIST"])
|
||||||
stream = Stream(auth = api.auth, listener = listener)
|
stream = Stream(auth = api.auth, listener = listener)
|
||||||
stream.filter(track = triggerWords, languages = ["fr"], stall_warnings = True, is_async = True)
|
stream.filter(track = triggerWords, languages = ["fr"], stall_warnings = True, is_async = True)
|
||||||
|
|
||||||
|
@ -366,6 +371,6 @@ if __name__ == "__main__":
|
||||||
triggerWords = permute(universalBase) # creation of a list of all the words (upper and lower case)
|
triggerWords = permute(universalBase) # creation of a list of all the words (upper and lower case)
|
||||||
|
|
||||||
# Loading environment variables and launching the bot
|
# Loading environment variables and launching the bot
|
||||||
keys = load(["TOKEN", "TOKEN_SECRET", "CONSUMER_KEY", "CONSUMER_SECRET", "PSEUDOS", "VERBOSE", "WHITELIST"])
|
keys = load(["TOKEN", "TOKEN_SECRET", "CONSUMER_KEY", "CONSUMER_SECRET", "PSEUDOS", "VERBOSE", "WHITELIST", "FORCELIST"])
|
||||||
print("") # just a newline
|
print("") # just a newline
|
||||||
main(keys["TOKEN"], keys["TOKEN_SECRET"], keys["CONSUMER_KEY"], keys["CONSUMER_SECRET"], keys["PSEUDOS"])
|
start()
|
||||||
|
|
Reference in a new issue