def log(text): if isinstance(text, str): def decorator(func): @functools.wraps(func) def wrapper(args, **kw): print('%s %s():' % (text, func.name)) return func(args, kw) return wrapper return decorator else: @functools.wraps(text) def wrapper(*args, kw): print('%s():' % text.name) return text(args, *kw) return wrapper