99 lines
2.4 KiB
Python
99 lines
2.4 KiB
Python
import math
|
||
from typing import Optional, List
|
||
|
||
from pydantic import BaseModel, ConfigDict
|
||
from pydantic.alias_generators import to_camel
|
||
|
||
|
||
class PageModel(BaseModel):
|
||
"""
|
||
分页模型
|
||
"""
|
||
offset: int
|
||
page_num: int
|
||
page_size: int
|
||
total: int
|
||
has_next: bool
|
||
|
||
|
||
class PageObjectResponse(BaseModel):
|
||
"""
|
||
用户管理列表分页查询返回模型
|
||
"""
|
||
rows: List = []
|
||
page_num: int
|
||
page_size: int
|
||
total: int
|
||
has_next: bool
|
||
|
||
|
||
class PageResponseModel(BaseModel):
|
||
"""
|
||
列表分页查询返回模型
|
||
"""
|
||
model_config = ConfigDict(alias_generator=to_camel)
|
||
|
||
rows: List = []
|
||
page_num: Optional[int] = None
|
||
page_size: Optional[int] = None
|
||
total: int
|
||
has_next: Optional[bool] = None
|
||
|
||
|
||
def get_page_info(offset: int, page_num: int, page_size: int, count: int):
|
||
"""
|
||
根据分页参数获取分页信息
|
||
:param offset: 起始数据位置
|
||
:param page_num: 当前页码
|
||
:param page_size: 当前页面数据量
|
||
:param count: 数据总数
|
||
:return: 分页信息对象
|
||
"""
|
||
has_next = False
|
||
if offset >= count:
|
||
res_offset_1 = (page_num - 2) * page_size
|
||
if res_offset_1 < 0:
|
||
res_offset = 0
|
||
res_page_num = 1
|
||
else:
|
||
res_offset = res_offset_1
|
||
res_page_num = page_num - 1
|
||
else:
|
||
res_offset = offset
|
||
if (res_offset + page_size) < count:
|
||
has_next = True
|
||
res_page_num = page_num
|
||
|
||
result = dict(offset=res_offset, page_num=res_page_num, page_size=page_size, total=count, has_next=has_next)
|
||
|
||
return PageModel(**result)
|
||
|
||
|
||
def get_page_obj(data_list: List, page_num: int, page_size: int):
|
||
"""
|
||
输入数据列表data_list和分页信息，返回分页数据列表结果
|
||
:param data_list: 原始数据列表
|
||
:param page_num: 当前页码
|
||
:param page_size: 当前页面数据量
|
||
:return: 分页数据对象
|
||
"""
|
||
# 计算起始索引和结束索引
|
||
start = (page_num - 1) * page_size
|
||
end = page_num * page_size
|
||
|
||
# 根据计算得到的起始索引和结束索引对数据列表进行切片
|
||
paginated_data = data_list[start:end]
|
||
has_next = True if math.ceil(len(data_list) / page_size) > page_num else False
|
||
|
||
result = PageResponseModel(
|
||
rows=paginated_data,
|
||
pageNum=page_num,
|
||
pageSize=page_size,
|
||
total=len(data_list),
|
||
hasNext=has_next
|
||
)
|
||
|
||
return result
|
||
|
||
|