Django原生sql也能使用Paginator分页的示例代码
django-pagination这是一个python包,来自github上的一个项目,很容易用。
不过这是一个懒人工具,好吧(工具理性)。不过当一个页面有多处需要采用分页的话,就行不通了,要么修改django-pagination的源码,改变它的url指向,不过我没研究,当工程涉及到迁移时,要知道要安装各种东西本来就是个缺点,还要再修改源码,那就得不偿失。因而转战django自带的分页插件——Paginator。
Paginator其实只需要实现两个方法`count`和`__getslice__`就可以自定义一个让Paginator支持的对象
0x00Django分页局限
使用Django肯定经常使用Paginator分页,很便捷。但是他可接受的分页对象必须是djangoorm的查询集或者list、tuple。
当需要使用原生sql查询数据且分页就无法使用Paginator。
0x01分页原理
其实分页就是传入数据集、每页数量、当前页数,然后计算(查询)数据总数量,根据每页数量计算出总页数,当前页的开始index和结束index,然后根据开始index和结束index获取本页数据返回。
请注意上面一句话的黑体字部分,它们就是计算分页的核心,那么Paginator其实只需要实现两个方法count和__getslice__就可以自定义一个让Paginator支持的对象,然后就可以使用Paginator分页了,不需要单独对原生sql写分页逻辑
0x02自定义分页
#coding=utf-8
fromdjango.core.paginatorimportPaginator
defpaginator(data_list,per_page,page_no):
"""封装Django分页"""
pages=Paginator(data_list,per_page)
#防止超出页数
ifnotpage_no>0:
page_no=1
ifpage_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
returnp.object_list,page_no,data
classQueryWrapper(object):
"""查询集包装器。实现djangoPaginator需要的必要方法,实现和query一样使用Paginator分页"""
def__init__(self,sql,params=None,db="default"):
"""
:paramsql:sql语句
:paramparams:sql语句的params参数
:paramdb:数据库名称(Django配置)
"""
self.db=db
self.sql=sql
self.params=params
defcount(self):
"""计算总页数"""
sql="""selectcount(*)from(%s)_count"""%self.sql
#sql封装方法请参考https://my.oschina.net/watcher/blog/1573503
returnfetchone_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
returnfetchall_to_dict((sql,self.params),db=self.db)#字典列表形式返回
defdemo_orm():
"""使用Django的ORM分页"""
#示例:查询status=1的用户分页,每页10条,取第2页数据(假设数据量足够)
status=1
per_page=10
page_no=2
#使用Django的ORM
fromdjango.contrib.auth.modelsimportUser
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"}]
printone_page_data_list
defdemo_raw():
"""使用原生sql实现相同分页"""
#示例:查询status=1的用户分页,每页10条,取第2页数据(假设数据量足够)
status=1
per_page=10
page_no=2
sql="selectid,username,first_namefromauth_userwherestatus=%(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获取数据一样
printone_page_data_list
if__name__=="__main__":
demo_orm()
demo_raw()
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。