pgkeyset 0.1.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
pgkeyset/__init__.py ADDED
@@ -0,0 +1,19 @@
1
+ """Strict keyset pagination for async SQLAlchemy on Postgres.
2
+
3
+ Code declares each paged read once as a :class:`Listing`, its name and total order, and
4
+ the listing does the rest::
5
+
6
+ CONTACTS = Listing("contacts", Sort.desc(contact.c.created_at), Sort.desc(contact.c.id))
7
+
8
+ stmt = CONTACTS.apply(select(contact), cursor) # ORDER BY + strictly-after
9
+ rows = (await conn.execute(stmt.limit(limit + 1))).mappings().all()
10
+ page, next_cursor = CONTACTS.paginate(rows, limit)
11
+
12
+ A cursor that is not a position in the listing raises :class:`InvalidCursor`, which the
13
+ server maps to its own bad-input error. :mod:`pgkeyset.testing` holds the walk test that
14
+ proves a listing never skips or repeats a row.
15
+ """
16
+
17
+ from .core import INVALID_CURSOR_MESSAGE, InvalidCursor, Listing, Sort
18
+
19
+ __all__ = ["INVALID_CURSOR_MESSAGE", "InvalidCursor", "Listing", "Sort"]
pgkeyset/core.py ADDED
@@ -0,0 +1,340 @@
1
+ """A listing's total order, the predicate that resumes it, and the cursor that names a position.
2
+
3
+ A :class:`Listing` is one paged read's order, declared once: a name, then its :class:`Sort`
4
+ columns from first to last, the last one unique. Everything a paged read needs follows from
5
+ that declaration, so it is written nowhere else:
6
+
7
+ * ``ORDER BY`` in the declared directions, with the NULL placement spelled out for a
8
+ nullable column instead of left to the database's default.
9
+ * "Strictly after this position", as a row comparison ``(a, id) < (x, y)`` when every
10
+ column runs the same direction and none can be NULL, which Postgres reads as an index
11
+ condition. Otherwise it is the explicit expansion: after on the first column, or equal on
12
+ it and after on the next, and so on, with each NULL band named, since a comparison against
13
+ NULL is NULL. Bound values are typed like their columns, so asyncpg gets a ``datetime``
14
+ for a ``timestamptz``.
15
+ * The cursor, ``base64url(JSON [listing, *key])``, with each timestamp as ISO-8601 carrying
16
+ its zone and each ``numeric`` as its decimal string. The listing tag means a cursor from
17
+ another read is refused rather than read as a position in an order it never came from.
18
+
19
+ Any cursor that does not decode as a position in this listing raises :class:`InvalidCursor`,
20
+ whose text is fixed: a parse failure's message never reaches the wire.
21
+ """
22
+
23
+ from __future__ import annotations
24
+
25
+ import base64
26
+ import binascii
27
+ import json
28
+ from collections.abc import Callable, Sequence
29
+ from dataclasses import dataclass
30
+ from datetime import UTC, datetime
31
+ from decimal import Decimal, InvalidOperation
32
+ from typing import Any, Literal
33
+
34
+ from sqlalchemy import ColumnElement, Select, and_, false, literal, or_, tuple_
35
+ from sqlalchemy.sql.elements import ColumnClause, Label
36
+ from sqlalchemy.types import (
37
+ BigInteger,
38
+ DateTime,
39
+ Float,
40
+ Integer,
41
+ Numeric,
42
+ SmallInteger,
43
+ String,
44
+ TypeEngine,
45
+ )
46
+
47
+ __all__ = ["INVALID_CURSOR_MESSAGE", "InvalidCursor", "Listing", "Sort"]
48
+
49
+ INVALID_CURSOR_MESSAGE = "The cursor is not a position in this listing."
50
+
51
+ # The value kinds a key can hold, and so the kinds a cursor can carry back.
52
+ _Kind = Literal["ts", "int", "str", "num"]
53
+
54
+
55
+ class InvalidCursor(ValueError):
56
+ """A cursor that does not decode as a position in the listing it was passed to.
57
+
58
+ The message is always :data:`INVALID_CURSOR_MESSAGE`, never the parse failure's, so a
59
+ server can put it on the wire as bad input. The way out is always the same: call again
60
+ without ``cursor``.
61
+ """
62
+
63
+ def __init__(self) -> None:
64
+ super().__init__(INVALID_CURSOR_MESSAGE)
65
+
66
+
67
+ @dataclass(frozen=True)
68
+ class Sort:
69
+ """One column of a listing's order: an expression, its direction, and where NULLs go.
70
+
71
+ ``nulls`` declares the column nullable and places its NULL band ``"first"`` or ``"last"``.
72
+ A table column declared ``nullable`` must say which; anything else is taken as never NULL.
73
+ Only a plain table column is checked, so a computed or labelled expression that can be NULL
74
+ must pass ``nulls`` itself.
75
+
76
+ ``value`` reads this column's value from a fetched row, for the cursor. It defaults to
77
+ ``row[<column key>]``, so a computed expression (a ``case`` band) needs one, unless it is
78
+ a labelled expression the query selects.
79
+ """
80
+
81
+ expr: ColumnElement[Any]
82
+ descending: bool = False
83
+ nulls: Literal["first", "last"] | None = None
84
+ value: Callable[[Any], Any] | None = None
85
+
86
+ @classmethod
87
+ def asc(
88
+ cls,
89
+ expr: ColumnElement[Any],
90
+ *,
91
+ nulls: Literal["first", "last"] | None = None,
92
+ value: Callable[[Any], Any] | None = None,
93
+ ) -> Sort:
94
+ return cls(expr, descending=False, nulls=nulls, value=value)
95
+
96
+ @classmethod
97
+ def desc(
98
+ cls,
99
+ expr: ColumnElement[Any],
100
+ *,
101
+ nulls: Literal["first", "last"] | None = None,
102
+ value: Callable[[Any], Any] | None = None,
103
+ ) -> Sort:
104
+ return cls(expr, descending=True, nulls=nulls, value=value)
105
+
106
+ def __post_init__(self) -> None:
107
+ _kind(self.expr.type) # refuse a type the cursor cannot carry, at declaration
108
+ if self.nulls is None and getattr(self.expr, "nullable", False):
109
+ raise ValueError(
110
+ f"{self.expr} is nullable: declare where its NULLs sort with nulls='first' "
111
+ "or nulls='last'"
112
+ )
113
+ if self.value is None and _row_key(self.expr) is None:
114
+ raise ValueError(f"{self.expr} is computed: pass value= to read it from a row")
115
+
116
+ @property
117
+ def kind(self) -> _Kind:
118
+ return _kind(self.expr.type)
119
+
120
+ @property
121
+ def nullable(self) -> bool:
122
+ return self.nulls is not None
123
+
124
+ def order(self) -> ColumnElement[Any]:
125
+ clause = self.expr.desc() if self.descending else self.expr.asc()
126
+ if self.nulls == "first":
127
+ return clause.nulls_first()
128
+ if self.nulls == "last":
129
+ return clause.nulls_last()
130
+ return clause
131
+
132
+ def read(self, row: Any) -> Any:
133
+ if self.value is not None:
134
+ return self.value(row)
135
+ return row[_row_key(self.expr)]
136
+
137
+ def bind(self, value: Any) -> ColumnElement[Any]:
138
+ return literal(value, self.expr.type)
139
+
140
+ def equals(self, value: Any) -> ColumnElement[bool]:
141
+ if value is None:
142
+ return self.expr.is_(None)
143
+ return self.expr == self.bind(value)
144
+
145
+ def beyond(self, value: Any) -> ColumnElement[bool] | None:
146
+ """Rows that sort strictly after ``value`` on this column alone, or None for none."""
147
+ if value is None:
148
+ # Inside the NULL band: everything after it is the non-NULL rows, or nothing.
149
+ return self.expr.is_not(None) if self.nulls == "first" else None
150
+ bound = self.bind(value)
151
+ past = self.expr < bound if self.descending else self.expr > bound
152
+ if self.nulls == "last":
153
+ return or_(past, self.expr.is_(None))
154
+ return past
155
+
156
+
157
+ class Listing:
158
+ """One paged read: a name for its cursors, and its total order.
159
+
160
+ The last :class:`Sort` must be unique and never NULL (the row's id), so no two rows share
161
+ a position and a page boundary never falls inside a tie.
162
+ """
163
+
164
+ def __init__(self, name: str, *order: Sort) -> None:
165
+ if not order:
166
+ raise ValueError("a listing needs at least one sort column")
167
+ if order[-1].nullable:
168
+ raise ValueError(
169
+ "a listing's last sort column is its unique tiebreak; it cannot be NULL"
170
+ )
171
+ self.name = name
172
+ self.order = order
173
+
174
+ def __repr__(self) -> str:
175
+ return f"Listing({self.name!r})"
176
+
177
+ # --- the query ---------------------------------------------------------- #
178
+
179
+ def apply[S: Select[Any]](self, stmt: S, cursor: str | None) -> S:
180
+ """Order ``stmt`` by this listing and, given a cursor, start strictly after it.
181
+
182
+ An empty or absent cursor is the first page. The caller adds ``.limit(limit + 1)``.
183
+ Raises :class:`InvalidCursor` for a cursor that is not a position in this listing, and
184
+ ``ValueError`` for a ``stmt`` that is already ordered: the listing owns the order, and an
185
+ ORDER BY ahead of it would page in an order the cursor does not follow.
186
+ """
187
+ if stmt._order_by_clauses:
188
+ raise ValueError(f"{self!r} owns the order: pass a statement without ORDER BY")
189
+ stmt = stmt.order_by(*(sort.order() for sort in self.order))
190
+ if cursor:
191
+ stmt = stmt.where(self.after(self.decode(cursor)))
192
+ return stmt
193
+
194
+ def after(self, key: Sequence[Any]) -> ColumnElement[bool]:
195
+ """The rows strictly after ``key`` in this order."""
196
+ if self._uniform():
197
+ exprs = [sort.expr for sort in self.order]
198
+ binds = [sort.bind(v) for sort, v in zip(self.order, key, strict=True)]
199
+ if self.order[0].descending:
200
+ return tuple_(*exprs) < tuple_(*binds)
201
+ return tuple_(*exprs) > tuple_(*binds)
202
+ terms: list[ColumnElement[bool]] = []
203
+ for i, (sort, value) in enumerate(zip(self.order, key, strict=True)):
204
+ beyond = sort.beyond(value)
205
+ if beyond is None:
206
+ continue
207
+ ties = [s.equals(v) for s, v in zip(self.order[:i], key[:i], strict=True)]
208
+ terms.append(and_(*ties, beyond) if ties else beyond)
209
+ return or_(*terms) if terms else false()
210
+
211
+ def _uniform(self) -> bool:
212
+ return not any(s.nullable for s in self.order) and (
213
+ len({s.descending for s in self.order}) == 1
214
+ )
215
+
216
+ # --- the page ----------------------------------------------------------- #
217
+
218
+ def paginate[T](self, rows: Sequence[T], limit: int) -> tuple[list[T], str | None]:
219
+ """Split a ``limit + 1`` fetch into ``(page, next_cursor)``.
220
+
221
+ The extra row exists only when another page does. ``next_cursor`` then encodes the last
222
+ row KEPT, read from the raw row, and is None on the last page.
223
+ """
224
+ page = list(rows[:limit])
225
+ if len(rows) > limit and page:
226
+ return page, self.encode(self.key(page[-1]))
227
+ return page, None
228
+
229
+ def key(self, row: Any) -> tuple[Any, ...]:
230
+ """A fetched row's position in this order."""
231
+ return tuple(sort.read(row) for sort in self.order)
232
+
233
+ # --- the cursor --------------------------------------------------------- #
234
+
235
+ def encode(self, key: Sequence[Any]) -> str:
236
+ """``base64url(JSON [listing, *key])``. A naive timestamp is a server bug, and raises."""
237
+ if len(key) != len(self.order):
238
+ raise ValueError(f"{self!r} keys have {len(self.order)} parts, got {len(key)}")
239
+ parts: list[Any] = [self.name]
240
+ for value in key:
241
+ if isinstance(value, datetime):
242
+ if value.tzinfo is None:
243
+ raise ValueError(f"{self!r}: a naive timestamp cannot name a position")
244
+ value = value.isoformat()
245
+ elif isinstance(value, Decimal):
246
+ # A JSON number is a float to most readers; the digits are the position.
247
+ value = str(value)
248
+ parts.append(value)
249
+ raw = json.dumps(parts, separators=(",", ":")).encode()
250
+ return base64.urlsafe_b64encode(raw).decode().rstrip("=")
251
+
252
+ def decode(self, cursor: str) -> tuple[Any, ...]:
253
+ """The position ``cursor`` names in this listing, or :class:`InvalidCursor`."""
254
+ try:
255
+ padded = cursor + "=" * (-len(cursor) % 4)
256
+ parts = json.loads(base64.urlsafe_b64decode(padded.encode()))
257
+ # A cursor of deeply nested arrays exhausts the parser's recursion limit.
258
+ except (binascii.Error, UnicodeError, ValueError, RecursionError):
259
+ raise InvalidCursor from None
260
+ if not isinstance(parts, list) or len(parts) != len(self.order) + 1:
261
+ raise InvalidCursor
262
+ if parts[0] != self.name:
263
+ raise InvalidCursor
264
+ try:
265
+ return tuple(_parse(v, s) for v, s in zip(parts[1:], self.order, strict=True))
266
+ except (TypeError, ValueError):
267
+ raise InvalidCursor from None
268
+
269
+
270
+ def _kind(type_: TypeEngine[Any]) -> _Kind:
271
+ if isinstance(type_, DateTime):
272
+ if not type_.timezone:
273
+ raise TypeError("a timestamp sort column must be timezone-aware (timestamptz)")
274
+ return "ts"
275
+ if isinstance(type_, Integer):
276
+ return "int"
277
+ if isinstance(type_, String):
278
+ return "str"
279
+ if isinstance(type_, Numeric) and not isinstance(type_, Float):
280
+ return "num"
281
+ raise TypeError(f"a cursor cannot carry a {type_!r} sort key")
282
+
283
+
284
+ def _row_key(expr: ColumnElement[Any]) -> str | None:
285
+ if isinstance(expr, ColumnClause | Label):
286
+ return expr.key
287
+ return None
288
+
289
+
290
+ def _parse(value: Any, sort: Sort) -> Any:
291
+ if value is None:
292
+ if not sort.nullable:
293
+ raise ValueError("NULL in a non-null position")
294
+ return None
295
+ if sort.kind == "ts":
296
+ if not isinstance(value, str):
297
+ raise TypeError("ts")
298
+ parsed = datetime.fromisoformat(value)
299
+ # A naive timestamp compared against timestamptz would be read in the session's zone.
300
+ if parsed.tzinfo is None:
301
+ raise ValueError("naive timestamp")
302
+ # The driver binds the instant in UTC, and an offset at either end of the calendar
303
+ # puts that instant outside the range a datetime holds.
304
+ try:
305
+ return parsed.astimezone(UTC)
306
+ except OverflowError:
307
+ raise ValueError("timestamp out of range") from None
308
+ if sort.kind == "int":
309
+ if not isinstance(value, int) or isinstance(value, bool):
310
+ raise TypeError("int")
311
+ # The driver binds an int at its column type's width and refuses one that does not fit.
312
+ type_ = sort.expr.type
313
+ bits = next((b for t, b in _INT_BITS.items() if isinstance(type_, t)), 32)
314
+ if not -(1 << (bits - 1)) <= value < (1 << (bits - 1)):
315
+ raise ValueError("int out of range")
316
+ return value
317
+ if sort.kind == "num":
318
+ if not isinstance(value, str):
319
+ raise TypeError("num")
320
+ try:
321
+ number = Decimal(value)
322
+ except InvalidOperation:
323
+ raise ValueError("not a decimal") from None
324
+ # NaN and the infinities parse, and neither is a position a numeric row holds.
325
+ if not number.is_finite():
326
+ raise ValueError("non-finite decimal")
327
+ return number
328
+ if not isinstance(value, str):
329
+ raise TypeError("str")
330
+ # Postgres text holds only UTF-8 with no NUL. Anything else the driver refuses at execute;
331
+ # an unencodable string (a lone surrogate) raises UnicodeEncodeError, a ValueError.
332
+ if "\x00" in value:
333
+ raise ValueError("NUL in a string key")
334
+ value.encode("utf-8")
335
+ return value
336
+
337
+
338
+ # Bind widths for the integer types narrower or wider than Postgres ``integer``, matched by
339
+ # ``isinstance`` so the uppercase ``SMALLINT`` and ``BIGINT`` subclasses resolve too.
340
+ _INT_BITS: dict[type, int] = {SmallInteger: 16, BigInteger: 64}
pgkeyset/py.typed ADDED
File without changes
pgkeyset/testing.py ADDED
@@ -0,0 +1,54 @@
1
+ """The walk test for a paged read: every row once, with rows inserted and removed mid-walk.
2
+
3
+ :func:`walk` follows ``next_cursor`` from the first page to the last and fails on a row met
4
+ twice, or, given the rows the walk must meet, on one it never met. ``between_pages`` runs
5
+ after each page that has a next one, which is where a test inserts and removes rows to show
6
+ the position does not move when other rows leave.
7
+ """
8
+
9
+ from __future__ import annotations
10
+
11
+ from collections.abc import Awaitable, Callable, Hashable, Iterable, Sequence
12
+
13
+ __all__ = ["walk"]
14
+
15
+
16
+ async def walk[T](
17
+ fetch: Callable[[str | None], Awaitable[tuple[Sequence[T], str | None]]],
18
+ *,
19
+ ident: Callable[[T], Hashable],
20
+ expected: Iterable[Hashable] | None = None,
21
+ between_pages: Callable[[int], Awaitable[None]] | None = None,
22
+ max_pages: int = 1000,
23
+ ) -> list[T]:
24
+ """Walk a listing to its end and return every row met, in order.
25
+
26
+ ``fetch(cursor)`` reads one page as ``(rows, next_cursor)``, passing the same filters
27
+ every call. ``ident`` names a row. ``expected``, when given, is every row the walk must
28
+ meet: those that existed when it began and still match. ``between_pages(n)`` runs after
29
+ page ``n`` (from 1) when another page follows.
30
+ """
31
+ walked: list[T] = []
32
+ seen: set[Hashable] = set()
33
+ cursor: str | None = None
34
+ for page_number in range(1, max_pages + 1):
35
+ rows, next_cursor = await fetch(cursor)
36
+ for row in rows:
37
+ row_id = ident(row)
38
+ assert row_id not in seen, f"page {page_number} repeats {row_id!r}"
39
+ seen.add(row_id)
40
+ walked.append(row)
41
+ if next_cursor is None:
42
+ break
43
+ assert rows, f"page {page_number} is empty but hands out a next_cursor"
44
+ if between_pages is not None:
45
+ await between_pages(page_number)
46
+ cursor = next_cursor
47
+ else:
48
+ raise AssertionError(f"the walk did not end within {max_pages} pages")
49
+ if expected is not None:
50
+ want = set(expected)
51
+ missing, extra = want - seen, seen - want
52
+ assert not missing, f"the walk never met {sorted(map(repr, missing))}"
53
+ assert not extra, f"the walk met rows it should not have: {sorted(map(repr, extra))}"
54
+ return walked
@@ -0,0 +1,256 @@
1
+ Metadata-Version: 2.5
2
+ Name: pgkeyset
3
+ Version: 0.1.0
4
+ Summary: Strict keyset pagination for async SQLAlchemy on Postgres, for APIs and agent tools. Refuses at declaration the orders that silently lose or repeat rows, tags each cursor with its listing, and fails a bad cursor with one safe message.
5
+ Project-URL: Homepage, https://github.com/NimbleBrainInc/pgkeyset
6
+ Project-URL: Issues, https://github.com/NimbleBrainInc/pgkeyset/issues
7
+ Project-URL: Changelog, https://github.com/NimbleBrainInc/pgkeyset/blob/main/CHANGELOG.md
8
+ Author: NimbleBrain, Inc.
9
+ License-Expression: Apache-2.0
10
+ License-File: LICENSE
11
+ License-File: NOTICE
12
+ Keywords: asyncio,cursor,keyset,mcp,pagination,postgres,sqlalchemy
13
+ Classifier: Development Status :: 3 - Alpha
14
+ Classifier: Framework :: AsyncIO
15
+ Classifier: Intended Audience :: Developers
16
+ Classifier: Operating System :: OS Independent
17
+ Classifier: Programming Language :: Python :: 3
18
+ Classifier: Programming Language :: Python :: 3.12
19
+ Classifier: Programming Language :: Python :: 3.13
20
+ Classifier: Topic :: Database
21
+ Classifier: Typing :: Typed
22
+ Requires-Python: >=3.12
23
+ Requires-Dist: sqlalchemy>=2.0.36
24
+ Description-Content-Type: text/markdown
25
+
26
+ # pgkeyset
27
+
28
+ Strict keyset pagination for async SQLAlchemy on Postgres, for APIs and agent tools.
29
+
30
+ - **Orders that lose rows are refused when you declare them.** A nullable sort column must
31
+ say where its NULLs go, a timestamp must carry a zone, a float cannot be a key, and the
32
+ last column (your unique id) cannot be NULL. These fail at import, not on page 40.
33
+ - **Cursors know where they came from.** Each cursor is tagged with the listing that issued
34
+ it, so a cursor passed to the wrong list is refused instead of read as a position in an
35
+ order it was never taken from.
36
+ - **A bad cursor has one way out.** Anything that is not a position in the listing raises
37
+ `InvalidCursor`, whose text is always the same safe sentence, never a parse error. Your
38
+ API answers "call again without `cursor`".
39
+ - **A walk test that proves it.** `pgkeyset.testing.walk` follows `next_cursor` to the end
40
+ while your test inserts and deletes rows between pages, and fails on any row met twice or
41
+ never met.
42
+
43
+ It is for Python services on async SQLAlchemy 2 that page through Postgres: HTTP APIs, and
44
+ MCP servers whose tools an agent calls page after page while it changes the same rows.
45
+
46
+ ```sh
47
+ pip install pgkeyset # or: uv add pgkeyset
48
+ ```
49
+
50
+ Python 3.12+, SQLAlchemy 2, Postgres. Tested with asyncpg on Postgres 14 and 17.
51
+
52
+ ## Why keyset, never offset
53
+
54
+ A caller works a list while it reads it: an agent completes a task, a reviewer rejects a
55
+ draft. Each removal shifts every later row up by one, so `OFFSET n` skips a row nobody saw.
56
+ A keyset position is a row's own sort key, and it does not move when other rows leave. It is
57
+ also an index range scan, where an offset reads and throws away every earlier row.
58
+
59
+ The price is that you can only go forward from a position, and the order must be total: two
60
+ rows must never share a position, or a page boundary inside the tie drops or repeats one.
61
+ Rows written in one transaction share `now()`, so a timestamp alone is never a total order.
62
+ Every listing ends with the row's unique id.
63
+
64
+ ## Usage
65
+
66
+ Declare each paged read once, next to the query that runs it:
67
+
68
+ ```python
69
+ from pgkeyset import InvalidCursor, Listing, Sort
70
+
71
+ CONTACTS = Listing("contacts", Sort.desc(contact.c.created_at), Sort.desc(contact.c.id))
72
+
73
+ async def list_contacts(conn, *, cursor: str | None, limit: int):
74
+ stmt = CONTACTS.apply(select(contact).where(...), cursor) # ORDER BY + strictly-after
75
+ rows = (await conn.execute(stmt.limit(limit + 1))).mappings().all()
76
+ page, next_cursor = CONTACTS.paginate(rows, limit) # encodes the last row kept
77
+ return [Contact.from_row(r) for r in page], next_cursor
78
+ ```
79
+
80
+ Fetch one row more than the page. If it exists there is a next page, and `next_cursor`
81
+ encodes the last row kept; on the last page it is `None`. An absent or empty `cursor` is the
82
+ first page. Pass the same filters with every page; the cursor carries a position, not a query.
83
+
84
+ `paginate` reads the key from the raw row, before you map it to a model, so a row your code
85
+ later skips as malformed still advances the walk.
86
+
87
+ ### Declaring the order
88
+
89
+ `Listing(name, *sorts)` takes the sort columns from first to last.
90
+
91
+ | Rule | Why | Refused with |
92
+ |---|---|---|
93
+ | The last column is the row's unique, non-NULL id. | Two rows sharing a position split across a page boundary. | `ValueError` if it is declared nullable |
94
+ | A nullable column says where its NULLs go: `Sort.asc(col, nulls="first")` or `"last"`. | A comparison against NULL is NULL, so a NULL band must be named on both sides of the cursor. | `ValueError` for a table column declared `nullable` without it |
95
+ | A timestamp column is `timestamptz` (`DateTime(timezone=True)`). | A naive value is read in the session's zone, so the same cursor means different instants. | `TypeError` |
96
+ | No float columns. | A float's cursor value is not the value the row holds. Use `numeric`. | `TypeError` |
97
+ | A computed expression passes `value=` to read its key from a row. | The cursor needs the value the row sorted by. | `ValueError` |
98
+ | The statement passed to `apply` has no ORDER BY of its own. | SQLAlchemy appends ORDER BY clauses, so an existing one would sort ahead of the listing's, and pages would follow an order the cursor does not. | `ValueError` from `apply` |
99
+
100
+ The library checks that the last column cannot be NULL. It does not check that it is
101
+ unique: declare your primary key there.
102
+
103
+ A computed column, such as a status band that puts drafts first:
104
+
105
+ ```python
106
+ DRAFT_FIRST = case((message.c.status == "draft", 0), else_=1)
107
+ REVIEW = Listing(
108
+ "review",
109
+ Sort.asc(DRAFT_FIRST, value=lambda row: 0 if row["status"] == "draft" else 1),
110
+ Sort.desc(message.c.drafted_at),
111
+ Sort.desc(message.c.id),
112
+ )
113
+ ```
114
+
115
+ A listing whose direction is a request argument declares one `Listing` per direction, each
116
+ with its own name, so a cursor from one direction is refused by the other.
117
+
118
+ ### The SQL
119
+
120
+ When every column runs one direction and none is nullable, "strictly after" is one row
121
+ comparison, which Postgres reads as an index condition:
122
+
123
+ ```sql
124
+ WHERE (created_at, id) < ($1::TIMESTAMP WITH TIME ZONE, $2::VARCHAR)
125
+ ORDER BY created_at DESC, id DESC
126
+ ```
127
+
128
+ Otherwise it is the explicit expansion: after on the first column, or equal on it and after
129
+ on the next, and so on, with each NULL band named. Bound values carry their column's type,
130
+ so asyncpg gets a `datetime` for a `timestamptz`.
131
+
132
+ ## The cursor
133
+
134
+ `base64url(JSON [listing, *key])`, unpadded. Timestamps are ISO-8601 with their zone,
135
+ `numeric` values are decimal strings (a JSON number would round-trip through a float), and
136
+ ints are checked against their column's width.
137
+
138
+ It is opaque to callers and not signed, because it grants nothing. It names a position, not a
139
+ right and not a filter: the read it resumes runs under the same authorization as any other,
140
+ so a cursor replayed by someone else returns only what they could already read.
141
+
142
+ ### A bad cursor
143
+
144
+ `decode` (and `apply`, which calls it) raises `InvalidCursor` for any cursor that is not a
145
+ position in that listing: malformed, truncated, from another listing, a naive timestamp, a
146
+ value out of its column's range. It is a `ValueError`, and its text is always
147
+ `INVALID_CURSOR_MESSAGE`:
148
+
149
+ > The cursor is not a position in this listing.
150
+
151
+ Map it to your bad-input error and tell the caller the way out: call again without `cursor`.
152
+ The parse failure itself never reaches the caller.
153
+
154
+ ```python
155
+ try:
156
+ contacts, next_cursor = await list_contacts(conn, cursor=cursor, limit=limit)
157
+ except InvalidCursor as exc:
158
+ raise BadRequest(str(exc), next_step="Call again without cursor to start from the top.")
159
+ ```
160
+
161
+ ### Stable across versions
162
+
163
+ A caller holds a cursor across your deploys. The encoding and the SQL a cursor resumes with
164
+ are frozen in this repository's golden tests, and a change to either is a breaking change.
165
+
166
+ ## What a walk guarantees
167
+
168
+ Every row that existed when the walk began, and still matches, comes back exactly once.
169
+
170
+ - A row written after the walk began sorts wherever its key puts it. In a newest-first walk
171
+ that is ahead of the cursor, so the walk does not meet it. Tell the caller on the last page
172
+ to start again without `cursor` to see new rows.
173
+ - A row whose sort key changes mid-walk moves in the order, and can be met twice or not at
174
+ all. Say so on any listing whose key can change, such as a status band.
175
+
176
+ ### The walk test
177
+
178
+ Test every paged read the same way: walk more than two pages to the end, insert and remove
179
+ rows between pages, and assert no gap and no duplicate. Seed the rows in one transaction so
180
+ the id tiebreak is what orders them.
181
+
182
+ ```python
183
+ from pgkeyset.testing import walk
184
+
185
+ walked = await walk(
186
+ lambda cursor: list_contacts(conn, cursor=cursor, limit=3),
187
+ ident=lambda c: c.id,
188
+ expected=seeded_ids, # every row that must be met
189
+ between_pages=insert_one_and_remove_one,
190
+ )
191
+ ```
192
+
193
+ `walk` fails on a row met twice, a row in `expected` never met, a row met that is not in
194
+ `expected`, an empty page that still hands out a cursor, or a walk that does not end.
195
+
196
+ ## Compared with sqlakeyset
197
+
198
+ [sqlakeyset](https://github.com/djrobstep/sqlakeyset) is the established keyset library for
199
+ SQLAlchemy, and it does more: ORM and Core, sync and async, paging backwards, and PostgreSQL,
200
+ MariaDB/MySQL and SQLite. If you need any of those, use it.
201
+
202
+ pgkeyset is narrower and stricter:
203
+
204
+ - sqlakeyset documents that rows with a NULL in their keyset are left out of the results and
205
+ asks you to keep sort columns `NOT NULL`. pgkeyset pages through NULLs, and refuses a
206
+ nullable column that does not say where they sort.
207
+ - sqlakeyset serializes floats into its bookmarks. pgkeyset refuses a float key and carries
208
+ `numeric` values as exact decimal strings.
209
+ - A pgkeyset cursor names its listing, and is refused anywhere else, with one fixed message.
210
+ - pgkeyset ships the walk test.
211
+
212
+ It pages forward only, targets Postgres only, and works on SQLAlchemy Core `select()`
213
+ statements. It builds statements and never runs them; it is tested on async engines with
214
+ asyncpg.
215
+
216
+ ## API
217
+
218
+ | Name | What it is |
219
+ |---|---|
220
+ | `Listing(name, *sorts)` | One paged read: its cursor tag and its total order. |
221
+ | `Listing.apply(stmt, cursor)` | `stmt`, which must be unordered, ordered by the listing and started strictly after `cursor` when one is given. |
222
+ | `Listing.paginate(rows, limit)` | Split a `limit + 1` fetch into `(page, next_cursor)`. |
223
+ | `Listing.after(key)`, `.key(row)`, `.encode(key)`, `.decode(cursor)` | The predicate, a row's key, and the codec, for custom flows such as a change feed's mark. |
224
+ | `Sort.asc(expr, *, nulls=None, value=None)`, `Sort.desc(...)` | One column of the order. |
225
+ | `InvalidCursor`, `INVALID_CURSOR_MESSAGE` | The refusal for a cursor that is not a position, and its fixed text. |
226
+ | `pgkeyset.testing.walk(fetch, *, ident, expected=None, between_pages=None, max_pages=1000)` | The walk test. |
227
+
228
+ ## Development
229
+
230
+ ```sh
231
+ docker run -d --rm -p 127.0.0.1:5432:5432 -e POSTGRES_HOST_AUTH_METHOD=trust postgres:17-alpine
232
+ make verify # ruff format check, ruff lint, ty, pytest
233
+ ```
234
+
235
+ The walk suites need a Postgres superuser at `PGKEYSET_TEST_URL` (default
236
+ `postgresql+asyncpg://postgres@127.0.0.1:5432/postgres`); they create and drop a throwaway
237
+ database. Without one they skip locally and fail under CI. The codec, declaration and golden
238
+ suites need no database.
239
+
240
+ `tests/golden/` holds frozen cursors and SQL. They are not regenerated by the tests. If one
241
+ fails, a cursor someone holds would decode differently or resume somewhere else.
242
+
243
+ ## Releasing
244
+
245
+ 1. In a pull request, set `version` in `pyproject.toml` and date its `CHANGELOG.md` entry.
246
+ 2. After it merges, tag the merge commit `v<version>` and push the tag.
247
+ 3. `release.yml` builds once, installs the built wheel into a clean environment on each
248
+ supported Python, runs the test suite the sdist ships against a real Postgres, and only then
249
+ publishes those same files to PyPI. A version can never be uploaded twice, so a red run
250
+ publishes nothing; fix it and release the next patch version.
251
+
252
+ Publishing uses trusted publishing (OIDC); there is no API token to rotate or leak.
253
+
254
+ ## License
255
+
256
+ Apache-2.0. See [LICENSE](LICENSE) and [NOTICE](NOTICE). Security reports: [SECURITY.md](SECURITY.md).
@@ -0,0 +1,9 @@
1
+ pgkeyset/__init__.py,sha256=5g9IqkMhJbIx6lNpWWWIuHNlo_3Ww7t8Le0jcc3z7YU,862
2
+ pgkeyset/core.py,sha256=SZSY6ie39lYdE-WOIObFzIDZ5cYgHS5lzKKEyRxkLgQ,14155
3
+ pgkeyset/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
4
+ pgkeyset/testing.py,sha256=qMmEo-9ZPeOLa0u4Sw6QbVxN4FJoX40j1DCdX7JroQE,2267
5
+ pgkeyset-0.1.0.dist-info/METADATA,sha256=Z7nvbOyP7AkNolwTFMvZNyi_1_4agq-BmSGHvsGyoik,12406
6
+ pgkeyset-0.1.0.dist-info/WHEEL,sha256=2iZrJkDIkeDh0k9SikKMqFuRP-ua6mb4Ir1AsER4BEw,87
7
+ pgkeyset-0.1.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
8
+ pgkeyset-0.1.0.dist-info/licenses/NOTICE,sha256=v2lmYs0_CJCfhxUPemY8tpWPR1OpG_LgVr6W8aAb7yk,105
9
+ pgkeyset-0.1.0.dist-info/RECORD,,
@@ -0,0 +1,4 @@
1
+ Wheel-Version: 1.0
2
+ Generator: hatchling 1.33.0
3
+ Root-Is-Purelib: true
4
+ Tag: py3-none-any
@@ -0,0 +1,201 @@
1
+ Apache License
2
+ Version 2.0, January 2004
3
+ http://www.apache.org/licenses/
4
+
5
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
6
+
7
+ 1. Definitions.
8
+
9
+ "License" shall mean the terms and conditions for use, reproduction,
10
+ and distribution as defined by Sections 1 through 9 of this document.
11
+
12
+ "Licensor" shall mean the copyright owner or entity authorized by
13
+ the copyright owner that is granting the License.
14
+
15
+ "Legal Entity" shall mean the union of the acting entity and all
16
+ other entities that control, are controlled by, or are under common
17
+ control with that entity. For the purposes of this definition,
18
+ "control" means (i) the power, direct or indirect, to cause the
19
+ direction or management of such entity, whether by contract or
20
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
21
+ outstanding shares, or (iii) beneficial ownership of such entity.
22
+
23
+ "You" (or "Your") shall mean an individual or Legal Entity
24
+ exercising permissions granted by this License.
25
+
26
+ "Source" form shall mean the preferred form for making modifications,
27
+ including but not limited to software source code, documentation
28
+ source, and configuration files.
29
+
30
+ "Object" form shall mean any form resulting from mechanical
31
+ transformation or translation of a Source form, including but
32
+ not limited to compiled object code, generated documentation,
33
+ and conversions to other media types.
34
+
35
+ "Work" shall mean the work of authorship, whether in Source or
36
+ Object form, made available under the License, as indicated by a
37
+ copyright notice that is included in or attached to the work
38
+ (an example is provided in the Appendix below).
39
+
40
+ "Derivative Works" shall mean any work, whether in Source or Object
41
+ form, that is based on (or derived from) the Work and for which the
42
+ editorial revisions, annotations, elaborations, or other modifications
43
+ represent, as a whole, an original work of authorship. For the purposes
44
+ of this License, Derivative Works shall not include works that remain
45
+ separable from, or merely link (or bind by name) to the interfaces of,
46
+ the Work and Derivative Works thereof.
47
+
48
+ "Contribution" shall mean any work of authorship, including
49
+ the original version of the Work and any modifications or additions
50
+ to that Work or Derivative Works thereof, that is intentionally
51
+ submitted to Licensor for inclusion in the Work by the copyright owner
52
+ or by an individual or Legal Entity authorized to submit on behalf of
53
+ the copyright owner. For the purposes of this definition, "submitted"
54
+ means any form of electronic, verbal, or written communication sent
55
+ to the Licensor or its representatives, including but not limited to
56
+ communication on electronic mailing lists, source code control systems,
57
+ and issue tracking systems that are managed by, or on behalf of, the
58
+ Licensor for the purpose of discussing and improving the Work, but
59
+ excluding communication that is conspicuously marked or otherwise
60
+ designated in writing by the copyright owner as "Not a Contribution."
61
+
62
+ "Contributor" shall mean Licensor and any individual or Legal Entity
63
+ on behalf of whom a Contribution has been received by Licensor and
64
+ subsequently incorporated within the Work.
65
+
66
+ 2. Grant of Copyright License. Subject to the terms and conditions of
67
+ this License, each Contributor hereby grants to You a perpetual,
68
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
69
+ copyright license to reproduce, prepare Derivative Works of,
70
+ publicly display, publicly perform, sublicense, and distribute the
71
+ Work and such Derivative Works in Source or Object form.
72
+
73
+ 3. Grant of Patent License. Subject to the terms and conditions of
74
+ this License, each Contributor hereby grants to You a perpetual,
75
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
76
+ (except as stated in this section) patent license to make, have made,
77
+ use, offer to sell, sell, import, and otherwise transfer the Work,
78
+ where such license applies only to those patent claims licensable
79
+ by such Contributor that are necessarily infringed by their
80
+ Contribution(s) alone or by combination of their Contribution(s)
81
+ with the Work to which such Contribution(s) was submitted. If You
82
+ institute patent litigation against any entity (including a
83
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
84
+ or a Contribution incorporated within the Work constitutes direct
85
+ or contributory patent infringement, then any patent licenses
86
+ granted to You under this License for that Work shall terminate
87
+ as of the date such litigation is filed.
88
+
89
+ 4. Redistribution. You may reproduce and distribute copies of the
90
+ Work or Derivative Works thereof in any medium, with or without
91
+ modifications, and in Source or Object form, provided that You
92
+ meet the following conditions:
93
+
94
+ (a) You must give any other recipients of the Work or
95
+ Derivative Works a copy of this License; and
96
+
97
+ (b) You must cause any modified files to carry prominent notices
98
+ stating that You changed the files; and
99
+
100
+ (c) You must retain, in the Source form of any Derivative Works
101
+ that You distribute, all copyright, patent, trademark, and
102
+ attribution notices from the Source form of the Work,
103
+ excluding those notices that do not pertain to any part of
104
+ the Derivative Works; and
105
+
106
+ (d) If the Work includes a "NOTICE" text file as part of its
107
+ distribution, then any Derivative Works that You distribute must
108
+ include a readable copy of the attribution notices contained
109
+ within such NOTICE file, excluding those notices that do not
110
+ pertain to any part of the Derivative Works, in at least one
111
+ of the following places: within a NOTICE text file distributed
112
+ as part of the Derivative Works; within the Source form or
113
+ documentation, if provided along with the Derivative Works; or,
114
+ within a display generated by the Derivative Works, if and
115
+ wherever such third-party notices normally appear. The contents
116
+ of the NOTICE file are for informational purposes only and
117
+ do not modify the License. You may add Your own attribution
118
+ notices within Derivative Works that You distribute, alongside
119
+ or as an addendum to the NOTICE text from the Work, provided
120
+ that such additional attribution notices cannot be construed
121
+ as modifying the License.
122
+
123
+ You may add Your own copyright statement to Your modifications and
124
+ may provide additional or different license terms and conditions
125
+ for use, reproduction, or distribution of Your modifications, or
126
+ for any such Derivative Works as a whole, provided Your use,
127
+ reproduction, and distribution of the Work otherwise complies with
128
+ the conditions stated in this License.
129
+
130
+ 5. Submission of Contributions. Unless You explicitly state otherwise,
131
+ any Contribution intentionally submitted for inclusion in the Work
132
+ by You to the Licensor shall be under the terms and conditions of
133
+ this License, without any additional terms or conditions.
134
+ Notwithstanding the above, nothing herein shall supersede or modify
135
+ the terms of any separate license agreement you may have executed
136
+ with Licensor regarding such Contributions.
137
+
138
+ 6. Trademarks. This License does not grant permission to use the trade
139
+ names, trademarks, service marks, or product names of the Licensor,
140
+ except as required for reasonable and customary use in describing the
141
+ origin of the Work and reproducing the content of the NOTICE file.
142
+
143
+ 7. Disclaimer of Warranty. Unless required by applicable law or
144
+ agreed to in writing, Licensor provides the Work (and each
145
+ Contributor provides its Contributions) on an "AS IS" BASIS,
146
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
147
+ implied, including, without limitation, any warranties or conditions
148
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
149
+ PARTICULAR PURPOSE. You are solely responsible for determining the
150
+ appropriateness of using or redistributing the Work and assume any
151
+ risks associated with Your exercise of permissions under this License.
152
+
153
+ 8. Limitation of Liability. In no event and under no legal theory,
154
+ whether in tort (including negligence), contract, or otherwise,
155
+ unless required by applicable law (such as deliberate and grossly
156
+ negligent acts) or agreed to in writing, shall any Contributor be
157
+ liable to You for damages, including any direct, indirect, special,
158
+ incidental, or consequential damages of any character arising as a
159
+ result of this License or out of the use or inability to use the
160
+ Work (including but not limited to damages for loss of goodwill,
161
+ work stoppage, computer failure or malfunction, or any and all
162
+ other commercial damages or losses), even if such Contributor
163
+ has been advised of the possibility of such damages.
164
+
165
+ 9. Accepting Warranty or Additional Liability. While redistributing
166
+ the Work or Derivative Works thereof, You may choose to offer,
167
+ and charge a fee for, acceptance of support, warranty, indemnity,
168
+ or other liability obligations and/or rights consistent with this
169
+ License. However, in accepting such obligations, You may act only
170
+ on Your own behalf and on Your sole responsibility, not on behalf
171
+ of any other Contributor, and only if You agree to indemnify,
172
+ defend, and hold each Contributor harmless for any liability
173
+ incurred by, or claims asserted against, such Contributor by reason
174
+ of your accepting any such warranty or additional liability.
175
+
176
+ END OF TERMS AND CONDITIONS
177
+
178
+ APPENDIX: How to apply the Apache License to your work.
179
+
180
+ To apply the Apache License to your work, attach the following
181
+ boilerplate notice, with the fields enclosed by brackets "[]"
182
+ replaced with your own identifying information. (Don't include
183
+ the brackets!) The text should be enclosed in the appropriate
184
+ comment syntax for the file format. We also recommend that a
185
+ file or class name and description of purpose be included on the
186
+ same "printed page" as the copyright notice for easier
187
+ identification within third-party archives.
188
+
189
+ Copyright [yyyy] [name of copyright owner]
190
+
191
+ Licensed under the Apache License, Version 2.0 (the "License");
192
+ you may not use this file except in compliance with the License.
193
+ You may obtain a copy of the License at
194
+
195
+ http://www.apache.org/licenses/LICENSE-2.0
196
+
197
+ Unless required by applicable law or agreed to in writing, software
198
+ distributed under the License is distributed on an "AS IS" BASIS,
199
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
200
+ See the License for the specific language governing permissions and
201
+ limitations under the License.
@@ -0,0 +1,4 @@
1
+ pgkeyset
2
+ Copyright 2026 NimbleBrain, Inc.
3
+
4
+ This product includes software developed at NimbleBrain, Inc.