Django原生sql也能使用Paginator分页
Django原生sql也能使用Paginator分页
张豪飞 发表于1周前
Django原生sql也能使用Paginator分页
  • 发表于 1周前
  • 阅读 80
  • 收藏 5
  • 点赞 1
  • 评论 0
摘要: Paginator其实只需要实现两个方法`count`和`__getslice__`就可以自定义一个让Paginator支持的对象

0x00 Django分页局限

使用Django肯定经常使用Paginator分页,很便捷。但是他可接受的分页对象必须是django orm的查询集或者list、tuple。

当需要使用原生sql查询数据且分页就无法使用Paginator。

0x01 分页原理

其实分页就是传入数据集、每页数量、当前页数,然后计算(查询)数据总数量,根据每页数量计算出总页数,当前页的开始index和结束index,然后根据开始index结束index获取本页数据返回。

请注意上面一句话的黑体字部分,它们就是计算分页的核心,那么Paginator其实只需要实现两个方法count__getslice__就可以自定义一个让Paginator支持的对象,然后就可以使用Paginator分页了,不需要单独对原生sql写分页逻辑

0x02 自定义分页

# coding=utf-8

from django.core.paginator import Paginator


def paginator(data_list, per_page, page_no):
    """封装Django分页"""
    pages = Paginator(data_list, per_page)

    # 防止超出页数
    if not page_no > 0:
        page_no = 1
    if page_no > pages.num_pages:
        page_no = pages.num_pages

    p = pages.page(page_no)  # 获取本页数据

    data = dict()  # 获取分页信息
    data['count'] = pages.count
    data['page_num'] = pages.num_pages
    data['per_page'] = per_page
    data['current'] = page_no
    data['start_index'] = p.start_index() - 1

    return p.object_list, page_no, data


class QueryWrapper(object):
    """查询集包装器。实现django Paginator需要的必要方法,实现和query一样使用Paginator分页"""

    def __init__(self, sql, params=None, db="default"):
        """
        :param sql: sql语句
        :param params: sql语句的params参数
        :param db: 数据库名称(Django配置)
        """
        self.db = db
        self.sql = sql
        self.params = params

    def count(self):
        """计算总页数"""
        sql = """select count(*) from (%s) _count""" % self.sql
        # sql封装方法请参考https://my.oschina.net/watcher/blog/1573503
        return fetchone_sql((sql, self.params), db=self.db, flat=True)  # 返回总页数

    def __getslice__(self, x, y):
        """ self.__getslice(x, y) = self[x:y]"""
        sql = self.sql + ' LIMIT {start}, {num}'.format(start=x, num=y - x)
        # sql封装方法请参考https://my.oschina.net/watcher/blog/1573503
        return fetchall_to_dict((sql, self.params), db=self.db)  # 字典列表形式返回


def demo_orm():
    """使用Django的ORM分页"""
    # 示例:查询status=1的用户分页,每页10条,取第2页数据(假设数据量足够)
    status = 1
    per_page = 10
    page_no = 2

    # 使用Django的ORM
    from django.contrib.auth.models import User

    query = User.objects.filter(status=status).values("id", "username", "first_name")
    one_page_data_list, page_no, page_data = paginator(query, per_page, page_no)
    # one_page_data_list 即为第二页数据,例如:[{"id": 1, "username": "111", "first_name": "aaa"}]
    print one_page_data_list


def demo_raw():
    """使用原生sql实现相同分页"""
    # 示例:查询status=1的用户分页,每页10条,取第2页数据(假设数据量足够)
    status = 1
    per_page = 10
    page_no = 2

    sql = "select id, username, first_name from auth_user where status=%(status)s"
    params = {"status": status}  # 使用params防止sql注入
    query = QueryWrapper(sql, params, "default")
    one_page_data_list, page_no, page_data = paginator(query, per_page, page_no)
    # one_page_data_list 同ORM获取数据一样
    print one_page_data_list


if __name__ == "__main__":
    demo_orm()
    demo_raw()

共有 人打赏支持
粉丝 26
博文 34
码字总数 17997
×
张豪飞
如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!
* 金额(元)
¥1 ¥5 ¥10 ¥20 其他金额
打赏人
留言
* 支付类型
微信扫码支付
打赏金额:
已支付成功
打赏金额: