Django模型关联数据动态提取与字典化：构建通用数据导出方法

from django.db import models
from django.utils.translation import gettext_lazy as _

# 假设 VIEW_TYPE_CHOICES 和 HEAT_TYPE_CHOICES 已经定义
VIEW_TYPE_CHOICES = [('web', 'Web'), ('mobile', 'Mobile')]
HEAT_TYPE_CHOICES = [('high', 'High'), ('medium', 'Medium')]

class Post(models.Model):
    title = models.CharField(max_length=200)
    content = models.TextField()
    # 其他字段...

    def __str__(self):
        return self.title

class ViewType(models.Model):
    post = models.ForeignKey(
        Post,
        on_delete=models.CASCADE,
        related_name="view_types",
        verbose_name=_("Post"),
    )
    view = models.CharField(
        max_length=20, choices=VIEW_TYPE_CHOICES, verbose_name=_("View")
    )
    # 其他字段...

    def dump(self):
        """返回此ViewType实例的关键数据"""
        return self.view

    def __str__(self):
        return f"{self.post.title} - View: {self.view}"

class HeatType(models.Model):
    post = models.ForeignKey(
        Post,
        on_delete=models.CASCADE,
        related_name="heat_types",
        verbose_name=_("Post"),
    )
    heat = models.CharField(
        max_length=30, choices=HEAT_TYPE_CHOICES, verbose_name=_("Heat")
    )
    # 其他字段...

    def dump(self):
        """返回此HeatType实例的关键数据"""
        return self.heat

    def __str__(self):
        return f"{self.post.title} - Heat: {self.heat}"

from django.db import models
from django.db.models.fields.reverse_related import ReverseManyToOneDescriptor
from django.utils.translation import gettext_lazy as _

# ... (Post, ViewType, HeatType 模型定义，与上文相同) ...

class Post(models.Model):
    title = models.CharField(max_length=200)
    content = models.TextField()
    # 其他字段...

    def dump_related_data(self):
        """
        动态获取所有反向外键关联的数据，并将其组织成字典。
        前提是关联模型定义了 'dump' 方法。
        """
        related_data = {}

        # 遍历Post模型的所有属性
        for attr_name, attr_value in Post.__dict__.items():
            # 识别反向外键描述符
            if isinstance(attr_value, ReverseManyToOneDescriptor):
                # 获取相关联的管理器（如 post.view_types.all()）
                # attr_name 将是 related_name，例如 "view_types", "heat_types"
                manager = getattr(self, attr_name)

                # 检查管理器是否可用（避免在实例未保存或无关联时出错）
                if manager:
                    # 获取所有关联实例，并调用它们的 dump() 方法
                    # 注意：这里会触发数据库查询
                    instances = manager.all()

                    # 仅当实例定义了 dump 方法时才尝试调用
                    if instances and hasattr(instances.first(), 'dump'):
                        related_data[attr_name] = [instance.dump() for instance in instances]
                    elif instances: # 如果没有dump方法，但有实例，可以考虑返回实例列表或空列表
                        related_data[attr_name] = [str(instance) for instance in instances] # 示例：返回实例的字符串表示
                    else:
                        related_data[attr_name] = [] # 没有关联实例
        return related_data

    def __str__(self):
        return self.title

# 在 Django shell 或 views.py 中
from myapp.models import Post, ViewType, HeatType

# 创建一些测试数据
post = Post.objects.create(title="我的第一篇博客", content="这是一篇测试文章。")
ViewType.objects.create(post=post, view='web')
ViewType.objects.create(post=post, view='mobile')
HeatType.objects.create(post=post, heat='high')

# 获取关联数据
all_related_data = post.dump_related_data()
print(all_related_data)
# 预期输出类似：
# {
#     'view_types': ['web', 'mobile'],
#     'heat_types': ['high']
# }

# 如果有其他关联模型，且它们也定义了 dump() 方法，它们的数据也会被自动包含进来。