o
    ˜¨Êh   ã                   @   sæ   d Z ddlmZ ddlmZmZmZmZ ddlm	Z	 ddl
mZ ddlmZmZ g d¢ZG dd	„ d	eƒZG d
d„ deeeƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deeƒZG dd„ deeƒZG dd„ deeƒZdS )z>
Classes to represent the definitions of aggregate functions.
é    )Ú
FieldError)ÚCaseÚFuncÚStarÚWhen)ÚIntegerField)ÚCoalesce)ÚFixDurationInputMixinÚNumericOutputFieldMixin)Ú	AggregateÚAvgÚCountÚMaxÚMinÚStdDevÚSumÚVariancec                       s¤   e Zd ZdZdZdZdZdZdZdZ	ddddœ‡ fdd„
Z
‡ fd	d
„Z‡ fdd„Z‡ fdd„Z	d‡ fdd„	Zedd„ ƒZddd„Z‡ fdd„Z‡ fdd„Z‡  ZS )r   z)%(function)s(%(distinct)s%(expressions)s)TNz%s FILTER (WHERE %%(filter)s)F)ÚdistinctÚfilterÚdefaultc                   sf   |r| j std| jj ƒ‚|d ur| jd urt| jj› d�ƒ‚|| _|| _|| _tƒ j	|i |¤Ž d S )Nz%s does not allow distinct.z does not allow default.)
Úallow_distinctÚ	TypeErrorÚ	__class__Ú__name__Úempty_result_set_valuer   r   r   ÚsuperÚ__init__)Úselfr   r   r   ÚexpressionsÚextra©r   © úM/var/www/html/env/lib/python3.10/site-packages/django/db/models/aggregates.pyr   "   s   
zAggregate.__init__c                    s   dd„ t ƒ  ¡ D ƒS )Nc                 S   s   g | ]}|j ‘qS r!   )Ú_output_field_or_none)Ú.0Úer!   r!   r"   Ú
<listcomp>0   s    z/Aggregate.get_source_fields.<locals>.<listcomp>)r   Úget_source_expressions©r   r    r!   r"   Úget_source_fields.   s   zAggregate.get_source_fieldsc                    s    t ƒ  ¡ }| jr|| jg S |S ©N)r   r'   r   )r   Úsource_expressionsr    r!   r"   r'   2   s   
z Aggregate.get_source_expressionsc                    s   | j o| ¡ | _ tƒ  |¡S r*   )r   Úpopr   Úset_source_expressions)r   Úexprsr    r!   r"   r-   8   s   z Aggregate.set_source_expressionsc                    sÜ   t ƒ  ||||¡}|jo|j ||||¡|_|sGt t|ƒ ¡ }t|ƒD ]#\}}	|	jrF|  ¡ | }
t|
dƒr8|
jnt	|
ƒ}t
d|j||f ƒ‚q#|j }d u rP|S t|dƒr]| ||||¡}d |_t|||jd�}|j|_|S )NÚnamez-Cannot compute %s('%s'): '%s' is an aggregateÚresolve_expression)Úoutput_field)r   r0   r   r   r'   Ú	enumerateÚcontains_aggregateÚhasattrr/   Úreprr   r   r   r#   Ú
is_summary)r   ÚqueryÚallow_joinsÚreuseÚ	summarizeÚfor_saveÚcr   ÚindexÚexprÚbefore_resolvedr/   r   Úcoalescer    r!   r"   r0   <   s6   ÿÿý
ÿÿù
zAggregate.resolve_expressionc                 C   sB   |   ¡ }t|ƒdkrt|d dƒrd|d j| j ¡ f S tdƒ‚)Né   r   r/   z%s__%sz$Complex expressions require an alias)r'   Úlenr4   r/   Úlowerr   )r   r   r!   r!   r"   Údefault_alias]   s   zAggregate.default_aliasc                 C   s   g S r*   r!   )r   Úaliasr!   r!   r"   Úget_group_by_colsd   s   zAggregate.get_group_by_colsc                    sì   | j rdnd|d< | jrk|jjr<| j ||¡\}}| j| d| j¡ }tƒ j||f||dœ|¤Ž\}}|g |¢|¢R fS |  	¡ }	d |	_|	 
¡ }
t| j|
d d�}|	 t|ƒg|
dd …  ¡ tt|	ƒj||fi |¤ŽS tƒ j||fi |¤ŽS )	Nz	DISTINCT Ú r   Útemplate)rH   r   r   )ÚthenrA   )r   r   ÚfeaturesÚ supports_aggregate_filter_clauseÚas_sqlÚfilter_templateÚgetrH   r   Úcopyr'   r   r-   r   r   )r   ÚcompilerÚ
connectionÚextra_contextÚ
filter_sqlÚfilter_paramsrH   ÚsqlÚparamsrO   r+   Ú	conditionr    r!   r"   rL   g   s8   ÿþü
û
ÿÿzAggregate.as_sqlc                    s.   t ƒ  ¡ }| jr| j|d< | jr| j|d< |S )Nr   r   )r   Ú_get_repr_optionsr   r   )r   Úoptionsr    r!   r"   rX   ‚   s   


