一个简单的分页
# -*- coding: utf-8 -*-
class PageResult(list):
def __init__(self, total=0, page_no=1, page_size=0, edge_size=0):
self.total = total
self.page_size = page_size if page_size > 0 else 0
self.edge_size = edge_size if edge_size > 0 else 0
if page_no <= 0:
self.no = 1
elif self.max >=1 and page_no > self.max:
self.no = self.max
else:
self.no = page_no
@property
def start(self):
return (self.no - 1) * self.page_size
@property
def max(self):
if self.page_size > 0:
return (self.total + self.page_size - 1) / self.page_size
else: #不分页，显示全部条目
return 1
@property
def has_prev(self):
return self.no > 1
@property
def has_next(self):
return self.no < self.max
@property
def slider(self):
if self.edge_size > 0:
start = max(self.no - self.edge_size, 1)
stop = min(self.no + self.edge_size, self.max)
else: #不滑窗，显示全部页码
start, stop = 1, self.max
return range(start, stop + 1)
def clear(self):
del self[:]
def page_dict(self):
page = {"curr_page":self.no, "data_count":self.total, "page_count":self.max}
page["has_prev"], page["has_next"] = self.has_prev, self.has_next
page["page_range"] = self.slider
return page
class Pagination(object):
def __init__(self, query):
self.query = query
def all(self, offset=0, limit=0):
if limit == 0:
return self.query.all()
else:
return self.query.offset(offset).limit(limit).all() #SQLAlchemy orm
return self.query.all()[offset : offset + limit] #Django orm
def count(self):
return self.query.count()
def head(self):
return self.query.first()
def paginate(self, page_no=1, page_size=0, edge_size=0):
result = PageResult(self.count(), page_no, page_size, edge_size)
result.extend( self.all(result.start, result.page_size) )
return result
