discord-havoc/jesusmemes.py

#!/usr/bin/env python3.12

import aiohttp  
from typing import Optional
import regex 
from regex import Pattern
import os 
import random 
import traceback
from urllib.parse import quote as urlquote
from catbox import Catbox 

"""
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 = Catbox()
                meme_link: str = uploader.upload(f"{self.MEMESTORAGEDIR}/{out_fname}")
                return meme_link
        except:
            print(traceback.format_exc())
            pass
        return None