kQj<dZddlmZddlZddlmZmZmZmZm Z m Z m Z ddl m Z e dZGdd eeeeZGd d e eeeZdd ZGd deeZGddeZGddeZGdde eZGddeZGddeZy)z"Lazy, resumable resource listings.) annotationsN)Any AsyncIteratorCallableGenericIteratorOptionalTypeVar)LOOPTc>eZdZdZdfd dZd dZd dZd dZd dZy) AsyncListinga(A lazy async iterator with inspectable pagination state. Returned by connection resource-listing methods. For example:: async for name in connection.list_views(page_limit=20): print(name) Requests happen during iteration; request errors terminate the iterator. c|SN)values T/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/lancedb/listing.pyzAsyncListing.suc ||_||_yr)_inner_convert)selfinnerconverts r__init__zAsyncListing.__init__s  rc|Srrrs r __aiter__zAsyncListing.__aiter__ rcpK|j|jjd{S7wr)rr __anext__rs rr#zAsyncListing.__anext__"s(}}4;;#8#8#::;;:s )64 6c6|jjSzDReturn the number of cached items available without another request.rnum_page_resultsrs rr'zAsyncListing.num_page_results%{{++--rc6|jjS)aReturn the token for the next request. Initially this is the supplied starting token (``None`` starts at the beginning). After fetching the final page it is ``None``, even if items remain cached. Drain the cache before saving a token to avoid skipping items when resuming. A failed request terminates iteration and retains its token for retrying with a new iterator. r page_tokenrs rr+zAsyncListing.page_token)s{{%%''rN)rrrzCallable[[Any], T])returnAsyncListing[T]r,r r,intr, Optional[str]) __name__ __module__ __qualname____doc__rr r#r'r+rrrrrs'BU <. (rrc8eZdZdZddZd dZd dZd dZd dZy) ListingzA lazy synchronous iterator with inspectable pagination state. For example:: for name in connection.list_views(page_limit=20): print(name) Use ``list(connection.list_views())`` to collect all results. c||_yr)rrrs rrzListing.__init__@s  rc|Srrrs r__iter__zListing.__iter__Cr!rc tj|jjS#t$rt dwxYwrr runrr#StopAsyncIteration StopIterationrs r__next__zListing.__next__F: *88DKK1134 4! *T ) * ,/Ac6|jjSr%r&rs rr'zListing.num_page_resultsLr(rc6|jjS)zReturn the next request's token. See [AsyncListing.page_token][lancedb.listing.AsyncListing.page_token] for initial, cached, final-page, and error behavior. r*rs rr+zListing.page_tokenPs {{%%''rN)rr-)r,z Listing[T]r.r/r1) r3r4r5r6rr<rBr'r+rrrr8r85s * .(rr8cT|&t|tr |dks|dkDr tdyy)zEValidate the legacy limit argument before passing it to the bindings.Nriz&limit must be between 0 and 2147483647) isinstancer0 ValueError)limits rvalidate_page_limitrKYs6  uc "eai593DABB4ErcpeZdZdZ ddd ddZeddZddZddZdd Z d Z d Z d Z dd Z y)_AsyncPagedNameListingzFAn async listing with an inspectable copy of its current backend page.NF)emptyc||_||_|rgnd|_d|_||_d|_t j|_y)Nr) r_token_page _position_done_failureasyncioLock_lock)rrr+rNs rrz_AsyncPagedNameListing.__init__ds?   05"4  15 \\^ rc|jS)z4The continuation token for the next backend request.)rPrs rr+z!_AsyncPagedNameListing.page_tokenos{{rcb|jdSt|j|jz S)z.Count the remaining names in the current page.r)rQlenrRrs rr'z'_AsyncPagedNameListing.num_page_resultsts'JJ&qLC Odnn,LLrcd|j td|jjS)Nz4Await or iterate the listing before reading its page)rQ RuntimeErrorcopyrs r _page_copyz!_AsyncPagedNameListing._page_copyxs) :: UV Vzz  rcVK |jjd{}|g|_d|_d|_d|_ y|d|_|d|_|jdu|_d|_ y7R#tj$rt$r}||_d|_d}~wwxYww)NTitemsr+r) r next_pagerUCancelledError BaseExceptionrTrSrQrPrR)rpageexcs r _fetch_pagez"_AsyncPagedNameListing._fetch_page}s ..00D <DJDKDJ gDJ|,DK,DJ1%%   DMDJ  s8B)A6A4A6AB)4A66B&B!!B&&B)cK|j4d{|j |j|j|jd{dddd{|S7R77 #1d{7swY|SxYwwr)rWrTrQrfrs r _load_initialz$_AsyncPagedNameListing._load_initials:: ) )}}(mm#zz!&&(((  ) )   ))  ) ) ) )  sVBA&B8A,A(A, BA* B(A,*B,A?2A5 3A?:Bc>|jjSr)rh __await__rs rrjz _AsyncPagedNameListing.__await__s!!#--//rc|Srrrs rr z _AsyncPagedNameListing.__aiter__r!rctK|j4d{|jt|j|j d{|jJ|j t |jk\r[|jrt|j d{|jJ|j t |jk\r[|j|j }|xj dz c_|cdddd{S777{7 #1d{7swYyxYww)Nr )rWrTr@rQrfrRrZrS)ritems rr#z _AsyncPagedNameListing.__anext__s::  }}(((zz!&&(((::) ))..C O3::,,&&(((zz--- ..C O3 ::dnn-D NNa N   ) )    soD8DD82D# D AD##D$4D#/D# D8D!D8D#D#!D8#D5)D, *D51D8r)rrr+r2rNboolr1r/r,z list[str])r,Noner,str)r3r4r5r6rpropertyr+r'r^rfrhrjr r#rrrrMrMasgP7; $MR $ $&3 $FJ $M! &0rrMc"eZdZdZeddZy)AsyncTableListingzAwaitable table iterator; awaiting loads one page. For example:: page = await db.list_tables(limit=20) print(page.tables, page.page_token) async for name in db.list_tables(limit=20): print(name) c"|jS)z=A copy of every table name in the most recently fetched page.r^rs rtableszAsyncTableListing.tables  rNror3r4r5r6rsrxrrrrurus!!rruc"eZdZdZeddZy)AsyncNamespaceListingzAwaitable namespace iterator; awaiting loads one page. For example:: page = await db.list_namespaces(limit=20) print(page.namespaces, page.page_token) c"|jS)z7A copy of every name in the most recently fetched page.rwrs r namespacesz AsyncNamespaceListing.namespacesryrNror3r4r5r6rsr~rrrr|r|s!!rr|c<eZdZddZdZddZed dZd dZy) _PagedNameListingcX||_tj|jyr)rr r?rhr:s rrz_PagedNameListing.__init__s  $$&'rc|Srrrs rr<z_PagedNameListing.__iter__r!rc tj|jjS#t$rt dwxYwrr>rs rrBz_PagedNameListing.__next__rCrDc.|jjSrr*rs rr+z_PagedNameListing.page_token{{%%%rc6|jjSrr&rs rr'z"_PagedNameListing.num_page_resultss{{++--rN)rrMrqr1r/) r3r4r5rr<rBrsr+r'rrrrrs*(* &&.rrc"eZdZdZeddZy) TableListingzTable iterator retaining the current page as a list. For example:: page = db.list_tables(limit=20) print(page.tables, page.page_token) for name in db.list_tables(limit=20): print(name) c.|jjSr)rrxrs rrxzTableListing.tabless{{!!!rNrorzrrrrrs""rrc"eZdZdZeddZy)NamespaceListingzNamespace iterator retaining the current page as a list. For example:: page = db.list_namespaces(limit=20) print(page.namespaces, page.page_token) c.|jjSr)rr~rs rr~zNamespaceListing.namespacesrrNrorrrrrrs&&rr)rJz Optional[int]r,rp)r6 __future__rrUtypingrrrrrr r background_loopr r rr8rKrrrMrur|rrrrrrrs)"UUU! CL"(=#WQZ"(J!(hqk71:!(HCK]3/K\!.!" !2 !. ., "$ " &( &r