mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-11-06 03:08:55 -05:00
25 lines
738 B
Python
25 lines
738 B
Python
from redbot.core import commands
|
|
from typing import Tuple
|
|
|
|
|
|
class CogOrCommand(commands.Converter):
|
|
async def convert(self, ctx: commands.Context, arg: str) -> Tuple[str]:
|
|
ret = ctx.bot.get_cog(arg)
|
|
if ret:
|
|
return "cogs", ret.__class__.__name__
|
|
ret = ctx.bot.get_command(arg)
|
|
if ret:
|
|
return "commands", ret.qualified_name
|
|
|
|
raise commands.BadArgument()
|
|
|
|
|
|
class RuleType(commands.Converter):
|
|
async def convert(self, ctx: commands.Context, arg: str) -> str:
|
|
if arg.lower() in ("allow", "whitelist", "allowed"):
|
|
return "allow"
|
|
if arg.lower() in ("deny", "blacklist", "denied"):
|
|
return "deny"
|
|
|
|
raise commands.BadArgument()
|