55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
from typing import Callable
|
|
from typing import Iterable
|
|
|
|
import traceback
|
|
from threading import Thread
|
|
from uuid import uuid4
|
|
from time import sleep
|
|
|
|
import logger
|
|
|
|
|
|
def _onionr_thread(func: Callable,
|
|
sleep_secs: int, initial_sleep, *args, **kwargs):
|
|
thread_id = str(uuid4())
|
|
if initial_sleep:
|
|
sleep(initial_sleep)
|
|
while True:
|
|
try:
|
|
func(*args, **kwargs)
|
|
except Exception as _: # noqa
|
|
logger.warn(
|
|
f"Onionr thread exception in {thread_id} \n" +
|
|
traceback.format_exc(),
|
|
terminal=True)
|
|
sleep(sleep_secs)
|
|
|
|
|
|
def add_onionr_thread(
|
|
func: Callable,
|
|
sleep_secs: int, *args, initial_sleep: int = 5, **kwargs):
|
|
"""Spawn a new onionr thread that exits when the main thread does.
|
|
|
|
Runs in an infinite loop with sleep between calls
|
|
Passes in an interable args and sleep variables"""
|
|
|
|
Thread(target=_onionr_thread,
|
|
args=(
|
|
func,
|
|
sleep_secs,
|
|
initial_sleep,
|
|
*args),
|
|
kwargs=kwargs,
|
|
daemon=True).start()
|
|
|
|
|
|
def add_delayed_thread(func: Callable, sleep_secs: int, *args, **kwargs):
|
|
assert sleep_secs > 0
|
|
def _do_delay_thread():
|
|
t = Thread(target=func, args=args, kwargs=kwargs, daemon=True)
|
|
sleep(sleep_secs)
|
|
t.start()
|
|
t = Thread(target=_do_delay_thread, daemon=True)
|
|
t.start()
|
|
return t
|