蓝鲸接口分页

最新推荐文章于 2023-07-23 19:49:58 发布

药引子1

最新推荐文章于 2023-07-23 19:49:58 发布

阅读量277

点赞数

分类专栏： python

本文链接：https://blog.csdn.net/weixin_43784418/article/details/111075264

版权

python 专栏收录该内容

23 篇文章 1 订阅

订阅专栏

分页器

class BkPageManage(object):
    def __init__(self, func, page=1, start=0, limit=200, no_page=False, **kwargs):
        self.func = func
        self.page = page
        self.start = start
        self.limit = limit
        self.no_page = no_page
        self.kwargs = kwargs
        self.__result = []
        self.__count = 0

    def __call__(self, *args, **kwargs):
        # 不分页
        if self.no_page:
            self.set_kwargs()
            self.__result, self.__count = self.func(**self.kwargs)
            for i in range(self.start, self.__count, self.limit):
                self.add_page()
                self.set_kwargs()
                result, count = self.func(**self.kwargs)
                self.__result.extend(result)
            return self
        # 分页
        for p in range(self.page - 1):
            self.add_page()
        self.set_kwargs()
        self.__result, self.__count = self.func(**self.kwargs)
        return self

    def set_kwargs(self):
        self.kwargs.update({
            'page': {
                'start': int(self.start),
                'limit': int(self.limit)
            }
        })

    def add_page(self):
        self.start += self.limit

    def get_result(self):
        return self.__result, self.__count

定义业务接口

from blueking.component.shortcuts import get_client_by_user


class BkService(object):
    def __init__(self, username='admin'):
        self.client = get_client_by_user(username)

    def search_business(self, **kwargs):
        info = self.client.cc.search_business(kwargs)
        return info.get('data').get('info'), info.get('data').get('count')

调用分页器

def search_business(request):
    bk_service = BkService(request.user.username)
    page = int(request.GET.get('page'))
    page_size = int(request.GET.get('page_size'))
    bk_page = BkPageManage(func=bk_service.search_business, page=page, limit=page_size, no_page=False)()
    result, count = bk_page.get_result()
    data = {
        'total': count,
        'data': result
    }
    return JsonResponse(data)