discord-havoc/util/jesusmemes.py

import aiohttp
from typing import Optional
import regex
from regex import Pattern
import os
import random
import logging
import traceback
from util.catbox import CatboxAsync

"""
Jesus Meme Generator
(requires Catbox uploader)
"""


class JesusMemeGenerator:
    def __init__(self) -> None:
        self.MEMEAPIURL = "https://apimeme.com/meme?meme="
        self.MEMESTORAGEDIR = os.path.join(os.path.expanduser("~"), "memes")
        self.top_line_regex: Pattern = regex.compile(
            r"[^\p{Letter}\p{Number}\p{Punctuation}\p{Horiz_Space}\p{Currency_Symbol}]"
        )
        self.bottom_line_regex: Pattern = regex.compile(
            r"[^\p{Letter}\p{Number}\p{Punctuation}\p{Horiz_Space}\p{Currency_Symbol}]"
        )
        self.url_regex_1: Pattern = regex.compile(r"\p{Horiz_Space}")
        self.url_regex_2: Pattern = regex.compile(r"#")

    async def create_meme(
        self, top_line: str, bottom_line: str, meme="Jesus-Talking-To-Cool-Dude"
    ) -> Optional[str]:
        """
        Create Meme

        Args:
            top_line (str): Top line of meme
            bottom_line (str): Bottom line of meme
            meme (str): The meme to use, defaults to Jesus-Talking-To-Cool-Dude

        Returns:
            Optional[str]

        """
        try:
            if not top_line or not bottom_line:
                return None
            top_line = self.top_line_regex.sub("", top_line.strip())
            bottom_line = self.bottom_line_regex.sub("", bottom_line.strip())
            out_fname: Optional[str] = None

            if len(top_line) < 1 or len(bottom_line) < 1:
                return None

            formed_url: str = (
                f"{self.MEMEAPIURL}{meme}&top={top_line.strip()}&bottom={bottom_line.strip()}"
            )
            formed_url = self.url_regex_1.sub(
                "+", self.url_regex_2.sub("%23", formed_url.strip())
            )
            timeout = aiohttp.ClientTimeout(total=15)
            async with aiohttp.ClientSession(timeout=timeout) as session:
                async with session.get(formed_url) as response:
                    UUID = f"{random.getrandbits(8)}-{random.getrandbits(8)}"
                    out_fname = f"{UUID}.jpg"
                    with open(f"{self.MEMESTORAGEDIR}/{out_fname}", "wb") as f:
                        f.write(await response.read())

            if not out_fname:
                uploader = CatboxAsync()
                meme_link: Optional[str] = await uploader.upload(
                    f"{self.MEMESTORAGEDIR}/{out_fname}"
                )
                if not meme_link:
                    logging.info("Meme upload failed!")
                    return None
                return meme_link
        except:
            print(traceback.format_exc())
            pass
        return None