python3 装饰器

Posted 2020-10-06

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了python3 装饰器相关的知识，希望对你有一定的参考价值。

如果想在程序中加入恶意代码,装饰器是不错的选择!

一来装饰器函数一般在主程序外定义,不易被发现;

二来装饰器函数在主程序运行之前执行,可以在主程序运行之前,获得系统分配的资源.

装饰器的语法规则很有趣,可以在不改变原有函数代码的前提下进行扩展,实现程序的友好迭代升级

其实装饰器的功能不只是更改函数,如果你愿意,装饰器还能"智能"的选择函数

在网站的后台,有这样一种需求:需要把前端页面的请求分类处理:把形似(1345.html, 4567.html)等以数字命名的网页逻辑交由一个函数处理;把形似(index.html, center.html)等以字母命名的网页交由另一个函数处理;

直白的讲,就是对不同页面的请求进行分类处理.

对数据进行分类匹配,正则表达式是首选的方式!

我们先定义两个负责处理不同逻辑的函数,并附带"正则表达式"匹配规则的装饰器

# 匹配"字符名称网页"的函数
@route(r"[a-z]+\.html")
def char(file_name, url=None):
    print("收到了‘名称为字符‘的网页")
    print("接收的文件名为",file_name)
    print("匹配的正则url表达式为:",url)
    print("~"*20)

#匹配"数字名称网页"的函数
@route(r"[\d]+\.html")
def num(file_name, url=None):
    print("收到了‘名称为数字‘的网页")
    print("接收的文件名为",file_name)
    print("匹配的正则url表达式为:",url)
    print("~"*20)

然后定义装饰器函数

# route用于匹配函数式
def route(url):
    # set_func用于接收函数的引用
    def set_func(func):
        # 将正则表达式与函数绑定
        route_dic[url] = func
        print("--为route_dic--添加了一个键值对")
        # call_func用于创建新的函数,并将新函数返回,替换被修饰的旧函数
        def call_func(file_name, url):
            return func(file_name, url)
        return call_func
    return set_func

基础的装饰器定义为两层函数嵌套,外层负责接受原函数(被添加装饰器的函数)的引用,内层负责扩展原函数功能,最后外层函数将内层函数返回,返回的内层函数将替代原函数(被添加装饰器的函数)

我们这里的装饰器函数在基础装饰器的层面上又多套了一层负责判定的函数,这样我们的装饰器就有了"分类功能"

测试代码:

import re

# 用于存储"函数的引用" 与 "url正则表达式"的键值对
route_dic = dict()

# route用于匹配函数式
def route(url):
    # set_func用于接收函数的引用
    def set_func(func):
        # 预加载,将正则表达式与函数绑定
        route_dic[url] = func
        print("--为route_dic--添加了一个键值对")
        # call_func用于创建新的函数,并将新函数返回,替换被修饰的旧函数
        def call_func(file_name, url):
            return func(file_name, url)
        return call_func
    return set_func
        
@route(r"[a-z]+\.html")
def char(file_name, url=None):
    print("收到了‘名称为字符‘的网页")
    print("接收的文件名为",file_name)
    print("匹配的正则url表达式为:",url)
    print("~"*20)

@route(r"[\d]+\.html")
def num(file_name, url=None):
    print("收到了‘名称为数字‘的网页")
    print("接收的文件名为",file_name)
    print("匹配的正则url表达式为:",url)
    print("~"*20)

def match_func(file_name):
    for url, call_func in route_dic.items():
        ret = re.match(url, file_name)
        if ret:
            print("找到了页面")
            call_func(file_name,url)
        else:
            pass

def main():
    print("route字典的值为", route_dic)
    # 测试"名称为字符"的网页
    match_func("index.html")
    # 测试"名称为数字"的网页
    match_func("123.html")
if __name__ == "__main__":
    main()