DType-Decorate
July 25, 2017 ยท View on GitHub
The DType-Decorate module defines two different decorators at the current state. These decorators can be used to constrain the attributes of the decorated function to specific data types. This can help to keep functions clean especially when they are written for a specific context. This is usually the case for scientific applications, where functionality is often more important than clean code.
The basic structure of this module was heavily inspired / extended on the basis of: https://stackoverflow.com/questions/15299878/how-to-use-python-decorators-to-check-function-arguments
Installation
You can either use `pip` to install the version from PyPI or git to install the probably more recent version from
github.
.. code-block:: bash
git clone http://github.com/mmaelicke/dtype-decorate.git
cd dtype-decorate
pip install -r requirements.txt
python setup.py install
.. code-block:: bash
pip install dtype-decorate
Usage
~~~~~
There are two decorators so far: `accept` and `enforce`. `accept` will restrict the attribute data types to the
the defined ones, while `enforce` will try to convert the given attribute to a desired data type.
Both can also be used together, where `accept` does only make sense to be used after `enforce`.
Define a function that does only accept an `int` and a `float`.
.. code-block:: python
import ddec
@ddec.accept(a=int, b=float)
def f(a, b):
pass
You can also specify more than one data type allowed. Any attribute not given in the decorator will just be
ignored.
.. code-block:: python
@ddec.accept(a=(int, float))
def f(a, be_any_type)
pass
f(5, 'mystr') # will run fine
f('mystr', 5) # will raise a TypeError
The `accept` decorator can also handle None type and callables like functions or lambda. These have to be specified
as a string.
.. code-block:: python
@ddec.accept(a='None', b=('None', 'callable'))
def f(a, b):
pass
f(None, None) # will run fine
f(None, lambda x: x) # will run fine
f(5, None) # will raise a TypeError