Django原生sql也能使用Paginator分页的示例代码

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()

来源：https://my.oschina.net/watcher/blog/1573523