1
|
|
|
# Copyright Pincer 2021-Present |
|
|
|
|
2
|
|
|
# Full MIT License can be found in `LICENSE` at the project root. |
3
|
|
|
|
4
|
|
|
from __future__ import annotations |
5
|
|
|
|
6
|
|
|
from collections import ChainMap |
7
|
|
|
|
8
|
|
|
from .. import client as _client |
9
|
|
|
from .chat_command_handler import ChatCommandHandler |
10
|
|
|
from .components.component_handler import ComponentHandler |
11
|
|
|
from ..objects.app.command import AppCommand, InteractableStructure |
12
|
|
|
|
13
|
|
|
|
14
|
|
|
INTERACTION_REGISTERS = ChainMap(ChatCommandHandler.register, ComponentHandler.register) |
15
|
|
|
|
16
|
|
|
|
17
|
|
|
class Interactable: |
18
|
|
|
""" |
19
|
|
|
Class that can register :class:`~pincer.commands.interactable.PartialInteractable` |
20
|
|
|
objects. Any class that subclasses this class can register Application Commands and |
21
|
|
|
Message Components. |
22
|
|
|
PartialInteractable objects are registered by running the register function and |
23
|
|
|
setting an attribute of the client to the result. |
24
|
|
|
""" |
25
|
|
|
|
26
|
|
|
def __init__(self): |
27
|
|
|
for value in vars(type(self)).values(): |
28
|
|
|
if isinstance(value, InteractableStructure): |
29
|
|
|
value.manager = self |
30
|
|
|
|
31
|
|
|
def __del__(self): |
32
|
|
|
self.unassign() |
33
|
|
|
|
34
|
|
|
def unassign(self): |
35
|
|
|
""" |
36
|
|
|
Removes this objects loaded commands from ChatCommandHandler and |
37
|
|
|
ComponentHandler and removes loaded events from the client. |
38
|
|
|
""" |
39
|
|
|
for value in vars(type(self)).values(): |
40
|
|
|
if isinstance(value, InteractableStructure): |
41
|
|
|
if isinstance(value.metadata, AppCommand): |
42
|
|
|
for key, _value in INTERACTION_REGISTERS.items(): |
43
|
|
|
if value is _value: |
44
|
|
|
INTERACTION_REGISTERS.pop(key) |
45
|
|
|
|
46
|
|
|
key = value.call.__name__.lower() |
47
|
|
|
|
48
|
|
|
event_or_list = _client._events.get(key) |
|
|
|
|
49
|
|
|
if isinstance(event_or_list, list): |
50
|
|
|
if value in event_or_list: |
51
|
|
|
event_or_list.remove(value) |
52
|
|
|
else: |
53
|
|
|
_client._events.pop(key, None) |
|
|
|
|
54
|
|
|
|