diff options
Diffstat (limited to 'thread_utils.py')
| -rw-r--r-- | thread_utils.py | 25 |
1 files changed, 12 insertions, 13 deletions
diff --git a/thread_utils.py b/thread_utils.py index 0130cdc..ad1f0bf 100644 --- a/thread_utils.py +++ b/thread_utils.py @@ -27,7 +27,7 @@ def is_current_thread_main_thread() -> bool: def background_thread( - _funct: Optional[Callable] + _funct: Optional[Callable], ) -> Tuple[threading.Thread, threading.Event]: """A function decorator to create a background thread. @@ -58,10 +58,11 @@ def background_thread( periodically check. If the event is set, it means the thread has been requested to terminate ASAP. """ + def wrapper(funct: Callable): @functools.wraps(funct) def inner_wrapper( - *a, **kwa + *a, **kwa ) -> Tuple[threading.Thread, threading.Event]: should_terminate = threading.Event() should_terminate.clear() @@ -72,10 +73,9 @@ def background_thread( kwargs=kwa, ) thread.start() - logger.debug( - f'Started thread {thread.name} tid={thread.ident}' - ) + logger.debug(f'Started thread {thread.name} tid={thread.ident}') return (thread, should_terminate) + return inner_wrapper if _funct is None: @@ -85,8 +85,8 @@ def background_thread( def periodically_invoke( - period_sec: float, - stop_after: Optional[int], + period_sec: float, + stop_after: Optional[int], ): """ Periodically invoke a decorated function. Stop after N invocations @@ -108,6 +108,7 @@ def periodically_invoke( print(f"Hello, {name}") """ + def decorator_repeat(func): def helper_thread(should_terminate, *args, **kwargs) -> None: if stop_after is None: @@ -130,14 +131,12 @@ def periodically_invoke( should_terminate.clear() newargs = (should_terminate, *args) thread = threading.Thread( - target=helper_thread, - args = newargs, - kwargs = kwargs + target=helper_thread, args=newargs, kwargs=kwargs ) thread.start() - logger.debug( - f'Started thread {thread.name} tid={thread.ident}' - ) + logger.debug(f'Started thread {thread.name} tid={thread.ident}') return (thread, should_terminate) + return wrapper_repeat + return decorator_repeat |