28 lines
698 B
Python
28 lines
698 B
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:
|
|||
|
|
"""
|
|||
|
|
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_set[offset: offset + page_size]
|
|||
|
|
handler(paragraph_list)
|