装饰器

合集 - python 基础(36)

1.流程控制语句2023-11-28 2.计算机基础2023-11-27 3.编程语言介绍2023-11-27 4.python语言介绍2023-11-27 5.python解释下载与安装2023-11-27 6.python知识补充+语法入门2023-11-27 7.基本数据类型2023-11-28 8.基本运算符2023-11-28 9.用户交互输入输出方法2023-11-28 10.字符编码2023-12-04 11.可变与不可变数据类型2023-12-04 12.文件操作的基本流程2023-12-05 13.文件的基本操作2023-12-05 14.异常处理2023-12-06 15.深浅拷贝2023-12-06 16.函数定义和调用2023-12-06 17.函数的调用2023-12-11 18.名称空间和作用域2023-12-11

19.装饰器2023-12-14

20.迭代器和三元运算符2023-12-14 21.模块与包2023-12-14 22.random、os、json模块2024-01-04 23.hashlib、time模块2024-01-04 24.正则2024-01-04 25.datetime、sys、logging模块2024-01-04 26.正则re模块2024-01-04 27.常用的内置函数2024-01-04 28.初识面向对象2024-01-04 29.面向对象进阶2024-01-04 30.封装、继承、多态2024-01-11 31.常用魔法方法和元类2024-01-11 32.网络编程初识2024-01-17 33.网络编程进阶2024-01-17 34.并发编程之多进程2024-01-17 35.并发编程之多线程2024-01-23 36.并发编程之协程2024-01-23

装饰器

1.装饰器

再不修改被装饰对象的源代码和调用方式的的前提下为装饰对象添加额外的功能

1.1装饰器分类

分为无参装饰器和有参装饰器
无参装饰器

# 无参装饰器模版
def outer(func):
   def inner():
      '''这里写调用 func 函数之前的逻辑'''
      res = func()
      '''这里写调用 func 函数之后的逻辑'''
      return res
    return inner

# 语法糖调用
@outer
def add():
  return 1

# 变量赋值调用
add = outer(add)
add()

# 示例
login_user_dict = {'username': 'serein', 'is_admin': True}
user_data_dict = {'serein': {'password': 521, 'role': 'admin'},
                  'formerly': {'password': 369, 'role': 'normal'}}

def check_admin(func):
  def inner():
    if login_user_dict['username'] and login_user_dict["is_admin"]:
      res = func()
      return res
    else:
      return False,"登陆失败"
    
    return inner

@check_admin
def get_money():
  return True,"取款成功"

# 第一种调用方法
# 调用check_admin方法，传入参数get_money
get_money = check_admin(get_money)
get_money()	# get_money加（）执行inner方法
print(get_money)	# 取款成功

# 第二种方法，语法糖，在get_money方法上面加@check_admin
get_money()

有参装饰器

# 有参数 装饰器模版
def wrapper_func(tag):
     def wrapper(func):
         def inner(*args, **kwargs):
             # tag : 表明当前需要处理的使用函数的逻辑
             if tag == None:
                 ...
             # 正常函数内需要的逻辑
             return func(*args, **kwargs)

         return inner

     return wrapper


 @wrapper_func(tag=None)
 def add():

# 案例1 有参数多个装饰器语法糖叠加使用
user_data = {'username': 'serein', 'password': '521'}
bank_data = {'serein': {'pay_pwd': '521', 'balance': 1000}}


# 先验证登录
def login_auth(func):
    def inner(*args, **kwargs):
        if user_data['username'] and user_data['password'] == "521":
            res = func(*args, **kwargs)
            return res
        else:
            return False, "未登陆"

    return inner

# 再验证 输入的金额 ---l 符合数字 / 余额充足
def check_balance(func):
    def inner(*args, **kwargs):
        name = input("请输入姓名：》》").strip()
        balance = input("请输入金额 :>>>> ").strip()
        pay_pwd = input("请输入取款密码：》》》").strip()
        if not balance.isdigit():
            return False, "非法输入"
        balance = int(balance)
        if balance > bank_data[name]['balance']:
            return False, "余额不足"
        if pay_pwd != bank_data[name]['pay_pwd']:
            return False, "取款密码错误"
        return func(name=name, balance=balance, pay_pwd=pay_pwd, *args, **kwargs)

    return inner


def add_balance(func):
    def inner(*args, **kwargs):
        name = input("请输入姓名：》》").strip()
        password = input("请输入登陆密码：")
        balance = input("请输入充值金额 :>>>> ").strip()
        if not balance.isdigit():
            return False, "请输入数字金额"
        return func(name=name, password=password, balance=balance, *args, **kwargs)
    return inner

# # 取款函数里面
@login_auth
@check_balance
def get_balance(*args, **kwargs):
    print(kwargs)
    name = kwargs.get('name')
    balance = kwargs.get('balance')
    bank_data[name]['balance'] -= balance
    print(bank_data)

@login_auth
@add_balance
def add_balance(*args, **kwargs):
    print(kwargs)
    name = kwargs.get('name')
    balance = kwargs.get('balance')
    balance = int(balance)
    bank_data[name]['balance'] += balance
    print(bank_data)

add_balance()


# 案例2，有参数合并装饰器，语法糖叠加使用
user_data = {'username': 'serein', 'password': '521'}
bank_data = {'serein': {'pay_pwd': '521', 'balance': 1000}}


