Python | 类型检查

最新推荐文章于 2024-09-22 17:25:23 发布

yimtcode

最新推荐文章于 2024-09-22 17:25:23 发布

阅读量2.5k

点赞数 1

分类专栏： Python 文章标签： python 类型检查类型检验

本文链接：https://blog.csdn.net/yimtcode/article/details/123958818

版权

Python 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

本文介绍了Python中使用类型检查进行参数验证的重要性，并提供了一个装饰器实现参数验证的示例。通过`type_check`装饰器，可以确保函数接收正确类型的参数，避免因类型错误导致的问题。示例中展示了如何对普通参数和对象类型进行检查，并在`main.py`中应用了这些检查。通常，我们只在对外暴露的方法上进行类型检查，以减少性能开销。

摘要由CSDN通过智能技术生成

Python | 类型检查

1. 简介

为什么需要参数类型检测

def add(a: int, b: int) -> int:
    return a + b


if __name__ == '__main__':
    print(add(1, 2))  # Output: 3
    print(add('Name: ', 'yimt'))  # Output: Name: yimt

2. 示例

2.1. 代码结构

hello-python
├── main.py
└── type_check.py

2.2. 代码

type_check.py

from typing import get_type_hints
from functools import wraps
from inspect import getfullargspec


def validate_input(obj, **kwargs):
    hints = get_type_hints(obj)
    for param_name, param_type in hints.items():
        if param_name == 'return':  # 返回值与传入参数无关，所以不检查
            continue

        if not isinstance(kwargs[param_name], param_type) and not issubclass(type(kwargs[param_name]), param_type):
            # 类型不匹配 and 没有继承关系
            raise TypeError(f'参数: {param_name} 应该是: {param_type}或子类')


def type_check(decorator):
    @wraps(decorator)
    def wrapped_decorator(*args, **kwargs):
        func_args = getfullargspec(decorator)[0]  # 提取传入参数
        kwargs.update(dict(zip(func_args, args)))  # 将数据更新到kwargs中
        validate_input(decorator, **kwargs)  # 验证参数
        return decorator(**kwargs)  # 调用被装饰方法

    return wrapped_decorator

main.py

from type_check import type_check


# 普通参数类型检查演示
@type_check
def add(a: int, b: int) -> int:
    return a + b


class A:
    def __init__(self, value: int):
        self.value = value


class B(A):
    def __init__(self, value: int):
        super().__init__(value)


# 对象类型检查演示
@type_check
def add_obj(a: A, b: A) -> int:
    return a.value + b.value


if __name__ == '__main__':
    print(add(1, 2))  # Output: 3
    print(add_obj(A(1), B(2)))  # Output: 3
    print(add_obj(A(1), 2))  # Output: TypeError: 参数: b 应该是: <class '__main__.A'>或子类