manage.py

import yaml

def new_tweeet(text, api):
    try:
        api.update_status(status=text)
        print( "\nSuccessfully tweeted ")
    except:
        print("\nUnsuccessfull ")
  

def del_tweet(api):
    del_count = int(input("How many tweets would you like to delete? "))
    with open("Data/grabbed_tweets.yaml", "r") as f:
        tweets_id = yaml.safe_load(f)
    tweets = list(tweets_id.keys())
    for i in range(del_count):
        tweet_index = tweets[i]
        tweet_id = tweets_id[tweet_index]['id']
        try:
            api.destroy_status(tweet_id)
            print("Deleted tweet with id: " + str(tweet_id))
        except:
            print("Unable to delete tweet with id: " + str(tweet_id))

def get_tweets(user_id, api):
    count = int(input("How many tweets would you like to grab? "))
    for i in range(count):
        try:
            print("\nFetching tweet " + str(i+1) + " of " + str(count))
            tweets = api.user_timeline(screen_name = user_id, count = count, exclude_replies=True, include_rts = False)
            print("Tweet fetched!\n")
        except:
            print("Unable to fetch tweets")

    
    grabbed_tweets = {}
    i = 1

    for tweet in tweets:
        j = str("tweet_" + str(i))
        grabbed_tweets[j] = {}
        grabbed_tweets[j]['id'] = tweet.id
        grabbed_tweets[j]['text'] = tweet.text
        grabbed_tweets[j]['name'] = tweet.user.name
        i += 1
    
    with open("Data/grabbed_tweets.yaml", "w") as f:
        f.write(yaml.dump(grabbed_tweets, sort_keys=False, default_flow_style=False))

    print("\nTweets saved to Data/grabbed_tweets.yaml")