play.callback

 1"""
 2This module contains the CallbackManager class and CallbackType enum.
 3"""
 4
 5from enum import Enum
 6
 7
 8class CallbackType(Enum):
 9    REPEAT_FOREVER = 0
10    WHEN_PROGRAM_START = 1
11    PRESSED_KEYS = 2
12    RELEASED_KEYS = 3
13    WHEN_CLICKED = 4
14    WHEN_CLICK_RELEASED = 5
15    WHEN_CLICKED_SPRITE = 6
16    WHEN_TOUCHING = 7
17    WHEN_STOPPED_TOUCHING = 8
18    WHEN_TOUCHING_WALL = 9
19    WHEN_STOPPED_TOUCHING_WALL = 10
20    WHEN_CONTROLLER_BUTTON_PRESSED = 11
21    WHEN_CONTROLLER_BUTTON_RELEASED = 12
22    WHEN_CONTROLLER_AXIS_MOVED = 13
23
24
25class CallbackManager:
26    def __init__(self):
27        """
28        A class to manage callbacks.
29        """
30        self.callbacks = {}
31
32    def add_callback(
33        self, callback_type, callback, callback_discriminator=None
34    ) -> None:
35        """
36        Add a callback to the callback manager.
37        :param callback_type: The type of callback.
38        :param callback: The callback function.
39        :param callback_discriminator: The discriminator for the callback.
40        :return: None
41        """
42        if callback_type not in self.callbacks:
43            if callback_discriminator is None:
44                self.callbacks[callback_type] = []
45            else:
46                self.callbacks[callback_type] = {}
47        if callback_discriminator is None:
48            self.callbacks[callback_type].append(callback)
49        else:
50            if callback_discriminator not in self.callbacks[callback_type]:
51                self.callbacks[callback_type][callback_discriminator] = []
52            self.callbacks[callback_type][callback_discriminator].append(callback)
53
54    def get_callbacks(self, callback_type) -> dict:
55        """
56        Get the callbacks of a certain type.
57        :param callback_type: The type of callback.
58        :return: The callbacks of the specified type.
59        """
60        return self.callbacks.get(callback_type, None)
61
62    def get_callback(self, callback_type, callback_discriminator=None) -> callable:
63        """
64        Get a callback of a certain type.
65        :param callback_type: The type of callback.
66        :param callback_discriminator: The discriminator for the callback.
67        :return: The callback(s) of the specified type.
68        """
69        if callback_discriminator is None:
70            return self.callbacks.get(callback_type, None)
71        return self.callbacks.get(callback_type, {}).get(callback_discriminator, None)
72
73
74callback_manager = CallbackManager()