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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
|
# -*- coding: utf-8 -*-
from functools import wraps
from typing import Any, Callable, Dict, Optional, Tuple, Type, TypeVar, Union
import warnings
from typing_extensions import Literal
from pyee.base import EventEmitter
UpliftingEventEmitter = TypeVar(name="UpliftingEventEmitter", bound=EventEmitter)
EMIT_WRAPPERS: Dict[EventEmitter, Callable[[], None]] = dict()
def unwrap(event_emitter: EventEmitter) -> None:
"""Unwrap an uplifted EventEmitter, returning it to its prior state."""
if event_emitter in EMIT_WRAPPERS:
EMIT_WRAPPERS[event_emitter]()
def _wrap(
left: EventEmitter,
right: EventEmitter,
error_handler: Any,
proxy_new_listener: bool,
) -> None:
left_emit = left.emit
left_unwrap: Optional[Callable[[], None]] = EMIT_WRAPPERS.get(left)
@wraps(left_emit)
def wrapped_emit(event: str, *args: Any, **kwargs: Any) -> bool:
left_handled: bool = left._call_handlers(event, args, kwargs)
# Do it for the right side
if proxy_new_listener or event != "new_listener":
right_handled = right._call_handlers(event, args, kwargs)
else:
right_handled = False
handled = left_handled or right_handled
# Use the error handling on ``error_handler`` (should either be
# ``left`` or ``right``)
if not handled:
error_handler._emit_handle_potential_error(event, args[0] if args else None)
return handled
def _unwrap() -> None:
warnings.warn(
DeprecationWarning(
"Patched ee.unwrap() is deprecated and will be removed in a "
"future release. Use pyee.uplift.unwrap instead."
)
)
unwrap(left)
def unwrap_hook() -> None:
left.emit = left_emit
if left_unwrap:
EMIT_WRAPPERS[left] = left_unwrap
else:
del EMIT_WRAPPERS[left]
del left.unwrap # type: ignore
left.emit = left_emit
unwrap(right)
left.emit = wrapped_emit
EMIT_WRAPPERS[left] = unwrap_hook
left.unwrap = _unwrap # type: ignore
_PROXY_NEW_LISTENER_SETTINGS: Dict[str, Tuple[bool, bool]] = dict(
forward=(False, True),
backward=(True, False),
both=(True, True),
neither=(False, False),
)
ErrorStrategy = Union[Literal["new"], Literal["underlying"], Literal["neither"]]
ProxyStrategy = Union[
Literal["forward"], Literal["backward"], Literal["both"], Literal["neither"]
]
def uplift(
cls: Type[UpliftingEventEmitter],
underlying: EventEmitter,
error_handling: ErrorStrategy = "new",
proxy_new_listener: ProxyStrategy = "forward",
*args: Any,
**kwargs: Any
) -> UpliftingEventEmitter:
"""A helper to create instances of an event emitter ``cls`` that inherits
event behavior from an ``underlying`` event emitter instance.
This is mostly helpful if you have a simple underlying event emitter
that you don't have direct control over, but you want to use that
event emitter in a new context - for example, you may want to ``uplift`` a
``EventEmitter`` supplied by a third party library into an
``AsyncIOEventEmitter`` so that you may register async event handlers
in your ``asyncio`` app but still be able to receive events from the
underlying event emitter and call the underlying event emitter's existing
handlers.
When called, ``uplift`` instantiates a new instance of ``cls``, passing
along any unrecognized arguments, and overwrites the ``emit`` method on
the ``underlying`` event emitter to also emit events on the new event
emitter and vice versa. In both cases, they return whether the ``emit``
method was handled by either emitter. Execution order prefers the event
emitter on which ``emit`` was called.
The ``unwrap`` function may be called on either instance; this will
unwrap both ``emit`` methods.
The ``error_handling`` flag can be configured to control what happens to
unhandled errors:
- 'new': Error handling for the new event emitter is always used and the
underlying library's non-event-based error handling is inert.
- 'underlying': Error handling on the underlying event emitter is always
used and the new event emitter can not implement non-event-based error
handling.
- 'neither': Error handling for the new event emitter is used if the
handler was registered on the new event emitter, and vice versa.
Tuning this option can be useful depending on how the underlying event
emitter does error handling. The default is 'new'.
The ``proxy_new_listener`` option can be configured to control how
``new_listener`` events are treated:
- 'forward': ``new_listener`` events are propagated from the underlying
- 'both': ``new_listener`` events are propagated as with other events.
- 'neither': ``new_listener`` events are only fired on their respective
event emitters.
event emitter to the new event emitter but not vice versa.
- 'backward': ``new_listener`` events are propagated from the new event
emitter to the underlying event emitter, but not vice versa.
Tuning this option can be useful depending on how the ``new_listener``
event is used by the underlying event emitter, if at all. The default is
'forward', since ``underlying`` may not know how to handle certain
handlers, such as asyncio coroutines.
Each event emitter tracks its own internal table of handlers.
``remove_listener``, ``remove_all_listeners`` and ``listeners`` all
work independently. This means you will have to remember which event
emitter an event handler was added to!
Note that both the new event emitter returned by ``cls`` and the
underlying event emitter should inherit from ``EventEmitter``, or at
least implement the interface for the undocumented ``_call_handlers`` and
``_emit_handle_potential_error`` methods.
"""
(
new_proxy_new_listener,
underlying_proxy_new_listener,
) = _PROXY_NEW_LISTENER_SETTINGS[proxy_new_listener]
new: UpliftingEventEmitter = cls(*args, **kwargs)
uplift_error_handlers: Dict[str, Tuple[EventEmitter, EventEmitter]] = dict(
new=(new, new), underlying=(underlying, underlying), neither=(new, underlying)
)
new_error_handler, underlying_error_handler = uplift_error_handlers[error_handling]
_wrap(new, underlying, new_error_handler, new_proxy_new_listener)
_wrap(underlying, new, underlying_error_handler, underlying_proxy_new_listener)
return new
|