kQjo$ddZddlmZddlmZmZmZmZmZm Z m Z ddl Z ddl m Z ddlmZddlmZmZmZerddlZdd lmZdd lmZmZd Ze eee ee eeffeeefdfZeZd Z e Gd dZ ddZ!dddZ"ddZ#ddZ$ddZ%ddZ&d dZ'GddZ(GddZ)y)!a~Materialized views: tables defined by a query over a source table and maintained by refresh. See ``DBConnection.create_materialized_view``. Local creation and refresh require the ``experimental-functions`` Cargo feature in a source build. Published packages disable it. Opening, reading, listing and dropping existing views and remote Cloud/Enterprise operations remain available. ) annotations) TYPE_CHECKINGDictListOptionalSequenceTupleUnionN) dataclass)LOOP)AsyncJobJob _typed_jobRefreshMaterializedViewResult) AsyncTable LanceTables mv.definitionceZdZUdZded<y)MaterializedViewDefinitiona)The query that defines a materialized view, as stored:: SELECT columns FROM [ns.]table [, function(args) AS alias | , UNNEST(column) AS alias] [WHERE predicate] [GROUP BY expr, ...] [LIMIT n] A Function in ``FROM`` position yields one row per element it returns. strqueryN)__name__ __module__ __qualname____doc____annotations__^/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/lancedb/materialized_view.pyrr/s JJr rc|jxsi}|jt}|td|dt t j ||S)NzTable 'z' is not a materialized view)metadatagetDEFINITION_META_KEY ValueError_definition_from_valuejsonloads)schemanamer#raws r!_definition_from_schemar->sP$"H ,,* +C {74&(DEFF !$**S/4 88r c@ttj||S)zBParse the definition native code hands over, in its stored layout.)r'r(r))r,r+s r!_definition_from_jsonr/Hs !$**S/4 88r c|jd}|:t|tr |tkDrt d|d|dt |dS|jd}|dvrt d|d |d t t |S) Nformatzmaterialized view 'z' is stored in format z., which this version of lancedb cannot refreshr)rkind)selectnamespaced_selectz' is stored in format kind 'z/', which this version of lancedb cannot refresh)r$ isinstanceintDEFINITION_FORMATNotImplementedErrorr _legacy_query)valuer+fmtr2s r!r'r'Ms ))H C #s#s->'>%%dV+A#G99 *g?? 99V D 22!!$'CD6J; ;   &M%,@ AAr cD|rtd|Dr|St|S)Nc3nK|]-}|dk(xs"|jxs|j/yw)_N)islowerisdigit).0cs r! z _legacy_ident..bs-IqAH: :qyy{:Is35)all_quote_identifierr+s r! _legacy_identrGas" IDII T ""r c|jdg}|reg}|D]L}|d|d}}||t|fvr|j|.|j|dt|Ndj |}nd}dj dg|jd g|d D}d |d |}|jd  |d|d z }|jd |d|dz }|S)zBRender the pre-format structured layout as the query it described. projectionsoutput expressionz AS z, *.c32K|]}t|ywN)rG)rAparts r!rCz _legacy_query..us  dssource_namespace source_tablezSELECT z FROM filterz WHERE limitz LIMIT )r$rEappendrGjoin) r:rIitemsprJrKcolumnstablers r!r9r9gs&))M2.K IA!"8a oJFf&7&?@@ Z( |4 f0E/FGH  I ))E" HHOeii 2B7O~9NO EgYfUG ,E yy& 75?+,, yy% 75>*++ Lr c2|jdd}d|dS)z:Quote a column name as a Lance SQL identifier (backticks).`z``)replace)r+escapeds r!rErEs ll3%G wiq>r c |yt|tr|g}t|trt|j Sg}|D]G}t|tr|j |t |f0|\}}|j ||fI|S)aL``select`` items may be a column name, an ``(alias, expression)`` pair, or a dict of the same. A bare name projects itself and is quoted, so any valid column name works; dict and pair entries are kept verbatim because their right side is an expression. A lone string is one column, not a sequence of its characters.N)r5rdictlistrWrUrE)r3 normalizeditemaliasrKs r!normalize_selectres~&#&$FLLN##J3 dC   t%6t%<= > $ E:   uj1 2 3 r ceZdZdZd dZd dZed dZeddZddZ ddd  dd Z ddd  dd Z y)AsyncMaterializedViewzA handle on a materialized view: its table plus its definition. Obtained from ``AsyncConnection.create_materialized_view`` or ``AsyncConnection.open_materialized_view``. c||_yrO_tableselfrZs r!__init__zAsyncMaterializedView.__init__s  r c"d|jdS)NzAsyncMaterializedView(name=)rFrls r!__repr__zAsyncMaterializedView.__repr__s,TYYM;;r c.|jjSrOrjr+rps r!r+zAsyncMaterializedView.name{{r c|jS)zThe view, as the table it is. Queries, indexes and search all apply; writes are not blocked, but a rebuild replaces them.rirps r!rZzAsyncMaterializedView.tables{{r c|K|jjjd{}t|S7w) The query that defines the view.N)rj_innermaterialized_view_definitionr/)rlr,s r! definitionz AsyncMaterializedView.definitions2KK&&CCEE$S))Fs (<:<FNfullsource_versionclK|jjj||d{S7w)aRecompute the view from its source. ``full=True`` forces a remote rebuild; otherwise the server chooses the refresh mode. Local views always rebuild, and ``source_version`` selects a source version instead of the latest. Remote SQL refreshes do not support source-version pinning. Remote results leave ``mode``, ``rows_written``, and ``source_version`` as ``None`` because the SQL job adapter does not expose these metrics. r}r|N)rjrxrefresh_materialized_viewrlr|r}s r!refreshzAsyncMaterializedView.refreshs:[[''AA)B    s +424cKddlm}t|jjj ||d{|j S7w)a&Submit a refresh and return its job without waiting. The job may already be complete for a local view. Remote refreshes are submitted through the SQL service. Options and result metrics follow [refresh][lancedb.materialized_view.AsyncMaterializedView.refresh]. r rrN)_lancedbrrrjrxrefresh_materialized_view_async from_json)rlr|r}rs r! refresh_asyncz#AsyncMaterializedView.refresh_asyncsP <++$$DD-DE  * 3 3    s6AA A)rZ 'AsyncTable'returnr)rrrrr|boolr} Optional[int]rz'RefreshMaterializedViewResult')r|rr}rrz)'AsyncJob[RefreshMaterializedViewResult]' rrrrrmrqpropertyr+rZrzrrrr r!rgrgs <   * $T  5B ( $$T  5B 2 r rgceZdZdZd dZd dZed dZeddZeddZ ddd  dd Z ddd  dd Z y)MaterializedViewzdSynchronous variant of [AsyncMaterializedView][lancedb.materialized_view.AsyncMaterializedView].cF||_t|j|_yrO)rjrg_asyncrks r!rmzMaterializedView.__init__s +ELL9 r c"d|jdS)NzMaterializedView(name=rorFrps r!rqzMaterializedView.__repr__s' }A66r c.|jjSrOrsrps r!r+zMaterializedView.namertr c|jS)zThe view, as the table it is.rirps r!rZzMaterializedView.tables{{r c\tj|jjS)rw)r runrrzrps r!rzzMaterializedView.definitions xx ..011r FNr{cbtj|jj||S)zRecompute the view from its source. See [AsyncMaterializedView.refresh][lancedb.materialized_view.AsyncMaterializedView.refresh].r{)r rrrrs r!rzMaterializedView.refreshs' xx ++n+UVVr ctttj|jj ||S)zSubmit a refresh and return its job without waiting. See [AsyncMaterializedView.refresh_async][lancedb.materialized_view.AsyncMaterializedView.refresh_async]. r{)rr rrrrs r!rzMaterializedView.refresh_asyncs4 HH ))tN)S   r )rZ 'LanceTable'r)rrrr)r|rr}rrz$'Job[RefreshMaterializedViewResult]'rrr r!rrsQ:7  22 $TWW5BW (W$T    5B  -  r r)r*z 'pa.Schema'r+rrr))r,rr+rrr)r:r`r+rrr)r+rrr)r:r`rr)r3 SelectArgrzOptional[List[Tuple[str, str]]])*r __future__rtypingrrrrrr r r( dataclassesr background_loopr jobrrrpyarrowparrrZrrr%rrMaterializedViewSourcer7rr-r/r'rGr9rErergrrr r!rs#NNN !!**7-&  U3c3h' ()cN   )   K K  K9 9"999 B(# 4 .= = @, , r