Implemented offset/limit pagination for views

Closes #70
This commit is contained in:
Simon Willison 2017-11-13 13:10:55 -08:00
commit 847f3e0c92
2 changed files with 37 additions and 12 deletions

View file

@ -382,8 +382,12 @@ class TableView(BaseView):
params = {} params = {}
next = special_args.get('_next') next = special_args.get('_next')
offset = ''
if next: if next:
if use_rowid: if is_view:
# _next is an offset
offset = ' offset {}'.format(int(next))
elif use_rowid:
where_clauses.append( where_clauses.append(
'rowid > :p{}'.format( 'rowid > :p{}'.format(
len(params), len(params),
@ -410,8 +414,13 @@ class TableView(BaseView):
if order_by: if order_by:
order_by = 'order by {} '.format(order_by) order_by = 'order by {} '.format(order_by)
sql = 'select {} from {} {}{}limit {}'.format( sql = 'select {select} from {table_name} {where}{order_by}limit {limit}{offset}'.format(
select, escape_sqlite_table_name(table), where_clause, order_by, self.page_size + 1, select=select,
table_name=escape_sqlite_table_name(table),
where=where_clause,
order_by=order_by,
limit=self.page_size + 1,
offset=offset,
) )
rows, truncated, description = await self.execute(name, sql, params, truncate=True) rows, truncated, description = await self.execute(name, sql, params, truncate=True)
@ -423,11 +432,16 @@ class TableView(BaseView):
rows = list(rows) rows = list(rows)
info = self.ds.inspect() info = self.ds.inspect()
table_rows = info[name]['tables'].get(table) table_rows = info[name]['tables'].get(table)
next = None next_value = None
next_url = None next_url = None
if len(rows) > self.page_size: if len(rows) > self.page_size:
next = path_from_row_pks(rows[-2], pks, use_rowid) if is_view:
next_url = urllib.parse.urljoin(request.url, path_with_added_args(request, {'_next': next})) next_value = int(next or 0) + self.page_size
else:
next_value = path_from_row_pks(rows[-2], pks, use_rowid)
next_url = urllib.parse.urljoin(request.url, path_with_added_args(request, {
'_next': next_value,
}))
return { return {
'database': name, 'database': name,
'table': table, 'table': table,
@ -443,7 +457,7 @@ class TableView(BaseView):
'sql': sql, 'sql': sql,
'params': params, 'params': params,
}, },
'next': next, 'next': next_value and str(next_value) or None,
'next_url': next_url, 'next_url': next_url,
}, lambda: { }, lambda: {
'database_hash': hash, 'database_hash': hash,

View file

@ -122,8 +122,11 @@ def test_table_with_slashes_in_name(app_client):
}] }]
def test_paginate(app_client): @pytest.mark.parametrize('path,expected_rows,expected_pages', [
path = '/test_tables/no_primary_key.jsono' ('/test_tables/no_primary_key.jsono', 201, 5),
('/test_tables/paginated_view.jsono', 201, 5),
])
def test_paginate_tables_and_views(app_client, path, expected_rows, expected_pages):
fetched = [] fetched = []
count = 0 count = 0
while path: while path:
@ -132,9 +135,11 @@ def test_paginate(app_client):
fetched.extend(response.json['rows']) fetched.extend(response.json['rows'])
path = response.json['next_url'] path = response.json['next_url']
if path: if path:
assert path.endswith(response.json['next']) assert response.json['next'] and path.endswith(response.json['next'])
assert 201 == len(fetched) assert count < 10, 'Possible infinite loop detected'
assert 5 == count
assert expected_rows == len(fetched)
assert expected_pages == count
def test_max_returned_rows(app_client): def test_max_returned_rows(app_client):
@ -190,6 +195,12 @@ WITH RECURSIVE
) )
INSERT INTO no_primary_key SELECT * from cnt; INSERT INTO no_primary_key SELECT * from cnt;
CREATE VIEW paginated_view AS
SELECT
content,
'- ' || content || ' -' AS content_extra
FROM no_primary_key;
CREATE TABLE "Table With Space In Name" ( CREATE TABLE "Table With Space In Name" (
pk varchar(30) primary key, pk varchar(30) primary key,
content text content text