o wi@ @sdZddlmZddlmZddlmZmZmZm Z ddl m Z ddl m Z e dZ Gd d d e ZGd d d e eeZGd dde eeZGdddeeeeZdS)zUtilities for client-side handling of "relay-style" GraphQL pagination. For formal specs and definitions, see https://relay.dev/graphql/connections.htm. ) annotations)Iterator)GenericLiteralOptionalTypeVar)NonNegativeInt) GQLResultNodeTc@s0eZdZUdZdZded<ded< ded<dS) PageInfozHPagination metadata returned by the server for a single page of results.zLiteral['PageInfo'] typename__ Optional[str] end_cursorbool has_next_pageN)__name__ __module__ __qualname____doc__r __annotations__rrN/home/ubuntu/.local/lib/python3.10/site-packages/wandb/_pydantic/pagination.pyr s  r c@eZdZUdZded<dS)Edgea>A wrapper around a single result item in a paginated response. In relay-style pagination, individual items are wrapped in "edges" which can carry additional metadata, e.g., per-item cursors. This base implementation only exposes the `node` (the actual result item, like a GraphQL `Run` or `Project`). r nodeNrrrrrrrrrrs rc@sZeZdZUdZded< ded< dZded< dd d ZedddZedddZ dS) Connectiona6A page of results from the response of a paginated GraphQL query. This follows the "Relay Connection" specification, which is a standard way to paginate large result sets in GraphQL. Instead of returning all results at once, the server returns one page at a time. Each "page" is represented by a `Connection` object that includes: - A list of `edges`, each wrapping a single result item (`node`). - A `page_info` object with metadata for fetching subsequent pages. - Optionally, a `total_count` of all results (not just this page). zlist[Edge[NodeT]]edgesr page_infoNzOptional[NonNegativeInt] total_countreturnIterator[NodeT]csfdd|jDS)z5Returns an iterator over the nodes in the connection.c3s|] }|jrVqdS)Nr).0edger#rr Csz#Connection.nodes..)rselfrr#rnodesAszConnection.nodesrcC|jjS)z.Returns True if there are more pages to fetch.)rrr'rrrhas_nextEzConnection.has_nextrcCr*)zEThe cursor value to pass as the `after` arg in the next page request.)rrr'rrr next_cursorJr,zConnection.next_cursor)r!r")r!r)r!r) rrrrrr r)propertyr+r-rrrrr+s     rc@r)ConnectionWithTotala+A `Connection` where the `totalCount` field must be present. Use this INSTEAD of `Connection` when the paginated query is expected to return a finite `totalCount` field, i.e. when `totalCount` is: - explicitly requested in the GraphQL query - non-nullable in the GraphQL schema rr Nrrrrrr/Ps r/N)r __future__rcollections.abcrtypingrrrrpydanticrbaser r r rrr/rrrrs      %