o i\@s&ddlmZddlZddlZddlmZddlmZmZm Z ddl m Z ddl mZddlmZddlmZerdddlmZdd lmZdd l mZdd lmZdd lmZe d ZedZedefZd4ddZd5ddZ d6ddZ!d7d d!Z"d8d#d$Z#Gd%d&d&Z$e$Z%d9d.d/Z&d:d2d3Z'dS);) annotationsN)wraps) TYPE_CHECKINGAnyTypeVar) functions)wrap_s)dtype_to_ffiname)Callable) ParamSpec)Series)PySeries)PolarsDataTypeTP.clstype[T]returncCst|dd}t|}t|D]5}|dsC|dkrCt||}t|rCt|}|jjd|jj}|||f|vrCt |rCt ||t |q|S)a Series/NameSpace class decorator that sets up expression dispatch. * Applied to the Series class, and/or any Series 'NameSpace' classes. * Walks the class attributes, looking for methods that have empty function bodies, with signatures compatible with an existing Expr function. * IFF both conditions are met, the empty method is decorated with @call_expr. _accessorN_Zplot) getattr _expr_lookupdir startswithcallable _undecorated__code__ co_varnames co_argcount_is_empty_methodsetattr call_expr)r namespaceZ expr_lookupnameattrargsr&J/home/app/Keep/.python/lib/python3.10/site-packages/polars/series/utils.py expr_dispatchs   r(r" str | None,set[tuple[str | None, str, tuple[str, ...]]]c Cst}d|_|durt||}t}t|D]2}|dsIzt||}Wn ty.Yqwt|rIt |}|j j d|j j }| |||fq|S)zACreate lookup of potential Expr methods (in the given namespace).Nr)plExprZ_pyexprrsetrrAttributeErrorrrrrradd)r"exprlookupr#mr%r&r&r'r=s$    rfunctionCallable[P, T]cCst|dr |j}t|ds|S)z1Return the given function without any decorators. __wrapped__)hasattrr5)r3r&r&r'rWs  rfunc SeriesMethodcs,td fd d }t|d t|S)z7Dispatch Series method to an expression implementation.selfrr%P.argskwargsP.kwargsrr cs\t|j}t|j}t|dd}durt||}t|j}|||i| S)Nr) rZ_sFcolr#r__name__Zto_frameZ select_seqZ to_series)r9r%r;sr0r"fr7r&r'wrapperas    zcall_expr..wrapper __signature__N)r9rr%r:r;r<rr )rr inspect signature)r7rCr&rBr'r!^s r!boolcCsB|j}|jtvo t|jdkr|jddup tjjdko |jdkS)z Confirm that the given function has no implementation. Definitions of empty: - only has a docstring (body is empty) - has no docstring and just contains 'pass' (or equivalent) NN)rco_code_EMPTY_BYTECODElen co_constssysflagsoptimize)r7Zfcr&r&r'rps  rc@s eZdZd ddZd dd Zd S) _EmptyBytecodeHelperrNonecCs*ddd}ddd}|jj|jjf|_dS)NrrScSsdS)Nr&r&r&r&r'_empty_with_docstringsz<_EmptyBytecodeHelper.__init__.._empty_with_docstringcSsdSrJr&r&r&r&r'_empty_without_docstringsz?_EmptyBytecodeHelper.__init__.._empty_without_docstringrrS)rrKempty_bytecode)r9rUrVr&r&r'__init__s   z_EmptyBytecodeHelper.__init__itembytesrGcCs ||jvSrJ)rX)r9rZr&r&r' __contains__s z!_EmptyBytecodeHelper.__contains__NrW)rZr[rrG)r? __module__ __qualname__rYr\r&r&r&r'rRs  rRr#strdtyperobjr Callable[..., Any] | NonecCs t|}|d|}t||dS)az Dynamically obtain the proper FFI function/ method. Parameters ---------- name function or method name where dtype is replaced by <> for example "call_foo_<>" dtype polars dtype. obj Object to find the method for. Returns ------- callable or None FFI function, or None if not found. z<>N)r replacer)r#r`raZffi_namefnamer&r&r' get_ffi_funcs  reCallable[..., Any]rcCsBddlm}z|d|}|dW|Sty |dw)Nr) check_lengthFT) polars._plrrg Exception)r7rgresultr&r&r'_with_no_check_lengths   rk)rrrr)r"r)rr*)r3r4rr4)r7r8rr8)r7r8rrG)r#r_r`rrar rrb)r7rfrr)( __future__rrErO functoolsrtypingrrrZpolars._reexportZ _reexportr+Zpolarsrr=Zpolars._utils.wraprZpolars.datatypesr collections.abcr r r rhr Zpolars._typingrrrr8r(rrr!rrRrLrerkr&r&r&r's6              #