Replace isMe with isOwner to allow other people to debug their own bot

pull/49/head
Stijn De Clercq 2021-04-14 18:07:09 +02:00
parent e3c0d2b444
commit 7118a80d5d
2 changed files with 9 additions and 8 deletions

View File

@ -89,7 +89,7 @@ class Train(commands.Cog):
return "".join(arr) return "".join(arr)
async def sendEmbed(self, ctx, embed): async def sendEmbed(self, ctx, embed):
if checks.allowedChannels(ctx): if await checks.allowedChannels(ctx):
await ctx.send(embed=embed) await ctx.send(embed=embed)
else: else:
await ctx.author.send(embed=embed) await ctx.author.send(embed=embed)

View File

@ -8,22 +8,23 @@ import requests
from functions.database import currency from functions.database import currency
# Checks if caller of a command is me # Checks if caller of a command is the owner of the bot
def isMe(ctx): async def isMe(ctx):
return str(ctx.author.id) == constants.myId # return str(ctx.author.id) == constants.myId
return await ctx.bot.is_owner(ctx.author)
# Checks if the caller of a command is an admin # Checks if the caller of a command is an admin
def isMod(ctx): async def isMod(ctx):
if ctx.guild is None: if ctx.guild is None:
return isMe(ctx) return await isMe(ctx)
return ctx.author.id in constants.mods[ctx.guild.id] return ctx.author.id in constants.mods[ctx.guild.id]
# Checks if a command is allowed to be used in this channel # Checks if a command is allowed to be used in this channel
def allowedChannels(ctx): async def allowedChannels(ctx):
return isMe(ctx) or ctx.channel.type == discord.ChannelType.private or int(ctx.channel.id) in constants.allowedChannels.values() return (await isMe(ctx)) or ctx.channel.type == discord.ChannelType.private or int(ctx.channel.id) in constants.allowedChannels.values()
# TODO find a better way to check for legit links because reddit posts return a 502 # TODO find a better way to check for legit links because reddit posts return a 502