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