python基础16_闭包_装饰器

不了解是否其他语言也有类似 python 装饰器这样的东西。

最近才发现ECMAScript6也是有生成器函数的，也有 yield  generator

装饰器的基础知识是闭包：

# 闭包：嵌套函数，内部函数调用外部函数的变量
def outer():
    a = 1
    def inner():
        print(a)
    # __closure__ 可以用来判断闭包
    # print(inner.__closure__)  # 打印 cell at ... 则表明它是一个闭包
    return inner  # 返回内部函数的内存地址，因为此函数用到了外部函数的变量，所以外部函数的变量也不会因函数的调用结束而消失
inn = outer()
inn()

from urllib.request import urlopen
# ret = urlopen('http://www.xiaohua100.cn/index.html').read()
# print(ret)

# def get_url():
#     url = 'http://www.xiaohua100.cn/index.html'
#     ret = urlopen(url).read()
#     print(ret)
#
# get_url()

# 闭包的简单示例：
# def get_url():
#     url = 'http://www.xiaohua100.cn/index.html'
#     def get():
#         ret = urlopen(url).read()
#         print(ret)
#     return get    # 得到 url 结果
#
# get_func = get_url()   # 将结果赋给变量，保存下来。
# get_func()
#

装饰器的作用与原则

import time

def say_hello():
    time.sleep(1)
    print('hello, everyone...')

# 定义一个闭包函数来增加功能
def tmr(f):  # 装修器函数
    def inner():
        start = time.time()
        f()  # 被装饰函数
        end = time.time()
        print(end - start)

    return inner  # 传参数的名字，不加括号

fun = tmr(say_hello)  # 使用闭包函数
fun()

# 装饰器的作用 —— 不想修改函数的调用方式 但是还想在原来的函数前后添加功能
# timmer就是一个装饰器函数，只是对一个函数 有一些装饰作用

# 原则： 开放封闭原则
#   开放 ： 对扩展是开放的
#   封闭 ： 对修改是封闭的

def deco(f):
    def inner(*args, **kwargs):
        print('装饰器在函数之前')
        ret = f(*args, **kwargs)
        print("装饰器在函数之后")

    return inner

@deco
def say_hi():
    time.sleep(1)
    print('Hi, good morning ...')

say_hi()

#!/usr/bin/env python
# coding:utf-8

# 复习一下函数作为参数传递。
# def f11():
#     print(12345)
#
#
# def f12(xxx):
#     xxx()
#
#
# f12(f11)

## 装修器：本质就是函数，功能是为其他函数添加附加功能
# 原则：
# 1. 不修改被装饰函数的源代码
# 2. 不修改被修饰函数的调用方式

## 装饰器 ＝ 高阶函数 + 函数嵌套 + 闭包

def outer(func):
    def inner():
        print('log')
        ret = func()
        print('after')
        return ret
    return inner

@outer
def f1():
    print('F1')
    return '砍你哦.'

@outer
def f2():
    print('F2')

@outer
def f100():
    print('F100')

在另一个文件中调用上面定义的函数，看看装饰器是否启作用：

#!/usr/bin/env python
# coding:utf-8

import outer01

outer01.f1()
outer01.f2()
outer01.f100()

装饰器简单原理：

#!/usr/bin/env python
# coding:utf-8

def outer(func):
    def inner():
        print('before')
        func()
        print('after')
    return inner

## 格式 @ + 函数名
## 功能：
#   1. 自动执行outer函数并且将其下面的函数名f1当作参数传递。
#   2. 将outer函数的返回值，重新赋值给f1
@outer
def f1():
    print('F1')

# 一旦函数被装饰器装饰，函数就会被重新赋值成装饰器函数的内层函数
f1()

换一个文件调用函数：

#!/usr/bin/env python
# coding:utf-8

import outer01

res = outer01.f1() # 也拿到了原函数的返回值。

print(res)

带有参数的装饰器:

#!/usr/bin/env python
# coding:utf-8

import time

def timmer(func):
    def wrapper(*args, **kwargs):
        start_time = time.clock()
        res = func(*args, **kwargs)
        stop_time = time.clock()
        print('运行时间是：%s' % (stop_time - start_time))
        return res

    return wrapper

@timmer
def test(name, age):
    time.sleep(1)
    print('test执行完毕 %s %s' % (name, age))
    return 'test complete.'

def test2(name, age, gender):
    time.sleep(1.2)
    print('测试2 %s %d %s' % (name, age, gender))

res = test('alex', 22)
print(res)

test2('jerry', 88, 'male')

带参数的装饰器:

#!/usr/bin/env python
# coding:utf-8

