Multimethod provides a decorator for adding multiple argument dispatching to functions. The decorator finds the multimethod of the same name, creating it if necessary, and registers the function with its annotations.
There are several multiple dispatch libraries on PyPI. This one aims for simplicity and speed. With caching of argument types, it should be the fastest pure Python implementation possible.
from multimethod import multimethod @multimethod def func(x: int, y: float): ...
func is now a multimethod which will delegate to the above function, when called with arguments of the specified types. Subsequent usage will register new types and functions to the existing multimethod of the same name. If an exact match can't be found, the next closest method is called (and cached). Candidate methods are ranked based on their subclass relationships. If no matches are found, a custom TypeError is raised.
A strict flag can also be set on the multimethod object, in which case finding multiple matches also raises a TypeError. Keyword arguments can be used when calling, but won't affect the dispatching.
The functools.singledispatch style syntax introduced in Python 3.4 is also supported. This requires creating a multidispatch object explicitly, and consequently doesn't rely on the name matching. The register method returns a decorator for given types, thereby supporting Python 2 and stacking of multiple signatures.
from multimethod import multidispatch @multidispatch def func(*args): ... @func.register(object, int) @func.register(int, object) def _(*args): ...
See tests for more example usage.
$ pip install multimethod
- Python ~=2.7, >=3.4
100% branch coverage.
$ pytest [--cov]
- Forward references allowed in type hints
- Multimethods can be defined inside a class
- Optimized dispatching
- Support for functools.singledispatch syntax
- Dispatch on Python 3 annotations