如何在Django模型查询中使用变量的字符串值动态调用对应字段

实现核心依赖Python内置的getattr()函数，该函数可以通过字符串形式的属性名，动态获取对象的对应属性。为了避免非法参数访问到非反应类字段，需要先对传入的reaction参数做白名单校验。

完整实现代码 #

# 预先定义所有合法的反应字段白名单，可根据你的模型字段调整
ALLOWED_REACTIONS = {"likes", "love", "angry", "haha", "sad", "care", "senti", "wow"}

def get_reaction_queryset(post_obj, reaction):
    # 先校验参数合法性
    if reaction not in ALLOWED_REACTIONS:
        # 可根据业务需求替换为抛出DRF参数异常、返回空查询集等逻辑
        raise ValueError(f"不支持的反应类型：{reaction}")
    # 动态获取对应多对多字段的管理器
    reaction_field = getattr(post_obj, reaction)
    # 执行查询返回结果
    return reaction_field.all()

调用示例 #

假设你已经获取到SocialPost的实例对象post，传入reaction参数即可得到对应查询集：

# 等价于 post.wow.all()
wow_users = get_reaction_queryset(post, "wow")

# 等价于 post.likes.all()
like_users = get_reaction_queryset(post, "likes")

扩展场景：动态添加/删除反应 #

如果需要实现用户添加/取消反应的逻辑，也可以用同样的方式实现：

# 给当前用户user添加对应反应
if reaction in ALLOWED_REACTIONS:
    getattr(post, reaction).add(user)

# 取消当前用户的对应反应
if reaction in ALLOWED_REACTIONS:
    getattr(post, reaction).remove(user)

内容的提问来源于stack exchange，提问作者Md Fahad Hossain