ReplyTweet.py

import tweepy
import time
import matplotlib.pyplot as plt
import importlib.util
import os
import gdown

spec = importlib.util.spec_from_file_location(
    "RandomPlotGenerator.py.py", "Randomness/RandomPlotGenerator.py"
)
foo = importlib.util.module_from_spec(spec)
spec.loader.exec_module(foo)

CONSUMER_KEY = os.environ.get("CONSUMER_KEY")
CONSUMER_SECRET = os.environ.get("CONSUMER_SECRET")
ACCESS_KEY = os.environ.get("ACCESS_KEY")
ACCESS_SECRET = os.environ.get("ACCESS_SECRET")

auth = tweepy.OAuthHandler(CONSUMER_KEY, CONSUMER_SECRET)
auth.set_access_token(ACCESS_KEY, ACCESS_SECRET)
api = tweepy.API(auth)

FILE_NAME = "lastSeenId.txt"


def retrieve_last_seen_id(file_name):
    f_read = open(file_name, "r")
    last_seen_id = int(f_read.read().strip())
    f_read.close()
    return last_seen_id


def store_last_seen_id(last_seen_id, file_name):
    f_write = open(file_name, "w")
    f_write.write(str(last_seen_id))
    f_write.close()
    return


def reply_to_tweet():
    print("replying...")
    feasible = True
    last_seen_id = retrieve_last_seen_id(FILE_NAME)

    mention = api.mentions_timeline(last_seen_id, tweet_mode="extended")

    for singleMention in reversed(mention):
        print(
            str(singleMention.id) + " - " + singleMention.full_text
        )  # printing all my tweets
        last_seen_id = singleMention.id
        store_last_seen_id(last_seen_id, FILE_NAME)
        if (
            "#batbot" in singleMention.full_text.lower()
            and "#help" in singleMention.full_text.lower()
        ):
            print("Help required")
            api.update_status(
                "@"
                + singleMention.user.screen_name
                + " Hi there! What would you like to simulate today?",
                singleMention.id,
            )
        elif (
            "#batbot" in singleMention.full_text.lower()
            and "plot" in singleMention.full_text.lower()
        ):
            if "model" in singleMention.full_text.lower():
                if "dfn" in singleMention.full_text.lower():
                    if "chen2020" in singleMention.full_text.lower():
                        (parameter_values, 
                        time) = foo.random_plot_generator(0, 0)
                    elif "marquis2019" in singleMention.full_text.lower():
                        (parameter_values, 
                        time) = foo.random_plot_generator(0, 1)
                    if "experiment" in singleMention.full_text.lower():
                        experiment = singleMention.full_text.split("[")
                        experiment = experiment[1].split("]")
                        experiment = experiment[0]
                        experiment = experiment.split(",")
                        print(experiment)
                        time, feasible = foo.random_plot_generator(
                            choice=1, cycle=experiment
                        )
                        print("yes")
                    if "drive cycle" in singleMention.full_text.lower():
                        tweet = singleMention.full_text.split()
                        link = tweet[-1]
                        output = 'drive_cycle.csv'
                        gdown.download(link, output, quiet=False)
                        time = foo.random_plot_generator(
                            choice=2
                        )


            if feasible:
                media = api.media_upload("replyFoo.png")
                test_string = (
                    "@"
                    + singleMention.user.screen_name
                    + " Here is your plot at time = "
                    + str(time)
                    + " seconds"
                )
                api.update_status(
                    status=test_string,
                    media_ids=[media.media_id],
                    in_reply_to_status_id=singleMention.id,
                )
            elif not feasible:
                test_string = (
                    "@"
                    + singleMention.user.screen_name
                    + " The experiment was not feasible"
                )
                api.update_status(
                    status=test_string,
                    in_reply_to_status_id=singleMention.id,
                )

            os.remove("replyFoo.png")
            os.remove("drive_cycle.csv")
            plt.clf()
        else:
            api.update_status(
                "@"
                + singleMention.user.screen_name
                + " I'm sorry, could you be a bit more specific "
                + "(remember this is only the testing phase)",
                singleMention.id,
            )


# Function to sync last seen ID
# def sync_last_seen_id():
#     last_seen_id = retrieve_last_seen_id(FILE_NAME)
#     mention = api.mentions_timeline(last_seen_id, tweet_mode="extended")

#     for singleMention in reversed(mention):
#         print(
#             str(singleMention.id) + " - " + singleMention.full_text
#         )  # printing all my tweets
#         last_seen_id = singleMention.id
#         store_last_seen_id(last_seen_id, FILE_NAME)

# sync_last_seen_id()

while True:
    reply_to_tweet()
    time.sleep(60)