lambda_function.py

import tweepy
from ArtInstance import Art
import SendEmail
import os
from datetime import date, timedelta


'''
lambda_function and package directory are zipped into lambda.zip.
we then use lambda.zip with AWS to run this program once a week for a year

lambda_function is the same as TwitterBot, just tweaked for AWSs
'''


def auth_client():
    # authentification
    client = tweepy.Client(
        bearer_token=os.environ['BEARER_TOKEN'],
        consumer_key=os.environ['API_KEY'],
        consumer_secret=os.environ['API_SECRET'],
        access_token=os.environ['ACCESS_TOKEN'],
        access_token_secret=os.environ['TOKEN_SECRET'],
        wait_on_rate_limit=True,
    )
    return client


def search_tweets():
    client = auth_client()

    query = "(@TrashTastePod OR #TrashTaste) (#trashtasteart OR #art OR #fanart OR #digitalart OR #illustration OR #drawing) -is:retweet has:media"
    # with Essential access, can only search through past 7 days of tweets, max of 100
    tweets = client.search_recent_tweets(query=query, tweet_fields=['entities'],
                                         media_fields=['url'], expansions=['attachments.media_keys', 'author_id'],
                                         max_results=100)
    return tweets


def query_tweets(tweets, client, art_class):
    media = {m.media_key: m for m in tweets.includes['media']}
    users = {user.id: user.username for user in tweets.includes['users']}

    try:
        # query tweets
        for tweet in tweets.data:
            media_keys = tweet.attachments['media_keys']
            media_urls = [media[key].url for key in media_keys]
            media_types = [media[key]['type'] for key in media_keys]

            author_id = tweet.author_id
            username = users[author_id]
            tweet_id = tweet.id
            tweet_url = tweet.entities['urls'][0]['display_url']

            # add data to dict
            art_class.add_art(author_id=author_id, tweet_id=tweet_id,
                              media_keys=media_keys, media_urls=media_urls, media_types=media_types, tweet_url=tweet_url, username=username)

            # liking and retweeting
            client.like(tweet.id)
            client.retweet(tweet.id)

    # if dict is empty, will return err
    except TypeError as e:
        print(f'There are no recent fan arts, try again in 7 days: {e}')

    else:
        print('Query complete')


def save_tweet_media(art_class):
    # saves the images into a folder
    # saving images at end so the api doesnt stay open for long periods
    art_class.save_all_users_media()

    # saves author_id, tweet_id, media_keys, media_urls, media_types, tweet_url, username to a json
    art_class.save_to_json()


def lambda_handler(event, context):
    # instance
    trash_taste_art = Art()

    client = auth_client()
    tweets = search_tweets()
    query_tweets(tweets, client, trash_taste_art)
    save_tweet_media(trash_taste_art)

    # directory variables
    tmp_folder = '../../tmp'
    data_folder = os.path.join(tmp_folder, 'data')
    zip_folder = os.path.join(tmp_folder, 'zip')

    # compress data dir as zip file
    today = date.today()
    last_week = date.today() - timedelta(days=7)
    zip_filename = f'TT-Data-{last_week}-{today}.zip'

    SendEmail.check_folder_exists(zip_folder)
    SendEmail.zip_dir(data_folder, zip_folder, zip_filename)

    SendEmail.send_email(zip_folder, zip_filename,
                         os.environ['EMAIL_ADDRESS'], os.environ['GMAIL_APP_PASS'], os.environ['CC_EMAIL_ADDRESS'])

    SendEmail.remove_folder_contents(data_folder)
    SendEmail.remove_folder_contents(zip_folder)