Ë
    z�h¿"  ã            
      ó*  — U d dl mZ d dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlm	Z	 d dlm
Z
 d d	lmZ d d
lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ erd dlm Z   edee«      Z!esejD                  dk\  rd dlm#Z$ nd dlm%Z$ nd dlm#Z$ g d ¢Z& ed!d"d#¬$«      Z'ed%   Z(d&e)d'<    ejT                  d(«      Z+d)e)d*<    G d+„ d,e$eef   «      Z, G d-„ d.e,eef   e$eef   «      Z- G d/„ d0e,eef   e$eef   «      Z. G d1„ d2e.eef   e$eee'f   «      Z/ G d3„ d4e/eee0f   e-eef   e$eef   «      Z1 G d5„ d6e.eef   e,eef   e$eeef   «      Z2y)7é    )ÚannotationsN)ÚTYPE_CHECKING)ÚAny)ÚCallable)ÚClassVar)ÚIterable)ÚIterator)ÚLiteral)ÚMapping)ÚSequence)ÚTypeVar)ÚCompliantDataFrameAny)ÚCompliantDataFrameT)ÚCompliantDataFrameT_co)ÚCompliantExprT_contra)ÚCompliantFrameT)ÚCompliantFrameT_co)ÚCompliantLazyFrameAny)ÚCompliantLazyFrameT)ÚDepthTrackingExprAny)ÚDepthTrackingExprT_contra)ÚEagerExprT_contra)ÚLazyExprT_contra)ÚNativeExprT_co)Úis_sequence_of)Ú	TypeAliasÚ_SameFrameT)é   é	   )ÚProtocol)ÚGeneric)ÚCompliantGroupByÚDepthTrackingGroupByÚEagerGroupByÚLazyGroupByÚNarwhalsAggregationÚNativeAggregationT_cozstr | Callable[..., Any]T)ÚboundÚ	covariant)
ÚsumÚmeanÚmedianÚmaxÚminÚstdÚvarÚlenÚn_uniqueÚcountr   r&   z(\w+->)zre.Pattern[str]Ú_RE_LEAF_NAMEc                  óJ   — e Zd ZU ded<   edd„«       Z	 	 	 	 	 	 	 	 dd„Zd	d„Zy)
r"   r   Ú_compliant_framec                ó   — | j                   S ©N)r6   ©Úselfs    úZ/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/narwhals/_compliant/group_by.pyÚ	compliantzCompliantGroupBy.compliantE   s   € à×$Ñ$Ð$ó    c               ó   — y r8   © )r:   Úcompliant_frameÚkeysÚdrop_null_keyss       r;   Ú__init__zCompliantGroupBy.__init__I   s   € ð r=   c                 ó   — y r8   r?   )r:   Úexprss     r;   ÚaggzCompliantGroupBy.aggR   ó   � r=   N)Úreturnr   )r@   r   rA   ú/Sequence[CompliantExprT_contra] | Sequence[str]rB   ÚboolrH   ÚNone)rE   r   rH   r   )Ú__name__Ú
__module__Ú__qualname__Ú__annotations__Úpropertyr<   rC   rF   r?   r=   r;   r"   r"   B   sK   … ØÓàò%ó ð%ðà+ðð >ðð ðð 
óô Lr=   r"   c                  ó   — e Zd Zdd„Zy)ÚDataFrameGroupByc                 ó   — y r8   r?   r9   s    r;   Ú__iter__zDataFrameGroupBy.__iter__Y   rG   r=   N)rH   z,Iterator[tuple[Any, CompliantDataFrameT_co]])rL   rM   rN   rT   r?   r=   r;   rR   rR   U   s   „ ô Lr=   rR   c                  ó>   — e Zd Z	 	 	 	 	 	 dd„Ze	 	 	 	 	 	 dd„«       Zy)ÚParseKeysGroupByc                ó„   — t        |t        «      rt        |«      }|||j                  «       fS | j	                  ||¬«      S )N)rA   )r   ÚstrÚlistÚcopyÚ_parse_expr_keys)r:   r@   rA   Úkeys_strs       r;   Ú_parse_keyszParseKeysGroupBy._parse_keys`   s@   € ô
 ˜$¤Ô$Ü˜D“zˆHØ" H¨h¯m©m«oÐ=Ð=à×(Ñ(¨¸tÐ(ÓDÐDr=   c                ó–  — dt        d„ | j                  D «       «      dz   z  } | j                  |Ž }t        ||«      D ��cg c]]  \  }}|j                  x}r5|j
                  j                  «       r|j                  j                  |«      n|j                  |› |› �«      ‘Œ_ }}} | j                  |Ž  | j                  |Ž |fS c c}}w )a#  Parses key expressions to set up `.agg` operation with correct information.

        Since keys are expressions, it's possible to alias any such key to match
        other dataframe column names.

        In order to match polars behavior and not overwrite columns when evaluating keys:

        - We evaluate what the output key names should be, in order to remap temporary column
            names to the expected ones, and to exclude those from unnamed expressions in
            `.agg(...)` context (see https://github.com/narwhals-dev/narwhals/pull/2325#issuecomment-2800004520)
        - Create temporary names for evaluated key expressions that are guaranteed to have
            no overlap with any existing column name.
        - Add these temporary columns to the compliant dataframe.
        Ú_c              3  óD   K  — | ]  }t        t        |«      «      –— Œ y ­wr8   )r1   rX   )Ú.0Úcs     r;   ú	<genexpr>z4ParseKeysGroupBy._parse_expr_keys.<locals>.<genexpr>}   s   è ø€ Ò!O°!¤#¤c¨!£f§+Ñ!Oùs   ‚ é   )r-   ÚcolumnsÚ_evaluate_aliasesÚzipÚ	_metadataÚexpansion_kindÚis_multi_outputÚnameÚsuffixÚaliasÚwith_columns)r@   rA   Úsuffix_tokenÚoutput_namesÚkeyÚnew_nameÚmetadataÚ	safe_keyss           r;   r[   z!ParseKeysGroupBy._parse_expr_keysk   sâ   € ð$ œcÑ!O°×7NÑ7NÔ!OÓOÐRSÑSÑTˆØ8�×8Ñ8¸$Ð?ˆô "% T¨<Ó!8÷
