mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 10:12:51 +00:00
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
# coding=utf-8
|
||
"""
|
||
@project: MaxKB
|
||
@Author：虎
|
||
@file： page_utils.py
|
||
@date：2024/11/21 10:32
|
||
@desc:
|
||
"""
|
||
from math import ceil
|
||
|
||
|
||
def page(query_set, page_size, handler, is_the_task_interrupted=lambda: False):
|
||
"""
|
||
|
||
@param query_set: 查询query_set
|
||
@param page_size: 每次查询大小
|
||
@param handler: 数据处理器
|
||
@param is_the_task_interrupted: 任务是否被中断
|
||
@return:
|
||
"""
|
||
query = query_set.order_by("id")
|
||
count = query_set.count()
|
||
for i in range(0, ceil(count / page_size)):
|
||
if is_the_task_interrupted():
|
||
return
|
||
offset = i * page_size
|
||
paragraph_list = query.all()[offset: offset + page_size]
|
||
handler(paragraph_list)
|
||
|
||
|
||
def page_desc(query_set, page_size, handler, is_the_task_interrupted=lambda: False):
|
||
"""
|
||
|
||
@param query_set: 查询query_set
|
||
@param page_size: 每次查询大小
|
||
@param handler: 数据处理器
|
||
@param is_the_task_interrupted: 任务是否被中断
|
||
@return:
|
||
"""
|
||
query = query_set.order_by("id")
|
||
count = query_set.count()
|
||
for i in sorted(range(0, ceil(count / page_size)), reverse=True):
|
||
if is_the_task_interrupted():
|
||
return
|
||
offset = i * page_size
|
||
paragraph_list = query.all()[offset: offset + page_size]
|
||
handler(paragraph_list)
|