main.py

import os
from dotenv import load_dotenv
import tweepy
import time
import spreadsheet
from locApi import loc
import datetime

load_dotenv()

google_map = loc()  # Object of loc class

services = ['oxygen', 'beds', 'bed']    # List of services available

token = os.getenv("TOKEN")                       # API Key-Tokens
token_secret = os.getenv("TOKEN_SECRET")
consumer_key = os.getenv("CONSUMER_KEY")
consumer_secret = os.getenv("CONSUMER_SECRET")

auth = tweepy.OAuthHandler(consumer_key, consumer_secret)       # Connecting to bot using creds
auth.set_access_token(token, token_secret)
api = tweepy.API(auth)

print("\nTwitter Bot is Up!")
print("made with <3 by Covid Warriors")

def retrieve_id(file_name):
    '''
    Function to retrieve tweet id from file
    and save it in a list of ids.
    '''

    ids = []
    with open(file_name, 'r') as file:
        f = file.readlines()
        for i in range(len(f)):
            if '\n' in f[i]:
                ids.append(f[i][:-1].strip())
            else:
                ids.append(f[i].strip())
    return ids

def store_id(id, file_name):
    '''
    Function to store tweet key into file.
    '''

    with open(file_name, 'a') as f:
        f.write(str(id) + '\n')
    return

def scrape(hashtag, date_since):
    
    tweets = tweepy.Cursor(api.search, q=hashtag, lang="en", since=date_since, tweet_mode='extended').items()

    tweets_list = [tweet for tweet in tweets]

    for tweet in reversed(tweets_list):
        id = tweet.id_str
        
        try:
            ids = retrieve_id('lastseen_id.txt')

            if id not in ids:
                text = tweet.full_text

                state = find_state(text.lower())
                service_need = find_service(text.lower())

                print(str(id) + ' - ' + text, flush=True)

                available = spreadsheet.get_data(text.split(), state, service_need)

                if not available:
                    available = spreadsheet.get_data(text.split(), state, service_need, statewise=True)

                tweet_toSend = spreadsheet.get_tweet(available, service_need)

                api.update_status('@' + tweet.user.screen_name + " " + tweet_toSend, id)

                store_id(id, 'lastseen_id.txt')
            else:
                print("Already replied to " + str(id))
        except:
            store_id(id, 'lastseen_id.txt')


def find_state(tweet):
    '''
    Find state from tweet
    Iterate through tweet words and put them in google map API
    fetch the state and return it.
    '''

    tweet_list = tweet.split()

    for t in tweet_list:
        try:
            address = google_map.locate(t)
            display_name = address['display_name'].split(',')
            country = display_name[-1].strip()
        except:
            country = "NA"

        if country == "India":
            try:
                state = display_name[-2].strip()
            except IndexError:
                continue
            try:
                temp = int(state)
                state = display_name[-3].strip()
                return state
            except ValueError:
                return state

def find_service(tweet):
    '''
    Find the service need from user's tweet
    '''
    tweet_list = tweet.split()

    for t in tweet_list:
        if t in services:
            if t == "oxygen":
                index = tweet_list.index(t)
                if tweet_list[index + 1] == "bed" or tweet_list[index + 1] == 'beds':
                    t = "oxygen bed"
            elif t == "beds":
                t = "bed"
            return t

def date():
    today = datetime.date.today()
    yesterday = today - datetime.timedelta(days=1)
    return str(yesterday)

while True:
    yesterday = date()
    scrape("#hospitalbeds", yesterday)
    #scrape("#CovidHelp", yesterday)
    #scrape("#NeedOxygen", yesterday)
    #scrape("#COVIDEmergency", yesterday)
    #scrape("#covidwarriorbottesting", yesterday)
    time.sleep(100)