dispather

mydispatch.py

#!/bin/env python3
# -*- coding: utf-8 -*-


class dispatch:
    class _handler:
        def __init__(self):
            self._handleTale = {}

        def register(self, argType):
            self._handleTale.setdefault(argType, None)

            def wrapper(handleFunc):
                self._handleTale[argType] = handleFunc

            return wrapper

        def __call__(self, arg):
            argType = type(arg)
            return self._handleTale[argType](arg)

    def __call__(self, method):
        method = self._handler()
        return method


@dispatch()
def handler(arg):
    print(arg)

@handler.register(str)
def _(arg:str):
    print(arg)

@handler.register(int)
def _(arg:int):
    print(arg)

handler("emacs")