From 885a21505fdb4ef6b28951874585747eb38f967d Mon Sep 17 00:00:00 2001 From: Andrei Date: Thu, 13 Jul 2017 13:42:23 -0700 Subject: [PATCH] [channel] add support for nsfw flag --- disco/types/channel.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/disco/types/channel.py b/disco/types/channel.py index e44567c..6003c30 100644 --- a/disco/types/channel.py +++ b/disco/types/channel.py @@ -123,6 +123,7 @@ class Channel(SlottedModel, Permissible): position = Field(int) bitrate = Field(int) recipients = AutoDictField(User, 'id') + nsfw = Field(bool) type = Field(enum(ChannelType)) overwrites = AutoDictField(PermissionOverwrite, 'id', alias='permission_overwrites') @@ -187,7 +188,7 @@ class Channel(SlottedModel, Permissible): """ Whether this channel is an NSFW channel. """ - return self.type == ChannelType.GUILD_TEXT and NSFW_RE.match(self.name) + return self.type == ChannelType.GUILD_TEXT and (self.nsfw or NSFW_RE.match(self.name)) @property def is_voice(self): @@ -399,6 +400,13 @@ class Channel(SlottedModel, Permissible): """ return self.client.api.channels_modify(self.id, position=position, reason=reason) + def set_nsfw(self, value, reason=None): + """ + Sets whether the channel is NSFW. + """ + assert (self.type == ChannelType.GUILD_TEXT) + return self.client.api.channels_modify(self.id, nsfw=value, reason=reason) + def set_bitrate(self, bitrate, reason=None): """ Sets the channels bitrate.