如何基于两个Django模型类在GET接口中按site_key过滤VCDUnavailAudit数据

两个模型通过user_key字段关联，你可以通过以下两种方式实现过滤：

方案1：不修改现有模型结构 #

先查询指定site_key对应的所有用户user_key集合，再用这个集合过滤审计表数据：

@api_view(['GET'])
def get_unavail_audit_records_by_site(request, site_key: int):
    # 获取对应站点的所有用户user_key
    target_user_keys = User.objects.filter(site_key=site_key).values_list('user_key', flat=True)
    # 过滤审计表记录
    availability_audit_records = VCDUnavailAudit.objects.filter(user_key__in=target_user_keys)
    serializer = VCDUnavailAuditSerializer(availability_audit_records, many=True)
    return Response(serializer.data, status=status.HTTP_200_OK)

方案2：添加外键关联（更符合Django ORM规范，推荐） #

首先修改模型，给VCDUnavailAudit添加外键关联User表，避免手动处理关联逻辑：

class VCDUnavailAudit(models.Model):
    # 用db_column指定数据库字段名保持和原来一致，不需要修改现有表结构
    user = models.ForeignKey(User, on_delete=models.DO_NOTHING, db_column='user_key', related_name='unavail_audits')
    user_name = models.CharField(max_length=40)

class User(models.Model):
    user_key = models.AutoField(primary_key=True)
    site_key = models.IntegerField() 

修改后视图可以直接跨表过滤，代码更简洁，ORM自动做关联查询性能更高：

@api_view(['GET'])
def get_unavail_audit_records_by_site(request, site_key: int):
    availability_audit_records = VCDUnavailAudit.objects.filter(user__site_key=site_key)
    serializer = VCDUnavailAuditSerializer(availability_audit_records, many=True)
    return Response(serializer.data, status=status.HTTP_200_OK)

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