ñ ��Xð  ŸM™MÐ)�Ð)¨x×/FÑ/F×/VÑ/VÔ/Xð �H‰H�O‰O˜LÔ)ð —‘˜h˜Z¨ ~Ð6Ó7ñ8ð
ˆ	ñ 
ð )ˆO×(Ñ(¨)Ð4Ø-ˆO×-Ñ-¨yÐ9Øð
ð 	
ùó
s   ÁA"CN)r@   r   rA   rI   rH   z,tuple[CompliantFrameT, list[str], list[str]])r@   r   rA   zSequence[CompliantExprT_contra]rH   z(tuple[_SameFrameT, list[str], list[str]])rL   rM   rN   r]   Ústaticmethodr[   r?   r=   r;   rV   rV   \   sS   „ ð	Eà(ð	Eð >ð	Eð 
6ó		Eð ð 
Ø$ð 
Ø,Kð 
à	1ò 
ó ñ 
r=   rV   c                  ód   — e Zd ZU dZded<   	 d	d„Zed
d„«       Ze	 	 	 	 dd„«       Zedd„«       Z	y)r#   zf`CompliantGroupBy` variant, deals with `Eager` and other backends that utilize `CompliantExpr._depth`.z+ClassVar[Mapping[NarwhalsAggregation, Any]]Ú_REMAP_AGGSc                ó²   — |D ]R  }| j                  |«      rŒ| j                  j                  j                  j	                  «       }d|›d�}t        |«      ‚ y )NzsNon-trivial complex aggregation found.

Hint: you were probably trying to apply a non-elementary aggregation with azú table.
Please rewrite your query such that group-by aggregations are elementary. For example, instead of:

    df.group_by('a').agg(nw.col('b').round(2).mean())

use:

    df.with_columns(nw.col('b').round(2)).group_by('a').agg(nw.col('b').mean())

)Ú
_is_simpler<   Ú_implementationrk   ÚlowerÚ
ValueError)r:   rE   Úexprrk   Úmsgs        r;   Ú_ensure_all_simplez'DepthTrackingGroupBy._ensure_all_simpleœ   sc   € Øò 	&ˆDØ—?‘? 4Õ(Ø—~‘~×5Ñ5×:Ñ:×@Ñ@ÓB�ðbà�hð jðjð ô ! “oÐ%ñ	&r=   c               ó`   — |j                  «       xr | j                  |«      | j                  v S )zNReturn `True` is we can efficiently use `expr` in a native `group_by` context.)Ú_is_elementaryÚ
_leaf_namerw   ©Úclsr}   s     r;   ry   zDepthTrackingGroupBy._is_simple¬   s*   € ð ×"Ñ"Ó$ÒP¨¯©¸Ó)=ÀÇÁÐ)PÐPr=   c               ó:   — | j                   j                  ||«      S )zÉReplace `name`, with some native representation.

        Arguments:
            name: Name of a `nw.Expr` aggregation method.

        Returns:
            A native compatible representation.
        )rw   Úget)r„   rk   s     r;   Ú_remap_expr_namez%DepthTrackingGroupBy._remap_expr_name±   s   € ð �‰×"Ñ" 4¨Ó.Ð.r=   c               óB   — t         j                  d|j                  «      S )z=Return the last function name in the chain defined by `expr`.Ú )r4   ÚsubÚ_function_namerƒ   s     r;   r‚   zDepthTrackingGroupBy._leaf_name¿   s   € ô × Ñ   T×%8Ñ%8Ó9Ð9r=   N)rE   z#Sequence[DepthTrackingExprT_contra]rH   rK   )r}   r   rH   rJ   )rk   úNarwhalsAggregation | AnyrH   r'   )r}   r   rH   rŒ   )
