1
0
Fork 0
mirror of https://github.com/uhIgnacio/EmoteManager.git synced 2024-08-15 02:23:13 +00:00
EmoteManager/cogs/meta.py

60 lines
1.7 KiB
Python

#!/usr/bin/env python3
# encoding: utf-8
import contextlib
import discord
from discord.ext import commands
class Meta(commands.Cog):
# TODO does this need to be configurable?
INVITE_DURATION_SECONDS = 60 * 60 * 3
MAX_INVITE_USES = 5
def __init__(self, bot):
self.bot = bot
@commands.command(aliases=['inv'])
async def invite(self, context):
"""Gives you a link to add me to your server."""
permissions = discord.Permissions()
permissions.update(**dict.fromkeys((
'read_messages',
'send_messages',
'add_reactions',
'external_emojis',
'manage_emojis',
'embed_links',
), True))
await context.send('<%s>' % discord.utils.oauth_url(self.bot.user.id, permissions))
@commands.command()
async def support(self, context):
"""Directs you to the support server."""
ch = self.bot.get_channel(self.bot.config['support_server_invite_channel'])
if ch is None:
await context.send('This command is temporarily unavailable. Try again later?')
return
invite = await ch.create_invite(max_age=self.INVITE_DURATION_SECONDS, max_uses=self.MAX_INVITE_USES)
try:
await context.author.send(f'Official support server invite: {invite}')
except discord.Forbidden:
with contextlib.suppress(discord.HTTPException):
await context.message.add_reaction(utils.SUCCESS_EMOJIS[True])
with contextlib.suppress(discord.HTTPException):
await context.send('Unable to send invite in DMs. Please allow DMs from server members.')
else:
try:
await context.message.add_reaction('📬')
except discord.HTTPException:
with contextlib.suppress(discord.HTTPException):
await context.send('📬')
def setup(bot):
bot.add_cog(Meta(bot))
if not bot.config.get('support_server_invite_channel'):
bot.remove_command('support')