test.py

import tweepy
import time
import logging
from connection import connect_account

logging.basicConfig(level=logging.INFO, filename='app.log',
                    filemode='w', format='%(name)s - %(levelname)s - %(message)s')
logger = logging.getLogger()


class Listener(tweepy.StreamListener):
     """
      This is a class which overrides tweepy.StreamListener and adds own code.
    """
    usernames = []

    def __init__(self, api):
        self.api = api
        self.me = api.me()

    def on_status(self, tweet):
        """
            Attributes: 
            self (object): Listener object. 
            tweet (object): Status object. 
        """
        print("On status block")
        logger.info(f"Processing tweet id {tweet.id}")
        # print(tweet.user.screen_name)
        self.usernames.append(tweet.user.screen_name)
        # print(len(tweets))

        if not tweet.favorited:
            # Like if not done yet
            try:
                tweet.favorite()
            except Exception as e:
                logger.error("Error on liking", exc_info=True)
            logger.info(f"Liked tweet {tweet.id}")
        user = self.api.get_user(tweet.user.screen_name)
        follow(self.api, user)
        # to increase the time taken for hits
        time.sleep(2.4)

    def on_error(self, status_code):
        """
            Attributes: 
            self (object): Listener object. 
            status_code (int): Error code. 
            """
        if status_code == 420:
            # returning False in on_error disconnects the stream
            return False


def follow_usernames(api, usernames):
    """
    Attributes: 
        api (object): tweepy.API() object which is the bot's account authenticated using the given keys. 
        usernames (list): List of usernames input which have to be followed. 
    """
    for name in usernames:
        try:
            user = api.get_user(name)
            follow(api, user)

        except Exception as e:
            logger.error("User doesnot exist", exc_info=True)


def follow(api, user):
    """
    Attributes: 
        api (object): tweepy.API() object which is the bot's account authenticated using the given keys. 
        user (object): User object. 
    """
    # print("Follow block")
    if not user.following:
        try:
            api.create_friendship(user.id)
        except Exception as e:
            logger.info(f"Already following {user.name}")
            logger.error("Error on following", exc_info=True)
    logger.info(f"Followed {user.name}")


def main():
    """
    Inputs from the command line or uses inbuilt lists.
    """
    choice = int(input("Input through command line or inbuilt ? 1 | 2 : "))
    if choice == 1:
        keywords = []
        usernames = []

        print("Input hashtags alongwith #")
        while True:
            ch = int(input("Continue? 1 for Y 0 for N"))
            if ch == 0:
                break
            keywords.append(input("Input the keywords"))

        print("Input usernames without @")
        while True:
            ch = int(input("Continue? 1 for Y 0 for N"))
            if ch == 0:
                break
            usernames.append(input("Input the screen_names"))
    else:
        keywords = ["#POSTPONEJEE_NEET", "#SCpostponeJEE_NEET",
                    "#jeeneetpostpone", "#neetjeepostpone"]
        usernames = ["DrRPNishank", "myogiadityanath", "ZeeNews"]

    print(keywords)
    print(usernames)
    """
    Connects to the bot's twitter developer account and uses the keys given in connection.py
    """
    api = connect_account()
    """
    Follows usernames in the list
    """
    follow_usernames(api, usernames)

    """
    Looks for tweets with the given hashtags in the stream
    """
    tweets_listener = Listener(api)
    stream = tweepy.Stream(api.auth, tweets_listener)
    stream.filter(track=keywords, languages=["en"])


if __name__ == "__main__":
    main()