Yet simple util to make wrapper with optional arguments
Project description
Yet simple util to make wrapper with optional arguments
Installation
Master branch
pip install git+https://github.com/ferrine/biwrap
Latest release
pip install biwrap
Problem outlined
Readability and transparent implementation is important. Wrappers is an advanced topic in programming and making them readable in some cases is difficult. The particular case is parametrizable wrapper. It can either be called with or without arguments. Implementation to handle this case is often tricky and looks weird (see SO thread). This package solves the problem and provides with simple and generic solution.
API
A minimal snippet to apply it to your problem.
import biwrap
@biwrap.biwrap
def wrapper(fn, arg1='default', arg2='default'):
def wrapped(*fn_args, **fn_kwargs):
... # depends on `arg1`, `arg2`
return wrapped
@wrapper # use defaults
def func1(a, b):
...
@wrapper(arg1='non default') # change defaults
def func2(a, b):
...
Example
Let’s discuss a case when it is needed to put functions into a registry. Some functions can have alias names.
Naive Solution
def register(alias=None):
def inner(fn):
if fn.__name__ not in register.storage:
register.storage[fn.__name__] = fn
elif register.storage[fn.__name__] is not fn:
raise KeyError('{} is already in storage'.format(fn.__name__))
if alias is not None and alias not in register.storage:
register.storage[alias] = fn
elif alias is not None:
raise KeyError('{} is already in storage'.format(alias))
return fn
return inner
register.storage = {}
@register()
def f1(a):
return a
print(register.storage)
#> {'f1': <function f1 at 0x11ff519d8>}
@register(alias='fn3')
def f2(a):
return a
print(register.storage)
#> {'f1': <function f1 at 0x11ff519d8>, 'f2': <function f2 at 0x10a87d0d0>, 'fn3': <function f2 at 0x10a87d0d0>}
Analysis
The above example shows redundancy in
decorator definition has double nesting (double def)
usage requires trailing parenthesis @register() even in case we do not use optional argument
More readable code should avoid these two points and look like:
def register(fn, alias=None):
...
@register
def f1(a):
return a
@register(alias='fn3') # <- (1)
def f2(a):
return a
Naive implementation of the above API won’t work. Line marked above as (1) will fail as first argument fn is not passed. But we want the output to be the same.
Better solution
import biwrap
@biwrap.biwrap
def register(fn, alias=None):
if fn.__name__ not in register.storage:
register.storage[fn.__name__] = fn
elif register.storage[fn.__name__] is not fn:
raise KeyError('{} is already in storage'.format(fn.__name__))
if alias is not None and alias not in register.storage:
register.storage[alias] = fn
elif alias is not None:
raise KeyError('{} is already in storage'.format(alias))
return fn
register.storage = {}
@register
def f1(a):
return a
print(register.storage)
#> {'f1': <function f1 at 0x10f45a048>}
@register(alias='fn3')
def f2(a):
return a
print(register.storage)
#> {'f1': <function f1 at 0x10f45a048>, 'f2': <function f2 at 0x10f45a488>, 'fn3': <function f2 at 0x10f45a488>}
Functionality Overview
Some corner cases may exist and custom coding can create a boilerplate for each usecase (see this SO thread). This package takes the best and implements yet simple but generic solution to resolve them all(?).
Setup
Let’s take a simple wrapper as an example. It will print hi or bye depending on parametrization, default is hi.
import biwrap
@biwrap.biwrap
def hiwrap(fn, hi=True):
def new(*args, **kwargs):
if hi:
print('hi')
else:
print('bye')
return fn(*args, **kwargs)
return new
Cases
Function wrapping
Defined wrapper can be used in both ways
@hiwrap
def fn(n):
print(n)
fn(1)
#> hi
#> 1
@hiwrap(hi=False)
def fn(n):
print(n)
fn(1)
#> bye
#> 1
Bound method wrapping
biwrap also works for bound methods. As seen in SO thread this can be a problem as first positional argument is self instead of a function.
class O:
@hiwrap(hi=False)
def fn(self, n):
print(n)
O().fn(1)
#> bye
#> 1
Class methods / properties wrapping
Implementation deals with these cases as well
class O:
def __init__(self, n):
self.n = n
@classmethod
@hiwrap
def fn(cls, n):
print(n)
@property
@hiwrap(hi=False)
def num(self):
return self.n
o = O(2)
o.fn(1)
#> hi
#> 1
print(o.num)
#> bye
#> 2
Wrapper as a function
Function like call is OK too
def fn(n):
print(n)
fn = hiwrap(fn, hi=False)
fn(1)
#> bye
#> 1
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.