11from typing import Callable , Union , List
22import RAT .rat_core
3- from RAT .rat_core import EventTypes , PlotEventData
3+ from RAT .rat_core import EventTypes , PlotEventData , ProgressEventData
44
55
6- def notify (event_type : EventTypes , data : Union [str , PlotEventData ]) -> None :
6+ def notify (event_type : EventTypes , data : Union [str , PlotEventData , ProgressEventData ]) -> None :
77 """Calls registered callbacks with the data when event type has
88 been triggered.
99
1010 Parameters
1111 ----------
1212 event_type : EventTypes
1313 The event type that was triggered.
14- data : str or PlotEventData
14+ data : str or PlotEventData or ProgressEventData
1515 The data sent by the event. The message event data is a string.
1616 """
1717 callbacks = __event_callbacks [event_type ]
1818 for callback in callbacks :
1919 callback (data )
2020
21- def get_event_callback (event_type : EventTypes ) -> List [Callable [[Union [str , PlotEventData ]], None ]]:
21+ def get_event_callback (event_type : EventTypes ) -> List [Callable [[Union [str , PlotEventData , ProgressEventData ]], None ]]:
2222 """Returns all callbacks registered for the given event type.
2323
2424 Parameters
@@ -28,20 +28,20 @@ def get_event_callback(event_type: EventTypes) -> List[Callable[[Union[str, Plot
2828
2929 Retuns
3030 ------
31- callback : Callable[[Union[str, PlotEventData]], None]
31+ callback : Callable[[Union[str, PlotEventData, ProgressEventData ]], None]
3232 The callback for the event type.
3333 """
3434 return list (__event_callbacks [event_type ])
3535
3636
37- def register (event_type : EventTypes , callback : Callable [[Union [str , PlotEventData ]], None ]) -> None :
37+ def register (event_type : EventTypes , callback : Callable [[Union [str , PlotEventData , ProgressEventData ]], None ]) -> None :
3838 """Registers a new callback for the event type.
3939
4040 Parameters
4141 ----------
4242 event_type : EventTypes
4343 The event type to register.
44- callback : Callable[[Union[str, PlotEventData]], None]
44+ callback : Callable[[Union[str, PlotEventData, ProgressEventData ]], None]
4545 The callback for when the event is triggered.
4646 """
4747 if not isinstance (event_type , EventTypes ):
@@ -60,4 +60,4 @@ def clear() -> None:
6060
6161
6262__event_impl = RAT .rat_core .EventBridge (notify )
63- __event_callbacks = {EventTypes .Message : set (), EventTypes .Plot : set ()}
63+ __event_callbacks = {EventTypes .Message : set (), EventTypes .Plot : set (), EventTypes . Progress : set () }
0 commit comments