§
    cŠtj³  ã            
      ó
  — d dl mZ d dlmZ d dl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 erd dlmZ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&dS ) é    )Úannotations)Úchain)ÚTYPE_CHECKINGÚAnyÚClassVarÚProtocolÚTypeVar)
ÚCompliantDataFrameTÚCompliantDataFrameT_coÚCompliantExprT_contraÚCompliantFrameTÚCompliantFrameT_coÚDepthTrackingExprAnyÚDepthTrackingExprT_contraÚEagerExprT_contraÚImplExprT_contraÚNarwhalsAggregation)Úis_sequence_of)ÚCallableÚIterableÚIteratorÚMappingÚSequence)ÚImplExpr)ÚCompliantGroupByÚDepthTrackingGroupByÚEagerGroupByÚNativeAggregationT_cozstr | Callable[..., Any]T)ÚboundÚ	covariantÚframer   Úexprsú(Iterable[ImplExpr[CompliantFrameT, Any]]Úreturnú	list[str]c               ób   ‡ — ˆ fd„|D ¦   «         }t          t          j        |¦  «        ¦  «        S )Nc              3  óB   •K  — | ]}|                      ‰¦  «        V — Œd S ©N©Ú_evaluate_aliases)Ú.0Úexprr!   s     €úZ/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/narwhals/_compliant/group_by.pyú	<genexpr>z$_evaluate_aliases.<locals>.<genexpr>$   s1   øè è € Ð	:Ð	:¨Dˆ$×
 Ò
  Ñ
'Ô
'Ð	:Ð	:Ð	:Ð	:Ð	:Ð	:ó    )Úlistr   Úfrom_iterable)r!   r"   Úits   `  r-   r*   r*   !   s7   ø€ ð 
;Ð	:Ð	:Ð	:°EÐ	:Ñ	:Ô	:€BÝ•Ô# BÑ'Ô'Ñ(Ô(Ð(r/   c                  óB   — e Zd ZU ded<   edd„¦   «         Zdd„Zdd„ZdS )r   r   Ú_compliant_framer$   r   c                ó   — | j         S r(   )r4   ©Úselfs    r-   Ú	compliantzCompliantGroupBy.compliant+   s   € àÔ$Ð$r/   Úcompliant_frameÚkeysú/Sequence[CompliantExprT_contra] | Sequence[str]Údrop_null_keysÚboolÚNonec              ó   — d S r(   © )r7   r9   r:   r<   s       r-   Ú__init__zCompliantGroupBy.__init__/   s	   € ð ˆsr/   r"   r   c                ó   — d S r(   r@   )r7   r"   s     r-   ÚaggzCompliantGroupBy.agg8   ó   € € € r/   N)r$   r   )r9   r   r:   r;   r<   r=   r$   r>   )r"   r   r$   r   )Ú__name__Ú
__module__Ú__qualname__Ú__annotations__Úpropertyr8   rA   rC   r@   r/   r-   r   r   (   s_   € € € € € € ØÐÐÑàð%ð %ð %ñ „Xð%ðð ð ð ð LÐKÐKÐKÐKÐKr/   r   c                  ó   — e Zd Zdd„ZdS )ÚDataFrameGroupByr$   ú,Iterator[tuple[Any, CompliantDataFrameT_co]]c                ó   — d S r(   r@   r6   s    r-   Ú__iter__zDataFrameGroupBy.__iter__?   rD   r/   N)r$   rL   )rE   rF   rG   rN   r@   r/   r-   rK   rK   ;   s   € € € € € ð LÐKÐKÐKÐKÐKr/   rK   c                  ó.   — e Zd Zdd„Zedd	„¦   «         Zd
S )ÚParseKeysGroupByr9   r   r:   ú*Sequence[ImplExprT_contra] | Sequence[str]r$   ú,tuple[CompliantFrameT, list[str], list[str]]c                ó¦   — t          |t          ¦  «        r&t          |¦  «        }|||                     ¦   «         fS |                      ||¬¦  «        S )N)r:   )r   Ústrr0   ÚcopyÚ_parse_expr_keys)r7   r9   r:   Úkeys_strs       r-   Ú_parse_keyszParseKeysGroupBy._parse_keysF   sO   € õ
 ˜$¥Ñ$Ô$ð 	>Ý˜D‘z”zˆHØ" H¨h¯mªm©o¬oÐ=Ð=Ø×$Ò$ _¸4Ð$Ñ@Ô@Ð@r/   úSequence[ImplExprT_contra]c                ó&  ‡ ‡‡‡— t          d„ ‰ j        D ¦   «         ¦  «        dz   Šdˆfd„Šˆ fd„|D ¦   «         }ˆˆfd„t          ||d	¬
