post.py

from os import environ

from jinja2 import Environment, select_autoescape, FileSystemLoader
import requests
import tweepy

from render import generate_image, generate_properties
import sheets

CONSUMER_KEY = environ["CONSUMER_KEY"]
CONSUMER_SECRET = environ["CONSUMER_SECRET"]
ACCESS_TOKEN = environ["ACCESS_TOKEN"]  # straight from the Twitter project (OAuth 1.0)
ACCESS_TOKEN_SECRET = environ["ACCESS_TOKEN_SECRET"]

auth = tweepy.OAuthHandler(CONSUMER_KEY, CONSUMER_SECRET)
auth.set_access_token(ACCESS_TOKEN, ACCESS_TOKEN_SECRET)
api = tweepy.API(auth)
env = Environment(loader=FileSystemLoader("templates"), autoescape=select_autoescape())


def update_status(text: str, media_id: int) -> dict:
    """Posts an update to the Twitter account with attached molecule."""
    response = api.update_status(status=text, media_ids=[media_id])
    return response


def upload_file(filename: str) -> int:
    """Uploads a media file (jpg) to the Twitter API."""
    media_object = api.media_upload(filename, chunked=False)
    return media_object.media_id


def build_status(molecule: dict, template: str) -> str:
    """Builds the tweet string based on the Jinja template."""
    template = env.get_template(template)
    molecule["properties"] = generate_properties(molecule.get("smiles"))
    tweet = template.render(**molecule)
    return tweet


def main(sheet: str = "malaria"):
    """Posts a tweet with a randomly selected molecule."""
    molecule = sheets.choose_molecule(sheet)
    smiles_string = molecule.get("smiles")
    if smiles_string == None:
        return  # fail early if we can't find a SMILES string

    tweet = build_status(molecule, f"{sheet}.html")
    new_filename = "molecule.png"
    generate_image(smiles_string)  # writes to new_filename
    media_id = upload_file(new_filename)
    update_status(tweet, media_id)


main()