python curry

curry.py

# -*- coding: utf-8 -*-
#author: Michał Nowotnik 2015
#public domain

def curry(func):
    argc = func.func_code.co_argcount
    def wrap(*args, **kwargs):
        if argc <= len(args)+len(kwargs):
            return func(*args, **kwargs)
        fncp = partial(wrap, *args, **kwargs)
        return fncp
    return wrap

def lazy_curry(func):
    argc = func.func_code.co_argcount
    def wrap(*args, **kwargs):
        if argc <= len(args)+len(kwargs):
            return partial(func,*args, **kwargs)
        fncp = partial(wrap, *args, **kwargs)
        return fncp
    return wrap

def partial(fnc, *args, **kwargs):
    def wrap(*argsp, **kwargsp):
        return fnc(*(args+argsp), **dict(kwargs,**kwargsp))
    return wrap


#Usage
'''
>>>@lazy_curry
... def f(a,b,c):
      return a+b+c

>>>f(1)(2)(3)()
6

>>>@curry
... def f(a,b,c):
      return a+b+c

>>>f(1)(2)(3)
6
'''