From 03041bdb0aabbefc820c4702452205698b6c420e Mon Sep 17 00:00:00 2001 From: Rapptz Date: Tue, 12 Apr 2016 13:10:43 -0400 Subject: [PATCH] Add Channel.overwrites_for function to get allow/deny pairs. --- discord/channel.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) diff --git a/discord/channel.py b/discord/channel.py index b63a27c8a..8353b1ba9 100644 --- a/discord/channel.py +++ b/discord/channel.py @@ -29,8 +29,11 @@ from .permissions import Permissions from .enums import ChannelType from collections import namedtuple from .mixins import Hashable +from .role import Role +from .member import Member Overwrites = namedtuple('Overwrites', 'id allow deny type') +PermissionOverwrite = namedtuple('PermissionOverwrite', 'allow deny') class Channel(Hashable): """Represents a Discord server channel. @@ -151,6 +154,34 @@ class Channel(Hashable): """Returns the channel's creation time in UTC.""" return utils.snowflake_time(self.id) + def overwrites_for(self, obj): + """Returns a namedtuple that gives you the channel-specific overwrites + for a member or a role. + + The named tuple is a tuple of (allow, deny) :class:`Permissions` + with the appropriately named entries. + + Parameters + ----------- + obj + The :class:`Role` or :class:`Member` or :class:`Object` denoting + whose overwrite to get. + """ + + if isinstance(obj, Member): + predicate = lambda p: p.type == 'member' + elif isinstance(obj, Role): + predicate = lambda p: p.type == 'role' + else: + predicate = lambda p: True + + for overwrite in filter(predicate, self._permission_overwrites): + if overwrite.id == obj.id: + return PermissionOverwrite(allow=Permissions(overwrite.allow), + deny=Permissions(overwrite.deny)) + + return PermissionOverwrite(allow=Permissions.none(), deny=Permissions.none()) + def permissions_for(self, member): """Handles permission resolution for the current :class:`Member`.