thread.py

from typing import Any, List, Callable, Tuple

from PyQt5 import QtCore  # type: ignore


class QtThread(QtCore.QThread):
    """
    Wrapper around QThread

    Parameters
    ----------

    fun: Callable[..., Any]
          Function that is called when ran

    finished: QtCore.pyqtSignal
          Signal emitted when callback is finished to return value to main thread

    *args: List[Any]
          Arguments that are passed into callback
    """

    args: Tuple[Any, ...]

    def __init__(self, fun, finished: QtCore.pyqtSignal, *args):
        super(QtThread, self).__init__(None)
        self.fun = fun
        self._finished = finished
        self.args = args

    def run(self) -> None:
        """
        Execute function and emit result
        """
        result: Any = self.fun(*self.args)
        self._finished.emit(result)

    def __del__(self):
        self.wait()