如何在Django模型子类中扩展父类get_info方法的返回字段

这其实很简单，核心思路就是先复用父类get_info方法生成的基础字典，再往里面添加子类特有的字段就行，我给你一步步演示具体实现：

1. 定义抽象父类 #

首先是你的抽象父类模型，包含公共字段和基础的get_info方法：

from django.db import models

class BaseInfoModel(models.Model):
    # 父类的公共字段示例
    created_at = models.DateTimeField(auto_now_add=True)
    
    def get_info(self):
        # 返回基础信息字典
        return {
            'created_at': self.created_at.strftime('%Y-%m-%d %H:%M:%S'),
            'model_type': self._meta.model_name
        }
    
    class Meta:
        abstract = True

2. 子类重写get_info方法 #

子类继承父类后，重写get_info方法时，先调用父类的方法拿到原有字典，再通过update方法或者字典解包的方式添加新字段：

方式一：使用update方法（直观易读） #

class UserModel(BaseInfoModel):
    username = models.CharField(max_length=50)
    email = models.EmailField()
    
    def get_info(self):
        # 调用父类方法获取基础信息字典
        base_info = super().get_info()
        # 添加子类特有的字段到字典中
        base_info.update({
            'username': self.username,
            'email': self.email
        })
        # 返回更新后的完整字典
        return base_info

方式二：字典解包（简洁写法） #

如果子类字段不多，也可以用字典解包的方式直接合并父类字典和子类字段：

class UserModel(BaseInfoModel):
    username = models.CharField(max_length=50)
    email = models.EmailField()
    
    def get_info(self):
        return {
            **super().get_info(),
            'username': self.username,
            'email': self.email
        }

3. 注意事项 #

• 务必使用super().get_info()调用父类方法，这样能确保父类后续新增的字段或逻辑会自动被子类继承，不用重复修改代码。
• 如果父类的get_info方法需要接收参数，记得在super调用时传递对应的参数，比如super().get_info(some_arg, another_arg)。
• 测试的时候，你可以实例化子类对象，调用get_info()就能看到包含父类和子类字段的完整字典了：

user = UserModel.objects.first()
print(user.get_info())
# 输出示例：{'created_at': '2024-05-20 14:30:00', 'model_type': 'usermodel', 'username': 'john_doe', 'email': 'john@example.com'}

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