[commands] Allow passing of a message to NoPrivateMessage again.

Prevents an accidental breaking change.
This commit is contained in:
Rapptz 2019-04-20 17:28:44 -04:00
parent 188bd4e708
commit 5a7b5cd14b
2 changed files with 5 additions and 4 deletions

View File

@ -1525,7 +1525,7 @@ def dm_only():
def predicate(ctx):
if ctx.guild is not None:
raise PrivateMessageOnly('This command can only be used in private messages.')
raise PrivateMessageOnly()
return True
return check(predicate)

View File

@ -159,7 +159,8 @@ class PrivateMessageOnly(CheckFailure):
This inherits from :exc:`CheckFailure`
"""
pass
def __init__(self, message=None):
super().__init__(message or 'This command can only be used in private messages.')
class NoPrivateMessage(CheckFailure):
"""Exception raised when an operation does not work in private message
@ -168,8 +169,8 @@ class NoPrivateMessage(CheckFailure):
This inherits from :exc:`CheckFailure`
"""
def __init__(self):
super().__init__('This command cannot be used in private messages.')
def __init__(self, message=None):
super().__init__(message or 'This command cannot be used in private messages.')
class NotOwner(CheckFailure):
"""Exception raised when the message author is not the owner of the bot.