#!/usr/bin/env python

# -*- coding:utf-8 -*-

# author: Changhua Gong

import time, functools


'''

实现装饰器：

1.函数即变量；

2.高阶函数；

    1）把一个函数名当做实参传给另一个函数（在不修改被装饰函数的源代码情况下为其添加功能）；

    2）返回值中，包含函数名（不修改函数的调用方试）。

3.嵌套函数（在一个函数中，用def去声明另一个函数）。

原则：

1.不能修改被装饰的函数的源代码；

2.不能修改被装饰的函数的调用方试。

补：

1.函数在调用前，只要其涉及到的变量均在调用前别解释完全（无论先后顺序）

  即可顺利执行。


'''



def timmer(func):

    # @functools.wraps(func)  #加上这个装饰后test的函数名就不是wrapper了，而是test，注意位置（http://www.liaoxuefeng.com/wiki/0014316089557264a6b348958f449949df42a6d3a2e542c000/0014318435599930270c0381a3b44db991cd6d858064ac0000）

    def wrapper(*args, **kv):

        time_b = time.time()

        return (func(*args, **kv))

        time_e = time.time()

        print("Function %s runned for %ss" % (func.__name__, (time_e - time_b)))


    return wrapper  # 注意这个return的位置



# 写在装饰的函数的前面，装饰多个函数，每个函数前面需加上@decorator

@timmer  # test = timmer(test) = wrapper, 调用时test() = wrapper()，这里说明参数的传输问题

# test(*args,**kv) = wrapper(*args,**kv)

def test():

    print("Function test begins running.")

    time.sleep(0.2)

    print("Function test ends running.")

    return "测试"


print("通过2层装饰器函数名更替为：%s" % test.__name__)

print(test())

print("-----------------------------------------")



def timmer(func):

    #@functools.wraps(func)

    def wrapper(*args, **kv):

        time_b = time.time()

        print("Begin calling.")

        rs = func(*args, **kv)  # 将函数返回值赋给rs

        print("End calling.")

        return rs  # 这里返回函数执行完后的返回值

        time_e = time.time()

        print("Function %s runned for %ss" % (func.__name__, (time_e - time_b)))


    return wrapper  # 注意这个return的位置



# 写在装饰的函数的前面，装饰多个函数，每个函数前面需加上@decorator

@timmer  # test = timmer(test) = wrapper, 调用时test() = wrapper()，这里说明参数的传输问题

# test(*args,**kv) = wrapper(*args,**kv)

def test():

    print("Function test begins running.")

    time.sleep(0.2)

    print("Function test ends running.")

    return "测试"


print("通过2层装饰器函数名更替为：%s" % test.__name__)

print(test())

print("-----------------------------------------")

'''

上述装饰器都是简单的两层。

如果装饰器本身需要传入参数，则需要3层（应用场景如：登录网页使用微信、QQ、还是邮箱）。

'''



def login(login_type):

    def timmer(func):

        #@functools.wraps(func)

        def wrapper(*args, **kv):

            time_b = time.time()

            if login_type == "QQ":

                print("Login by QQ.")

                return (func(*args, **kv))

            else:

                print("Login by Others.")

                return (func(*args, **kv))

            time_e = time.time()

            print("Function %s runned for %ss" % (func.__name__, (time_e - time_b)))

        return wrapper  # 注意这个return的位置

    return timmer # 注意这个return的位置


# 写在装饰的函数的前面，装饰多个函数，每个函数前面需加上@decorator

@login("QQ")  # test = timmer(test) = wrapper, 调用时test() = wrapper()，这里说明参数的传输问题

# test(*args,**kv) = wrapper(*args,**kv)

def test():

    print("Function test begins running.")

    time.sleep(0.2)

    print("Function test ends running.")

    return "测试"



print("通过3层装饰器函数名更替为：%s" % test.__name__)

print(test())