#!/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())