2525from abc import ABC
2626from asyncio import Future
2727import asyncio
28- from typing import Any , Callable , Self , Type , TypeVar
28+ from typing import Any , Callable , Self , TypeVar
2929
3030from .state import ConnectionState
3131
@@ -42,31 +42,31 @@ async def __load__(cls, data: Any, state: ConnectionState) -> Self | None:
4242
4343class EventEmitter :
4444 def __init__ (self , state : ConnectionState ) -> None :
45- self ._listeners : dict [Type [Event ], list [Callable ]] = {}
46- self ._events : dict [str , list [Type [Event ]]]
47- self ._wait_fors : dict [Type [Event ], list [Future ]] = {}
45+ self ._listeners : dict [type [Event ], list [Callable ]] = {}
46+ self ._events : dict [str , list [type [Event ]]]
47+ self ._wait_fors : dict [type [Event ], list [Future ]] = {}
4848 self ._state = state
4949
50- def add_event (self , event : Type [Event ]) -> None :
50+ def add_event (self , event : type [Event ]) -> None :
5151 try :
5252 self ._events [event .__event_name__ ].append (event )
5353 except KeyError :
5454 self ._events [event .__event_name__ ] = [event ]
5555
56- def remove_event (self , event : Type [Event ]) -> list [Type [Event ]] | None :
56+ def remove_event (self , event : type [Event ]) -> list [type [Event ]] | None :
5757 return self ._events .pop (event .__event_name__ , None )
5858
59- def add_listener (self , event : Type [Event ], listener : Callable ) -> None :
59+ def add_listener (self , event : type [Event ], listener : Callable ) -> None :
6060 try :
6161 self ._listeners [event ].append (listener )
6262 except KeyError :
6363 self .add_event (event )
6464 self ._listeners [event ] = [listener ]
6565
66- def remove_listener (self , event : Type [Event ], listener : Callable ) -> None :
66+ def remove_listener (self , event : type [Event ], listener : Callable ) -> None :
6767 self ._listeners [event ].remove (listener )
6868
69- def add_wait_for (self , event : Type [T ]) -> Future [T ]:
69+ def add_wait_for (self , event : type [T ]) -> Future [T ]:
7070 fut = Future ()
7171
7272 try :
@@ -76,7 +76,7 @@ def add_wait_for(self, event: Type[T]) -> Future[T]:
7676
7777 return fut
7878
79- def remove_wait_for (self , event : Type [Event ], fut : Future ) -> None :
79+ def remove_wait_for (self , event : type [Event ], fut : Future ) -> None :
8080 self ._wait_fors [event ].remove (fut )
8181
8282 async def emit (self , event_str : str , data : Any ) -> None :
0 commit comments