import logging import random from twython import Twython, TwythonError from bot.aws_secrets import get_secret logger = logging.getLogger() logger.setLevel(logging.INFO) class Winston: def __init__(self): self.bot = Twython( app_key=get_secret("CONSUMER_KEY"), app_secret=get_secret("CONSUMER_SECRET"), oauth_token=get_secret("ACCESS_TOKEN_KEY"), oauth_token_secret=get_secret("ACCESS_TOKEN_SECRET"), ) self.potential_tweets = [ "@PlayOverwatch I didn't pay my taxes!", "I'm wanted in over 60 countries!", "Overwatch.", "Echo!", "@PlayOverwatch #LetWinstonWallRide", "@PlayOverwatch #LetWinstonWallClimb", "Please Delete Echo\n\nSincerely, Winston From Overwatch\n\n@PlayOverwatch", "Year of the Winston Event When? @PlayOverwatch", "Is This On?", "How Embarrassing!", "Winston From Overwatch.", "Is it too much to ask for some peanut butter covered toes? @PlayOverwatch", "I'm holding Sigma hostage in Paris.\nFor every hour Echo isn't nerfed, I will remove one of his toes.\n\n@PlayOverwatch", "My new year's resolutions: Less peanut butter, more... bananas.", "I'm feeling unstoppable!", "I have a crippling addiction to peanut butter." ] self.potential_tweets_with_images = [ ["Are You With Me? @PlayOverwatch", "winston_grin.jpg"], ["You won't like me when I'm angry.", "winston_angry.jpg"], ["Look out world! I'm on a rampage!", "winston_primal_rage.jpg"], ["Holy Shit.", "overwatch_meme.jpg"] ] def send_tweet(self, tweet_text): """Sends a tweet to Twitter""" self.bot.update_status(status=tweet_text) logger.info(f"Tweet Sent: '{tweet_text}'") def send_random_tweet(self): """Tweet something random from potential tweets""" result = random.choice([True, False]) if result: random_tweet = random.choice(self.potential_tweets) self.bot.update_status(status=random_tweet) logger.info(f"Random Tweet Sent: '{random_tweet}'") else: random_tweet_with_image = random.choice(self.potential_tweets_with_images) self.tweet_with_media(random_tweet_with_image) logger.info( f"Random Tweet With Image Sent:\nTweet: '{random_tweet_with_image[0]}'\nImage Filename: '{random_tweet_with_image[1]}'") def tweet_with_media(self, text_and_media): """Tweet with media + optional text""" text = text_and_media[0] filename = text_and_media[1] path = f"./media/{filename}" media = open(path, 'rb') try: response = self.bot.upload_media(media=media) self.bot.update_status(status=text, media_ids=[response['media_id']]) logger.info(f"Tweet Sent With Image:\nTweet: {text}\nImage Name: {filename}") except TwythonError as error: print(error.msg) def follow_someone(self, username): """Follows someone based on given id""" self.bot.create_friendship(screen_name=username) logger.info(f"Followed User: " f'{username}') def like_tweet(self, tweet_id): """Likes a tweet based on it's ID""" self.bot.create_favorite(id=tweet_id) logger.info(f"Liked Tweet, Tweet ID: " f'{tweet_id}')