From c693945a46b5ee3885b9b9a0f36ac67a0f767cf6 Mon Sep 17 00:00:00 2001 From: Rapptz Date: Tue, 8 Jun 2021 09:27:55 -0400 Subject: [PATCH] [types] Split PartialVoiceState for proper gateway type --- discord/types/guild.py | 4 ++-- discord/types/voice.py | 10 +++++++--- 2 files changed, 9 insertions(+), 5 deletions(-) diff --git a/discord/types/guild.py b/discord/types/guild.py index 339438fb9..2fdf8975f 100644 --- a/discord/types/guild.py +++ b/discord/types/guild.py @@ -25,7 +25,7 @@ DEALINGS IN THE SOFTWARE. from typing import List, Literal, Optional, TypedDict from .snowflake import Snowflake from .channel import GuildChannel -from .voice import PartialVoiceState +from .voice import GuildVoiceState from .welcome_screen import WelcomeScreen from .activity import PartialPresenceUpdate from .role import Role @@ -56,7 +56,7 @@ class _GuildOptional(TypedDict, total=False): joined_at: Optional[str] large: bool member_count: int - voice_states: List[PartialVoiceState] + voice_states: List[GuildVoiceState] members: List[Member] channels: List[GuildChannel] presences: List[PartialPresenceUpdate] diff --git a/discord/types/voice.py b/discord/types/voice.py index af0a92834..06bd25a9b 100644 --- a/discord/types/voice.py +++ b/discord/types/voice.py @@ -32,8 +32,7 @@ class _PartialVoiceStateOptional(TypedDict, total=False): self_stream: bool -class PartialVoiceState(_PartialVoiceStateOptional): - channel_id: Optional[Snowflake] +class _VoiceState(_PartialVoiceStateOptional): user_id: Snowflake session_id: str deaf: bool @@ -44,7 +43,12 @@ class PartialVoiceState(_PartialVoiceStateOptional): suppress: bool -class VoiceState(PartialVoiceState, total=False): +class GuildVoiceState(_VoiceState): + channel_id: Snowflake + + +class VoiceState(_VoiceState, total=False): + channel_id: Optional[Snowflake] guild_id: Snowflake