diff --git a/qtapputils/managers/taskmanagers.py b/qtapputils/managers/taskmanagers.py index a6dd9af..7ca3175 100644 --- a/qtapputils/managers/taskmanagers.py +++ b/qtapputils/managers/taskmanagers.py @@ -35,7 +35,11 @@ def __init__(self): def _get_method(self, task: str): # Try direct, then fallback to underscore-prefixed (for backward # compatibility with older version of qtapputils). - return getattr(self, task, getattr(self, '_' + task)) + try: + method = getattr(self, task) + except AttributeError: + method = getattr(self, '_' + task) + return method def add_task(self, task_uuid4: Any, task: str, *args, **kargs): """ diff --git a/qtapputils/managers/tests/test_taskmanagers.py b/qtapputils/managers/tests/test_taskmanagers.py index a6ad75a..f4b8911 100644 --- a/qtapputils/managers/tests/test_taskmanagers.py +++ b/qtapputils/managers/tests/test_taskmanagers.py @@ -41,7 +41,7 @@ def _set_something(index, value): DATA[index] = value worker = WorkerBase() - worker._get_something = _get_something + worker.get_something = _get_something worker._set_something = _set_something return worker