Python中如何为泛型序列参数实现单一子类型约束的类型标注

Absolutely! You can enforce a homogeneous sequence of a single BaseModel subclass using Python's generic type hints, paired with strict mode in your type checker (like pyright or mypy). Here's how to get this working:

Solution Code #

For Python 3.12+ (PEP 695 Generic Syntax) #

from collections.abc import Sequence
from pydantic import BaseModel, ConfigDict

class ExampleModel(BaseModel):
    id: int
    name: str
    model_config = ConfigDict(validate_default=True)

class NewModel(BaseModel):
    id: int
    name: str

# Generic function with BaseModel constraint
def func[T: BaseModel](b: Sequence[T]) -> T:
    return b[0]

# Valid inputs (single subclass sequences)
l1 = [ExampleModel(id=1, name="Test1"), ExampleModel(id=2, name="Test2")]
l2 = [NewModel(id=3, name="NewTest1"), NewModel(id=4, name="NewTest2")]

# Invalid input (mixed subclasses - triggers type checker error)
l3 = [ExampleModel(id=5, name="Test3"), NewModel(id=6, name="NewTest3")]

print(func(l1))
print(func(l2))
print(func(l3))  # Type checker will flag this line

For Python 3.11 and Earlier (Traditional TypeVar) #

from typing import TypeVar
from collections.abc import Sequence
from pydantic import BaseModel, ConfigDict

# Define a TypeVar bound to BaseModel
T = TypeVar("T", bound=BaseModel)

class ExampleModel(BaseModel):
    id: int
    name: str
    model_config = ConfigDict(validate_default=True)

class NewModel(BaseModel):
    id: int
    name: str

def func(b: Sequence[T]) -> T:
    return b[0]

# Rest of the code identical to the 3.12+ example

How It Works #

The generic type constraint (T: BaseModel for PEP 695, or bound=BaseModel for TypeVar) tells the type checker that T must be a single, concrete subtype of BaseModel.

• For homogeneous lists like l1 or l2, the checker infers T as ExampleModel or NewModel respectively, which fits the constraint perfectly.
• For the mixed list l3, the checker would normally infer T as ExampleModel | NewModel (a union type). However, with strict mode enabled, type checkers reject this because the union doesn’t represent one specific BaseModel subclass—exactly what you want to avoid.

Enable Strict Mode in Your Type Checker #

To enforce this behavior, configure your tool:

• Pyright: Add strict = true to your pyproject.toml or pyrightconfig.json.
• Mypy: Run mypy with the --strict flag, or add strict = true to your mypy.ini or pyproject.toml.

Why Overloads Aren’t a Good Fit #

Your original overload approach isn’t scalable because it requires manual updates for every new BaseModel subclass you create. The generic solution works for any subclass of BaseModel without additional code, which aligns with your requirement for broad generic compatibility.

内容的提问来源于stack exchange，提问作者Frostyfeet909