mirror of
https://github.com/simonw/datasette.git
synced 2025-12-10 16:51:24 +01:00
parent
7dac1c05cd
commit
847f3e0c92
2 changed files with 37 additions and 12 deletions
|
|
@ -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,
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue