From 1222bce271cf736b4db8c1eecb2823edd22f85dc Mon Sep 17 00:00:00 2001 From: Chris <toadawes12@gmail.com> Date: Sun, 27 Jan 2019 10:44:11 -0500 Subject: [PATCH] Add CategoryChannel.text_channels and voice_channels --- discord/channel.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/discord/channel.py b/discord/channel.py index 6a630388f..045d16243 100644 --- a/discord/channel.py +++ b/discord/channel.py @@ -624,6 +624,24 @@ class CategoryChannel(discord.abc.GuildChannel, Hashable): ret.sort(key=comparator) return ret + @property + def text_channels(self): + """List[:class:`TextChannel`]: Returns the text channels that are under this category.""" + ret = [c for c in self.guild.channels + if c.category_id == self.id + and isinstance(c, TextChannel)] + ret.sort(key=lambda c: (c.position, c.id)) + return ret + + @property + def voice_channels(self): + """List[:class:`VoiceChannel`]: Returns the text channels that are under this category.""" + ret = [c for c in self.guild.channels + if c.category_id == self.id + and isinstance(c, VoiceChannel)] + ret.sort(key=lambda c: (c.position, c.id)) + return ret + class DMChannel(discord.abc.Messageable, Hashable): """Represents a Discord direct message channel.