f130c3e67f
* Unneeded requirements
* Typo and While True
* Empty requirements
* audio tag
* Correct description, `audioset notify` alert, unneeded lavaplayer line
* unneeded requirements
* extra slash
* Add CPU / RAM / Disk warning
* Unneeded requirements
* Proper coglint requirements
* Default for `"Hi {}, I'm Dad"` set to enabled
* unneeded requirements
* Add list command, fix no_pm to guild_only
* Remove some comments and rename variable
* Fix to bugs
* unneeded requirements
* unneeded requirements
* remove pass_context legacy code
* More accurate instructions
* fix custom emojis
* less comments
* Unneeded requirements
* Unneeded requirements
* proper description
* Fix description wrap
* Unneeded requirements
* Unneeded requirements
* Missing await
* Info updates
* `await ctx.embed_requested`
* no requirements necessary
* markdown
* missing help
* Auto_help handles it
* Leaverset channel description
* Embeds and optional nicknames
* text clarification
* test alternative path
* Gotta upload the fixes
* more fix
* another test
* Steal skyrim
* Undo skyrim test
* Forgot await
* Use .json files, proper init to use files, move cooldown to seed
102 lines
3.0 KiB
Python
102 lines
3.0 KiB
Python
import pathlib
|
|
|
|
import aiohttp
|
|
import discord
|
|
from MyQR import myqr
|
|
from PIL import Image
|
|
from redbot.core import Config, commands
|
|
from redbot.core.bot import Red
|
|
from redbot.core.data_manager import cog_data_path
|
|
from typing import Any
|
|
|
|
Cog: Any = getattr(commands, "Cog", object)
|
|
|
|
|
|
class QRInvite(Cog):
|
|
"""
|
|
Create custom QR codes for server invites
|
|
"""
|
|
|
|
def __init__(self, bot: Red):
|
|
self.bot = bot
|
|
self.config = Config.get_conf(self, identifier=9811198108111121, force_registration=True)
|
|
default_global = {}
|
|
default_guild = {}
|
|
|
|
self.config.register_global(**default_global)
|
|
self.config.register_guild(**default_guild)
|
|
|
|
@commands.command()
|
|
async def qrinvite(
|
|
self,
|
|
ctx: commands.Context,
|
|
invite: str = None,
|
|
colorized: bool = False,
|
|
image_url: str = None,
|
|
):
|
|
"""
|
|
Create a custom QR code invite for this server
|
|
"""
|
|
if invite is None:
|
|
try:
|
|
invite = await ctx.channel.create_invite()
|
|
except discord.Forbidden:
|
|
try:
|
|
invite = await ctx.channel.invites()
|
|
invite = invite[0]
|
|
except discord.Forbidden:
|
|
await ctx.send("No permission to get an invite, please provide one")
|
|
return
|
|
invite = invite.code
|
|
|
|
if image_url is None:
|
|
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]
|
|
|
|
path: pathlib.Path = cog_data_path(self)
|
|
image_path = path / (ctx.guild.icon + "." + extension)
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get(image_url) as response:
|
|
image = await response.read()
|
|
|
|
with image_path.open("wb") as file:
|
|
file.write(image)
|
|
|
|
if extension == "webp":
|
|
new_path = convert_png(str(image_path))
|
|
else:
|
|
new_path = str(image_path)
|
|
|
|
myqr.run(
|
|
invite,
|
|
picture=new_path,
|
|
save_name=ctx.guild.icon + "_qrcode.png",
|
|
save_dir=str(cog_data_path(self)),
|
|
colorized=colorized,
|
|
)
|
|
|
|
png_path: pathlib.Path = path / (ctx.guild.icon + "_qrcode.png")
|
|
with png_path.open("rb") as png_fp:
|
|
await ctx.send(file=discord.File(png_fp.read(), "qrcode.png"))
|
|
|
|
|
|
def convert_png(path):
|
|
im = Image.open(path)
|
|
im.load()
|
|
alpha = im.split()[-1]
|
|
im = im.convert("RGB").convert("P", palette=Image.ADAPTIVE, colors=255)
|
|
mask = Image.eval(alpha, lambda a: 255 if a <= 128 else 0)
|
|
im.paste(255, mask)
|
|
new_path = path.replace(".webp", ".png")
|
|
im.save(new_path, transparency=255)
|
|
return new_path
|