Updates for 1.4.0

Author: bryanforbes

discord-stubs/enums.pyi

@@ -182,12 +182,15 @@ class UserFlags(Enum):
     partner: int
     hypesquad: int
     bug_hunter: int
+    mfa_sms: int
+    premium_promo_dismissed: int
     hypesquad_bravery: int
     hypesquad_brilliance: int
     hypesquad_balance: int
     early_supporter: int
     team_user: int
     system: int
+    has_unread_urgent_messages: int
     bug_hunter_level_2: int
     verified_bot: int
     verified_bot_developer: int

discord-stubs/errors.pyi

@@ -24,9 +24,13 @@ class InvalidArgument(ClientException): ...
 class LoginFailure(ClientException): ...
 
 class ConnectionClosed(ClientException):
-    code: int
+    code: Optional[int]
     reason: str
     shard_id: Optional[int]
     def __init__(
-        self, socket: aiohttp.ClientWebSocketResponse, *, shard_id: Optional[int]
+        self,
+        socket: aiohttp.ClientWebSocketResponse,
+        *,
+        shard_id: Optional[int],
+        code: Optional[int] = ...,
     ) -> None: ...

discord-stubs/ext/commands/cooldowns.pyi

@@ -25,6 +25,7 @@ class Cooldown:
     type: BucketType
     def __init__(self, rate: int, per: float, type: BucketType) -> None: ...
     def get_tokens(self, current: Optional[int] = ...) -> int: ...
+    def get_retry_after(self, current: Optional[float] = ...) -> float: ...
     def update_rate_limit(self, current: Optional[float] = ...) -> Optional[float]: ...
     def reset(self) -> None: ...
     def copy(self) -> Cooldown: ...

discord-stubs/ext/commands/core.pyi

@@ -117,6 +117,7 @@ class Command(_BaseCommand, Generic[_CT]):
     async def prepare(self, ctx: _CT) -> None: ...
     def is_on_cooldown(self, ctx: _CT) -> bool: ...
     def reset_cooldown(self, ctx: _CT) -> None: ...
+    def get_cooldown_retry_after(self, ctx: _CT) -> float: ...
     async def invoke(self, ctx: _CT) -> None: ...
     async def reinvoke(self, ctx: _CT, *, call_hooks: bool = ...) -> None: ...
     def error(self, coro: _C) -> _C: ...

discord-stubs/ext/commands/help.pyi

@@ -24,7 +24,7 @@ import discord
 from .cog import Cog
 from .context import Context
 from .cooldowns import Cooldown
-from .core import Command, Group
+from .core import Command, Group, _CheckPredicate
 from .errors import CommandError
 
 _T = TypeVar('_T')
@@ -45,6 +45,7 @@ class _CommandAttrs(TypedDict, total=False):
     ignore_extra: bool
     cooldown: Cooldown
     parent: Command[Context]
+    checks: List[_CheckPredicate[Any]]
 
 class _PaginatorProtocol(Protocol):
     prefix: Optional[str]
@@ -96,6 +97,8 @@ class HelpCommand(Generic[_CT]):
         command_attrs: _CommandAttrs = ...,
     ) -> None: ...
     def copy(self: _HC) -> _HC: ...
+    def add_check(self, func: _CheckPredicate[_CT]) -> None: ...
+    def remove_check(self, func: _CheckPredicate[_CT]) -> None: ...
     def get_bot_mapping(self) -> Dict[Optional[Cog[_CT]], List[Command[_CT]]]: ...
     @property
     def clean_prefix(self) -> str: ...

poetry.lock

@@ -22,7 +22,7 @@ typing-extensions = "^3.7.4"
 
 [tool.poetry.dev-dependencies]
 black = "^19.10b0"
-"discord.py" = {git = "https://github.com/Rapptz/discord.py.git", rev = "ae1ec8072f9ef26acec107824eed42fdbb897cc2"}
+"discord.py" = {git = "https://github.com/Rapptz/discord.py.git", rev = "9048914756b2e0542af015c48ae52bb468ef8ea9"}
 flake8 = "^3.8.3"
 flake8-pyi = "^20.5.0"
 isort = "^5.3.0"

pyproject.toml

@@ -222,14 +222,7 @@ discord.state.ConnectionState.store_user
 discord.state.ConnectionState.voice_clients
 discord.team.Team.__init__
 discord.team.TeamMember.__init__
-discord.template.T
 discord.template.Template.__init__
-discord.template.a
-discord.template.e
-discord.template.l
-discord.template.m
-discord.template.p
-discord.template.t
 discord.user.BaseUser.__init__
 discord.user.ClientUser.__init__
 discord.user.Profile.__doc__