Django模型中修改QuerySet查询参数适配哈希字段的问题

1. 自定义QuerySet处理哈希逻辑 #

核心是重写QuerySet的查询方法，自动将传入的name和value参数哈希后再执行查询。先定义一个和你save方法中一致的哈希函数：

import hashlib
from django.db import models

def hash_value(val):
    # 这里用你save方法中使用的哈希算法，比如SHA256，务必和保存时的逻辑一致
    return hashlib.sha256(val.encode('utf-8')).hexdigest()

class MyModelQuerySet(models.QuerySet):
    def filter(self, **kwargs):
        # 检查并处理name参数
        if 'name' in kwargs:
            kwargs['name'] = hash_value(kwargs['name'])
        # 检查并处理value参数
        if 'value' in kwargs:
            kwargs['value'] = hash_value(kwargs['value'])
        # 交给父类方法执行实际查询
        return super().filter(**kwargs)
    
    def get(self, **kwargs):
        # get方法和filter逻辑一致，处理参数后调用父类方法
        if 'name' in kwargs:
            kwargs['name'] = hash_value(kwargs['name'])
        if 'value' in kwargs:
            kwargs['value'] = hash_value(kwargs['value'])
        return super().get(**kwargs)

2. 将自定义QuerySet绑定到模型Manager #

给MyModel替换默认的管理器，用上上面定义的QuerySet：

class MyModelManager(models.Manager):
    def get_queryset(self):
        return MyModelQuerySet(self.model, using=self._db)

class MyModel(models.Model):
    name = models.CharField(max_length=255)
    value = models.CharField(max_length=255)
    
    # 替换默认的objects管理器
    objects = MyModelManager()
    
    def save(self, *args, **kwargs):
        # 保留你原来的哈希保存逻辑
        self.name = hash_value(self.name)
        self.value = hash_value(self.value)
        super().save(*args, **kwargs)

3. 进阶：覆盖更多查询场景（可选） #

如果需要使用exclude或者Q对象查询，直接重写_filter_or_exclude方法统一处理：

class MyModelQuerySet(models.QuerySet):
    # 保留上面的filter、get方法
    
    def exclude(self, **kwargs):
        # 处理exclude的参数
        if 'name' in kwargs:
            kwargs['name'] = hash_value(kwargs['name'])
        if 'value' in kwargs:
            kwargs['value'] = hash_value(kwargs['value'])
        return super().exclude(**kwargs)
    
    def _filter_or_exclude(self, negate, *args, **kwargs):
        # 处理关键字参数
        if 'name' in kwargs:
            kwargs['name'] = hash_value(kwargs['name'])
        if 'value' in kwargs:
            kwargs['value'] = hash_value(kwargs['value'])
        # 递归处理Q对象中的参数
        new_args = []
        for q in args:
            if isinstance(q, models.Q):
                processed_children = []
                for child in q.children:
                    if child[0] == 'name':
                        processed_children.append(('name', hash_value(child[1])))
                    elif child[0] == 'value':
                        processed_children.append(('value', hash_value(child[1])))
                    else:
                        processed_children.append(child)
                new_q = models.Q(*processed_children, _negated=q.negated)
                new_args.append(new_q)
            else:
                new_args.append(q)
        return super()._filter_or_exclude(negate, *new_args, **kwargs)

关键注意事项 #

• 哈希函数必须和save方法中的逻辑完全一致，否则哈希后的字符串不匹配会导致查询失败。
• 如果还有其他查询方法需要覆盖（比如get_or_create），可以参照上述逻辑，先处理参数再调用父类方法。

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