python全栈闯关--11-装饰器初识

Posted 2020-11-16 熊熊闯深林

1、装饰器形成

当不想修改原函数，未函数前后添加功能时，就可以使用装饰器，在函数前后增加功能。

装饰器的初始形成

import time

def timer(f):
    def inner():
        print("我是装饰器，我来装饰了！！！")
        start = time.time()
        f()
        end = time.time()
        print(start - end)
    return inner  # 返回inner由于f实现了闭包，直接调用了程序

def func():
    time.sleep(1)
    print("我是小鱼，要作作作。。。有本事来装饰我！")

# 通过inner返回，在func前后进行了函数的功能的扩展
t = timer(func)
t()  # 虽然实现了功能，但是函数的调用名修改了

# 为了不改变调用名，可以赋值给func
func = timer(func)
func()  # 通过装饰器，实现了不修改函数名的装饰

语法糖

在函数前面加，加上@函数名，就可以实现装饰器，叫做语法糖

def timer(f):
    def inner():
        print("我是装饰器，我来装饰了！！！")
        start = time.time()
        f()
        end = time.time()
        print(start - end)
    return inner  # 返回inner由于f实现了闭包，直接调用了程序

@timer
def func():
    time.sleep(1)
    print("我是小鱼，要作作作。。。有本事来装饰我！")

 

2、带参数和返回值的装饰器

import time

def timer(f):
    def inner(a,b):  # inner接收传入的参数
        print("我是装饰器，我来装饰了！！！")
        start = time.time()
        ret = f(a,b)
        end = time.time()
        print(start - end)
        return  ("我是装饰器的返回值！！！",start - end)  # 在闭包内部，实现返回值
    return inner  # 返回inner由于f实现了闭包，直接调用了程序

@timer
def func(a,b):
    time.sleep(1)
    print("我是小鱼，要作作作。。。有本事来装饰我！")
    print("a=%s,b=%s" % (a,b))

ret = func(1,2)  # 参数相当于传递给inner
print(ret)

执行顺序

 

 

 

3、传递任意参数的装饰器

通过*args和**kwargs传递任意参数

def warpper(f):
    def inner(*args, **kwargs):
        ret = f(*args, **kwargs)
        return ret
    return inner

@warpper
def func(*args, **kwargs):  #*和**打散
    print("我是被装饰的函数！")
    print(args)
    print(kwargs)

func()  # 传空参数
func(1,2,3,4,b=1, a=2)  # 传任意参数
dic1 = {"a":1, "b":2}
func(*(1,2,3,4),**dic1)  # 打散传递

4、参数位置随感

个人观察结论：*是按顺序打散，形参位置：*args中，args打散后为1 2 3 4，所以args为(1,2,3,4)

def outer(*args):
    print(args)  # (1, 2, 3, 4)
    print(*args)  # 1 2 3 4  元祖被打散
    def inner(*args):
        print(\'innner\', args)
    inner(*args)  # 打散传入inner(1,2,3,4)

outer = outer(1,2,3,4)

 

5、warps

函数的__doc__和__name__可以查看函数的说明和名字。

装饰器后，函数名和名字都变为装饰器的内部函数

def warpeer(f):
    def inner(*args, **kwargs):
        \'\'\'
        this is warpper inner
        :param args:
        :param kwargs:
        :return:
        \'\'\'
        print("this is innser")
        ret = f(*args, ** kwargs);
        return ret
    return inner

@warpeer
def func():
    print("this is func")

print(func.__name__)
print(func.__doc__)

 

 加上wraps后，可以识别到函数自己的doc和name

from functools import wraps

def warpeer(f):
    @wraps(f)
    def inner(*args, **kwargs):
        \'\'\'
        this is warpper inner
        :param args:
        :param kwargs:
        :return:
        \'\'\'
        print("this is innser")
        ret = f(*args, ** kwargs);
        return ret
    return inner

@warpeer
def func():
    \'\'\'
    this is func doc
    :return:
    \'\'\'
    print("this is func")

print(func.__name__)
print(func.__doc__)

 

 

 

6、练习

# 1.编写装饰器，为多个函数加上认证的功能（用户的账号密码来源于文件），
# 要求登录成功一次，后续的函数都无需再输入用户名和密码

uname = "小鱼"
upwd = "qwe"
LOGTYPE = False


def login(func):
    def in_login():
        global LOGTYPE
        # print(LOGTYPE)
        if LOGTYPE:
            # print(LOGTYPE)
            func()
            return
        username = input("请输入用户名：")
        userword = input("请输入密码：")
        if username.strip() == uname and userword.strip() == upwd:
            ret = func()
            LOGTYPE = True
            return ret
        else:
            print("用户名或者密码错误！")
        pass

    return in_login


@login
def shoplist_add():
    print("增加一件商品！")


@login
def shoplist_del():
    print("删除一件商品！")


shoplist_add()
shoplist_del()

 

# 1.编写下载网页内容的函数，要求功能是：用户传入一个url，函数返回下载页面的结果
# 2.为题目1编写装饰器，实现缓存网页内容的功能：
# 具体：实现下载的页面存放于文件中，如果文件内有值（文件大小不为0），就优先从文件中读取网页内容，否则，就去下载，然后存到文件中

import os
from urllib.request import urlopen as uop

def catche(fun):
    def in_c(*args, **kwargs):
        file = "catche.txt"
        if not os.path.exists(file):
            with open(file, \'wb\') as f:
                f.write("")
        if os.path.getsize(file) :
            with open(file, \'rb\') as f:
                return f.read()
        else:
            ret = fun(*args, **kwargs)
            with open(file, \'wb\') as f:
                f.write(b\'****catche***\' + ret)
            return ret
    return in_c

@catche
def get_url(url):
    ret = uop(url).read()
    return ret


print(get_url("http://www.baidu.com"))
print(get_url("http://www.baidu.com"))