user_list=[
    {'name':'alex','passwd':''},
    {'name':'tom','passwd':''},
    {'name':'jerry','passwd':''},
    {'name':'kang','passwd':''},
]

current_dic = {'username':None,'login':True}

def auth(auth_type='filedb'):
    def auth_func(func):
        def wapper(*args, **kwargs):
            print('认证类型是:',auth_type)
            if auth_type =='filedb':
                if current_dic['username'] and current_dic['login']:
                    res = func(*args, **kwargs)
                    return res
                username=input('请输入用户名：')
                passwd=input('请输入密码：')
                for user_dic in user_list:
                    if username ==user_dic['name'] and passwd == user_dic['passwd']:
                        current_dic['username'] = username
                        current_dic['login'] = True
                        res = func(*args, **kwargs)
                        return res
                else:
                    print('用户名或密码错误.')
            elif auth_type =='ldap':
                print('LDAP方式')
                res = func(*args, **kwargs)
                return res
            else:
                print('不知道什么方式.')
                res = func(*args, **kwargs)
                return res
        return wapper
    return auth_func

@auth()
def login():
    print("欢迎登录。" )

@auth(auth_type='ldap')
def shopping_cart(name):
    print('%s的购物车里有%s %s %s' % (name,'衣服','电池','牛奶'))

@auth(auth_type='dsdsds')
def home(name):
    print("%s 的主页" % name)

login()
shopping_cart('alex')
home('alex')

小练习: 模拟网站增加登录状态验证:

#!/usr/bin/env python
# coding:utf-8

user_dic={'username':None,'login':False}

def auth_func(func):
    def wapper(*args, **kwargs):
        if user_dic['username'] and user_dic['login']:
            res = func(*args, **kwargs)
            return res
        username=input('请输入用户名：')
        passwd=input('请输入密码：')
        if username =='user' and passwd == '123':
            user_dic['username'] = username
            user_dic['login'] = True
            res = func(*args, **kwargs)
            return res
        else:
            print('用户名或密码错误.')
    return wapper

@auth_func
def login():
    print("欢迎登录。" )

@auth_func
def shopping_cart(name):
    print('%s的购物车里有%s %s %s' % (name,'衣服','电池','牛奶'))

@auth_func
def home(name):
    print("%s 的主页" % name)

login()
shopping_cart('alex')
home('alex')

functools 模块中的 wraps

from functools import wraps

def wraper(f):
    @wraps(f)         # 此内置装饰器保证被装饰的函数的 __name__ 和 __doc__依然不变
    def inner(*args,**kwargs):
        '''
        装饰器内部函数 inner
        '''
        print('装饰在函数之前')
        ret = f(*args,**kwargs)
        print('装饰在函数之后')
        return ret
    return inner

@wraper
def holiday(day):
    '''
    这是一个放假函数
    '''
    print('公司决定放假{}天'.format(day))
    return 'very happy.'

# 获取函数名 和 函数文档
print(holiday.__name__)     # 装饰器添加之后，结果变成了 inner
print(holiday.__doc__)      # 取函数的注释文档
ret = holiday(7)
print(ret)

用装饰器为函数加上调用记录功能，记录每次调用的函数名写入文件

# 用装饰器为函数加上调用记录功能，记录每次调用的函数名写入文件
import time

def recd(func):
    def inner(*args,**kwargs):
        with open('log','a',encoding='utf8') as f:
            f.write(time.strftime('%Y-%m-%d %X',time.localtime()) + '  ')
            f.write(func.__name__+'\n')
        ret = func(*args,**kwargs)
        return ret
    return inner

@recd
def shop_add():
    print('增加了一件商品')

@recd
def shop_del():
    print('删除了某个商品')

# print(time.asctime())
# print(time.strftime('%Y-%m-%d %X',time.localtime()))

shop_add()
shop_del()

假设有2个装饰器，装饰同一个函数。

那么执行顺序是先执行第一个装饰器的前半部分，再执行第二个装饰器的前半部分，再执行被装饰的函数主体。

主体执行完成后，再执行第二个装饰器的后半部分，最后执行第一个装饰器的后半部分。

#多个装饰器装饰一个函数
def wrapper1(func):
    def inner1():
        print('wrapper1 ,before func')
        ret = func()
        print('wrapper1 ,after func')
        return ret
    return inner1

def wrapper2(func):
    def inner2():
        print('wrapper2 ,before func')
        ret = func()
        print('wrapper2 ,after func')
        return ret
    return inner2

def wrapper3(func):
    def inner3():
        print('wrapper3 ,before func')
        ret = func()
        print('wrapper3 ,after func')
        return ret
    return inner3

@wrapper3
@wrapper2
@wrapper1
def f():
    print('in f')
    return '哈哈哈'

print(f())