Onionr/src/onionrthreads/__init__.py

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