rL   rM   rN   Ú__doc__rO   r   Úclassmethodry   r‡   r‚   r?   r=   r;   r#   r#   �   sf   … ñ qà<Ó<ðó&ð  òQó ðQð ð/Ø,ð/à	ò/ó ð/ð ò:ó ñ:r=   r#   c                  ó   — e Zd Zy)r$   N)rL   rM   rN   r?   r=   r;   r$   r$   Å   s   „ ð r=   r$   c                  ó:   — e Zd ZU ded<   ded<   dd„Z	 	 	 	 dd„Zy)	r%   z	list[str]Ú_keysÚ_output_key_namesc             #  ó¤  K  — |j                  | j                  «      }|j                  r|j                  |«      n|} || j                  «      }|j                  «       rIh | j                  £| j
                  £}t        |||«      D ]  \  }}}||vsŒ|j                  |«      –— Œ  y t        ||«      D ]  \  }}|j                  |«      –— Œ y ­wr8   )Ú_evaluate_output_namesr<   Ú_alias_output_namesÚ_is_multi_output_unnamedr‘   r’   rg   rm   )	r:   r}   rp   ÚaliasesÚnative_exprsÚexcludeÚnative_exprrk   rm   s	            r;   Ú_evaluate_exprzLazyGroupBy._evaluate_exprÔ   sØ   è ø€ Ø×2Ñ2°4·>±>ÓBˆð ×'Ò'ð ×$Ñ$ \Ô2àð 	ñ
 ˜DŸN™NÓ+ˆØ×(Ñ(Ô*Ø<˜Ÿ
™
Ð< T×%;Ñ%;Ð<ˆGÜ,/°¸lÈGÓ,Tò 3Ñ(�˜T 5Ø˜wÒ&Ø%×+Ñ+¨EÓ2Ó2ñ3ô '*¨,¸Ó&@ò /Ñ"�˜UØ!×'Ñ'¨Ó.Ó.ñ/ùs   ‚BCÂ>Cc             #  óP   K  — |D ]  }| j                  |«      E d {  –—†  Œ y 7 Œ­wr8   )r›   )r:   rE   r}   s      r;   Ú_evaluate_exprszLazyGroupBy._evaluate_exprså   s.   è ø€ ð ò 	1ˆDØ×*Ñ*¨4Ó0×0Ñ0ñ	1Ø0ús   ‚&œ$�&N)r}   r   rH   úIterator[NativeExprT_co])rE   zIterable[LazyExprT_contra]rH   rž   )rL   rM   rN   rO   r›   r�   r?   r=   r;   r%   r%   Ì   s*   … ð
 ÓØ Ó ó/ð"1Ø/ð1à	!ô1r=   r%   )3Ú
__future__r   ÚreÚsysÚtypingr   r   r   r   r   r	   r
   r   r   r   Únarwhals._compliant.typingr   r   r   r   r   r   r   r   r   r   r   r   r   Únarwhals.utilsr   Útyping_extensionsr   r   Úversion_infor    Ú
Protocol38r!   Ú__all__r'   r&   rO   Úcompiler4   r"   rR   rV   r#   rX   r$   r%   r?   r=   r;   ú<module>rª      sÛ  ðÞ "ã 	Û 
Ý  Ý Ý Ý Ý Ý Ý Ý Ý Ý å <Ý :Ý =Ý <Ý 6Ý 9Ý <Ý :Ý ;Ý @Ý 8Ý 7Ý 5Ý )áÝ+á˜-Ð)>Ð@UÓV€Kñ Ø
×Ñ˜6Ò!Þ1æ0õ .ò€ñ  ØÐ#=ÈôÐ ð ")ØSñ"Ð �Yó ð
 ", §¡¨JÓ!7€ˆÓ 7ôL�zÐ"4Ð6KÐ"KÑLô Lô&LØÐ+Ð-BÐBÑCØÐ%Ð'<Ð<Ñ=ôLô0
Ø�_Ð&;Ð;Ñ<ØˆÐ 5Ð5Ñ6ô0
ôf3:Ø�_Ð&?Ð?Ñ@ØˆÐ 9Ð;PÐPÑQô3:ôlØÐ,Ð.?ÀÐDÑEØÐ(Ð*;Ð;Ñ<ØÐ"Ð$5Ð5Ñ6ôô1ØÐ(Ð*:Ð:Ñ;ØÐ(Ð*:Ð:Ñ;ØÐ"Ð$4°nÐDÑEõ1r=   