From 6733665dacf961e46804b6a555b8ca5935b571f5 Mon Sep 17 00:00:00 2001 From: Sander van Grieken Date: Tue, 18 Apr 2023 14:47:01 +0200 Subject: [PATCH] qml: simplify path_protect decorator --- electrum/gui/qml/auth.py | 33 +++++++++++++++------------------ 1 file changed, 15 insertions(+), 18 deletions(-) diff --git a/electrum/gui/qml/auth.py b/electrum/gui/qml/auth.py index 49c68542a..f9896efcf 100644 --- a/electrum/gui/qml/auth.py +++ b/electrum/gui/qml/auth.py @@ -4,24 +4,21 @@ from PyQt5.QtCore import pyqtSignal, pyqtSlot, pyqtProperty from electrum.logging import get_logger -def auth_protect(message='', method='pin', reject=None): - - def decorator(func=None): - if func is None: - return partial(auth_protect, reject=reject, method=method) - - @wraps(func) - def wrapper(self, *args, **kwargs): - _logger = get_logger(__name__) - _logger.debug(f'{str(self)}.{func.__name__}') - if hasattr(self, '__auth_fcall'): - _logger.debug('object already has a pending authed function call') - raise Exception('object already has a pending authed function call') - setattr(self, '__auth_fcall', (func,args,kwargs,reject)) - getattr(self, 'authRequired').emit(method, message) - - return wrapper - return decorator +def auth_protect(func=None, reject=None, method='pin', message=''): + if func is None: + return partial(auth_protect, reject=reject, method=method, message=message) + + @wraps(func) + def wrapper(self, *args, **kwargs): + _logger = get_logger(__name__) + _logger.debug(f'{str(self)}.{func.__name__}') + if hasattr(self, '__auth_fcall'): + _logger.debug('object already has a pending authed function call') + raise Exception('object already has a pending authed function call') + setattr(self, '__auth_fcall', (func, args, kwargs, reject)) + getattr(self, 'authRequired').emit(method, message) + + return wrapper class AuthMixin: _auth_logger = get_logger(__name__)