bradbase / xlfunctions Goto Github PK

def trunc(number, num_digits=0):
    """Truncate a number to the specified number of digits."""

    if isinstance(number, (int, float)):
        # Simple case. We want to make sure to return an integer in this
        # case.
        if num_digits == 0:
            return math.trunc(number)
        return math.trunc(number * 10**num_digits) / 10**num_digits
    else:
        return ExcelError(
            '#VALUE!', '{} must be a number. You gave me a {}'.format(
                number, type(number)))


xlcalculator.evaluator.evaluator.TRUNC = trunc
xlfunctions.TRUNC = trunc
xlfunctions.SUPPORTED_FUNCTIONS['TRUNC'] = 'TRUNC'

class ExcelFunctions(dict):

    def register(self, func, name=None):
        if name is None:
           name = func.__name__
        self[name.upper()] = func

    def __getattr__(self, name):
        return self[name]

fn  = ExcelFunctions()

# A quick decorator
def register(name=None):
    def reg(func):
        fn.register(func, name)
    return reg

# Example use:

@register('TRUNC')
def trunc(arg):
    return int(arg)

eval(model.cells['A1'].formula.python_code, locals={'fn': fn})