zAggregate._get_repr_options)NTNFFr*   )r   Ú
__module__Ú__qualname__rH   r3   r/   rM   Úwindow_compatibler   r   r   r)   r'   r-   r0   ÚpropertyrD   rF   rL   rX   Ú__classcell__r!   r!   r    r"   r      s(    ÿÿ!

r   c                   @   ó   e Zd ZdZd ZdZdS )r   ÚAVGTN©r   rZ   r[   Úfunctionr/   r   r!   r!   r!   r"   r   ‹   ó    r   c                       s4   e Zd ZdZd Zeƒ ZdZdZd‡ fdd„	Z	‡  Z
S )r   ÚCOUNTTr   Nc                    sD   |dkrt ƒ }t|t ƒr|d urtdƒ‚tƒ j|fd|i|¤Ž d S )NÚ*z8Star cannot be used with filter. Please specify a field.r   )r   Ú
isinstanceÚ
ValueErrorr   r   )r   Ú
expressionr   r   r    r!   r"   r   ˜   s
   zCount.__init__r*   )r   rZ   r[   rb   r/   r   r1   r   r   r   r^   r!   r!   r    r"   r   ‘   s    r   c                   @   ó   e Zd ZdZd ZdS )r   ÚMAXN©r   rZ   r[   rb   r/   r!   r!   r!   r"   r       ó    r   c                   @   ri   )r   ÚMINNrk   r!   r!   r!   r"   r   ¥   rl   r   c                       ó.   e Zd Zd Zd‡ fdd„	Z‡ fdd„Z‡  ZS )r   Fc                    ó&   |rdnd| _ tƒ j|fi |¤Ž d S )NÚSTDDEV_SAMPÚ
STDDEV_POP©rb   r   r   ©r   rh   Úsampler   r    r!   r"   r   ­   ó   zStdDev.__init__c                    ó   i t ƒ  ¡ ¥d| jdki¥S )Nrt   rp   ©r   rX   rb   r(   r    r!   r"   rX   ±   ó   zStdDev._get_repr_options©F©r   rZ   r[   r/   r   rX   r^   r!   r!   r    r"   r   ª   ó    r   c                   @   r_   )r   ÚSUMTNra   r!   r!   r!   r"   r   µ   rc   r   c                       rn   )r   Fc                    ro   )NÚVAR_SAMPÚVAR_POPrr   rs   r    r!   r"   r   ¾   ru   zVariance.__init__c                    rv   )Nrt   r}   rw   r(   r    r!   r"   rX   Â   rx   zVariance._get_repr_optionsry   rz   r!   r!   r    r"   r   »   r{   r   N)Ú__doc__Údjango.core.exceptionsr   Údjango.db.models.expressionsr   r   r   r   Údjango.db.models.fieldsr   Ú%django.db.models.functions.comparisonr   Ú!django.db.models.functions.mixinsr	   r
   Ú__all__r   r   r   r   r   r   r   r   r!   r!   r!   r"   Ú<module>   s    r