Django模型动态关联检查：高效管理复杂关系

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

# 假设所有需要动态检查关联的模型都继承自一个基础模型
class BaseModel(models.Model):
    # ... 其他通用字段或方法 ...

    class Meta:
        abstract = True # 声明为抽象模型

    def has_relation(self, ignore_models=None) -> bool:
        """
        检查当前实例是否被其他模型关联。
        :param ignore_models: 一个模型列表，其中的模型将被忽略，不参与关联检查。
                              例如：[Ticket, User]
        :return: True 表示存在关联，False 表示不存在关联。
        """
        if ignore_models is None:
            ignore_models = []

        try:
            # 遍历所有指向当前模型的反向关系
            for related_obj_descriptor in self._meta.related_objects:
                # related_obj_descriptor 是 ManyToOneRel, ManyToManyRel, OneToOneRel 的实例
                # 提取关联字段的名称和关联模型的类
                # 注意：obj.identity[0].name 和 obj.identity[0].model 是内部实现细节，
                # 更通用的方式可能是使用 related_obj_descriptor.field.name 和 related_obj_descriptor.related_model
                # 但根据提供的代码，我们沿用此方式。
                field_name = related_obj_descriptor.identity[0].name
                related_model = related_obj_descriptor.identity[0].model

                # 如果关联模型在忽略列表中，则跳过
                if related_model in ignore_models:
                    continue

                # 构建查询字典，查找关联到当前实例的记录
                # 假设所有关联模型都有一个 'is_deleted' 字段，用于软删除
                lookup = {
                    "is_deleted": False, # 根据实际业务逻辑调整或移除
                    field_name: self.pk  # 使用当前实例的主键进行关联查询
                }

                # 查询关联模型中是否存在记录
                relation_count = related_model.objects.filter(**lookup).count()

                # 如果找到任何关联记录，则立即返回 True
                if relation_count > 0:
                    return True

            # 遍历所有反向关系后仍未找到关联，返回 False
            return False
        except Exception:
            # 捕获异常，通常情况下，如果发生异常，我们可能倾向于认为存在关联
            # 以避免误删或不当操作。但在生产环境中，建议更精确地处理异常。
            return True

# 示例模型
class A(BaseModel):
    name = models.CharField(_('Name'), max_length=255)

class OtherModel1(models.Model):
    a = models.ForeignKey(A, on_delete=models.PROTECT, related_name='other_model1_set')
    description = models.TextField()
    is_deleted = models.BooleanField(default=False)

class OtherModel2(models.Model):
    main_a = models.ForeignKey(A, on_delete=models.PROTECT, related_name='other_model2_set')
    value = models.IntegerField()
    is_deleted = models.BooleanField(default=False)

# 未来可能添加的更多模型
# class OtherModel3(models.Model):
#     a_ref = models.ForeignKey(A, on_delete=models.PROTECT, related_name='other_model3_set')
#     is_deleted = models.BooleanField(default=False)

# 创建一个A的实例
a_instance = A.objects.create(name="主实例A")

# 创建一个关联到a_instance的OtherModel1实例
OtherModel1.objects.create(a=a_instance, description="关联描述1")

# 检查a_instance是否存在关联
if a_instance.has_relation():
    print(f"实例 '{a_instance.name}' 存在关联。")
else:
    print(f"实例 '{a_instance.name}' 不存在关联。")

# 假设我们不想检查OtherModel1的关联
if a_instance.has_relation(ignore_models=[OtherModel1]):
    print(f"实例 '{a_instance.name}' (忽略OtherModel1后) 存在关联。")
else:
    print(f"实例 '{a_instance.name}' (忽略OtherModel1后) 不存在关联。")

# 创建一个OtherModel2实例，但不关联到a_instance
OtherModel2.objects.create(value=100)