+j!$dZddlmZddlZddlZddlmZddlmZmZddl m Z m Z Gdde Z d Zd Zd Zed GddZdZd-d.dZd-d/dZ d0d1d!Zd2d$Zd3d&Zed'd(d(ej)Zd4d*Zd5d,ZdS)6uOpaque keyset-pagination cursor for /api/assets. Payload JSON uses short keys to keep the encoded length small: {"s": , "v": , "id": , "o": } The `o` key binds the cursor to the sort direction it was minted under, so replaying a `desc` cursor against an `asc` request fails with ``INVALID_CURSOR`` rather than silently walking the wrong direction. `o` is mandatory on every payload — a cursor without it is rejected as malformed. Encoding is base64url with no padding. Cursors are opaque tokens: the payload format is internal to this server, and clients must treat a cursor as a black box handed back via `next_cursor`. No byte-level compatibility with any other implementation is required. Time values are serialized as Unix microseconds (UTC) — microsecond precision is sufficient to round-trip the timestamps stored by the database without rounding rows in the same millisecond bucket. ) annotationsN) dataclass)datetimetimezone)IterableOptionalceZdZdZdS)InvalidCursorErrorzRaised on a malformed, oversized, or unsupported-sort-field cursor. Map to a 400 response with code ``INVALID_CURSOR`` at the handler. N)__name__ __module__ __qualname____doc__:/home/wildlama/comfy/ComfyUI/app/assets/services/cursor.pyr r srr i iT)frozenc8eZdZUded<ded<ded<ded<dS) CursorPayloadstr sort_fieldvalueidorderN)r r r __annotations__rrrrr:s4OOOJJJ GGGJJJJJrr)ascdescrrrrrrreturnc|tvrtdtd||stdt|tkrtdt|tkrtd||||d}t j|dd }tj| d  d  d S) a Encode a cursor payload as a base64url (no-padding) string. `order` binds the cursor to the sort direction it was minted under so a later request with a flipped `order` query parameter is rejected with ``INVALID_CURSOR`` rather than silently walking the wrong direction. zorder must be one of z, got zid must be non-emptyid exceeds maximum lengthvalue exceeds maximum length)svro),:F) separators ensure_asciizutf-8=ascii) _VALID_ORDERSr lenMAX_CURSOR_ID_LENGTHMAX_CURSOR_VALUE_LENGTHjsondumpsbase64urlsafe_b64encodeencoderstripdecode)rrrrpayloadraws r encode_cursorr8Es M!! !W!W!We!W!WXXX 9 !7888 2ww%%% !<=== 5zz+++ !?@@@U"5AAG *W% H H HC  #CJJw$7$7 8 8 ? ? E E L LW U UUrtrc|jtdt|tj}t |t|||S)aEncode a time-typed cursor at Unix microsecond precision. Accepts an aware datetime (any timezone) and normalizes to UTC. Naive datetimes are rejected so callers can't accidentally encode the local wall-clock value of a UTC-stored timestamp. Nz2encode_cursor_from_time requires an aware datetime)r)tzinfo ValueError_datetime_to_unix_micros astimezonerutcr8r)rr9rrmicross rencode_cursor_from_timerA_sS xMNNN %all8<&@&@ A AF S[["E B B BBrcursorallowed_sort_fields Iterable[str]expected_order str | Nonect|tkrtd dt| dzz}tj||z}n6#t tjjf$r}td||d}~wwxYw tj |}n1#tj tf$r}td||d}~wwxYwt|tstd|d}|d }|d } |d } t|tr*t|trt| tstd | d krtdt| t krtdt|t"krtd||vrtd|t| tstd| t$vrtd| || |krtd| d|t'||| | S)u0Parse an opaque cursor. ``allowed_sort_fields`` is the endpoint's accepted sort-field list — a cursor carrying a field outside this set is rejected so a cursor minted for one column can't be replayed against another (e.g. a ``created_at`` timestamp string compared against a ``name`` column). ``expected_order`` (``"asc"``/``"desc"``), when supplied, must match the payload's ``o`` field. ``o`` is required on every payload; a cursor missing it is rejected as malformed. Passing no allowed fields rejects every cursor. zcursor exceeds maximum length=z encoding: Nz payload: zpayload: expected objectr"r#rr$z%payload: missing or non-string s/v/idz missing idr r!zunsupported sort field zmissing or non-string ozunsupported order z cursor order z does not match request order )rrrr)r,MAX_ENCODED_CURSOR_LENGTHr r1urlsafe_b64decoder<binasciiErrorr/loadsJSONDecodeErrorUnicodeDecodeError isinstancedictgetrr-r.r+r) rBrCrEpaddingr7edecodedrrrrs r decode_cursorrXls$ 6{{... !@AAA:#f++)*&v'788 - .::: !1a!1!1229:9*S//  "4 5999 Q11q89 gt $ $= !;<<<S!!J KK  E T  B KK  E j# & &Jj.D.DJJWY[^L_L_J !HIII Rxx ... 2ww%%% !<=== 5zz+++ !?@@@,,, !I:!I!IJJJ eS ! !< !:;;; M!! !?e!?!?@@@!e~&=&= UE U U> U U    Je% P P PPs/-AB 2BB B##C9C  Cr6Optional[CursorPayload]c|td t|j}n%#t$r}td||d}~wwxYw t |S#t t tf$r}td||d}~wwxYw)zDParse a time-typed cursor value as Unix microseconds, returning UTC.Nnil cursor payloadz value is not a valid timestamp: z%value is out of representable range: )r intrr<_unix_micros_to_datetime OverflowErrorOSError)r6r@rVs rdecode_cursor_timer`s !5666PW]## PPP !GA!G!GHHaOPU'/// 7J /UUU!!L!L!LMMSTTUs,( A AA AB 4BB r\c|td t|jS#t$r}td||d}~wwxYw)z*Parse a cursor value as a base-10 integer.Nr[zvalue is not a valid integer: )r r\rr<)r6rVs rdecode_cursor_intrbsh !5666N7=!!! NNN !E!!E!EFFAMNs' A AA i)r;cP|tz }|jdz|jzdz|jzS)zBConvert an aware UTC datetime to Unix microseconds (integer math).iQ@B)_EPOCHdaysseconds microseconds)r9deltas rr=r=s, JE J %- /9 rus*#""""" !!!!!!''''''''%%%%%%%%,! $ VVVVV4 C C C C C "&@Q@Q@Q@Q@QF U U U U NNNN $1X\ 2 2 2RRRR aaaaaar