diff --git a/bot/winston.py b/bot/winston.py index 7618902..841eae2 100644 --- a/bot/winston.py +++ b/bot/winston.py @@ -1,3 +1,4 @@ +import filetype as filetype from twython import Twython from bot.aws_secrets import get_secret @@ -12,7 +13,39 @@ class Winston: get_secret("ACCESS_TOKEN_SECRET") ) self.potential_tweets = [ - "Hello! I'm Winston From Overwatch!", - "Winston! From! Overwatch!", - "Winston? From Overwatch?" + "I didn't pay my taxes!", + "I'm wanted in over 60 countries! I'm still on the run :D", + "Overwatch", + ] + + def send_tweet(self, tweet_text): + """Sends a tweet to Twitter""" + + self.bot.update_status(status=tweet_text) + + 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') + + if filetype.is_image(media): + response = self.bot.upload_media(media=media.read()) + self.bot.update_status(status=text if text else "Test", media_ids=[response['media_id']]) + elif filetype.is_video(media): + response = self.bot.upload_video(media=media, media_type='video/mp4') + self.bot.update_status(media_ids=[response['media_id']]) + + def follow_someone(self, username): + """Follows someone based on given id""" + + self.bot.create_friendship(screen_name=username) + + def like_tweet(self, tweet_id): + """Likes a tweet based on it's ID""" + + self.bot.create_favorite(id=tweet_id)