Ë
    ðmxiŠ  ã            
      ó¸  — d dl mZ d dlmZ d dlmZmZmZmZm	Z	m
Z
 d dlmZmZmZmZmZmZmZmZmZmZ d dlmZmZ erd dlmZmZmZmZ d dlmZ g d¢Z  e
d	d
d¬«      Z!	 	 	 	 	 	 dd„Z" G d„ de	eef   «      Z# G d„ de#eef   e	eef   «      Z$ G d„ de#eef   e	eef   «      Z% G d„ de%eef   e	eee!f   «      Z& G d„ de&eee!f   e$eef   e	eee!f   «      Z'y)é    )Úannotations)Úchain)ÚTYPE_CHECKINGÚAnyÚCallableÚClassVarÚProtocolÚTypeVar)
ÚCompliantDataFrameTÚCompliantDataFrameT_coÚCompliantExprT_contraÚCompliantFrameTÚCompliantFrameT_coÚDepthTrackingExprAnyÚDepthTrackingExprT_contraÚEagerExprT_contraÚImplExprT_contraÚNarwhalsAggregation)Úis_sequence_ofÚ
zip_strict)ÚIterableÚIteratorÚMappingÚSequence)ÚImplExpr)ÚCompliantGroupByÚDepthTrackingGroupByÚEagerGroupByÚNativeAggregationT_cozstr | Callable[..., Any]T)ÚboundÚ	covariantc               óV   ‡ — ˆ fd„|D «       }t        t        j                  |«      «      S )Nc              3  ó@   •K  — | ]  }|j                  ‰«      –— Œ y ­w©N)Ú_evaluate_aliases)Ú.0ÚexprÚframes     €úS/home/htdocs/ttos/venv/lib/python3.12/site-packages/narwhals/_compliant/group_by.pyú	<genexpr>z$_evaluate_aliases.<locals>.<genexpr>$   s   øè ø€ Ò	:¨Dˆ$×
 Ñ
  ×
'Ñ	:ùs   ƒ)Úlistr   Úfrom_iterable)r(   ÚexprsÚits   `  r)   r%   r%   !   s%   ø€ ó 
;°EÔ	:€BÜ”×#Ñ# BÓ'Ó(Ð(ó    c                  ó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 r$   )r1   ©Úselfs    r)   Ú	compliantzCompliantGroupBy.compliant+   s   € à×$Ñ$Ð$r/   c               ó   — y r$   © )r4   Úcompliant_frameÚkeysÚdrop_null_keyss       r)   Ú__init__zCompliantGroupBy.__init__/   s   € ð r/   c                 ó   — y r$   r7   )r4   r-   s     r)   ÚaggzCompliantGroupBy.agg8   ó   � r/   N)Úreturnr   )r8   r   r9   z/Sequence[CompliantExprT_contra] | Sequence[str]r:   Úboolr?   ÚNone)r-   r   r?   r   )Ú__name__Ú
__module__Ú__qualname__Ú__annotations__Úpropertyr5   r;   r=   r7   r/   r)   r   r   (   sK   … ØÓàò%ó ð%ðà+ðð >ðð ðð 
óô Lr/   r   c                  ó   — e Zd Zdd„Zy)ÚDataFrameGroupByc                 ó   — y r$   r7   r3   s    r)   Ú__iter__zDataFrameGroupBy.__iter__?   r>   r/   N)r?   z,Iterator[tuple[Any, CompliantDataFrameT_co]])rB   rC   rD   rJ   r7   r/   r)   rH   rH   ;   s   „ ô Lr/   rH   c                  ó>   — e Zd Z	 	 	 	 	 	 dd„Ze	 	 	 	 	 	 dd„«       Zy)ÚParseKeysGroupByc                ó„   — t        |t        «      rt        |«      }|||j                  «       fS | j	                  ||¬«      S )N)r9   )r   Ústrr+   ÚcopyÚ_parse_expr_keys)r4   r8   r9   Úkeys_strs       r)   Ú_parse_keyszParseKeysGroupBy._parse_keysF   s@   € ô
 ˜$¤Ô$Ü˜D“zˆHØ" H¨h¯m©m«oÐ=Ð=Ø×$Ñ$ _¸4Ð$Ó@Ð@r/   c           
     ó  ‡	— t        d„ | j                  D «       «      dz   Š	dˆ	fd„}|D �cg c]  }|j                  | «      ‘Œ }}t        ||«      D ��cg c]b  \  }}|j                  x}r5|j
                  j                  «       r|j                  j                  |«      n|j                   ||d   «      «      ‘Œd }}} | j                  |Ž t        | |«      t        t        j                  |«      «      fS c c}w 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r$   )ÚlenrN   )r&   Úcs     r)   r*   z4ParseKeysGroupBy._parse_expr_keys.<locals>.<genexpr>b   s   è ø€ ÒK¨aœc¤# a£&ŸkÑKùs   ‚ é   c                óN   •— t        | «      }d|› dd‰t        |«      z
  dz
  z  › �S )NÚ_Ú_tmpé   )rN   rU   )ÚkeyÚkey_strÚtmp_name_lengths     €r)   Ú_temporary_namez:ParseKeysGroupBy._parse_expr_keys.<locals>._temporary_named   s3   ø€ ä˜#“hˆGØ�w�i˜t C¨?¼SÀ»\Ñ+IÈAÑ+MÑ$NÐ#OÐPÐPr/   r   )r\   rN   r?   rN   )ÚmaxÚcolumnsr%   r   Ú	_metadataÚexpansion_kindÚis_multi_outputÚnameÚmapÚaliasÚwith_columnsr+   r   r,   )
