diff --git a/app/cogs/canvas_tools.py b/app/cogs/canvas_tools.py new file mode 100644 index 00000000..ed7e4796 --- /dev/null +++ b/app/cogs/canvas_tools.py @@ -0,0 +1,80 @@ +from discord.ext import commands +import core_utils +import discord +import random +import requests +import os + + +class Canvass(commands.Cog): + def __init__(self, bot): + self.bot: commands.Bot = bot + + @commands.command(name="simp") + async def simp(self, ctx: commands.Context, image_link=None): + if not image_link: + return await ctx.send("You must provide a link to an image") + response = requests.get( + "https://some-random-api.ml/canvas/simpcard/?avatar=%s" % image_link + ) + with open("/app/simp.jpg", "wb") as f: + f.write(response.content) + await ctx.message.delete() + await ctx.send(file=discord.File("/app/simp.jpg")) + os.remove("/app/simp.jpg") + + @commands.command(name="horny") + async def horny(self, ctx: commands.Context, image_link=None): + if not image_link: + return await ctx.send("You must provide a link to an image") + response = requests.get( + "https://some-random-api.ml/canvas/horny/?avatar=%s" % image_link + ) + with open("/app/horny.jpg", "wb") as f: + f.write(response.content) + await ctx.message.delete() + await ctx.send(file=discord.File("/app/horny.jpg")) + os.remove("/app/horny.jpg") + + @commands.command(name="wasted") + async def wasted(self, ctx: commands.Context, image_link=None): + if not image_link: + return await ctx.send("You must provide a link to an image") + response = requests.get( + "https://some-random-api.ml/canvas/wasted/?avatar=%s" % image_link + ) + with open("/app/wasted.jpg", "wb") as f: + f.write(response.content) + await ctx.message.delete() + await ctx.send(file=discord.File("/app/wasted.jpg")) + os.remove("/app/wasted.jpg") + + @commands.command(name="pixelate") + async def pixelate(self, ctx: commands.Context, image_link=None): + if not image_link: + return await ctx.send("You must provide a link to an image") + response = requests.get( + "https://some-random-api.ml/canvas/pixelate/?avatar=%s" % image_link + ) + with open("/app/pixelate.jpg", "wb") as f: + f.write(response.content) + await ctx.message.delete() + await ctx.send(file=discord.File("/app/pixelate.jpg")) + os.remove("/app/pixelate.jpg") + + @commands.command(name="blur") + async def blur(self, ctx: commands.Context, image_link=None): + if not image_link: + return await ctx.send("You must provide a link to an image") + response = requests.get( + "https://some-random-api.ml/canvas/blur/?avatar=%s" % image_link + ) + with open("/app/blur.jpg", "wb") as f: + f.write(response.content) + await ctx.message.delete() + await ctx.send(file=discord.File("/app/blur.jpg")) + os.remove("/app/blur.jpg") + + +def setup(bot): + bot.add_cog(Canvass(bot)) diff --git a/app/cogs/cheeky_functions.py b/app/cogs/cheeky_functions.py index 5b5e7e4a..d842d822 100644 --- a/app/cogs/cheeky_functions.py +++ b/app/cogs/cheeky_functions.py @@ -16,45 +16,6 @@ class Cheeky(commands.Cog): await ctx.reply(decide.decide(ctx.message.content)) - @commands.command(name="simp") - async def simp(self, ctx: commands.Context): - if len(ctx.message.content.split()) == 1: - return await ctx.send("You must provide a link to an image") - response = requests.get( - "https://some-random-api.ml/canvas/simpcard/?avatar=%s" - % ctx.message.content.split()[-1] - ) - with open("/app/simp.jpg", "wb") as f: - f.write(response.content) - await ctx.message.delete() - await ctx.send(file=discord.File("/app/simp.jpg")) - - @commands.command(name="horny") - async def horny(self, ctx: commands.Context): - if len(ctx.message.content.split()) == 1: - return await ctx.send("You must provide a link to an image") - response = requests.get( - "https://some-random-api.ml/canvas/horny/?avatar=%s" - % ctx.message.content.split()[-1] - ) - with open("/app/horny.jpg", "wb") as f: - f.write(response.content) - await ctx.message.delete() - await ctx.send(file=discord.File("/app/horny.jpg")) - - @commands.command(name="wasted") - async def wasted(self, ctx: commands.Context): - if len(ctx.message.content.split()) == 1: - return await ctx.send("You must provide a link to an image") - response = requests.get( - "https://some-random-api.ml/canvas/wasted/?avatar=%s" - % ctx.message.content.split()[-1] - ) - with open("/app/wasted.jpg", "wb") as f: - f.write(response.content) - await ctx.message.delete() - await ctx.send(file=discord.File("/app/wasted.jpg")) - @commands.command(name="verify") async def verify(self, ctx: commands.Context): message = list(await ctx.message.channel.history(limit=2).flatten())[-1] diff --git a/format_code.sh b/format_code.sh index 63bf36e8..87a39e44 100755 --- a/format_code.sh +++ b/format_code.sh @@ -1 +1 @@ -docker run --rm -ti -v $(pwd):/tmp/python/app python:alpine sh -c "cd /tmp/python ; pip install black; black ." +docker run --rm -ti -v $(pwd):/tmp/python/app python:alpine sh -c "cd /tmp/python ; pip install -q black; black ."