velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Disallow unsupported guild channel types

This commit is contained in:
Luna 2020-05-02 01:28:34 -03:00
parent 6cf971ac1b
commit aac0fd10ea
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
litecord/storage.py
View File

@ -405,10 +405,14 @@ class Storage:
return str_(last_msg) return str_(last_msg)
async def _channels_extra(self, row) -> Dict: async def _channels_extra(self, row) -> Dict:
"""Fill in more information about a channel.""" """Fill in more information about a channel.
channel_type = row["type"]
Only works with guild channels, as they have
base data and extra data.
"""
channel_type = row["type"]
chan_type = ChannelType(channel_type) chan_type = ChannelType(channel_type)
assert chan_type in (ChannelType.GUILD_TEXT, ChannelType.GUILD_VOICE)
if chan_type == ChannelType.GUILD_TEXT: if chan_type == ChannelType.GUILD_TEXT:
ext_row = await self.db.fetchrow( ext_row = await self.db.fetchrow(
@ -423,7 +427,6 @@ class Storage:
drow = dict(ext_row) drow = dict(ext_row)
last_msg = await self.chan_last_message_str(row["id"]) last_msg = await self.chan_last_message_str(row["id"])
drow["last_message_id"] = last_msg drow["last_message_id"] = last_msg
return {**row, **drow} return {**row, **drow}
@ -440,9 +443,6 @@ class Storage:
return {**row, **dict(vrow)} return {**row, **dict(vrow)}
log.warning("unknown channel type: {}", chan_type)
return row
async def get_chan_type(self, channel_id: int) -> int: async def get_chan_type(self, channel_id: int) -> int:
"""Get the channel type integer, given channel ID.""" """Get the channel type integer, given channel ID."""
return await self.db.fetchval( return await self.db.fetchval(