overloadable 1.0.2

Create an overloaded function around a core choosing function.

Overview

Create an overloaded function around a core choosing function.

Installation

To install overloadable, you can use pip. Open your terminal and run:

pip install overloadable

Implementation

import functools
import typing

__all__ = ["overloadable"]

class Holder:...

def overloadable(old, /):
    holder = Holder()
    @functools.wraps(old)
    def new(*args, **kwargs):
        key = old(*args, **kwargs)
        value = holder._data.lookup[key]
        ans = value(*args, **kwargs)
        return ans
    holder._data = new
    new.lookup = dict()
    new.overload = functools.partial(tool, data=new)
    return new

def tool(key=None, **kwargs):
    return functools.partial(decorator, key=key, **kwargs)

def decorator(old, /, *, data, key):
    typing.overload(old)
    data.lookup[key] = old
    return data

Example

from overloadable import overloadable

class Bar:
    def __init__(self, addon) -> None:
        self.addon = addon

    @overloadable
    def foo(self, x):
        if type(x) is int:
            return "int"

    @foo.overload("int")
    def foo(self, x):
        return x * x + self.addon

    @foo.overload() # key=None
    def foo(self, x):
        return str(x)[::-1]

bar = Bar(42)
print(bar.foo(1)) # prints 43
print(bar.foo(3.14)) # prints 41.3
print(bar.foo("baz")) # prints zab

License

This project is licensed under the MIT License.

Links

• Documentation

• Download

• Source

Credits

• Author: Johannes

• Email: [e-mail]

Thank you for using overloadable!