r8   r9   r_   r'   Úkeys_aliasesr\   Ú	new_namesÚmetadataÚ	safe_keysr^   s
            @r)   rP   z!ParseKeysGroupBy._parse_expr_keysP   s  ø€ ô$ ÑK°?×3JÑ3JÔKÓKÈaÑOˆõ	Qð
 MQÖQÀD˜×.Ñ.¨Õ?ÐQˆÐQô #-¨T°<Ó"@÷
ñ ��Yð  ŸM™MÐ)�Ð)¨x×/FÑ/F×/VÑ/VÔ/Xð �H‰H�L‰L˜Ô)ð —‘™?¨9°Q©<Ó8Ó9ñ:ð
ˆ	ñ 
ð )ˆO×(Ñ(¨)Ð4Ü˜o¨yÓ9Ü”×$Ñ$ \Ó2Ó3ð
ð 	
ùò Rùó
s   «C6ÁA'C;N)r8   r   r9   z*Sequence[ImplExprT_contra] | Sequence[str]r?   ú,tuple[CompliantFrameT, list[str], list[str]])r8   r   r9   zSequence[ImplExprT_contra]r?   rm   )rB   rC   rD   rR   ÚstaticmethodrP   r7   r/   r)   rL   rL   B   sS   „ ðAà(ðAð 9ðAð 
6ó	Að ð%
Ø(ð%
Ø0Jð%
à	5ò%
ó ñ%
r/   rL   c                  óv   — 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	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_simpler5   Ú_implementationre   ÚlowerÚ
ValueError)r4   r-   r'   re   Ú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_namerp   ©Úclsr'   s     r)   rr   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.
        )rp   Úget)r|   re   s     r)   Ú_remap_expr_namez%DepthTrackingGroupBy._remap_expr_name›   s   € ð �‰×"Ñ" 4¨Ó.Ð.r/   c               ó\   — t        |j                  j                  «       «      j                  S )z=Return the last function name in the chain defined by `expr`.)Únextrb   Úop_nodes_reversedre   r{   s     r)   rz   zDepthTrackingGroupBy._leaf_name¦   s"   € ô �D—N‘N×4Ñ4Ó6Ó7×<Ñ<Ð<r/   c               ó\   — t        |j                  j                  «       «      j                  S )z?Return the last function kwargs in the chain defined by `expr`.)r�   rb   r‚   Úkwargsr{   s     r)   Ú_kwargszDepthTrackingGroupBy._kwargs«   s"   € ô �D—N‘N×4Ñ4Ó6Ó7×>Ñ>Ð>r/   N)r-   z#Sequence[DepthTrackingExprT_contra]r?   rA   )r'   r   r?   r@   )re   úNarwhalsAggregation | Anyr?   r   )r'   r   r?   r†   )r'   r   r?   zdict[str, Any])rB   rC   rD   Ú__doc__rE   rw   Úclassmethodrr   r   rz   r…   r7   r/   r)   r   r   y   sz   … ñ qà<Ó<ðó&ð  òQó ðQð ð/Ø,ð/à	ò/ó ð/ð ò=ó ð=ð ò?ó ñ?r/   r   c                  ó   — e Zd Zy)r   N)rB   rC   rD   r7   r/   r)   r   r   ±   s   „ ð r/   r   N)r(   r   r-   z(Iterable[ImplExpr[CompliantFrameT, Any]]r?   z	list[str])(Ú
__future__r   Ú	itertoolsr   Útypingr   r   r   r   r	   r
   Únarwhals._compliant.typingr   r   r   r   r   r   r   r   r   r   Únarwhals._utilsr   r   Úcollections.abcr   r   r   r   Únarwhals._compliant.exprr   Ú__all__r   r%   r   rH   rL   r   r   r7   r/   r)   ú<module>r’      s;  ðÝ "å ß L× L÷÷ ÷ ÷ 7áßEÓEå1ò G€áØÐ#=ÈôÐ ð
)Øð)Ø#Kð)àó)ôL�xÐ 2Ð4IÐ IÑJô Lô&LØÐ+Ð-BÐBÑCØÐ#Ð%:Ð:Ñ;ôLô4
Ø�_Ð&6Ð6Ñ7Øˆ_Ð.Ð.Ñ/ô4
ôn5?Ø�_Ð&?Ð?Ñ@Øˆ_Ð7Ð9NÐNÑOô5?ôpØÐ,Ð.?ÐAVÐVÑWØÐ(Ð*;Ð;Ñ<ØÐ Ð"3Ð5JÐJÑKõr/   