82 lines
3.2 KiB
Python
82 lines
3.2 KiB
Python
# This is a rewrite port of a cog from Anismash:
|
|
# https://github.com/Anismash/Ani-Cogs/blob/master/retrosign/retrosign.py
|
|
|
|
import aiohttp
|
|
from bs4 import BeautifulSoup as bs
|
|
import discord
|
|
from redbot.core import commands
|
|
from io import BytesIO
|
|
import os
|
|
from random import choice
|
|
import re
|
|
import unicodedata
|
|
|
|
|
|
class Retrosign(commands.Cog):
|
|
"""Make an 80s retro sign. Originally by Anismash"""
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.session = aiohttp.ClientSession()
|
|
|
|
@commands.cooldown(1, 15, discord.ext.commands.BucketType.guild)
|
|
@commands.command(name="retrosign")
|
|
async def retrosign(self, ctx, *, content: str):
|
|
"""Make a retrosign with 3 words seperated by ';' or with one word in the middle."""
|
|
texts = [t.strip() for t in content.split(";")]
|
|
if len(texts) == 1:
|
|
lenstr = len(texts[0])
|
|
if lenstr <= 15:
|
|
data = dict(
|
|
bcg=choice([1, 2, 3, 4, 5]),
|
|
txt=choice([1, 2, 3, 4]),
|
|
text1="",
|
|
text2=texts[0],
|
|
text3="",
|
|
)
|
|
else:
|
|
return await ctx.send("\N{CROSS MARK} Your line is too long (14 character limit)")
|
|
elif len(texts) == 3:
|
|
texts[0] = unicodedata.normalize('NFD', texts[0]).encode('ascii', 'ignore')
|
|
texts[0] = texts[0].decode('UTF-8')
|
|
texts[0] = re.sub(r'[^A-Za-z0-9 ]', '', texts[0])
|
|
if len(texts[0]) >= 15:
|
|
return await ctx.send(
|
|
"\N{CROSS MARK} Your first line is too long (14 character limit)"
|
|
)
|
|
if len(texts[1]) >= 13:
|
|
return await ctx.send(
|
|
"\N{CROSS MARK} Your second line is too long (12 character limit)"
|
|
)
|
|
if len(texts[2]) >= 26:
|
|
return await ctx.send(
|
|
"\N{CROSS MARK} Your third line is too long (25 character limit)"
|
|
)
|
|
data = dict(
|
|
bcg=choice([1, 2, 3, 4, 5]),
|
|
txt=choice([1, 2, 3, 4]),
|
|
text1=texts[0],
|
|
text2=texts[1],
|
|
text3=texts[2],
|
|
)
|
|
else:
|
|
return await ctx.send(
|
|
"\N{CROSS MARK} please provide three words seperated by ';' or one word"
|
|
)
|
|
|
|
async with ctx.channel.typing():
|
|
async with self.session.post(
|
|
"https://photofunia.com/effects/retro-wave", data=data
|
|
) as response:
|
|
if response.status == 200:
|
|
soup = bs(await response.text(), "html.parser")
|
|
download_url = soup.find("div", class_="downloads-container").ul.li.a["href"]
|
|
async with self.session.request("GET", download_url) as image_response:
|
|
if image_response.status == 200:
|
|
image_data = await image_response.read()
|
|
with BytesIO(image_data) as temp_image:
|
|
image = discord.File(fp=temp_image, filename="image.png")
|
|
await ctx.channel.send(file=image)
|
|
|
|
def cog_unload(self):
|
|
self.bot.loop.create_task(self.session.close())
|