import types


def deco(*args):
    def _deco(func):
        def __deco():
            print("before %s called [%s]." % (func.__name__, args))
            func()
            print("  after %s called [%s]." % (func.__name__, args))

        return __deco

    # 当直接使用 @deco 定义的时候第一个参数为函数
    if len(args) == 1 and type(args[0]) is types.FunctionType:
        return _deco(args[0])

    return _deco


def bfunc(func):
    print("before login() called.")
    func()
    print("  after login() called.")
    return func


@deco
def login():
    print(" login() called.")


login()
login()