Skip to content

Commit

Permalink
Revert enabled argument for transient function wrapper as not really …
Browse files Browse the repository at this point in the history
…useful for that use case.
  • Loading branch information
GrahamDumpleton committed Jul 25, 2023
1 parent 922d5b2 commit 26b1a95
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/wrapt/wrappers.py
Original file line number Diff line number Diff line change
Expand Up @@ -902,7 +902,7 @@ def _wrapper(wrapper):
return wrap_object(module, name, FunctionWrapper, (wrapper, enabled))
return _wrapper

def transient_function_wrapper(module, name, enabled=None):
def transient_function_wrapper(module, name):
def _decorator(wrapper):
def _wrapper(wrapped, instance, args, kwargs):
target_wrapped = args[0]
Expand All @@ -914,14 +914,14 @@ def _wrapper(wrapped, instance, args, kwargs):
target_wrapper = wrapper.__get__(instance, type(instance))
def _execute(wrapped, instance, args, kwargs):
(parent, attribute, original) = resolve_path(module, name)
replacement = FunctionWrapper(original, target_wrapper, enabled)
replacement = FunctionWrapper(original, target_wrapper)
setattr(parent, attribute, replacement)
try:
return wrapped(*args, **kwargs)
finally:
setattr(parent, attribute, original)
return FunctionWrapper(target_wrapped, _execute, enabled)
return FunctionWrapper(wrapper, _wrapper, enabled)
return FunctionWrapper(target_wrapped, _execute)
return FunctionWrapper(wrapper, _wrapper)
return _decorator

# A weak function proxy. This will work on instance methods, class
Expand Down

0 comments on commit 26b1a95

Please sign in to comment.