diff options
Diffstat (limited to 'discord/ext/commands/bot.py')
| -rw-r--r-- | discord/ext/commands/bot.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/discord/ext/commands/bot.py b/discord/ext/commands/bot.py index 64dd8062..de5821e5 100644 --- a/discord/ext/commands/bot.py +++ b/discord/ext/commands/bot.py @@ -140,6 +140,7 @@ class BotBase(GroupMixin): self._after_invoke = None self.description = inspect.cleandoc(description) if description else '' self.pm_help = pm_help + self.owner_id = options.get('owner_id') self.command_not_found = options.pop('command_not_found', 'No command called "{}" found.') self.command_has_no_subcommands = options.pop('command_has_no_subcommands', 'Command {0.name} has no subcommands.') @@ -275,6 +276,26 @@ class BotBase(GroupMixin): return (yield from discord.utils.async_all(f(ctx) for f in self._checks)) + @asyncio.coroutine + def is_owner(self, user): + """Checks if a :class:`User` or :class:`Member` is the owner of + this bot. + + If an :attr:`owner_id` is not set, it is fetched automatically + through the use of :meth:`application_info`. + + Parameters + ----------- + user: :class:`abc.User` + The user to check for. + """ + + if self.owner_id is None: + app = yield from self.application_info() + self.owner_id = owner_id = app.owner.id + return user.id == owner_id + return user.id == self.owner_id + def before_invoke(self, coro): """A decorator that registers a coroutine as a pre-invoke hook. @@ -815,6 +836,10 @@ class Bot(BotBase, discord.Client): subcommand but the command does not have any subcommands. Defaults to ``"Command {0.name} has no subcommands."``. The first format argument is the :class:`Command` attempted to get a subcommand and the second is the name. + owner_id: Optional[int] + The ID that owns the bot. If this is not set and is then queried via + :meth:`is_owner` then it is fetched automatically using + :meth:`application_info`. """ pass |