fd845ecb40
FORK MAINTAINERS: I'm so, so sorry, but this was planned since forever. If you need help integrating this, feel free to contact me.
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
from discord.ext import commands
|
|
from discord.ext.commands import Cog
|
|
from helpers.checks import check_if_collaborator
|
|
import config
|
|
import json
|
|
|
|
|
|
class Invites(Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
@commands.command()
|
|
@commands.guild_only()
|
|
@commands.check(check_if_collaborator)
|
|
async def invite(self, ctx):
|
|
welcome_channel = self.bot.get_channel(config.welcome_channel)
|
|
author = ctx.message.author
|
|
reason = f"Created by {str(author)} ({author.id})"
|
|
invite = await welcome_channel.create_invite(
|
|
max_age=0, max_uses=1, temporary=True, unique=True, reason=reason
|
|
)
|
|
|
|
with open("data/invites.json", "r") as f:
|
|
invites = json.load(f)
|
|
|
|
invites[invite.id] = {
|
|
"uses": 0,
|
|
"url": invite.url,
|
|
"max_uses": 1,
|
|
"code": invite.code,
|
|
}
|
|
|
|
with open("data/invites.json", "w") as f:
|
|
f.write(json.dumps(invites))
|
|
|
|
await ctx.message.add_reaction("🆗")
|
|
try:
|
|
await ctx.author.send(f"Created single-use invite {invite.url}")
|
|
except discord.errors.Forbidden:
|
|
await ctx.send(
|
|
f"{ctx.author.mention} I could not send you the \
|
|
invite. Send me a DM so I can reply to you."
|
|
)
|
|
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Invites(bot))
|