fabiommendes/sidekick

"""
A builtins replacement. Should it be included in sidekick.api?
"""
import builtins as _builtins
from functools import wraps as _wraps

from sidekick.functions import fn

_execute_with = lambda **kwargs: lambda f: f(**kwargs) or f
_flipped = lambda f: _wraps(f)(lambda x, y: f(y, x))
_getattr = getattr
_dir = dir


# Arity-1 functions
# abs, all, any, ascii, bin, bool, callable, chr, classmethod, dict, dir, float
# frozenset, hash, help, hex, id, input, int, iter, len, list, memoryview
# next, oct, ord, repr, reversed, round, set, sorted, staticmethod, str, sum
# tuple, type, vars


@fn.curry(2)
def int_base(base, x):
    """
    Convert string x representing a number in some base to integer.
    """
    return _builtins.int(x, base)


@fn.curry(3)
def getattr_or(default, attr, obj):
    """
    Return given attribute of object or the default argument if attr is not
    given.
    """
    return _builtins.getattr(obj, attr, default)


@fn.curry(2)
@_wraps(setattr)
def setattr(value, attr, obj):
    """
    Like setattr, but with the arguments flipped.
    """
    return _builtins.setattr(obj, attr, value)


max = fn(max)
min = fn(min)
print = fn(print)


# What should we do with those functions?
def _raise(*args, **kwargs):
    raise NotImplementedError("use the builtin function for now!")


bytearray = (
    bytes
) = compile = eval = exec = pow = open = property = range = slice = super = zip = _raise


#
# Patch module to include other functions
#
@_execute_with(
    mod=_builtins,
    ns=globals(),
    arities={"complex": 2, "divmod": 2, "globals": 0, "locals": 0, "object": 0},
    flipped={
        "delattr",
        "enumerate",
        "format",
        "getattr",
        "hasattr",
        "isinstance",
        "issubclass",
    },
    blacklist={
        "breakpoint",
        "copyright",
        "credits",
        "display",
        "get_ipython",
        "license",
    },
)
def _create_fn_functions(mod, ns, arities=None, flipped=(), blacklist=()):
    arities = arities or {}
    for k in _dir(mod):
        if k.startswith("_") or k in ns or k in blacklist:
            continue
        v = _getattr(mod, k)

        if callable(v):
            if k in flipped:
                v = fn.curry(2, _flipped(v))
            else:
                v = fn.curry(arities.get(k, 1), v)
        ns[k] = v