adding abilitie to have multiple accounts

This commit is contained in:
Mylloon 2021-08-04 00:04:38 +02:00
parent 49ae084324
commit 7f7b66727e
2 changed files with 18 additions and 9 deletions

View file

@ -9,7 +9,7 @@ TOKEN | Token d'accès disponible dans la section `Authentication Toke
TOKEN_SECRET | Token d'accès secret disponible dans la section `Authentication Tokens` sous la sous-rubrique `Access Token and Secret` TOKEN_SECRET | Token d'accès secret disponible dans la section `Authentication Tokens` sous la sous-rubrique `Access Token and Secret`
CONSUMER_KEY | Clé API disponible dans la section `Consumer Keys` CONSUMER_KEY | Clé API disponible dans la section `Consumer Keys`
CONSUMER_SECRET | Clé secrète API disponible dans la section `Consumer Keys` CONSUMER_SECRET | Clé secrète API disponible dans la section `Consumer Keys`
PSEUDO | Pseudo du compte que vous voulez écouter pour le snipe PSEUDOS | Pseudos du ou des compte.s que vous voulez écouter pour le snipe (a séparer avec une virgule **sans** espaces)
Ensuite installe les dépendances avec `pip install -r requirements.txt`. Ensuite installe les dépendances avec `pip install -r requirements.txt`.

25
main.py
View file

@ -12,17 +12,20 @@ def load(variables):
load_dotenv() # load .env file load_dotenv() # load .env file
for var in variables: for var in variables:
try: try:
keys[var] = environ[var] res = environ[var]
if var == "PSEUDOS":
res = list(set(res.split(',')) - {""}) # create a list for the channels and remove blank channels and doubles
keys[var] = res
except KeyError: except KeyError:
print(f"Please set the environment variable {var} (.env file supported)") print(f"Please set the environment variable {var} (.env file supported)")
exit(1) exit(1)
return keys return keys
class Listener(StreamListener): class Listener(StreamListener):
def __init__(self, api = None, user = None): def __init__(self, api = None, users = None):
super(Listener, self).__init__() super(Listener, self).__init__()
self.api = api self.api = api
self.listOfFriendsID = api.friends_ids(user) self.listOfFriendsID = getFriendsID(api, users)
def on_status(self, status): def on_status(self, status):
"""Answer to tweets.""" """Answer to tweets."""
@ -37,6 +40,12 @@ class Listener(StreamListener):
print(f"Error happens! {error}") print(f"Error happens! {error}")
pass pass
def getFriendsID(api, users: list):
liste = []
for user in users:
liste.extend(api.friends_ids(user))
return liste
def seniority(date: str): def seniority(date: str):
datetimeObject = datetime.strptime(date, '%a %b %d %H:%M:%S +0000 %Y') # Convert String format to datetime format datetimeObject = datetime.strptime(date, '%a %b %d %H:%M:%S +0000 %Y') # Convert String format to datetime format
datetimeObject = datetimeObject.replace(tzinfo = timezone('UTC')) # Twitter give us an UTC time datetimeObject = datetimeObject.replace(tzinfo = timezone('UTC')) # Twitter give us an UTC time
@ -63,17 +72,17 @@ def permute(array: list):
quoiListe.append(temp) quoiListe.append(temp)
return quoiListe return quoiListe
def main(accessToken, accessTokenSecret, consumerKey, consumerSecret, user): def main(accessToken: str, accessTokenSecret: str, consumerKey: str, consumerSecret: str, users: list):
"""Main method.""" """Main method."""
auth = OAuthHandler(consumerKey, consumerSecret) auth = OAuthHandler(consumerKey, consumerSecret)
auth.set_access_token(accessToken, accessTokenSecret) auth.set_access_token(accessToken, accessTokenSecret)
api = API(auth_handler = auth, wait_on_rate_limit = True) api = API(auth_handler = auth, wait_on_rate_limit = True)
listener = Listener(api, user) listener = Listener(api, users)
stream = Stream(auth = api.auth, listener = listener) stream = Stream(auth = api.auth, listener = listener)
print(f"Scroll sur Twitter avec les abonnements de @{user}...") print(f"Scroll sur Twitter avec les abonnements de @{', @'.join(users)}...")
stream.filter(track = quoi, languages = ["fr"], is_async = True) stream.filter(track = quoi, languages = ["fr"], is_async = True)
if __name__ == '__main__': if __name__ == '__main__':
@ -87,5 +96,5 @@ if __name__ == '__main__':
""" """
quoi = permute(["quoi", "koi"]) quoi = permute(["quoi", "koi"])
feur = ["feur", "(feur)", "FEUR", "feur lol", "https://twitter.com/shukuzi62/status/1422611919538724868/video/1"] feur = ["feur", "(feur)", "FEUR", "feur lol", "https://twitter.com/shukuzi62/status/1422611919538724868/video/1"]
keys = load(["TOKEN", "TOKEN_SECRET", "CONSUMER_KEY", "CONSUMER_SECRET", "PSEUDO"]) keys = load(["TOKEN", "TOKEN_SECRET", "CONSUMER_KEY", "CONSUMER_SECRET", "PSEUDOS"])
main(keys["TOKEN"], keys["TOKEN_SECRET"], keys["CONSUMER_KEY"], keys["CONSUMER_SECRET"], keys["PSEUDO"]) main(keys["TOKEN"], keys["TOKEN_SECRET"], keys["CONSUMER_KEY"], keys["CONSUMER_SECRET"], keys["PSEUDOS"])