34 lines
1.3 KiB
Python
34 lines
1.3 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 {author.name}#{author.discriminator} ({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(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))
|