79 lines
2.9 KiB
Python
79 lines
2.9 KiB
Python
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
|