104 lines
3.0 KiB
Python
104 lines
3.0 KiB
Python
"""数据库通用查询方法"""
|
||
from typing import Optional
|
||
|
||
from tortoise import connections, models
|
||
|
||
|
||
class DbHelper:
|
||
def __init__(self, model: models.Model):
|
||
"""
|
||
初始化
|
||
:param model: 模型类
|
||
"""
|
||
self.model = model
|
||
|
||
def __filter(self, kwargs: dict):
|
||
"""
|
||
过滤数据,默认过滤数据
|
||
:param kwargs:
|
||
:return:
|
||
"""
|
||
return self.model.filter(**kwargs)
|
||
|
||
async def select(self, kwargs: dict = None) -> Optional[models.Model]:
|
||
"""
|
||
查询符合条件的第一个对象, 查无结果时返回None
|
||
:param kwargs: kwargs: {"name:"7y", "id": 1}
|
||
:return: select * from model where name = "7y" and id = 1 limit 1
|
||
"""
|
||
if kwargs is None:
|
||
kwargs = {}
|
||
return await self.__filter(kwargs).first()
|
||
|
||
async def update(self, filters: dict = None, updates: dict = None):
|
||
"""
|
||
更新单条数据
|
||
:param filters: 条件字典 {"id":1,"status__not": 9}
|
||
:param updates: 待更新数据 {"status": 5}
|
||
:return: 0 失败， 1 成功
|
||
"""
|
||
return await self.__filter(filters).update(**updates)
|
||
|
||
async def delete(self, pk: int) -> int:
|
||
"""
|
||
逻辑删除单条数据, status -> 9
|
||
:param pk: 数据id
|
||
:return: 0 是删除 失败， 1是删除成功
|
||
"""
|
||
filters = {"id": pk}
|
||
updates = dict(status=9)
|
||
return await self.update(filters=filters, updates=updates)
|
||
|
||
async def insert(self, data: dict):
|
||
"""
|
||
新增一条数据
|
||
:param data: 模型字典
|
||
:return: 新增之后的对象
|
||
"""
|
||
return await self.model.create(**data)
|
||
|
||
async def selects(
|
||
self, offset: int, limit: int, kwargs: dict = None, order_by: str = None
|
||
) -> dict:
|
||
"""
|
||
条件分页查询数据列表, 支持排序
|
||
Args:
|
||
offset: 偏移量
|
||
limit: 数量
|
||
kwargs: 条件 {}
|
||
order_by: 排序，默认为None， 传入 -字段名 降序 字段名升序
|
||
SQL => select * from model where xx=xx ... order by xx limit offset, limit
|
||
Returns:
|
||
{"items": Model列表, "total": "数量"}
|
||
"""
|
||
if kwargs is None:
|
||
kwargs = {}
|
||
objs = self.__filter(kwargs).all()
|
||
if order_by is not None:
|
||
objs = objs.order_by(order_by)
|
||
|
||
return dict(
|
||
items=await objs.offset(offset).limit(limit), total=await objs.count()
|
||
)
|
||
|
||
async def inserts(self, objs: list[models.Model]):
|
||
"""
|
||
批量新增数据
|
||
:param objs: 模型列表
|
||
:return:
|
||
"""
|
||
await self.model.bulk_create(objs)
|
||
|
||
@classmethod
|
||
async def raw_sql(cls, sql: str, args: list = None):
|
||
"""
|
||
手动执行SQL
|
||
:param sql:
|
||
:param args: sql参数
|
||
:return:
|
||
"""
|
||
db = connections.get("default")
|
||
if args is None:
|
||
args = []
|
||
return await db.execute_query_dict(sql, args)
|