🔄Refactor(voice_client.py): on_voice_state_update function to handle …

…voice state updates more efficiently.
Nat1anWasTaken · Dec 24, 2023 · cfe3bba · cfe3bba
1 parent 95eeb17
commit cfe3bba
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/lava/voice_client.py b/lava/voice_client.py
@@ -32,12 +32,12 @@ async def on_voice_state_update(self, data):
             'd': data
         }
 
-        channel = get(self.channel.guild.voice_channels, id=int(data['channel_id']))
-        self.channel = channel
-
-        await self.lavalink.voice_update_handler(lavalink_data)
-
-        if not data['channel_id']:
+        if data['channel_id']:
+            channel = get(self.channel.guild.voice_channels, id=int(data['channel_id']))
+            self.channel = channel
+            await self.lavalink.voice_update_handler(lavalink_data)
+        else:
+            await self.channel.guild.change_voice_state(channel=None)
             self.cleanup()
 
     async def connect(self, *, timeout: float, reconnect: bool, self_deaf: bool = False,