# 合并装饰器
def Decorator(tag_func=None, tag=None):
    # tag-func = login 走登陆装饰器
    if tag_func == 'login':
        def wraper(func):
            def inner(*args, **kwargs):
                if user_data['username'] and user_data['password'] == "521":
                    return func(*args, **kwargs)
                return False, "未登陆"

            return inner
        return wraper

    elif tag_func == "balance":
        def wraper(func):
            def inner(*args, **kwargs):
                name = input("请输入用户名")
                balance = input("请输入金额")
                balance = int(balance)
                pay_pwd = ''
                if tag != "add_balance":
                    pay_pwd = input("请输入密码：")
                    if pay_pwd != bank_data[name]['pay_pwd']:
                        return False, "密码错误"
                    if balance > bank_data[name]['balance']:
                        return False, '余额不足'
                return func(name=name, balance=balance, pay_pwd=pay_pwd, *args, **kwargs)
            return inner
        return wraper


# 取钱
@Decorator(tag_func='login')
@Decorator(tag_func='balance', tag='get_balance')
def get_balance(*args, **kwargs):
    print(kwargs)
    name = kwargs.get('name')
    balance = kwargs.get('balance')
    bank_data[name]['balance'] -= balance
    print(bank_data)


# 存钱
@Decorator(tag_func='login')
@Decorator(tag_func='balance', tag='add_balance')
def add_balance(*args, **kwargs):
    print(kwargs)
    name = kwargs.get('name')
    balance = kwargs.get('balance')
    balance = int(balance)
    bank_data[name]['balance'] += balance
    print(bank_data)


print(get_balance())

1.2 伪装装饰器

1.2.1查看装饰器(help)

可以使用help(函数名)来查看函数的文档注释，本质就是查看函数的doc属性
- 但对于被装饰之后的函数，查看文档注释

import time


# 定义外层函数
def wrapper(func):
    # 定义内层函数：通过参数接收外部的值
    def inner(*args, **kwargs):
        start_time = time.time()
        res = func(*args, **kwargs)
        stop_time = time.time()
        print('run time is %s' % (stop_time - start_time))

    return inner


@wrapper
def home(name):
    '''
    home page function
    :param name: str
    :return: None
    '''
    time.sleep(5)
    print('Welcome to the home page', name)


print(help(home))
'''
打印结果：

Help on function wrapper in module __main__:

wrapper(*args, **kwargs)

None
'''

在被装饰之后home=wrapper
- 查看home.name也可以发现home的函数名确实是wrapper
- 想要保留原函数的文档和函数名属性，需要修正装饰器

import time


# 定义外层函数
def wrapper(func):
    # 定义内层函数：通过参数接收外部的值
    def inner(*args, **kwargs):
        start_time = time.time()
        res = func(*args, **kwargs)
        stop_time = time.time()
        print('run time is %s' % (stop_time - start_time))

    inner.__doc__ = func.__doc__
    inner.__name__ = func.__name__
    return inner


@wrapper
def home(name):
    '''
    home page function
    :param name: str
    :return: None
    '''
    time.sleep(5)
    print('Welcome to the home page', name)


print(help(home))
'''
打印结果：

Help on function home in module __main__:

home(*args, **kwargs)
    home page function
    :param name: str
    :return: None

None
'''

1.2.2 伪装装饰器

functools模块下提供一个装饰器wraps专门用来帮我们实现这件事

import time
from functools import wraps


# 定义外层函数
def wrapper(func):
    # 伪装内层函数
    @wraps(func)
    # 定义内层函数：通过参数接收外部的值
    def inner(*args, **kwargs):
        start_time = time.time()
        res = func(*args, **kwargs)
        stop_time = time.time()
        print('run time is %s' % (stop_time - start_time))

    return inner


@wrapper
def home(name):
    '''
    home page function
    :param name: str
    :return: None
    '''
    time.sleep(5)
    print('Welcome to the home page', name)


print(help(home))
'''
打印结果：

Help on function home in module __main__:

home(name)
    home page function
    :param name: str
    :return: None

None
'''

posted @ 2023-12-14 21:02 Formerly0^0 阅读(16) 评论(0) 编辑收藏举报

刷新页面返回顶部

登录后才能查看或发表评论，立即登录或者逛逛博客园首页

相关博文：

· 函数的调用

· 常用的内置函数

· 装饰器、语法糖、修复技术

· 装饰器、装饰器推导、装饰器语法糖

· 装饰器（笔记整理）

阅读排行：
· winform 绘制太阳，地球，月球运作规律
· 震惊！C++程序真的从main开始吗？99%的程序员都答错了
· 【硬核科普】Trae如何「偷看」你的代码？零基础破解AI编程运行原理
· 超详细：普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 上周热点回顾（3.3-3.9）

公告

昵称： Formerly0^0
园龄： 1年3个月
粉丝： 0
关注： 2

+加关注

2025年3月

日

一

二

三

四

五

六

Formerly

装饰器

装饰器

1.装饰器

1.1装饰器分类

1.2 伪装装饰器

1.2.1查看装饰器(help)

1.2.2 伪装装饰器

公告

搜索

常用链接

我的标签

合集

随笔档案

阅读排行榜