Currying decorator

curry.py

from functools import partial

def curry(fn, argcount=None):
    if not callable(fn):
        return partial(curry, argcount=fn)
    argcount = argcount or fn.__code__.co_argcount
    if argcount < 2:
        return fn
    def wrapper(first):
        return curry(partial(fn, first), argcount-1)
    return wrapper


@curry
def test1(a, b, c):
    return a, b, c


@curry(4)
def test2(a, *rest):
    return a, rest

if __name__ == '__main__':
    print(test1(1)(2)(3))
    print(test2(1)(2)(3)(4))