|
|
@ -2,13 +2,10 @@ import pathlib
|
|
|
|
|
|
|
|
|
|
|
|
import aiohttp
|
|
|
|
import aiohttp
|
|
|
|
import discord
|
|
|
|
import discord
|
|
|
|
|
|
|
|
from MyQR import myqr
|
|
|
|
from PIL import Image
|
|
|
|
from PIL import Image
|
|
|
|
|
|
|
|
from redbot.core import Config, commands
|
|
|
|
from redbot.core import Config, checks, commands
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from redbot.core.bot import Red
|
|
|
|
from redbot.core.bot import Red
|
|
|
|
|
|
|
|
|
|
|
|
from MyQR import myqr
|
|
|
|
|
|
|
|
from redbot.core.data_manager import cog_data_path
|
|
|
|
from redbot.core.data_manager import cog_data_path
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -46,6 +43,11 @@ class QRInvite:
|
|
|
|
if image_url is None:
|
|
|
|
if image_url is None:
|
|
|
|
image_url = ctx.guild.icon_url
|
|
|
|
image_url = ctx.guild.icon_url
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if image_url == "": # Still
|
|
|
|
|
|
|
|
await ctx.send(
|
|
|
|
|
|
|
|
"Could not get an image, please provide one. *(`{}help qrinvite` for details)*".format(ctx.prefix))
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
|
|
extension = pathlib.Path(image_url).parts[-1].replace('.', '?').split('?')[1]
|
|
|
|
extension = pathlib.Path(image_url).parts[-1].replace('.', '?').split('?')[1]
|
|
|
|
|
|
|
|
|
|
|
|
path: pathlib.Path = cog_data_path(self)
|
|
|
|
path: pathlib.Path = cog_data_path(self)
|
|
|
@ -69,6 +71,7 @@ class QRInvite:
|
|
|
|
with png_path.open("rb") as png_fp:
|
|
|
|
with png_path.open("rb") as png_fp:
|
|
|
|
await ctx.send(file=discord.File(png_fp.read(), "qrcode.png"))
|
|
|
|
await ctx.send(file=discord.File(png_fp.read(), "qrcode.png"))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def convert_png(path):
|
|
|
|
def convert_png(path):
|
|
|
|
im = Image.open(path)
|
|
|
|
im = Image.open(path)
|
|
|
|
im.load()
|
|
|
|
im.load()
|
|
|
@ -76,6 +79,6 @@ def convert_png(path):
|
|
|
|
im = im.convert('RGB').convert('P', palette=Image.ADAPTIVE, colors=255)
|
|
|
|
im = im.convert('RGB').convert('P', palette=Image.ADAPTIVE, colors=255)
|
|
|
|
mask = Image.eval(alpha, lambda a: 255 if a <= 128 else 0)
|
|
|
|
mask = Image.eval(alpha, lambda a: 255 if a <= 128 else 0)
|
|
|
|
im.paste(255, mask)
|
|
|
|
im.paste(255, mask)
|
|
|
|
newPath = path.replace(".webp",".png")
|
|
|
|
new_path = path.replace(".webp", ".png")
|
|
|
|
im.save(newPath, transparency=255)
|
|
|
|
im.save(new_path, transparency=255)
|
|
|
|
return newPath
|
|
|
|
return new_path
|
|
|
|