Skip to content

Commit 2bbb850

Browse files
committed
Refactor: always create self.f attribute, and refactor path-decision using that invariant.
1 parent 1284d17 commit 2bbb850

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

pywisetransfer/deprecation.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ def _message(message=None, *args, **kwargs):
1616
return message
1717

1818
def __init__(self, *args, **kwargs):
19+
self.f = None
1920
if len(args) == 1 and callable(args[0]):
2021
global f
2122
f = orig = args[0]
@@ -40,9 +41,10 @@ def _emit_warning(self):
4041
warnings.warn(self.message, DeprecationWarning, stacklevel=3)
4142

4243
def __call__(self, *args, **kwargs):
43-
if len(args) == 1 and callable(args[0]):
44+
if self.f:
45+
return self.f(*args, **kwargs)
46+
else:
4447
return deprecated(args[0], message=self.message).f
45-
return self.f(*args, **kwargs)
4648

4749
def __repr__(self):
4850
return repr(self.f)

0 commit comments

Comments
 (0)