i vddlmZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ddl mZeje5ddlmZddde rddlmZdd lmZdd lmZd gZdd d d d d d d d  ddZddZedd dddZddZd dddZy#1swYhxYw)) annotationsN) lru_cache)Path) TYPE_CHECKINGAny)parse_into_list_of_expressions) wrap_expr)Iterable)Expr)IntoExprregister_plugin_functionF)kwargsis_elementwisechanges_lengthreturns_scalarcast_to_supertypeinput_wildcard_expansionpass_name_to_apply use_abs_pathc t|} t|} t|| }tt j t ||| | ||||| | S)a Register a plugin function. See the `user guide `_ for more information about plugins. Parameters ---------- plugin_path Path to the plugin package. Accepts either the file path to the dynamic library file or the path to the directory containing it. function_name The name of the Rust function to register. args The arguments passed to this function. These get passed to the `input` argument on the Rust side, and have to be expressions (or be convertible to expressions). kwargs Non-expression arguments to the plugin function. These must be JSON serializable. is_elementwise Indicate that the function operates on scalars only. This will potentially trigger fast paths. changes_length Indicate that the function will change the length of the expression. For example, a `unique` or `slice` operation. returns_scalar Automatically explode on unit length if the function ran as final aggregation. This is the case for aggregations like `sum`, `min`, `covariance` etc. cast_to_supertype Cast the input expressions to their supertype. input_wildcard_expansion Expand wildcard expressions before executing the function. pass_name_to_apply If set to `True`, the `Series` passed to the function in a group-by operation will ensure the name is set. This is an extra heap allocation per group. use_abs_path If set to `True`, the path will be resolved to an absolute path. The path to the dynamic library is relative to the virtual environment by default. Returns ------- Expr Warnings -------- This is highly unsafe as this will call the C function loaded by `plugin::function_name`. The parameters you set dictate how Polars will handle the function. Make sure they are correct! r) plugin_path function_nameargsrrrrrrr)r_serialize_kwargs_resolve_plugin_pathr plrr str) rrrrrrrrrrrpyexprsserialized_kwargss P/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/polars/plugins.pyr r sdF-T2G)&1&{NK  $$K('$)%=)/1)   c6|syddl}|j|dS)z+Serialize the function's keyword arguments.r"rN)protocol)pickledumps)rr&s r!rros  <<< ++r")maxsizerct|ts t|}|jr t||S|j D]}t |st||cSd|}t |)z.Get the file path of the dynamic library file.rz"no dynamic library found at path: ) isinstanceris_file_resolve_file_pathiterdir_is_dynamic_libFileNotFoundError)pathrpmsgs r!rr{sr dD !Dz ||~!$\BB \\^D 1 %alC CD /tf 5C C  r"cB|jxr|jdvS)N)z.soz.dllz.pyd)r,suffix)r1s r!r/r/s <<> Ddkk-DDDr"cttj}|r|jS |j |}|S#t $r|j}Y|SwxYw)N)rsysprefixresolve relative_to ValueError)r1r venv_path file_paths r!r-r-s^SZZ I||~ '((3I  ' I  'sAAA)r Path | strrrrzIntoExpr | Iterable[IntoExpr]rdict[str, Any] | Nonerboolrr@rr@rr@rr@rr@rr@returnr )rr?rAbytes)r1r>rr@rAr)r1rrAr@)r1rrr@rAr) __future__r contextlibr7 functoolsrpathlibrtypingrrpolars._utils.parserpolars._utils.wrapr suppress ImportError polars._plr_plrrcollections.abcr polarsr polars._typingr __all__r rrr/r-r"r!rSs1" %>(Z%(' % &%)   #%*$TTT ( T " T  TTTT#TTT Tn , 2CH ! ! E