mirror of https://github.com/stijndcl/didier
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
from typing import Optional
|
|
|
|
from aiohttp import ClientSession
|
|
|
|
import settings
|
|
from database.schemas.relational import MemeTemplate
|
|
from didier.exceptions.missing_env import MissingEnvironmentVariable
|
|
|
|
__all__ = ["generate_meme"]
|
|
|
|
|
|
def generate_boxes(meme: MemeTemplate, fields: list[str]) -> list[dict[str:str]]:
|
|
"""Generate the template boxes for Imgflip"""
|
|
# If a meme only has 1 field, join all the arguments together into one string
|
|
if meme.field_count == 1:
|
|
fields = [" ".join(fields)]
|
|
|
|
fields = fields[: min(20, meme.field_count)]
|
|
return [{"text": text} for text in fields]
|
|
|
|
|
|
async def generate_meme(http_session: ClientSession, meme: MemeTemplate, fields: list[str]) -> Optional[str]:
|
|
"""Make a request to Imgflip to generate a meme"""
|
|
name, password = settings.IMGFLIP_NAME, settings.IMGFLIP_PASSWORD
|
|
|
|
# Ensure credentials exist
|
|
if name is None:
|
|
raise MissingEnvironmentVariable("IMGFLIP_NAME")
|
|
|
|
if password is None:
|
|
raise MissingEnvironmentVariable("IMGFLIP_PASSWORD")
|
|
|
|
boxes = generate_boxes(meme, fields)
|
|
payload = {"template_id": meme.template_id, "username": name, "password": password, "boxes": boxes}
|
|
|
|
async with http_session.post("https://api.imgflip.com/caption_image", data=payload) as response:
|
|
if response.status != 200:
|
|
return None
|
|
|
|
data = await response.json()
|
|
|
|
return data["data"]["url"]
|