71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
||
|
||
from typing import Any
|
||
from pydantic import ConfigDict, Field, BaseModel
|
||
from pydantic.alias_generators import to_camel
|
||
|
||
from app.common.constant import RET
|
||
from app.core.exceptions import CustomException
|
||
|
||
|
||
class PageResultSchema(BaseModel):
|
||
"""分页查询结果模型"""
|
||
model_config = ConfigDict(alias_generator=to_camel, from_attributes=True)
|
||
|
||
page_no: int | None = Field(default=None, ge=1, description="页码，默认为1")
|
||
page_size: int | None = Field(default=None, ge=1, description="页面大小，默认为10")
|
||
total: int = Field(default=0, ge=0, description="总记录数")
|
||
has_next: bool | None = Field(default=False, description="是否有下一页")
|
||
items: list[Any] = Field(default_factory=list, description="分页后的数据列表")
|
||
|
||
|
||
class PaginationService:
|
||
"""分页服务类"""
|
||
|
||
@staticmethod
|
||
async def paginate(data_list: list[Any], page_no: int | None = None, page_size: int | None = None) -> dict[str, Any]:
|
||
"""
|
||
分页数据处理。
|
||
输入数据列表和分页信息，返回分页数据列表结果。
|
||
未传入 page_no 和 page_size 时，使用默认值进行分页。
|
||
|
||
参数:
|
||
- data_list (list[Any]): 原始数据列表。
|
||
- page_no (int | None): 当前页码，默认 None。
|
||
- page_size (int | None): 每页数据量，默认 None。
|
||
|
||
返回:
|
||
- dict[str, Any]: 分页数据对象。
|
||
|
||
异常:
|
||
- CustomException: 当分页参数不合法时抛出。
|
||
"""
|
||
total = len(data_list)
|
||
|
||
# 设置默认值
|
||
if page_no is None:
|
||
page_no = 1
|
||
if page_size is None:
|
||
page_size = 10
|
||
|
||
# 验证分页参数
|
||
if page_no < 1 or page_size < 1:
|
||
raise CustomException(code=RET.ERROR.code, msg="分页参数不合法")
|
||
|
||
# 计算起始索引和结束索引
|
||
start = (page_no - 1) * page_size
|
||
end = min(start + page_size, total)
|
||
|
||
# 根据计算得到的起始索引和结束索引对数据列表进行切片
|
||
paginated_data = data_list[start:end]
|
||
|
||
# 判断是否有下一页
|
||
has_next = end < total
|
||
|
||
return {
|
||
"items": paginated_data,
|
||
"total": total,
|
||
"page_no": page_no,
|
||
"page_size": page_size,
|
||
"has_next": has_next
|
||
} |