kQj+)dZddlmZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZddlmZmZdd lmZmZdd lmZmZdd lmZGd d eeZGddeeZGddZGddZdddddddede ede e eeeeffde ede ede edefdZ dddddddede ede e eeeeffde ede ede edefdZ!y)zCRemote catalogs manage databases through a server's root namespace.)cached_property) timedelta)Any AsyncIteratorIteratorOptionalUnion)_lancedb)LOOP)AsyncAuthorization Authorization)AsyncConnection DBConnection) ClientConfig OAuthConfig)RemoteDBConnectioncbeZdZdZdej fdZd dZdefdZ de fdZ de efdZ y ) AsyncDatabaseNameszAn async iterator of database names with pagination state. Returned by [list_databases][lancedb.catalog.AsyncCatalog.list_databases]. innerc||_yN_innerselfrs T/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/lancedb/catalog.py__init__zAsyncDatabaseNames.__init__  returnc|Srrs r __aiter__zAsyncDatabaseNames.__aiter__ r cRK|jjd{S7wr)r __anext__r$s rr(zAsyncDatabaseNames.__anext__s[[**,,,,s '%'c6|jjSzBReturn the number of names available without another REST request.rnum_page_resultsr$s rr,z#AsyncDatabaseNames.num_page_results!{{++--r c6|jjS)aReturn the token for the next REST request. Before the first request, this is the supplied starting token (``None`` starts at the beginning). After the final page is fetched, it is ``None``, even if names remain cached. Drain the cache before saving a token to avoid skipping those names when resuming. A failed request terminates iteration and leaves its token available for resuming a new iterator. r page_tokenr$s rr0zAsyncDatabaseNames.page_token%s{{%%''r N)r!r)__name__ __module__ __qualname____doc__r DatabaseNamesrr%strr(intr,rr0r#r rrrsF h44--.#. (HSM (r rcNeZdZdZdefdZd dZdefdZde fdZ de efdZ y ) r5zA synchronous iterator of database names with pagination state. Returned by [Catalog.list_databases][lancedb.catalog.Catalog.list_databases]. rc||_yrrrs rrzDatabaseNames.__init__7rr r!c|Srr#r$s r__iter__zDatabaseNames.__iter__:r&r c tj|jjS#t$rt dwxYwr)r runrr(StopAsyncIteration StopIterationr$s r__next__zDatabaseNames.__next__=s: *88DKK1134 4! *T ) *s ,/Ac6|jjSr*r+r$s rr,zDatabaseNames.num_page_resultsCr-r c6|jjS)zReturn the token for the next REST request. See [page_token][lancedb.catalog.AsyncDatabaseNames.page_token] for initial, cached, final-page, and error behavior. r/r$s rr0zDatabaseNames.page_tokenGs {{%%''r N)r!r5) r1r2r3r4rrr;r6r@r7r,rr0r#r rr5r51sA 0*#* .#.(HSM(r r5c eZdZdZdej fdZedefdZ e de fdZ ddd e d e defd Zd e defd Zd d ddee deedefdZdddd e de de dd fdZy ) AsyncCatalogaAn asynchronous remote catalog returned by [connect_catalog_async][lancedb.connect_catalog_async]. Create/connect return ordinary [AsyncConnection][lancedb.db.AsyncConnection] instances. Drop uses restricted behavior: remove the database's tables first. rc||_yrrrs rrzAsyncCatalog.__init__Xrr r!c@t|jjSz;Authorization sharing this catalog's scope and credentials.)r rauthzr$s rrHzAsyncCatalog.authz[s"$++"3"344r c.|jjSz&The catalog's root namespace endpoint.rurir$s rrLzAsyncCatalog.uri`{{r Fexist_oknamerOcjKt|jj||d{S7w)BCreate a database, or open an existing one when ``exist_ok=True``.rNN)rrcreate_database)rrPrOs rrSzAsyncCatalog.create_databasees2++--dX-F F  Fs &31 3cfKt|jj|d{S7w)4Connect to an existing database by its logical name.N)rrconnect_databaserrPs rrVzAsyncCatalog.connect_databasems&T[[%A%A$%GGHHGs $1/ 1Nr0 page_limitr0rYcNt|jj||S)aIterate lazily over all database names using ``async for``. ``page_token`` resumes from a saved token; ``None`` starts at the beginning. ``page_limit`` sets the maximum names per REST response (not the total); ``None`` uses the server default. Request errors are raised during iteration. Examples -------- ```python async for name in catalog.list_databases(): print(name) ``` rX)rrlist_databasesrr0rYs rr[zAsyncCatalog.list_databasesqs( " KK & &* & T  r ignore_missingcascader^r_c\K|jj|||d{y7wzDrop a database. A nonempty database is an error unless ``cascade`` is set, which drops everything in it first; the name stays taken until the server finishes cleaning up their storage.r]N)r drop_databaserrPr^r_s rrbzAsyncCatalog.drop_databases0 kk'' (   s ",*,)r1r2r3r4r Catalogrrr rHpropertyr6rLboolrrSrVrr7rr[rbr#r rrDrDPsh..5)55S .3  &*  I3I?I .2t %c] ?G}  *49%  ,0 CG  r rDc eZdZdZddddddedeedeefdZe de fd Z e defd Z d d d ededefdZd edefdZd ededefdZddddeedeedefdZd d dd edededdfdZy)rda^A synchronous remote catalog returned by [connect_catalog][lancedb.connect_catalog]. Examples -------- ```python catalog = lancedb.connect_catalog("https://my-server.example", api_key="secret") db = catalog.create_database("analytics", exist_ok=True) for name in catalog.list_databases(): print(name) ``` Napi_key client_configsql_host_override oauth_configrrkrlcJ||_||_||_||_||_yr)r_api_key_client_config_sql_host_override _oauth_config)rrrirjrkrls rrzCatalog.__init__s*  +"3)r r!c@t|jjSrG)rrrHr$s rrHz Catalog.authzsT[[..//r c.|jjSrJrKr$s rrLz Catalog.urirMr FrNrPrOctj|jj||}|j ||S)rRrN)r r=rrS_wrap_database)rrPrOrs rrSzCatalog.create_databases744TH4MN""4//r c~|j|tj|jj |S)rU)rur r=rrVrWs rrVzCatalog.connect_databases-""4$++2N2Nt2T)UVVr c tj|||j|j|j|j |j Sr)r _from_catalogrLrnrorqrp)rrPrs rruzCatalog._wrap_databasesD!//   HH MM        # #  r rXr0rYcNt|jj||S)aJIterate lazily over database names, fetching more results as needed. ``page_token`` resumes from a saved token; ``None`` starts at the beginning. ``page_limit`` sets the maximum names per REST response (not the total); ``None`` uses the server default. Request errors are raised during iteration. rX)r5rr[r\s rr[zCatalog.list_databasess( KK & &* & T  r r]r^r_cftj|jj|||yra)r r=rrbrcs rrbzCatalog.drop_databases.  KK % %^W &  r )r1r2r3r4rDrr6rrrrrHrerLrfrrSrVrrur7r5r[rbr#r rrdrds "+/.2 * * $C= *{+ *0}00S>C0C0d0|0 WSW\W  3    <  .2t  %c]  ?G}    49%    ,0  CG    r rdNrirjrkread_consistency_intervalrlendpointrirjrkr|rlr!cKt|tr tdi|}| t}tj||||||j nd|d{}t |S7w)aConnect to an HTTP(S) server's root catalog. Root requests omit database-selection headers. API key, client configuration, OAuth, and table read consistency settings are inherited by opened databases. Database names containing slashes remain single logical names. Set ``sql_host_override`` to the SQL service endpoint to execute SQL through returned connections when the catalog endpoint uses HTTPS. Nr{r#) isinstancedictrr connect_catalog total_secondsrD)r}rirjrkr|rlrs rconnect_catalog_asyncrs{"-&$5}5 $ **#+)4 & 3 3 5!  E   sAA,A*A,c fttjt||||||||||S)zConnect synchronously to an HTTP(S) server's root catalog. See [connect_catalog_async][lancedb.connect_catalog_async] for options. Local filesystem and object-store catalogs are not supported. r{rh)rdr r=r)r}rirjrkr|rls rrr sE   !+"3*C)   #+! r )"r4 functoolsrdatetimertypingrrrrr r background_loopr rHr rdbrrremoterr remote.dbrr6rr5rDrdrrrr#r rrsiJ%@@!4--)(s+(>(HSM(>= = @R R p"CG'+59*.!!c]!E,S#X">?@ !  } ! ( 2 !;'!!N"CG'+59*.c]E,S#X">?@   }  ( 2 ;' r