Return None if region isn't set

This commit is contained in:
Luna 2020-04-01 20:28:58 -03:00
parent 6a74a7f990
commit b67de9a62d
1 changed files with 6 additions and 1 deletions

View File

@ -136,7 +136,10 @@ class LVSPManager:
) )
def get_health(self, hostname: str) -> float: def get_health(self, hostname: str) -> float:
"""Get voice server health, given hostname.""" """Get voice server health, given hostname.
Returns -1 if the given hostname is not connected.
"""
try: try:
conn = self.conns[hostname] conn = self.conns[hostname]
except KeyError: except KeyError:
@ -152,6 +155,8 @@ class LVSPManager:
hostname = self.assign[guild_id] hostname = self.assign[guild_id]
except KeyError: except KeyError:
region = await self.guild_region(guild_id) region = await self.guild_region(guild_id)
if region is None:
return None
# sort connected servers by health # sort connected servers by health
sorted_servers = sorted(self.servers[region], key=self.get_health) sorted_servers = sorted(self.servers[region], key=self.get_health)