MongoEngine自定义Queryset

BaseModel:

class BaseModel(Document):
    is_delete = BooleanField(default=False)  # 软删除
    is_block = BooleanField(default=False)  # 禁用
    create_time = DateTimeField(default=datetime.now)  # 创建时间
    update_time = DateTimeField()  # 更新时间
    meta = {
        'allow_inheritance': True,
        'abstract': True,
        'queryset_class': CustomerQuerySet,
    }

``
用了is_delete逻辑判断数据是否已经被删除,如果在每次Model.objects查询时加上is_delete=False比较麻烦,而且容易忽略掉;因此可以重写queryset对象的查询方法
在这里插入图片描述底层BaseQuerySet的__call__方法会返回所查询的queryset对象,所以我们可以自定义CustomerQuerySet并集成mongonengine的QuerySet类,重写call方法,把需要的条件参数加到kwargs里面,进而实现自定义的筛选功能

class CustomerQuerySet(QuerySet):
    def __call__(self, *args, **kwargs):
        # 过滤掉已经逻辑删除的数据
        kwargs['is_delete'] = False
        return super().__call__(*args, **kwargs)

写好该方法后,需要在BaseModel的meta里面重新定义queryset_class,使用自定义的QuerySet类

    meta = {
        'allow_inheritance': True,
        'abstract': True,
        'queryset_class': CustomerQuerySet,
    }
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值