Skip to content

purefunctions/pymultidispatch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pymultimethods

Provides multiple dispatch or multimethods as described in https://en.wikipedia.org/wiki/Multiple_dispatch

multimethods can be used to effectively solve the Expression Problem described in https://en.wikipedia.org/wiki/Expression_problem

It can find uses in a lot of cases. Some examples are given below, but the uses are not limited to only those

  • Functions where we process inputs differently based on isinstance checks of one or more input arguments
  • The basic case of processing differently based on the type of one instance is what is used in Object oriented programming based on the self parameter. This basic methodology is what is called Single Dispatch
  • Functions where we process inputs differently based on the type of a field.
  • Example: Different handler for different versions of payload based on a "version" field in a dict, etc

More important than those is the ability that multimethods provide of NOT having to change the dispatcher for every single type of 'new' type that we come up with.

Usage is best described with an example:

Let's say we handle a payload based on a version number and type in the dict.

payload = {
"type": "init"
"version": 3,
...
}

In the payload above we want to dispatch to a handler based on "type" and "version"

Typical two ways to do this are:

def handle(payload):
  type, version = payload["type"], payload["version"]
  if ("init", 3) == (type, version):
  return handle_init_v3(payload)
  elif ...

def handle_init_v3(payload):
...

Better that above

HANDLERS = {
("init", 3): handle_init_v3
...
}

def handle(payload):
    type, version = payload["type"], payload["version"]
    return HANDLERS[(type, version)](payload)

The above two suffer from the problem of having to change existing files when a new version or handler is added. What if we could do this?

# Register a handler function using a @multimethod decorator. Arg to the decorator
# is the dispatch key generation function. Body of the function decorated is
# the default handler to be called if a handler wasn't registered for the
# dispatch key generated by the dispatch key generation function

from pymultidispatch.multimethods import multimethod

@multimethod(lambda payload: (payload["type"], payload["version"]))
def handle_message(payload):
    # A optional default handler function to be called if a handler wasn't registered for the
    # dispatch key generated by the above dispatch key generation function
    pass

# In the same file, or any other file
@handle_message.register(("init", 3))
def _(payload):  # Can be of any name, but leave out a name and use _ as convention
    # payload handler for init v3
    ...

# In the same file or any other file
@handle_message.register(("init", 4))
def _(payload):
    # payload handler for init v4
    ...

# We can also have more 'keys' passed into register to handle cases where the handler is the same
@handle_message.register(("init", 5), ("init", 6))
def _(payload):
    # payload handler for init v5 and init v6
    ...

As can be seen from above, there is no need to change the dispatcher function or any other existing file

This implementation of multimethods is inspired by the Clojure language's implementation of the concept

Usage:

The same as mentioned above

The @multimethod(<dispatch_key_gen_fn>) decorator is used to define a multimethod dispatch function. In the example above, the dispatcher is based on the payload type and payload version. It should be noted that the assumptions is that whatever input is given to this function is the same that is given to the dispatched functions. The dispatched functions are defined on this multimethod using the <fn>.register(<key>) decorator, where <fn> is the function decorated using @multimethod() decorator and <key> is the key for which we are registering <fn> as a handler. Note: Additional keys can also be passed onto to <fn>.register() function as follows:

<fn>.register(<key1>, <key2>, <key3>)

This will register the function for all the given keys and invoke the function when the dispatch_key_gen_fn returns andy of the given keys

The dispatch_key is the result of calling <dispatch_key_gen_fn> of the @multimethod decorator

Default params: Default params can be used, but only in the default handler (i.e. the function decorated with @multimethod decorator). The other registered handler uses the default params (if any) in the default handler that was decorated with @multimethod decorator. Any default params overwritten at the time of the call still takes precedence just like regular python functions do

About

Python multimethods

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages