ɯei8 dZddlmZmZmZmZmZmZddlm cm cm cm cm Z ddlmZmZddlmZmZmZmZddlmZddlmZmZmZddlmZmZdd lm Z m!Z!d d l"m#Z#d d l$m%Z%d d l&m'Z'm(Z(m)Z)GddZ*Gdde*Z+dee,efdefdZ-dee,ee,e*fdededefdZ.dede#de#deeeeffdZ/de+de#dee%deeeffdZ0y) z(Contains table function related classes.)DictIterableListOptionalTupleUnionN) Ascending SortOrder)NamedArgumentsTableFunctionPosArgumentsTableFunctionTableFunctionExpression$TableFunctionPartitionSpecDefinition) ColumnOrName) publicapi quote_namevalidate_object_name)Column_to_col_if_str) ArrayTypeMapType) SnowflakePlan)SelectStatement)build_expr_from_python_val+build_expr_from_snowpark_column_or_col_namewith_src_positionceZdZdZeddddeeeefdede e jde d ed df d Z d Zedddd de eeeefde eeeefde d dfdZedddede d dfdZeZy)TableFunctionCallaRepresents a table function call. A table function call has the function names, positional arguments, named arguments and the partitioning information. The constructor of this class is not supposed to be called directly. Instead, use :func:`~snowflake.snowpark.function.call_table_function`, which will create an instance of this class. Or use :func:`~snowflake.snowpark.function.table_function` to create a ``Callable`` object and call it to create an instance of this class. NT)_ast _emit_ast func_namefunc_argumentsrr func_named_argumentsreturnc|r |r td||_||_||_||_d|_d|_d|_d|_d|_ ||_ y)Nz8A table function shouldn't have both args and named argsF) ValueErrornameuser_visible_name argumentsnamed_arguments_over _partition_by _order_by_aliases_api_call_sourcer)selfr!rr r"r#s c/var/www/html/glpi_dashboard/venv/lib/python3.12/site-packages/snowflake/snowpark/table_function.py__init__zTableFunctionCall.__init__,sl 2WX X" &/   !  !15 $ c||_yN)r/)r0api_call_sources r1_set_api_call_sourcez&TableFunctionCall._set_api_call_sourceFs /r3) partition_byorder_byr r8r9ct|ttfr|f}n| t|}nd}|r t |dkDr|D] }t |d}|r|j tj}t|j}|jj|j |t|ttfr@d|j_t|jj j#|nFd|j_|D]0}t|jj j#|2|t|ttfr@d|j$_t|j$j j#|nFd|j$_|D]0} t|j$j j#| 2t'|j(g|j*d|i|j,} d| _t|ttfr|f} n| t|} nd} | r?| Dcgc]3}t|tr |j0nt|j05c}nd} | | _t|ttfr|f}n| t|}nd}|r8g} t |dkDr!|D]}| j5t || | _| Scc}w)aSpecify the partitioning plan for this table function call when you lateral join this table function. When a query does a lateral join on a table function, the query feeds data to the table function row by row. Before rows are passed to table functions, the rows can be grouped into partitions. Partitioning has two main benefits: - Partitioning allows Snowflake to divide up the workload to improve parallelization and thus performance. - Partitioning allows Snowflake to process all rows with a common characteristic as a group. You can return results that are based on all rows in the group, not just on individual rows. Refer to `table functions and partitions `__ for more information. Args: partition_by: Specify the partitioning column(s). It tells the table function to partition by these columns. order_by: Specify the ``order by`` column(s). It tells the table function to process input rows with this order within a partition. Note that if this function is called but both ``partition_by`` and ``order_by`` are ``None``, the table function call will put all input rows into a single partition. If this function isn't called at all, the Snowflake database will use implicit partitioning. NrTFr) isinstancestrrtuplelen_create_order_by_expressionrprotoExprrtable_fn_call_overlhsCopyFromr8variadicrargsaddr9rr'r)r*r+ _expressionr,appendr-)r0r8r9r order_by_tupleeastexprpartition_clause order_clausenew_table_functionpartition_by_tuplepartition_spec order_specs r1overzTableFunctionCall.overIs6 hf .&[N  !"8_N!N >"Q&'3A/23 .**,C$S%;%;"Q&'FA%%&A!&DEF+5  (!!+ s8K9)r aliasescd}|rtj}t|j}|jj |j d|j_|D]0}t|jjj|2|Dcgc] }t|}}tt|t|k7r td||_||_|Scc}w)aLAlias the output columns from the output of this table function call. Args: aliases: An iterable of unique column names that do not collide with column names after join with the main table. Raises: ValueError: Raises error when the aliases are not unique after being canonicalized. NTz6All output column names after aliasing must be unique.)r@rArtable_fn_call_aliasrCrDrrUrErrFrGrr>setr&r.)r0r rUrLrMargcol canon_aliasess r1aliaszTableFunctionCall.aliass **,C$S%<%<=D HH  dii ($(DLL ! I*4<<+<+<+@+@+BCH I5<Q4 Cr3rc0eZdZdZdededdffd ZxZS)_ExplodeFunctionCallz[Internal class to identify explode function call as a special instance of TableFunctionCallrZouterr$NcFt|d||||_d|_y)Nflatten)inputrfexplode)superr2rZr()r0rZrf __class__s r1r2z_ExplodeFunctionCall.__init__s& #U;&/r3)r]r^r_r`rr2 __classcell__)rls@r1reres%e0F060d00r3rerKr$c.t|tr'tt|jt St|trDt|jtr |jSt|jt St d)NzDOrder By columns must be of column names in str, or a Column object.)r;r<r rrHr TypeError)rKs r1r?r?sj!S.. << Av  ammY /== Q]]IK8 8 R  r3funcrF named_argsc  d}d}d}d}d}|r |r tdt|tr|}nt|tr$|D] } t | dj |}nt|t rp|s|r td|j}|j}|j}|j}|j}|j}|j}|j}n td|r/t!||D cgc]} t#| dj$c} } nBt'||j)D cic]\} } | t#| dj$c} } } |r t+|||nd| _|| _|| _| Scc} wcc} } w)Nz9A table function shouldn't have both args and named args..zN'args' and 'named_args' shouldn't be used if a TableFunction instance is used.z'func' should be a function name in str, a list of strs that have all or a part of the fully qualified name, or a TableFunctionCall instance.ztable function)r&r;r<listrjoinrr'r)r*r+r,r-r.r/ror rrHr itemsrrRrUr6) rprFrqrTr8r9rUr6fqdnnrYtable_function_expressionarg_names r1!_create_table_function_expressionr{s DLHGO TUU$ D$  $A  # $xx~ D+ , :` yy~~)) zz)) >>--// \   $= JN O3^C!1 2 > > O% ! %@ &0%5%5%7 !Hc..>?KKK % !  -T<J , )0%0?- $$# P  s 8F 6!F func_expr current_plan join_plancdtdttfd}||}||}|j}|t |d}|Dcgc]}t |j } }|r'|Dcgc]}t |j c}ng} |rmt |t |k7rtd|d|t||D cgc]-\} } t | j| j /} } } n&|Dcgc]}t |j } }| | | fScc}wcc}wcc} } wcc}w)Nplanr$cT|jDcgc]}|jc}Scc}wr5)outputr')rattrs r1get_column_names_from_planz>_get_cols_after_join_table..get_column_names_from_plan%s&*kk2d 222s%zxThe number of aliases should be same as the number of cols added by table function. Columns added by table function are z and aliases given are ) rrr<rUr>r_namedr&zipr\)r|r}r~rcols_before_joincols_after_joinrUnew_cols_namesrZold_cols alias_colsr' alias_namenew_colss r1_get_cols_after_join_tabler sP 334932,?0;OG$S)9%:%<=N0@As ""$AHA>E':3&+$$&:2J ~ #g, .77E6FF]^e]fh  %($@  j 4L  z * 1 1 3  5CCSF3K&&(CC Xz ))!B: Ds D-. D2 2D7 D=rselect_statementc|rN|j|jjgj}|jdj }n@|jj j}|jj|dgd}g}|j}|r'|Dcgc]}t|jc}ng}t|tr|rct|dk7rtdt||j!tdj#|dj||fS|j!tdj||fSt|t$r|rt|dk7rtdt||j'tdj)|djtdj)|djg||fS|j'tdjtdjg||fStd|jd |cc}w) Nrrz>Invalid number of aliases given for explode. Expecting 1, got VALUEz>Invalid number of aliases given for explode. Expecting 2, got KEYz Invalid column type for explode(z*). Expected ArrayType() or MapType(), got )selectrZrsnowflake_planrdatatyperHr' output_dictgetr.rr;rr>r&rIr\rextendrb) rprrexplode_col_typecol_namecolsrUrZrs r1_get_cols_after_explode_joinrAs9 &&(9':;JJ;;q>2288'',,++//4&A!D DmmG>E':3&+$$&:2J"I. 7|q  TUXY`UaTbc KKw--gaj9@@B C*  ' KKw..0 1&  % $g . 7|q  TUXY`UaTbc KK5M%%gaj188:7O'' 3::<    KK--/1G1G1IJ K  .txxj8bcsbt u  3;s% I+)1r`typingrrrrrr4snowflake.snowpark._internal.proto.generated.ast_pb2snowpark _internalr@ generatedast_pb25snowflake.snowpark._internal.analyzer.sort_expressionr r 4snowflake.snowpark._internal.analyzer.table_functionr r r r'snowflake.snowpark._internal.type_utilsr"snowflake.snowpark._internal.utilsrrrsnowflake.snowpark.columnrrsnowflake.snowpark.typesrr!_internal.analyzer.snowflake_planr#_internal.analyzer.select_statementr_internal.ast.utilsrrrrrer<r?r{rrrcr3r1rs, /??DDDV A =7<@jjZ0,0  5f#5  )  7% T#Y 11 27% 7%7% 7%t*&*** 4t  *B* * */* 4: *r3