Wh5 LUddlmZddlmZddlmZddlZddlmZddlm Z ddl m Z m Z m Z ddlmZer0ddlmZmZdd lmZmZdd lmZdd lmZdd lmZdd lmZddlmZmZddddddddddd Z dddZ!dddZ"ejFZ$ ejJZ& ejNZ( ejRZ* dd d;d!Z+ dd&Z/ed'(d?d)Z0e jbjdZ2ie2jfejhe2jjejle2jnejpe2jrejte2jvejxe2jzej|e2j~eje2jeje2jeje2jeje2jeje2je d*e2jeje2jeje2jeje2jeje2je d+ZSd,eTd-<ejejejejd.ZYd/eTd0<e2je2jfZ\ d@d1Z]dAd2Z^dBd3Z_dCd4Z` dDd5Za dEddd6d7 dFd8Zb dGd9ZcdHd:Zdy)I) annotations) lru_cache) TYPE_CHECKINGN) DuckDBPyType)Versionisinstance_or_issubclass zip_strict)ColumnNotFoundError)MappingSequence)DuckDBPyRelation Expression)CompliantLazyFrameAny)DuckDBLazyFrame) DuckDBExpr)DType) IntoDTypeTimeUnityearquartermonthdayhourminutesecond millisecond microsecond nanosecond) yqmodhmsmsusnsdescasc)TFz nulls lastz nulls first) separatorcJ|rtdt|g|Stdg|S)aConcatenate many strings, NULL inputs are skipped. Wraps [concat] and [concat_ws] `FunctionExpression`(s). Arguments: exprs: Native columns. separator: String that will be used to separate the values of each column. Returns: A new native expression. [concat]: https://duckdb.org/docs/stable/sql/functions/char.html#concatstring- [concat_ws]: https://duckdb.org/docs/stable/sql/functions/char.html#concat_wsseparator-string- concat_wsconcat)Flit)r,exprss P/var/www/html/jupyter_env/lib/python3.12/site-packages/narwhals/_duckdb/utils.py concat_strr45s,6?1[#i. 15 1WAhDWQVDWWc4g}|D]}|j|}|j|}|j|j|}t|t|k7rd|dt|d}t ||j t |||S)Nz!Internal error: got output names z, but only got z results)_call_evaluate_output_names_alias_output_nameslenAssertionErrorextendzip)dfr2native_resultsexprnative_series_list output_namesmsgs r3evaluate_exprsrDGs46NE!ZZ^2226  # # /33LAL | $6 7 75l^?SVWiSjRkkstC % %c,0BCDE r5c:eZdZUdZdZded<ddZeddZy) DeferredTimeZoneaObject which gets passed between `native_to_narwhals_dtype` calls. DuckDB stores the time zone in the connection, rather than in the dtypes, so this ensures that when calculating the schema of a dataframe with multiple timezone-aware columns, that the connection's time zone is only fetched once. Note: we cannot make the time zone a cached `DuckDBLazyFrame` property because the time zone can be modified after `DuckDBLazyFrame` creation: ```python df = nw.from_native(rel) print(df.collect_schema()) rel.query("set timezone = 'Asia/Kolkata'") print(df.collect_schema()) # should change to reflect new time zone ``` Nz str | None_cached_time_zonec||_yN)_rel)selfrels r3__init__zDeferredTimeZone.__init__ks  r5cf|jt|j|_|jS)z;Fetch relation time zone (if it wasn't calculated already).)rGfetch_rel_time_zonerJ)rKs r3 time_zonezDeferredTimeZone.time_zonens-  ! ! )%8%CD "%%%r5)rLr returnNone)rQstr) __name__ __module__ __qualname____doc__rG__annotations__rMpropertyrPr5r3rFrFWs-"%)z(&&r5rFc J|j}|j}|dk(r&|jt|j||S|dk(rM|j }|j |Dcgc]&}|j|dt|d||(c}S|dk(r|j \}}|dg}|djdk(r:|dj \}}|jd|d|djdk(r:t|d||} |j| t|S|dk(rF|tjur|jS|j dd} |j| S|d k(r|j|j St!||Scc}w) Nliststructr)namedtypearray)innershapeenum) categoriesztimestamp with time zone)rP)iddtypesListnative_to_narwhals_dtypechildchildrenStructFieldinsertArraytuplerV1EnumDatetimerP$_non_nested_native_to_narwhals_dtype) duckdb_dtypeversiondeferred_time_zoneduckdb_dtype_idrgrkrjsizercrbres r3ririvs#ooO ^^F& {{ $\%7%7BT U  ("((}} &     q258WFXY   '!"++ t G9AhkkW$(++KE4 LLDG $AhkkW$)q7  FLLN   6;;=   FNN$   6==?  FMMO  V]]_  FLLN  &.."  !  6==?     vs+  -! " V__&# $->>#OO%>>#  / 0 c/6>>+,1-r5INT128UINT128z"Mapping[type[DType], DuckDBPyType]NW_TO_DUCKDB_DTYPES)r%r&r'r(zMapping[TimeUnit, DuckDBPyType]TIME_UNIT_TO_TIMESTAMPc |j}|j}tj|x}r|St ||j rZ|t jur d}t|t||j rtd|jSd}t|t ||jrn|j}|j}|s t |S|dk7rd|d}t|||jx} k7rd| d|d }t|t"j$St ||j&r'|jx}dk7rd |d}t"j(St ||j*r,t-|j.||} t1j2| St ||j4rN|j6D cic]$} | j8t-| j:||&} } t1j<| St||j>ry|} t| |j>r#| j.} t| |j>r#t-| ||}d jAd |jBD}t||StE|tFrd |jHd}t|d|}tK|cc} w)Nz9Converting to Enum is not supported in narwhals.stable.v1ENUMz9Can not cast / initialize Enum without categories presentr'zUOnly microsecond precision is supported for timezone-aware `Datetime` in DuckDB, got z precisionzOnly the connection time zone z is supported, got: .z6Only microsecond-precision Duration is supported, got r+c3(K|] }d|d yw)[]NrZ).0items r3 z+narwhals_to_native_dtype..s"G4QtfA;"GszConverting to z# dtype is not supported for DuckDB.zUnknown dtype: )&rg base_typerrrrrrrqNotImplementedError isinstancerre ValueErrorrs time_unitrPr duckdb_dtypes TIMESTAMP_TZrINTERVALrhnarwhals_to_native_dtyperbduckdb list_typerlfieldsr_r` struct_typerojoinrc issubclassUNSUPPORTED_DTYPESrTr;)r`rvrwrgr duckdb_typerCtutzrel_tzrbfieldrnw_inner duckdb_innerduckdb_shape_fmts r3rrs^^F!I)--i88{8v{{3 gjj MC%c* * eV[[ )$u'7'7&: ;< <Iov7 __ __)"- - :ijlimmwxCS/ ! .888& 926(:NrdRSTCS/ !)))v7// !Bd *J2$jYC%%%v{{3(g?QR&&v}}5  JJ0gGYZ Z  !!&))%&#6<<0~~H6<<0/'CUV 77"G5;;"GG|n-=,>?@@)/0y1122UV!#&& E7 #C  ! s )Kc<t|tr t|S|SrI)rrScol)into_expressions r3parse_into_expressionrs#-os#C3 XXr5cn|sydj|Dcgc]}t|c}}d|Scc}w)Nr+, z partition by )rr) partition_byxby_sqls r3generate_partition_by_sqlr#s@  YY|L!0345L MF 6( ##Ms2c2djd|DS)Nrc3DK|]}tt|ywrI)rSr)rr_s r3rz$join_column_names..+s6ST^6s )r)namess r3join_column_namesr*s 9966 66r5cX|sydjdt|||D}d|S)Nr+,c3fK|])\}}}t|dt|dt|+yw) N)rDESCENDING_TO_ORDERNULLS_LAST_TO_NULLS_POS)rr _descending _nulls_lasts r3rz(generate_order_by_sql..3sD 'A{K ! # $A&9+&F%GqI`alImHnos/1z order by )rr ) descending nulls_lastorder_byrs r3generate_order_by_sqlr.s=  XX+5h J+WFvh r5F)rr ignore_nullsc  ddlm}t |} |xsdgt |z}|xsdgt |z}t|||d} | | d| d|d} n|d |d} n |d| d } nd } |rt|jd d n t|}||d| d| d| d S#t$r$} dtj} t | | d} ~ wwxYw)Nr) SQLExpressionz=1.3.0 is required for this operation. Found: DuckDB F)rrz rows between z preceding and z followingz%rows between unbounded preceding and z" preceding and unbounded followingr+)z ignore nulls)z over (r) rrModuleNotFoundError __version__rrr:rrS removesuffix)r@rr rows_startrows_endrrrrexcrCpbobrowsfuncs r3window_expressionr:s0( #L 1B6wX6J6wX6J ZJ WB("6 {m?8*JO  6xj K   {m+MN=Ic$i$$S) *. 9sSWyD D6AbT4&: ;;% 0LVM_M_L`a!#&C/0sB$$ C-C  Ccttjr4tfddDr t j |j SS)Nc38K|]}|tvywrI)rS)rrC exceptions r3rz)catch_duckdb_exception..as"=  s9~=s)znot found in FROM clausez5this column cannot be referenced before it is defined)available_columns)rrBinderExceptionanyr from_available_column_namescolumns)rframes` r3catch_duckdb_exceptionr^sO)V334= =:#>>#mm   r5cJ|dk(r|djSt|g|S)Nisnullr)rr0)r_argss r3functionros) xAw~~ T>D>r5)r2rr,rSrQr)r>rr2rrQzlist[tuple[str, Expression]])rurrvrrwrFrQr)rLzduckdb.DuckDBPyRelationrQrS)rxrSrvrrQr)r`rrvrrwrFrQr)rstr | ExpressionrQr)rr rQrS)rrSrQrS)rr rSequence[bool]rr rQrS)rZrZNN)r@rrSequence[str | Expression]rr r int | Nonerr rSequence[bool] | Nonerr rboolrQr)r ExceptionrrrQzColumnNotFoundError | Exception)r_rSrrrQr)e __future__r functoolsrtypingrr duckdb.typingrrnarwhals._utilsrrr narwhals.exceptionsr collections.abcr r r rnarwhals._compliant.typingrnarwhals._duckdb.dataframernarwhals._duckdb.exprrnarwhals.dtypesrnarwhals.typingrr UNITS_DICTrrColumnExpressionrConstantExpressionr1CaseExpressionwhenFunctionExpressionr0r4rDrFrirOrtMAINrgrDOUBLErFLOATrBLOBrVARCHARrBOOLEANrDATErTIMErTINYINTrSMALLINTrINTEGERrBIGINTrrUTINYINTr USMALLINTrUINTEGERrUBIGINTrrrX TIMESTAMP_S TIMESTAMP_MS TIMESTAMP TIMESTAMP_NSrr CategoricalrrrrrrrrrrZr5r3r7s" %&II313@:0%3            $E2!-mD * , ( ,57X$  $. !  &&>,J,J)0,JFV,J ,J^ 2--:   ; NNM((; NNM''; MM=%%; MM=(( ;  NNM)) ;  KK## ; KK##; KK&&; LL-((; LL-''; LL-&&; MM<); LL-((; MM=**; MM=));  MM=((!;" NNL+#;7(  " "  $ $  ! !  $ $ ;7 nnf&8&894 4&4