smartondev / gwbackupy / deb2aa08-a890-4101-ad8b-bbfac3177150

from __future__ import annotations

import logging

import signal

import threading

import time

from datetime import datetime

is_killed_handling: bool = False

is_killed_value: bool = False

is_killed_lock = threading.RLock()

def is_killed_reset():

    with is_killed_lock:

        is_killed_value = False

def is_killed() -> bool:

    with is_killed_lock:

        if not is_killed_handling:

            signal.signal(signal.SIGINT, is_killed_handling_func)

            signal.signal(signal.SIGTERM, is_killed_handling_func)

            is_killed_handling = True

        return is_killed_value

def is_killed_handling_func(*args):

    with is_killed_lock:

        logging.info("Handle kill signal")

        is_killed_value = True

def sleep_with_check(seconds: float, sleep_step: float = 0.1) -> bool:

    start = datetime.now().timestamp()

    while datetime.now().timestamp() - start < seconds:

        time.sleep(sleep_step)

        if is_killed():

            return False

    return True

def await_all_futures(futures: [], sleep_step: float = 0.1) -> bool:

    while not is_killed():

        if not sleep_with_check(1, sleep_step=sleep_step):

            return False

        has_not_done = False

        for f in futures:

            if not f.done():

        if has_not_done:

            break

    return not is_killed()