background.py

import logging
import multiprocessing as mp
import queue
from datetime import datetime

from utils import LogOrigin, save_algorithm


class BackgroundProcess(mp.Process):
    def __init__(
        self,
        export_queue,
        log_queue,
        close_event,
        log_levels,
    ):
        super().__init__()
        self.export_queue = export_queue
        self.log_queue = log_queue
        self.close_event = close_event
        self.log_levels = log_levels

    def is_closed(self):
        return self.close_event.is_set()

    def run(self):
        try:
            while not self.is_closed():
                if self.export_queue is not None:
                    while not self.is_closed():
                        try:
                            name, algo = self.export_queue.get(timeout=0.01)
                        except queue.Empty:
                            break
                        else:
                            dt = datetime.now().isoformat().replace(':', '-')
                            fn = f'{dt}_{name}.esi'
                            save_algorithm(algo, fn)
                            self.log_queue.put(
                                (LogOrigin.FILE_HANDLER, logging.DEBUG, f'{name} exported to {fn}')
                            )
                            self.export_queue.task_done()

        except KeyboardInterrupt:
            return