如何在Django中对User与Address模型实现布尔搜索？

1. 先修正模型中的笔误 #

你的模型代码存在几处拼写错误，先修正避免后续运行报错：

from django.contrib.auth.models import AbstractUser
from django.db import models

class Address(models.Model):
    city = models.CharField(max_length=100)
    state = models.CharField(max_length=100)
    country = models.CharField(max_length=100)
    # 关联User模型，对应你提供的Address数据中的user字段
    user = models.OneToOneField('User', on_delete=models.CASCADE, related_name='user_address')

class User(AbstractUser):
    first_name = models.CharField(max_length=100)
    last_name = models.CharField(max_length=100)
    # 修正ForeignKey拼写和on_delete参数值
    address = models.ForeignKey(Address, on_delete=models.CASCADE, related_name='users')
    department = models.CharField(max_length=50)

2. 核心实现思路 #

利用Django的Q对象实现复杂布尔查询：

• Q()对象原生支持&（对应布尔AND）、|（对应布尔OR）、~（对应布尔NOT）操作
• 通过外键关联查询语法address__city（双下划线）实现User与Address模型的跨表查询
• 将用户输入的布尔表达式转换为Q对象的组合逻辑

3. 具体代码实现 #

3.1 定义字段匹配的Q生成函数 #

先写两个辅助函数，分别生成匹配用户姓名、地址信息的Q对象：

from django.db.models import Q

def match_user_name(term):
    # 匹配first_name或last_name包含搜索词（不区分大小写）
    return Q(first_name__icontains=term) | Q(last_name__icontains=term)

def match_address(term):
    # 匹配city/state/country包含搜索词（不区分大小写）
    return Q(address__city__icontains=term) | Q(address__state__icontains=term) | Q(address__country__icontains=term)

3.2 解析布尔表达式并构建查询 #

将用户输入的布尔表达式（如John AND (Ohio OR London)）转换为Q对象组合，这里做了安全限制，只允许合法操作符和术语：

def parse_boolean_search(search_str):
    # 替换布尔关键字为Q对象支持的操作符
    expr = search_str.upper().replace('AND', '&').replace('OR', '|')
    # 提取所有搜索术语
    import re
    terms = re.findall(r'\b[A-Za-z]+\b', search_str)
    # 将每个术语替换为对应的Q对象调用（同时匹配姓名和地址，可根据需求调整）
    for term in terms:
        expr = expr.replace(term, f"(match_user_name('{term}') | match_address('{term}'))")
    # 执行表达式生成查询对象（确保输入可信，避免恶意代码）
    try:
        query = eval(expr)
        return query
    except Exception as e:
        raise ValueError(f"无效的搜索表达式: {str(e)}")

3.3 执行查询示例 #

调用上述函数结合User模型执行查询：

# 示例1：查询姓名含John，且地址含Ohio或London的用户
search_expr = "John AND (Ohio OR London)"
query = parse_boolean_search(search_expr)
users = User.objects.filter(query)
# 结果：ID=3的John Cena（地址为London）

# 示例2：查询姓名含Jon或John，且地址含Delhi或London的用户
search_expr = "(Jon OR John) AND (Delhi OR London)"
query = parse_boolean_search(search_expr)
users = User.objects.filter(query)
# 结果：ID=2的Jon Doe（Delhi）、ID=3的John Cena（London）

4. 安全提示 #

• 上述eval方法存在一定安全风险，如果搜索表达式来自不可信用户，建议使用pyparsing库构建专门的语法解析器
• 若需要精确匹配而非模糊包含，可将__icontains替换为__exact

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