1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
from enum import IntEnum
from typing import Any, Callable, Dict, List, Tuple
from eduvpn_common.state import State, StateType
# The attribute that callback functions get
EDUVPN_CALLBACK_PROPERTY = "_eduvpn_property_callback"
def class_state_transition(state: int, state_type: StateType) -> Callable:
"""A decorator to be internally by classes to register the event
:param state: int: The state of the transition
:param state_type: StateType: The type of transition
:meta private:
"""
def wrapper(func):
"""
:param func: The function to set the internal attribute for
"""
setattr(func, EDUVPN_CALLBACK_PROPERTY, (state, state_type))
return func
return wrapper
class EventHandler(object):
"""The class that neatly handles event callbacks"""
def __init__(self):
self.handlers: Dict[Tuple[int, StateType], List[Callable]] = {}
def change_class_callbacks(self, cls: Any, add: bool = True) -> None:
"""The function that is used to change class callbacks
:param cls: Any: The class to change the callbacks for
:param add: bool: (Default value = True): Whether or not to add or remove the event. If true the event gets added
:meta private:
"""
# Loop over method names
for method_name in dir(cls):
try:
# Get the method
method = getattr(cls, method_name)
except Exception as e:
# Unable to get a value, go to the next
continue
# If it has a callback defined, add it to the events
method_value = getattr(method, EDUVPN_CALLBACK_PROPERTY, None)
if method_value:
state, state_type = method_value
if add:
self.add_event(state, state_type, method)
else:
self.remove_event(state, state_type, method)
def remove_event(self, state: State, state_type: StateType, func: Callable) -> None:
"""Removes an event
:param state: State: The state to remove the event for
:param state_type: StateType: The state type to remove the event for
:param func: Callable: The function that needs to be removed from the event
:meta private:
"""
for key, values in self.handlers.copy().items():
if key == (state, state_type):
values.remove(func)
if not values:
del self.handlers[key]
else:
self.handlers[key] = values
def add_event(self, state: State, state_type: StateType, func: Callable) -> None:
"""Adds an event
:param state: State: The state to add the event for
:param state_type: StateType: The state type to add the event for
:param func: Callable: The function that needs to be added to the event
:meta private:
"""
if (state, state_type) not in self.handlers:
self.handlers[(state, state_type)] = []
self.handlers[(state, state_type)].append(func)
def run_state(
self, state: State, other_state: State, state_type: StateType, data: str
) -> bool:
"""The function that runs the callback for a specific event
:param state: State: The state of the event
:param other_state: State: The other state of the event
:param state_type: StateType: The state type of the event
:param data: str: The data that gets passed to the function callback when the event is ran
:meta private:
"""
if (state, state_type) not in self.handlers:
return False
for func in self.handlers[(state, state_type)]:
func(other_state, data)
return True
def run(self, old_state: State, new_state: State, data: Any) -> bool:
"""Run a specific event.
It converts the data and then runs the event for all state types
:param old_state: State: The previous state for running the event
:param new_state: State: The new state for running the event
:param data: Any: The data that gets passed to the event
:param convert: bool: (Default value = True): Whether or not to convert the data further
"""
# First run leave transitions, then enter
self.run_state(old_state, new_state, StateType.LEAVE, data)
# We decide handled based on enter transitions
handled = self.run_state(new_state, old_state, StateType.ENTER, data)
return handled
|
