q-Ph XddlmZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ddl mZeje5ddlmZdddn #1swxYwYe rddlmZdd lmZdd lmZd gZdd d d d d d d d d+d Zd,d"Zed#$d d%d-d(Zd.d)Zd d%d/d*ZdS)0) 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_path plugin_path Path | str function_namestrargsIntoExpr | Iterable[IntoExpr]rdict[str, Any] | Nonerboolrrrrrrreturnr c t|} t|} t|| }tt jt ||| | ||||| | 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) rrrrrrrrrr)r_serialize_kwargs_resolve_plugin_pathr plrr r) rrrrrrrrrrrpyexprsserialized_kwargss N/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/polars/plugins.pyr r sF-T22G)&11&{NNNK  $K(('$)%=)/1)    bytesc@|sdSddl}||dS)z+Serialize the function's keyword arguments.r'rN)protocol)pickledumps)rr,s r&r!r!os0 sMMM <<< + ++r')maxsizer pathrc.t|tst|}|rt||S|D]$}t |rt||cS%d|}t |)z.Get the file path of the dynamic library file.r z"no dynamic library found at path: ) isinstanceris_file_resolve_file_pathiterdir_is_dynamic_libFileNotFoundError)r0rpmsgs r&r"r"{s dD ! !Dzz ||~~C!$\BBBB \\^^DD 1   D%alCCC C C C D 6t 5 5C C  r'c<|o|jdvS)N)z.soz.dllz.pyd)r3suffix)r0s r&r6r6s <<>> Ddk-DDDr'cttj}|r|S ||}n$#t $r|}YnwxYw|S)N)rsysprefixresolve relative_to ValueError)r0r venv_path file_paths r&r4r4suSZ  I'||~~ '((33II ' ' ' III ' sAA('A()rrrrrrrrrrrrrrrrrrrrrrrr )rrrr()r0rrrrr)r0rrr)r0rrrrr) __future__r contextlibr= functoolsrpathlibrtypingrrpolars._utils.parserpolars._utils.wrapr suppress ImportError polars.polarspolarsr#collections.abcr r polars._typingr __all__r r!r"r6r4r'r&rSs""""""" %%%%%%%%>>>>>>((((((Z%%                 ((((((('''''' % &%)   #%*$TTTTTTn , , , , 2CH ! ! ! ! ! ! EEEE