Python类参数类型注解未生效：如何强制Transaction.concept为str

Python的类型注解仅作为静态提示，运行时不会自动校验参数类型，所以你传入int类型的1不会被拦截。以下是几种可行的解决方案：

1. 手动在__init__中添加类型检查 #

直接在初始化方法里对concept参数做类型校验，不符合则抛出TypeError：

class Transaction:
    def __init__(self, date, value, concept: str = ""):
        self.date = date
        self.value = value
        # 强制校验concept类型
        if not isinstance(concept, str):
            raise TypeError(f"concept必须是str类型，实际传入{type(concept).__name__}")
        self.concept = concept

if __name__ == "__main__":
    # 正常情况：传入str
    trans1 = Transaction(date=1, value=1, concept="餐饮支出")
    print(type(trans1.concept))  # <class 'str'>
    
    # 错误情况：传入int，会抛出TypeError
    try:
        trans2 = Transaction(date=1, value=1, concept=1)
    except TypeError as e:
        print(e)  # concept必须是str类型，实际传入int

2. 使用dataclasses结合__post_init__校验（Python 3.7+） #

如果用dataclasses定义类，可以在__post_init__方法中完成类型校验：

from dataclasses import dataclass

@dataclass
class Transaction:
    date: any
    value: any
    concept: str = ""

    def __post_init__(self):
        if not isinstance(self.concept, str):
            raise TypeError(f"concept必须是str类型，实际传入{type(self.concept).__name__}")

if __name__ == "__main__":
    # 错误示例
    try:
        trans = Transaction(date=1, value=1, concept=1)
    except TypeError as e:
        print(e)

3. 使用pydantic实现自动类型校验 #

pydantic是专门用于数据校验的库，会自动根据类型注解校验参数，不符合时抛出清晰的错误：

首先安装pydantic：

pip install pydantic

然后定义类：

from pydantic import BaseModel

class Transaction(BaseModel):
    date: any
    value: any
    concept: str = ""

if __name__ == "__main__":
    # 错误示例：传入int会触发校验错误
    try:
        trans = Transaction(date=1, value=1, concept=1)
    except Exception as e:
        print(e)  # 1 is not a valid string

内容的提问来源于stack exchange，提问作者Abel Gutiérrez