o i @s ddlmZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ddl mZeeddlmZWdn1sDwYe r]ddlmZdd lmZdd lmZd gZdd d d d d d d d d/d d Zd0d"d#Zed$d%d d&d1d)d*Zd2d+d,Zd d&d3d-d.ZdS)4) 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]r dict[str, Any] | Nonerboolrrrrrrreturnr c CsBt|} t|} t|| d}ttjt||| | ||||| |d 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) rrrr rrrrrr)r_serialize_kwargs_resolve_plugin_pathrplrr r) rrrr rrrrrrrZpyexprsZserialized_kwargsr"E/home/app/Keep/.python/lib/python3.10/site-packages/polars/plugins.pyr s"C bytescCs|sdSddl}|j|ddS)z+Serialize the function's keyword arguments.rN)protocol)pickledumps)r r(r"r"r#rosr)maxsizerpathrcCs^t|ts t|}|rt||dS|D]}t|r%t||dSqd|}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)r,rpmsgr"r"r#r {s    r cCs|o|jdvS)N)z.soz.dllz.pyd)r.suffix)r,r"r"r#r1sr1cCsDttj}|r |Sz||}W|Sty!|}Y|Sw)N)rsysprefixresolve relative_to ValueError)r,rZ venv_path file_pathr"r"r#r/s   r/)rrrrrrr rrrrrrrrrrrrrrrrr )r rrr$)r,rrrrr)r,rrr)r,rrrrr) __future__r contextlibr6 functoolsrpathlibrtypingrrZpolars._utils.parserZpolars._utils.wraprsuppress ImportErrorZ polars._plrZ_plrr!collections.abcr Zpolarsr Zpolars._typingr __all__r rr r1r/r"r"r"r#s<          W