Django通用工具函数动态Model实例的类型提示问题
Django工具函数的类型提示定义
对于这个动态获取Model实例的工具函数,类型提示可以按以下方式定义:
参数model的类型
model是Django模型类本身(而非实例),因此用Type[Model]标注,其中Model是Django所有模型的基类。如果要让类型提示更精准,支持具体模型的类型推断,可以结合泛型TypeVar实现。
返回值的类型
返回的是传入模型类的实例对象,通过泛型变量关联输入的模型类,能让类型检查工具自动识别具体的模型实例类型。
完整带类型提示的代码
from django.db.models import Model from typing import Type, TypeVar, Dict # 定义泛型变量,绑定到Model的子类 T = TypeVar('T', bound=Model) def get_or_create_object(self, model: Type[T], data: Dict) -> T: try: obj, _ = model.objects.get_or_create(**data) except model.MultipleObjectsReturned: obj = model.objects.filter(**data).first() return obj # 调用示例(类型检查工具会自动识别返回值为ModelName的实例) get_or_create_object(self, ModelName, data)
补充说明
Type[T]:明确model参数必须是Django模型类(T为Model的子类),而非模型实例。-> T:指定返回值为对应模型类的实例,这样调用函数后,类型检查工具能提供该模型属性、方法的补全提示。- 如果不需要精准的泛型推断,也可以简化为:
model: Type[Model],返回值-> Model,但泛型方式的类型提示更友好。
内容的提问来源于stack exchange,提问作者Iceman
相关产品推荐
相关产品推荐