¦  «        D ¦   «         } ‰ j        |Ž t	          ‰ |¦  «        t          t          j        |¦  «        ¦  «        fS )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  óN   K  — | ] }t          t          |¦  «        ¦  «        V — Œ!d S r(   )ÚlenrT   )r+   Úcs     r-   r.   z4ParseKeysGroupBy._parse_expr_keys.<locals>.<genexpr>b   s.   è è € ÐKÐK¨a�c¥# a¡&¤&™kœkÐKÐKÐKÐKÐKÐKr/   é   ÚkeyrT   r$   c                ó^   •— t          | ¦  «        }d|› dd‰t          |¦  «        z
  dz
  z  › �S )NÚ_Ú_tmpé   )rT   r\   )r_   Úkey_strÚtmp_name_lengths     €r-   Ú_temporary_namez:ParseKeysGroupBy._parse_expr_keys.<locals>._temporary_named   s9   ø€ å˜#‘h”hˆGØP�wÐPÐP C¨?½SÀ¹\¼\Ñ+IÈAÑ+MÑ$NÐPÐPÐPr/   c                ó:   •— g | ]}|                      ‰¦  «        ‘ŒS r@   r)   )r+   r,   r9   s     €r-   ú
<listcomp>z5ParseKeysGroupBy._parse_expr_keys.<locals>.<listcomp>i   s'   ø€ ÐQÐQÐQÀD˜×.Ò.¨Ñ?Ô?ÐQÐQÐQr/   c                óÖ   •— g | ]e\  }}|j         xŠr3‰j                             ¦   «         r|j                             ‰¦  «        n#|                      ‰|d          ¦  «        ¦  «        ‘ŒfS )r   )Ú	_metadataÚexpansion_kindÚis_multi_outputÚnameÚmapÚalias)r+   r_   Ú	new_namesrf   Úmetadatas      €€r-   rh   z5ParseKeysGroupBy._parse_expr_keys.<locals>.<listcomp>j   s„   ø€ ð 
ð 
ð 
ñ ��Yð  œMÐ)�ð:Ø/7Ô/F×/VÒ/VÑ/XÔ/Xð:ˆCŒH�LŠL˜Ñ)Ô)Ð)ð —’˜?˜?¨9°Q¬<Ñ8Ô8Ñ9Ô9ð
ð 
ð 
r/   T)Ústrict)r_   rT   r$   rT   )ÚmaxÚcolumnsÚzipÚwith_columnsr*   r0   r   r1   )r9   r:   Úkeys_aliasesÚ	safe_keysrf   rq   re   s   `   @@@r-   rV   z!ParseKeysGroupBy._parse_expr_keysP   sæ   øøøø€ õ$ ÐKÐK°?Ô3JÐKÑKÔKÑKÔKÈaÑOˆð	Qð 	Qð 	Qð 	Qð 	Qð 	Qð
 RÐQÐQÐQÈDÐQÑQÔQˆð
ð 
ð 
ð 
ð 
õ #& d¨LÀÐ"FÑ"FÔ"Fð
ñ 
ô 
ˆ	ð )ˆOÔ(¨)Ð4Ý˜o¨yÑ9Ô9Ý•Ô$ \Ñ2Ô2Ñ3Ô3ð
ð 	
r/   N)r9   r   r:   rQ   r$   rR   )r9   r   r:   rY   r$   rR   )rE   rF   rG   rX   ÚstaticmethodrV   r@   r/   r-   rP   rP   B   sL   € € € € € ðAð Að Að Að ð%
ð %
ð %
ñ „\ð%
ð %
ð %
r/   rP   c                  óˆ   — 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
dS )r   zf`CompliantGroupBy` variant, deals with `Eager` and other backends that utilize `CompliantExpr._depth`.z+ClassVar[Mapping[NarwhalsAggregation, Any]]Ú_REMAP_AGGSr"   ú#Sequence[DepthTrackingExprT_contra]r$   r>   c                óª   — |D ]O}|                       |¦  «        s8| j        j        j                             ¦   «         }d|›d�}t          |¦  «        ‚ŒPd S )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_simpler8   Ú_implementationrm   ÚlowerÚ
ValueError)r7   r"   r,   rm   Úmsgs        r-   Ú_ensure_all_simplez'DepthTrackingGroupBy._ensure_all_simple†   s{   € Øð 	&ð 	&ˆDØ—?’? 4Ñ(Ô(ð &Ø”~Ô5Ô:×@Ò@ÑBÔB�ðjàðjð jð jð õ ! ‘o”oÐ%ð&ð	&ð 	&r/   r,   r   r=   c               ób   — |                      ¦   «         o|                      |¦  «        | j        v S )zNReturn `True` is we can efficiently use `expr` in a native `group_by` context.)Ú_is_elementaryÚ
_leaf_namer{   ©Úclsr,   s     r-   r~   zDepthTrackingGroupBy._is_simple–   s.   € ð ×"Ò"Ñ$Ô$ÐP¨¯ª¸Ñ)=Ô)=ÀÄÐ)PÐPr/   rm   úNarwhalsAggregation | Anyr   c               ó<   — | j                              ||¦  «        }|S )z‡Replace `name`, with some native representation.

        Arguments:
            name: Name of a `nw.Expr` aggregation method.
        )r{   Úget)rˆ   rm   Úresults      r-   Ú_remap_expr_namez%DepthTrackingGroupBy._remap_expr_name›   s    € ð ),¬×(;Ò(;¸DÀ$Ñ(GÔ(GˆØˆr/   c               óX   — t          |j                             ¦   «         ¦  «        j        S )z=Return the last function name in the chain defined by `expr`.)Únextrj   Úop_nodes_reversedrm   r‡   s     r-   r†   zDepthTrackingGroupBy._leaf_name§   s#   € õ �D”N×4Ò4Ñ6Ô6Ñ7Ô7Ô<Ð<r/   údict[str, Any]c               óX   — t          |j                             ¦   «         ¦  «        j        S )z?Return the last function kwargs in the chain defined by `expr`.)r�   rj   r�   Úkwargsr‡   s     r-   Ú_kwargszDepthTrackingGroupBy._kwargs¬   s#   € õ �D”N×4Ò4Ñ6Ô6Ñ7Ô7Ô>Ð>r/   N)r"   r|   r$   r>   )r,   r   r$   r=   )rm   r‰   r$   r   )r,   r   r$   r‰   )r,   r   r$   r‘   )rE   rF   rG   Ú__doc__rH   rƒ   Úclassmethodr~   r�   r†   r”   r@   r/   r-   r   r   y   sÈ   € € € € € € ð qÐpà<Ð<Ð<Ñ<ðð&ð &ð &ð &ð  ðQð Qð Qñ „[ðQð ð	ð 	ð 	ñ „[ð	ð ð=ð =ð =ñ „[ð=ð ð?ð ?ð ?ñ „[ð?ð ?ð ?r/   r   c                  ó   — e Zd ZdS )r   N)rE   rF   rG   r@   r/   r-   r   r   ²   s   € € € € € ð €3r/   r   N)r!   r   r"   r#   r$   r%   )'Ú
__future__r   Ú	itertoolsr   Útypingr   r   r   r   r	   Únarwhals._compliant.typingr
   r   r   r   r   r   r   r   r   r   Únarwhals._utilsr   Úcollections.abcr   r   r   r   r   Únarwhals._compliant.exprr   Ú__all__r   r*   r   rK   rP   r   r   r@   r/   r-   ú<module>r       sâ  ðØ "Ð "Ð "Ð "Ð "Ð "à Ð Ð Ð Ð Ð Ø BÐ BÐ BÐ BÐ BÐ BÐ BÐ BÐ BÐ BÐ BÐ BÐ BÐ Bðð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð +Ð *Ð *Ð *Ð *Ð *àð 2ØOÐOÐOÐOÐOÐOÐOÐOÐOÐOÐOÐOÐOÐOà1Ð1Ð1Ð1Ð1Ð1ð GÐ
FÐ
F€à˜ØÐ#=Èðñ ô Ð ð
)ð )ð )ð )ðLð Lð Lð Lð L�xÐ 2Ð4IÐ IÔJñ Lô Lð Lð&Lð Lð Lð Lð LØÐ+Ð-BÐBÔCØÐ#Ð%:Ð:Ô;ñLô Lð Lð4
ð 4
ð 4
ð 4
ð 4
Ø�_Ð&6Ð6Ô7Øˆ_Ð.Ð.Ô/ñ4
ô 4
ð 4
ðn6?ð 6?ð 6?ð 6?ð 6?Ø�_Ð&?Ð?Ô@Øˆ_Ð7Ð9NÐNÔOñ6?ô 6?ð 6?ðrð ð ð ð ØÐ,Ð.?ÐAVÐVÔWØÐ(Ð*;Ð;Ô<ØÐ Ð"3Ð5JÐJÔKñô ð ð ð r/   