zl程序教程

您现在的位置是:首页 >  后端

当前栏目

Python编程:数据库分页取数据

2023-09-14 09:07:14 时间

简单版分页

def paginate(page, paginate_by=20):
    """
    表分页
    """
    if page > 0:
        page -= 1

    return {
        "limit": paginate_by,
        "offset": page * paginate_by
    }


if __name__ == "__main__":
   for i in range(1, 10):
       ret = paginate(i)
       print(ret["limit"], ret["offset"])
   
   """
   (20, 0)
   (20, 20)
   (20, 40)
   (20, 60)
   (20, 80)
   (20, 100)
   (20, 120)
   (20, 140)
   (20, 160)
   """

分页函数参考peewee模块

加强版分页函数

由于前端传回来的参数,都是字符串,后端还要转为int,每次都转很麻烦,所以直接封装到分页函数里

# -*- coding: utf-8 -*-

def paginate(page, size=20):
    """
    数据库 分页 和 翻页 功能函数
    @param page: int or str 页面页数
    @param size: int or str 分页大小
    @return: dict
    {
        'limit': 20,   所取数据行数
        'offset': 0,   跳过的行数
        'before': 0,   前一页页码
        'current': 1,  当前页页码
        'next': 2      后一页页码
    }
    """

    if not isinstance(page, int):
        try:
            page = int(page)
        except TypeError:
            page = 1

    if not isinstance(size, int):
        try:
            size = int(size)
        except TypeError:
            size = 20

    if page > 0:
        page -= 1

    data = {
        "limit": size,
        "offset": page * size,
        "before": page,
        "current": page + 1,
        "next": page + 2
    }

    return data


if __name__ == '__main__':
    result = paginate(None, None)
    print(result)

    result = paginate(0, 20)
    print(result)

    result = paginate(1, 50)
    print(result)

    result = paginate(3, 20)
    print(result)

    result = paginate("3", "30")
    print(result)

    """
    {'limit': 20, 'offset': 0, 'before': 0, 'current': 1, 'next': 2}
    {'limit': 20, 'offset': 0, 'before': 0, 'current': 1, 'next': 2}
    {'limit': 50, 'offset': 0, 'before': 0, 'current': 1, 'next': 2}
    {'limit': 20, 'offset': 40, 'before': 2, 'current': 3, 'next': 4}
    {'limit': 30, 'offset': 60, 'before': 2, 'current': 3, 'next': 4}
    """