I tried implementing the overload operator in pure python which works in much similar fashion using this following script:

Python Script

from inspect import getfullargspec, getcallargs

global_map = {}

class Dummy:
    def __init__(self, name):
        self.func_name = name

    def __call__(self, *args, **kwargs):
        func_map_list = global_map.get(self.func_name, False)
        if not func_map_list:
            raise Exception("Function not defined")
        for item in func_map_list:
            func, key = item
            try:
                ann_dict = getcallargs(func, *args, **kwargs)
            except:
                continue
            f = True
            for k, v in ann_dict.items():
                if not key.annotations.get(k, False):
                    f = False
                    break
                else:
                    if type(v) != key.annotations.get(k):
                        f = False
                        break
            if f:
                return func(*args, **kwargs)
        raise Exception("Function not found with matching signature")


def overload(f):
    dummy = Dummy(f.__name__)
    f_list = global_map.get(f.__name__, [])
    f_list.append((f, getfullargspec(f)))
    global_map[f.__name__] = f_list
    return dummy

@overload
def foo(a:int, b:int):
    return a*b

@overload
def foo(a:int):
    return a**2

print(foo(2), foo(2, 10))

It seems to be working fine, though it might require a bit of more testing.

Figure out how to handle generic functions #141

Description

Metadata

Metadata

Assignees

Labels

Type

Projects

Milestone

Relationships

Development

Issue actions