19 lines
813 B
Python
19 lines
813 B
Python
|
"""该文件主要是小工具，简化接口处理函数写的代码太多"""
|
|||
|
|
from ninja import Schema
|
|||
|
|
from django.db.models import QuerySet
|
|||
|
|
|
|||
|
|
def conditionNoneToBlank(condition: Schema):
|
|||
|
|
"""将BaseModel/Schema对象中None变为空字符串"""
|
|||
|
|
for attr, value in condition.__dict__.items():
|
|||
|
|
if getattr(condition, attr) is None:
|
|||
|
|
setattr(condition, attr, '')
|
|||
|
|
|
|||
|
|
def model_retrieve(condition: Schema, qs: QuerySet, exclude_field) -> QuerySet:
|
|||
|
|
"""该函数合并上面功能，传入qs以及排除字段，全部添加__icontains进行查询"""
|
|||
|
|
conditionNoneToBlank(condition)
|
|||
|
|
search_obj = {}
|
|||
|
|
for k, v in condition.dict(exclude_none=True).items():
|
|||
|
|
if k not in exclude_field:
|
|||
|
|
search_obj["".join([k, "__icontains"])] = v
|
|||
|
|
return qs.filter(**search_obj)
|