make sure wrap() returns the new result

This commit is contained in:
Damien Elmes 2008-12-10 16:46:49 +09:00
parent d035806723
commit 963d79069f

View file

@ -46,10 +46,10 @@ def wrap(old, new, pos="after"):
def repl(*args, **kwargs): def repl(*args, **kwargs):
if pos == "after": if pos == "after":
old(*args, **kwargs) old(*args, **kwargs)
new(*args, **kwargs) return new(*args, **kwargs)
elif pos == "before": elif pos == "before":
new(*args, **kwargs) new(*args, **kwargs)
old(*args, **kwargs) return old(*args, **kwargs)
else: else:
new(_old=old, *args, **kwargs) return new(_old=old, *args, **kwargs)
return repl return repl