QR invite fixes

flask-develop
bobloy 6 years ago
parent eecfb821a0
commit 50a61398e6

@ -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

Loading…
Cancel
Save