Python中的生成器和装饰器

1 生成式

（1）列表生成式

• 生成十个随机的验证码（）四个字母组成

import random ,string
result=["".join(random.sample(string.ascii_letters,4)) for i in range(10)]
print(result)
输出：
['rtHL', 'PNTz', 'kAXt', 'XzQl', 'YbmG', 'xFOA', 'WtvJ', 'BhZk', 'lSAf', 'AOzE']

（2） 集合生成式

• 将0~6的平方输出到集合中

result={i**2 for i in range(6)}
print(result)
输出：
{0, 1, 4, 9, 16, 25}

（3）字典生成式

result={i:i**2 for i in range(6)}
print(result)
输出：
{0: 0, 1: 1, 2: 4, 3: 9, 4: 16, 5: 25}

2 生成器

（1）生成式改写

nums=(num for num in range(100) if num%2==0)
print(nums) ## <generator object <genexpr> at 0x000001EE90678740>
print(next(nums)) ## 0
print(next(nums)) ## 2

（2）yield关键字

• yield:遇到yield则停止执行代码， 当再次调用next方法时，会从上次停止的地方继续执行，遇到yield停止
• 如果函数里面有yield关键字，函数的返回值就是一个生成器

def login():
    print('step 1')
    yield 1
    print('step 2')
    yield 2
    print('step 3')
    yield 3
g=login()
print(g) ##<generator object login at 0x00000181C7598740>
print(next(g))
输出：
step 1
1
print(next(g))
输出：
step 2
2

3 闭包

1. 函数里面嵌套函数
2. 外部函数的返回值是内部函数的引用
3. 内部函数可以使用外部函数的变量

def info(name):
    def wapper():
        print(name)

    print('info')
    return wapper

result = info('tom')
result()

调试函数：

• 调用info函数，传参
• info函数内部的wapper函数没有被条用，所以不会执行
• 输出info

• 将wapper函数的引用作为info函数的返回值

• 此时result实质上就是wrapper函数

• 调用result函数实质上就是wrapper函数

• 执行wapper函数，打印形参name的值

4 装饰器

1. 装饰器: 用来装饰函数的工具。
2. 功能: 在不改变源代码的情况下， 添加额外功能(eg: 计算运行时间, 记录日志，权限判断)的工具.
3. 装饰器是基于闭包实现的

（1）求两个数的和，并输出运算的时间

import time
##装饰器timer
def timer(f): ## 1 传入f=add函数
    def wapper(x, y):  ## 4 执行wapper函数 
        start_time = time.time() ## 5计算开始时间
        answer = f(x, y) ## 6 调用f=add函数，并将结果返回给answer
        end_time = time.time() ## 8 计算计算完成后的时间
        print(end_time - start_time)  ## 9 打印计算的时间
        return answer  ## 将add函数运算的结过返回给result

    return wapper ## 2 将wapper函数的引用作为timer的返回值

@timer ## 魔法糖
def add(x, y):
    return x + y ## 7将计算结果返回给result

result = add(3, 4) ## 3 result实质上就是wrapper函数，调用wapper函数并将结果返回给result
print(result) ## dayinadd的结果

（2） 装饰器的万能模板:

def 装饰器名称(f):
    @wraps(f)  # 保留被装饰函数的属性信息和帮助文档
    def wrapper(*args, **kwargs):
        # 执行函数之前做的事情
        result = f(*args, **kwargs)
        # 执行函数之后做的事情
        return  result
    return  wrapper

• 计算爬虫下载一张图片所用的时间（含参数的装饰器）

import time
from functools import wraps
def timer(f):
    @wraps(f)
    def wapper(*args, **kwargs):
        start_time = time.time()
        answer = f(*args,**kwargs)
        end_time = time.time()
        print(end_time - start_time)
        return answer

    return wapper

@timer
def crawl():
    import requests
    url='http://www.petsnet.cn/wp-content/uploads/2017/01/e76bc1db5286ebad2b8a5e3489fe1586.jpg'
    content=requests.get(url).content
    with open('img.jpg','wb') as f:
        f.write(content)
        print("下载成功")

crawl() ##调用此函数相当于调用wapper函数

（3）多装饰器

• 规则: 执行装饰器内容是从上到下， 被装饰的顺序是从下到上。

from functools import  wraps
def is_login(f):
    # @wraps(f)
    def wrapper1(*args, **kwargs):
        print('is_login, 用户是否登录')
        result = f(*args, **kwargs)
        return  result
    return  wrapper1

def is_permission(f):
    # @wraps(f)
    def wrapper2(*args, **kwargs):
        print('is_permission, 用户是否有权限')
        result = f(*args, **kwargs)
        return  result
    return  wrapper2
@is_login           # show_hosts=is_login(wrapper2)   show_hosts=wrapper1
@is_permission      # show_hosts = is_permission(show_hosts) show_hosts=wrapper2
def show_hosts():
    print("显示所有的主机")
show_hosts()
输出：
is_login, 用户是否登录
is_permission, 用户是否有权限
显示所有的主机