12 lines
296 B
Python
12 lines
296 B
Python
from tenacity import RetryCallState
|
|
from tenacity.stop import stop_base
|
|
|
|
from .shutdown_listener import should_exit
|
|
|
|
|
|
class stop_if_should_exit(stop_base):
|
|
"""Stop if the should_exit flag is set."""
|
|
|
|
def __call__(self, retry_state: 'RetryCallState') -> bool:
|
|
return should_exit()
|