Python装饰器练习，判断函数参数类型

最新推荐文章于 2022-05-18 16:40:51 发布

sura_1988

最新推荐文章于 2022-05-18 16:40:51 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/weixin_43314056/article/details/86618575

本文介绍了一种Python装饰器的设计，用于确保函数接收的参数符合指定的类型要求。该装饰器能够灵活地应用于多种场景，如限制参数为整数、浮点数、字符串或列表等类型，有助于增强代码的健壮性和类型安全性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目要求：

 编写装饰器required_types, 条件如下:
#     1). 当装饰器为@required_types(int,float)确保函数接收到的
每一个参数都是int或者float类型;
#     2). 当装饰器为@required_types(list)确保函数接收到的每一个
          参数都是list类型;
#     3). 当装饰器为@required_types(str,int)确保函数接收到的每
一个参数都是str或者int类型;
#     4). 如果参数不满足条件， 打印 TypeError:参数必须为xxxx类
型

import functools

def required_types(*types):
    def required_int(fun):
        @functools.wraps(fun)
        def wrapper(*args,**kwargs):
            for i in args:
                if not isinstance(i,types):
                    print('TypeError',types)
                    #break
            else:
                inspect_res = fun(*args,**kwargs)
                return inspect_res
        return wrapper
    return required_int
@required_types(str,int)
def add(*args,**kwargs):
    print(args)
print(add(1,2,3,4,4,4,5,56,66,6,6,66,6,65,5,1))