Django原生sql也能使用Paginator分页的示例
程序员文章站
2022-06-16 16:27:18
django-pagination这是一个python包,来自github上的一个项目,很容易用。
不过这是一个懒人工具,好吧(工具理性)。不过当一个页面有多处需要采...
django-pagination这是一个python包,来自github上的一个项目,很容易用。
不过这是一个懒人工具,好吧(工具理性)。不过当一个页面有多处需要采用分页的话,就行不通了,要么修改django-pagination的源码,改变它的url指向,不过我没研究,当工程涉及到迁移时,要知道要安装各种东西本来就是个缺点,还要再修改源码,那就得不偿失。因而转战django自带的分页插件——Paginator。
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()
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。
上一篇: shell脚本实现猜数游戏