You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Enso-Bot/venv/Lib/site-packages/discord/permissions.py

598 lines
19 KiB
Python

4 years ago
# -*- coding: utf-8 -*-
"""
The MIT License (MIT)
Copyright (c) 2015-2016 Rapptz
4 years ago
Permission is hereby granted, free of charge, to any person obtaining a
copy of this software and associated documentation files (the "Software"),
to deal in the Software without restriction, including without limitation
the rights to use, copy, modify, merge, publish, distribute, sublicense,
and/or sell copies of the Software, and to permit persons to whom the
Software is furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
DEALINGS IN THE SOFTWARE.
"""
class Permissions:
4 years ago
"""Wraps up the Discord permission value.
Supported operations:
+-----------+------------------------------------------+
| Operation | Description |
+===========+==========================================+
| x == y | Checks if two permissions are equal. |
+-----------+------------------------------------------+
| x != y | Checks if two permissions are not equal. |
+-----------+------------------------------------------+
| x <= y | Checks if a permission is a subset |
| | of another permission. |
+-----------+------------------------------------------+
| x >= y | Checks if a permission is a superset |
| | of another permission. |
+-----------+------------------------------------------+
| x < y | Checks if a permission is a strict |
| | subset of another permission. |
+-----------+------------------------------------------+
| x > y | Checks if a permission is a strict |
| | superset of another permission. |
+-----------+------------------------------------------+
| hash(x) | Return the permission's hash. |
+-----------+------------------------------------------+
| iter(x) | Returns an iterator of (perm, value) |
| | pairs. This allows this class to be used |
| | as an iterable in e.g. set/list/dict |
| | constructions. |
+-----------+------------------------------------------+
The properties provided are two way. You can set and retrieve individual bits using the properties as if they
were regular bools. This allows you to edit permissions.
4 years ago
Attributes
-----------
value
The raw value. This value is a bit array field of a 32-bit integer
4 years ago
representing the currently available permissions. You should query
permissions via the properties rather than using this raw value.
"""
__slots__ = [ 'value' ]
4 years ago
def __init__(self, permissions=0, **kwargs):
self.value = permissions
def __eq__(self, other):
return isinstance(other, Permissions) and self.value == other.value
def __ne__(self, other):
return not self.__eq__(other)
def __hash__(self):
return hash(self.value)
def _perm_iterator(self):
for attr in dir(self):
# check if it's a property, because if so it's a permission
is_property = isinstance(getattr(self.__class__, attr), property)
if is_property:
yield (attr, getattr(self, attr))
def __iter__(self):
return self._perm_iterator()
4 years ago
def is_subset(self, other):
"""Returns True if self has the same or fewer permissions as other."""
4 years ago
if isinstance(other, Permissions):
return (self.value & other.value) == self.value
else:
raise TypeError("cannot compare {} with {}".format(self.__class__.__name__, other.__class__name))
4 years ago
def is_superset(self, other):
"""Returns True if self has the same or more permissions as other."""
4 years ago
if isinstance(other, Permissions):
return (self.value | other.value) == self.value
else:
raise TypeError("cannot compare {} with {}".format(self.__class__.__name__, other.__class__name))
4 years ago
def is_strict_subset(self, other):
"""Returns True if the permissions on other are a strict subset of those on self."""
4 years ago
return self.is_subset(other) and self != other
def is_strict_superset(self, other):
"""Returns True if the permissions on other are a strict superset of those on self."""
4 years ago
return self.is_superset(other) and self != other
__le__ = is_subset
__ge__ = is_superset
__lt__ = is_strict_subset
__gt__ = is_strict_superset
@classmethod
def none(cls):
"""A factory method that creates a :class:`Permissions` with all
permissions set to False."""
4 years ago
return cls(0)
@classmethod
def all(cls):
"""A factory method that creates a :class:`Permissions` with all
permissions set to True."""
return cls(0b01111111111101111111110011111111)
4 years ago
@classmethod
def all_channel(cls):
"""A :class:`Permissions` with all channel-specific permissions set to
True and the server-specific ones set to False. The server-specific
4 years ago
permissions are currently:
- manager_server
4 years ago
- kick_members
- ban_members
- administrator
- change_nicknames
4 years ago
- manage_nicknames
"""
return cls(0b00110011111101111111110001010001)
@classmethod
def general(cls):
"""A factory method that creates a :class:`Permissions` with all
"General" permissions from the official Discord UI set to True."""
return cls(0b01111100000000000000000010111111)
4 years ago
@classmethod
def text(cls):
"""A factory method that creates a :class:`Permissions` with all
"Text" permissions from the official Discord UI set to True."""
4 years ago
return cls(0b00000000000001111111110001000000)
@classmethod
def voice(cls):
"""A factory method that creates a :class:`Permissions` with all
"Voice" permissions from the official Discord UI set to True."""
return cls(0b00000011111100000000000000000000)
4 years ago
def update(self, **kwargs):
"""Bulk updates this permission object.
4 years ago
Allows you to set multiple attributes by using keyword
arguments. The names must be equivalent to the properties
listed. Extraneous key/value pairs will be silently ignored.
Parameters
------------
\*\*kwargs
A list of key/value pairs to bulk update permissions with.
"""
for key, value in kwargs.items():
try:
is_property = isinstance(getattr(self.__class__, key), property)
except AttributeError:
continue
if is_property:
4 years ago
setattr(self, key, value)
def _bit(self, index):
return bool((self.value >> index) & 1)
def _set(self, index, value):
if value == True:
self.value |= (1 << index)
elif value == False:
self.value &= ~(1 << index)
else:
raise TypeError('Value to set for Permissions must be a bool.')
4 years ago
def handle_overwrite(self, allow, deny):
# Basically this is what's happening here.
# We have an original bit array, e.g. 1010
# Then we have another bit array that is 'denied', e.g. 1111
# And then we have the last one which is 'allowed', e.g. 0101
# We want original OP denied to end up resulting in
# whatever is in denied to be set to 0.
# So 1010 OP 1111 -> 0000
# Then we take this value and look at the allowed values.
# And whatever is allowed is set to 1.
# So 0000 OP2 0101 -> 0101
# The OP is base & ~denied.
# The OP2 is base | allowed.
self.value = (self.value & ~deny) | allow
@property
4 years ago
def create_instant_invite(self):
"""Returns True if the user can create instant invites."""
return self._bit(0)
@create_instant_invite.setter
def create_instant_invite(self, value):
self._set(0, value)
4 years ago
@property
4 years ago
def kick_members(self):
"""Returns True if the user can kick users from the server."""
return self._bit(1)
4 years ago
@kick_members.setter
def kick_members(self, value):
self._set(1, value)
@property
4 years ago
def ban_members(self):
"""Returns True if a user can ban users from the server."""
return self._bit(2)
@ban_members.setter
def ban_members(self, value):
self._set(2, value)
4 years ago
@property
4 years ago
def administrator(self):
"""Returns True if a user is an administrator. This role overrides all other permissions.
4 years ago
This also bypasses all channel-specific overrides.
"""
return self._bit(3)
4 years ago
@administrator.setter
def administrator(self, value):
self._set(3, value)
@property
4 years ago
def manage_channels(self):
"""Returns True if a user can edit, delete, or create channels in the server.
This also corresponds to the "manage channel" channel-specific override."""
return self._bit(4)
@manage_channels.setter
def manage_channels(self, value):
self._set(4, value)
4 years ago
@property
def manage_server(self):
"""Returns True if a user can edit server properties."""
return self._bit(5)
4 years ago
@manage_server.setter
def manage_server(self, value):
self._set(5, value)
4 years ago
@property
4 years ago
def add_reactions(self):
"""Returns True if a user can add reactions to messages."""
return self._bit(6)
4 years ago
@add_reactions.setter
def add_reactions(self, value):
self._set(6, value)
4 years ago
@property
def view_audit_logs(self):
"""Returns True if a user can view the server's audit log."""
return self._bit(7)
4 years ago
@view_audit_logs.setter
def view_audit_logs(self, value):
self._set(7, value)
4 years ago
# 2 unused
4 years ago
@property
def read_messages(self):
"""Returns True if a user can read messages from all or specific text channels."""
return self._bit(10)
4 years ago
@read_messages.setter
def read_messages(self, value):
self._set(10, value)
4 years ago
@property
4 years ago
def send_messages(self):
"""Returns True if a user can send messages from all or specific text channels."""
return self._bit(11)
@send_messages.setter
def send_messages(self, value):
self._set(11, value)
4 years ago
@property
4 years ago
def send_tts_messages(self):
"""Returns True if a user can send TTS messages from all or specific text channels."""
return self._bit(12)
4 years ago
@send_tts_messages.setter
def send_tts_messages(self, value):
self._set(12, value)
4 years ago
@property
def manage_messages(self):
"""Returns True if a user can delete messages from a text channel. Note that there are currently no ways to edit other people's messages."""
return self._bit(13)
4 years ago
@manage_messages.setter
def manage_messages(self, value):
self._set(13, value)
4 years ago
@property
4 years ago
def embed_links(self):
"""Returns True if a user's messages will automatically be embedded by Discord."""
return self._bit(14)
4 years ago
@embed_links.setter
def embed_links(self, value):
self._set(14, value)
@property
4 years ago
def attach_files(self):
"""Returns True if a user can send files in their messages."""
return self._bit(15)
@attach_files.setter
def attach_files(self, value):
self._set(15, value)
4 years ago
@property
4 years ago
def read_message_history(self):
"""Returns True if a user can read a text channel's previous messages."""
return self._bit(16)
4 years ago
@read_message_history.setter
def read_message_history(self, value):
self._set(16, value)
4 years ago
@property
def mention_everyone(self):
"""Returns True if a user's @everyone will mention everyone in the text channel."""
return self._bit(17)
4 years ago
@mention_everyone.setter
def mention_everyone(self, value):
self._set(17, value)
4 years ago
@property
def external_emojis(self):
"""Returns True if a user can use emojis from other servers."""
return self._bit(18)
4 years ago
@external_emojis.setter
def external_emojis(self, value):
self._set(18, value)
4 years ago
# 1 unused
4 years ago
@property
4 years ago
def connect(self):
"""Returns True if a user can connect to a voice channel."""
return self._bit(20)
@connect.setter
def connect(self, value):
self._set(20, value)
4 years ago
@property
4 years ago
def speak(self):
"""Returns True if a user can speak in a voice channel."""
return self._bit(21)
4 years ago
@speak.setter
def speak(self, value):
self._set(21, value)
@property
4 years ago
def mute_members(self):
"""Returns True if a user can mute other users."""
return self._bit(22)
@mute_members.setter
def mute_members(self, value):
self._set(22, value)
4 years ago
@property
4 years ago
def deafen_members(self):
"""Returns True if a user can deafen other users."""
return self._bit(23)
@deafen_members.setter
def deafen_members(self, value):
self._set(23, value)
4 years ago
@property
4 years ago
def move_members(self):
"""Returns True if a user can move users between other voice channels."""
return self._bit(24)
4 years ago
@move_members.setter
def move_members(self, value):
self._set(24, value)
@property
4 years ago
def use_voice_activation(self):
"""Returns True if a user can use voice activation in voice channels."""
return self._bit(25)
@use_voice_activation.setter
def use_voice_activation(self, value):
self._set(25, value)
4 years ago
@property
4 years ago
def change_nickname(self):
"""Returns True if a user can change their nickname in the server."""
return self._bit(26)
4 years ago
@change_nickname.setter
def change_nickname(self, value):
self._set(26, value)
@property
4 years ago
def manage_nicknames(self):
"""Returns True if a user can change other user's nickname in the server."""
return self._bit(27)
@manage_nicknames.setter
def manage_nicknames(self, value):
self._set(27, value)
4 years ago
@property
4 years ago
def manage_roles(self):
"""Returns True if a user can create or edit roles less than their role's position.
4 years ago
This also corresponds to the "manage permissions" channel-specific override.
4 years ago
"""
return self._bit(28)
4 years ago
@manage_roles.setter
def manage_roles(self, value):
self._set(28, value)
4 years ago
@property
4 years ago
def manage_webhooks(self):
"""Returns True if a user can create, edit, or delete webhooks."""
return self._bit(29)
@manage_webhooks.setter
def manage_webhooks(self, value):
self._set(29, value)
4 years ago
@property
4 years ago
def manage_emojis(self):
"""Returns True if a user can create, edit, or delete emojis."""
return self._bit(30)
@manage_emojis.setter
def manage_emojis(self, value):
self._set(30, value)
4 years ago
# 1 unused
# after these 32 bits, there's 21 more unused ones technically
def augment_from_permissions(cls):
cls.VALID_NAMES = { name for name in dir(Permissions) if isinstance(getattr(Permissions, name), property) }
4 years ago
# make descriptors for all the valid names
for name in cls.VALID_NAMES:
4 years ago
# god bless Python
def getter(self, x=name):
4 years ago
return self._values.get(x)
def setter(self, value, x=name):
4 years ago
self._set(x, value)
prop = property(getter, setter)
setattr(cls, name, prop)
return cls
@augment_from_permissions
class PermissionOverwrite:
"""A type that is used to represent a channel specific permission.
4 years ago
Unlike a regular :class:`Permissions`\, the default value of a
permission is equivalent to ``None`` and not ``False``. Setting
a value to ``False`` is **explicitly** denying that permission,
while setting a value to ``True`` is **explicitly** allowing
that permission.
The values supported by this are the same as :class:`Permissions`
with the added possibility of it being set to ``None``.
Supported operations:
4 years ago
+-----------+------------------------------------------+
| Operation | Description |
+===========+==========================================+
| iter(x) | Returns an iterator of (perm, value) |
| | pairs. This allows this class to be used |
| | as an iterable in e.g. set/list/dict |
| | constructions. |
+-----------+------------------------------------------+
4 years ago
Parameters
-----------
\*\*kwargs
Set the value of permissions by their name.
"""
def __init__(self, **kwargs):
self._values = {}
for key, value in kwargs.items():
setattr(self, key, value)
def _set(self, key, value):
if value not in (True, None, False):
raise TypeError('Expected bool or NoneType, received {0.__class__.__name__}'.format(value))
self._values[key] = value
def pair(self):
"""Returns the (allow, deny) pair from this overwrite.
The value of these pairs is :class:`Permissions`.
"""
allow = Permissions.none()
deny = Permissions.none()
4 years ago
for key, value in self._values.items():
if value is True:
setattr(allow, key, True)
elif value is False:
setattr(deny, key, True)
return allow, deny
@classmethod
def from_pair(cls, allow, deny):
"""Creates an overwrite from an allow/deny pair of :class:`Permissions`."""
ret = cls()
for key, value in allow:
if value is True:
setattr(ret, key, True)
for key, value in deny:
if value is True:
setattr(ret, key, False)
return ret
def is_empty(self):
"""Checks if the permission overwrite is currently empty.
An empty permission overwrite is one that has no overwrites set
to True or False.
4 years ago
"""
return all(x is None for x in self._values.values())
def update(self, **kwargs):
"""Bulk updates this permission overwrite object.
4 years ago
Allows you to set multiple attributes by using keyword
arguments. The names must be equivalent to the properties
listed. Extraneous key/value pairs will be silently ignored.
Parameters
------------
\*\*kwargs
A list of key/value pairs to bulk update with.
"""
for key, value in kwargs.items():
if key not in self.VALID_NAMES:
continue
setattr(self, key, value)
def __iter__(self):
for key in self.VALID_NAMES:
4 years ago
yield key, self._values.get(key)