From afee2cac107c668e0defec574f61633f561abf45 Mon Sep 17 00:00:00 2001
From: andrei <b1naryth1ef@gmail.com>
Date: Mon, 16 Oct 2017 10:11:35 -0700
Subject: [PATCH] Add methods for creating sub-channels on a category

---
 disco/types/channel.py   | 28 ++++++++++++++++++++++++++++
 examples/basic_plugin.py |  4 ++--
 2 files changed, 30 insertions(+), 2 deletions(-)

diff --git a/disco/types/channel.py b/disco/types/channel.py
index 23d0b3c..f8fbc6d 100644
--- a/disco/types/channel.py
+++ b/disco/types/channel.py
@@ -441,6 +441,34 @@ class Channel(SlottedModel, Permissible):
             parent_id=to_snowflake(parent) if parent else parent,
             reason=reason)
 
+    def create_text_channel(self, *args, **kwargs):
+        """
+        Creates a sub-text-channel in this category. See `Guild.create_text_channel`
+        for arguments and more information.
+        """
+        if self.type != ChannelType.GUILD_CATEGORY:
+            raise ValueError('Cannot create a sub-channel on a non-category channel')
+
+        kwargs['parent_id'] = self.id
+        return self.guild.create_text_channel(
+            *args,
+            **kwargs
+        )
+
+    def create_voice_channel(self, *args, **kwargs):
+        """
+        Creates a sub-voice-channel in this category. See `Guild.create_voice_channel`
+        for arguments and more information.
+        """
+        if self.type != ChannelType.GUILD_CATEGORY:
+            raise ValueError('Cannot create a sub-channel on a non-category channel')
+
+        kwargs['parent_id'] = self.id
+        return self.guild.create_voice_channel(
+            *args,
+            **kwargs
+        )
+
 
 class MessageIterator(object):
     """
diff --git a/examples/basic_plugin.py b/examples/basic_plugin.py
index 4a02338..71c864b 100644
--- a/examples/basic_plugin.py
+++ b/examples/basic_plugin.py
@@ -15,8 +15,8 @@ class BasicPlugin(Plugin):
     @Plugin.command('create-some-channels')
     def on_create_some_channels(self, event):
         category = event.guild.create_category('My Category')
-        event.guild.create_text_channel('text-channel', parent_id=category.id)
-        event.guild.create_voice_channel('voice-channel', parent_id=category.id)
+        category.create_text_channel('text-channel')
+        category.create_voice_channel('voice-channel')
         event.msg.reply('Ok, created some channels')
 
     @Plugin.command('ratelimitme')