U
    Ha=Z                     @   s  d dl Z d dlZd dlZ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 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 G dd dZG dd dee	ZG dd deZG dd dZG dd deZ G dd deeZ!ej"G dd deeZ#ej"G dd deZ$ej"G dd deeZ%ej"G dd deeZ&ej"G dd  d eeZ'ej"G d!d" d"eeZ(G d#d$ d$Z)ej"G d%d& d&e)e&Z*ej"G d'd( d(e)e'Z+ej"G d)d* d*e eZ,G d+d, d,eZ-ej"G d-d. d.e-Z.ej"G d/d0 d0e.Z/ej"G d1d2 d2e-Z0ej"G d3d4 d4e0Z1ej"G d5d6 d6e-Z2ej"G d7d8 d8e2Z3ej"G d9d: d:e eZ4ej"G d;d< d<eZ5ej"G d=d> d>eZ6ej"G d?d@ d@e6Z7G dAdB dBeZ8G dCdD dDe8e#Z9G dEdF dFe8e%Z:G dGdH dHe8e&Z;G dIdJ dJe8e'Z<G dKdL dLe8e(Z=G dMdN dNZ>ej"G dOdP dPe>e$Z?ej"G dQdR dRe>e.Z@ej"G dSdT dTe>e/ZAej"G dUdV dVe>e0ZBej"G dWdX dXe>e1ZCej"G dYdZ dZe>e2ZDej"G d[d\ d\e>e3ZEdS )]    N)copy)EmptyResultSet)CaseExistsFuncValueWhen)	CharFieldDateTimeFieldFieldIntegerField	UUIDField)RegisterLookupMixin)
OrderedSet)RemovedInDjango40Warning)cached_property)make_hashablec                   @   s   e Zd ZdZdZdZdd Zdd Zd,dd	Zd
d Z	dd Z
dd Zdd Zd-ddZdd Zdd Zdd Zd.ddZdd Zdd Zed d! Zed"d# Zed$d% Zed&d' Zd(d) Zd*d+ ZdS )/LookupNTFc                 C   s`   || | _ | _|  | _t| j dr0| j  }ng }|rVddlm} t||rVtd|| _	d S )Nget_bilateral_transformsr   QueryzBBilateral transformations on nested querysets are not implemented.)
lhsrhsget_prep_lookuphasattrr   django.db.models.sql.queryr   
isinstanceNotImplementedErrorbilateral_transforms)selfr   r   r   r    r    </tmp/pip-unpacked-wheel-3jxiddxt/django/db/models/lookups.py__init__   s    

zLookup.__init__c                 C   s   | j D ]}||}q|S N)r   )r   valueZ	transformr    r    r!   apply_bilateral_transforms(   s    

z!Lookup.apply_bilateral_transformsc                 C   s   |d kr| j }| jrrg g  }}|D ]L}t|| jjd}| |}||j}||\}}	|	| |
|	 q"n$| ||\}
}dgt| | }}||fS )Noutput_field%s)r   r   r   r   r'   r%   resolve_expressionquerycompileappendextendget_db_prep_lookuplen)r   compiler
connectionr   sqlssqls_paramspr$   sqlZ
sql_params_paramsr    r    r!   batch_process_rhs-   s    


zLookup.batch_process_rhsc                 C   s   |   r| jgS | j| jgS r#   )rhs_is_direct_valuer   r   r   r    r    r!   get_source_expressions>   s    zLookup.get_source_expressionsc                 C   s(   t |dkr|d | _n|\| _| _d S )N   r   )r/   r   r   )r   Z	new_exprsr    r    r!   set_source_expressionsC   s    zLookup.set_source_expressionsc                 C   s<   t | jdr| jS | jr6t | jjdr6| jj| jS | jS Nr)   get_prep_value)r   r   prepare_rhsr   r'   r?   r:   r    r    r!   r   I   s
    zLookup.get_prep_lookupc                 C   s
   d|gfS )Nr(   r    )r   r$   r1   r    r    r!   r.   P   s    zLookup.get_db_prep_lookupc                 C   s*   |p| j }t|dr ||j}||S )Nr)   )r   r   r)   r*   r+   )r   r0   r1   r   r    r    r!   process_lhsS   s    

zLookup.process_lhsc                 C   s^   | j }| jr:|  r$t|| jjd}| |}||j}t	|drN|
|S | ||S d S )Nr&   as_sql)r   r   r9   r   r   r'   r%   r)   r*   r   r+   r.   )r   r0   r1   r$   r    r    r!   process_rhsY   s    


zLookup.process_rhsc                 C   s   t | jd S NrB   )r   r   r:   r    r    r!   r9   g   s    zLookup.rhs_is_direct_valuec                 C   s4   t | }|j||_t|jdr0|j||_|S )Nrelabeled_clone)r   r   rE   r   r   )r   Zrelabelsnewr    r    r!   rE   j   s
    zLookup.relabeled_clonec                 C   s*   | j  }t| jdr&|| j  |S )Nget_group_by_cols)r   rG   r   r   r-   )r   aliascolsr    r    r!   rG   q   s    
zLookup.get_group_by_colsc                 C   s   t d S r#   r   r   r0   r1   r    r    r!   rB   w   s    zLookup.as_sqlc                 C   sf   d}g }| j | jfD ]0}t|tr:tt|dddd}d}|| q|rVt| | n| }|||S )NFT)Zthen)default)	r   r   r   r   r   r   r,   typerB   )r   r0   r1   wrappedexprsexprlookupr    r    r!   	as_oraclez   s    
zLookup.as_oraclec                 C   s   | j jpt| jddS )Ncontains_aggregateF)r   rS   getattrr   r:   r    r    r!   rS      s    zLookup.contains_aggregatec                 C   s   | j jpt| jddS )Ncontains_over_clauseF)r   rU   rT   r   r:   r    r    r!   rU      s    zLookup.contains_over_clausec                 C   s   | j jpt| jddS )N
is_summaryF)r   rV   rT   r   r:   r    r    r!   rV      s    zLookup.is_summaryc                 C   s   | j | j| jfS r#   )	__class__r   r   r:   r    r    r!   identity   s    zLookup.identityc                 C   s   t |tstS | j|jkS r#   )r   r   NotImplementedrX   )r   otherr    r    r!   __eq__   s    
zLookup.__eq__c                 C   s   t t| jS r#   )hashr   rX   r:   r    r    r!   __hash__   s    zLookup.__hash__)N)N)N)__name__
__module____qualname__lookup_namer@   Zcan_use_none_as_rhsr"   r%   r8   r;   r=   r   r.   rA   rC   r9   rE   rG   rB   rR   r   rS   rU   propertyrV   rX   r[   r]   r    r    r    r!   r      s6   






r   c                   @   s,   e Zd ZdZdZdZedd Zdd ZdS )		Transformz
    RegisterLookupMixin() is first so that get_lookup() and get_transform()
    first examine self and then check output_field.
    Fr<   c                 C   s   |   d S Nr   )r;   r:   r    r    r!   r      s    zTransform.lhsc                 C   s2   t | jdr| j }ng }| jr.|| j |S )Nr   )r   r   r   	bilateralr,   rW   )r   r   r    r    r!   r      s    z"Transform.get_bilateral_transformsN)	r^   r_   r`   __doc__re   arityrb   r   r   r    r    r    r!   rc      s   
rc   c                       s.   e Zd Zd fdd	Zdd Zdd Z  ZS )	BuiltinLookupNc                    sb   t  |||\}}| jj }| jjj|d}|j||| }|j| j	|| }|t
|fS )N)r1   )superrA   r   r'   Zget_internal_typedb_typeopsZfield_cast_sqlZlookup_castra   list)r   r0   r1   r   lhs_sqlr7   Zfield_internal_typerj   rW   r    r!   rA      s     zBuiltinLookup.process_lhsc                 C   sF   |  ||\}}| ||\}}|| | ||}d||f |fS Nz%s %s)rA   rC   r-   
get_rhs_op)r   r0   r1   rm   r7   rhs_sql
rhs_paramsr    r    r!   rB      s
    
zBuiltinLookup.as_sqlc                 C   s   |j | j | S r#   	operatorsra   r   r1   r   r    r    r!   rp      s    zBuiltinLookup.get_rhs_op)N)r^   r_   r`   rA   rB   rp   __classcell__r    r    rn   r!   rh      s   	rh   c                   @   s   e Zd ZdZdZdd ZdS )FieldGetDbPrepValueMixinzZ
    Some lookups require Field.get_db_prep_value() to be called on their
    inputs.
    Fc                    sT   t | jjdd }t |dd p$| jjjd| jrB fdd|D n| ddgfS )NZtarget_fieldget_db_prep_valuer(   c                    s   g | ]}| d dqS )Tpreparedr    ).0vr1   rx   r    r!   
<listcomp>   s     z?FieldGetDbPrepValueMixin.get_db_prep_lookup.<locals>.<listcomp>Try   )rT   r   r'   rx   $get_db_prep_lookup_value_is_iterable)r   r$   r1   fieldr    r}   r!   r.      s    z+FieldGetDbPrepValueMixin.get_db_prep_lookupN)r^   r_   r`   rf   r   r.   r    r    r    r!   rw      s   rw   c                       sB   e Zd ZdZdZdd Z fddZdd Zd fd
d	Z  Z	S ) FieldGetDbPrepValueIterableMixinzg
    Some lookups require Field.get_db_prep_value() to be called on each value
    in an iterable.
    Tc                 C   s^   t | jdr| jS g }| jD ]<}t |dr,n"| jrNt | jjdrN| jj|}|| q|S r>   )r   r   r@   r   r'   r?   r,   )r   Zprepared_values	rhs_valuer    r    r!   r      s    

z0FieldGetDbPrepValueIterableMixin.get_prep_lookupc                    s&   |   r| ||S t ||S d S r#   )r9   r8   ri   rC   rK   rn   r    r!   rC      s    z,FieldGetDbPrepValueIterableMixin.process_rhsc                 C   s<   |g}t |dr||j}t |dr4||\}}||fS )Nr)   rB   )r   r)   r*   r+   )r   r0   r1   r5   paramr7   r    r    r!   resolve_expression_parameter   s    

z=FieldGetDbPrepValueIterableMixin.resolve_expression_parameterNc                    sJ   t   |}t fddt| D  \}}tj|}|t|fS )Nc                 3   s"   | ]\}}  ||V  qd S r#   )r   )r{   r5   r   r0   r1   r   r    r!   	<genexpr>
  s   zEFieldGetDbPrepValueIterableMixin.batch_process_rhs.<locals>.<genexpr>)ri   r8   zip	itertoolschainfrom_iterabletuple)r   r0   r1   r   Zpre_processedr5   r7   rn   r   r!   r8     s    z2FieldGetDbPrepValueIterableMixin.batch_process_rhs)N)
r^   r_   r`   rf   r   r   rC   r   r8   rv   r    r    rn   r!   r      s   r   c                   @   s   e Zd ZdZdZdd ZdS )PostgresOperatorLookupz*Lookup defined by operators on PostgreSQL.Nc                 C   sD   |  ||\}}| ||\}}t|t| }d|| j|f |fS )Nz%s %s %s)rA   rC   r   postgres_operator)r   r0   r1   r   
lhs_paramsr   rr   r7   r    r    r!   as_postgresql  s    z$PostgresOperatorLookup.as_postgresql)r^   r_   r`   rf   r   r   r    r    r    r!   r     s   r   c                       s,   e Zd ZdZ fddZ fddZ  ZS )Exactexactc                    sZ   ddl m} t| j|rL| j rD| jjsL| j  | jdg ntdt	 
||S )Nr   r   pkzSThe QuerySet value for an exact lookup must be limited to one result using slicing.)r   r   r   r   Zhas_limit_onehas_select_fieldsclear_select_clause
add_fields
ValueErrorri   rC   )r   r0   r1   r   rn   r    r!   rC   !  s    

zExact.process_rhsc                    s`   t | jtrRt| jddrR|j| jrR| ||\}}| jrBdnd}|| |fS t 	||S )NZconditionalFr(   zNOT %s)
r   r   boolrT   r   rk   Z0conditional_expression_supported_in_where_clauserA   ri   rB   )r   r0   r1   rm   r7   templatern   r    r!   rB   /  s    
zExact.as_sql)r^   r_   r`   ra   rC   rB   rv   r    r    rn   r!   r     s   r   c                       s$   e Zd ZdZdZ fddZ  ZS )IExactZiexactFc                    s2   t  ||\}}|r*|j|d |d< ||fS rd   )ri   rC   rk   Zprep_for_iexact_queryr   qnr1   r   r7   rn   r    r!   rC   C  s    zIExact.process_rhs)r^   r_   r`   ra   r@   rC   rv   r    r    rn   r!   r   >  s   r   c                   @   s   e Zd ZdZdS )GreaterThangtNr^   r_   r`   ra   r    r    r    r!   r   J  s   r   c                   @   s   e Zd ZdZdS )GreaterThanOrEqualZgteNr   r    r    r    r!   r   O  s   r   c                   @   s   e Zd ZdZdS )LessThanltNr   r    r    r    r!   r   T  s   r   c                   @   s   e Zd ZdZdS )LessThanOrEqualZlteNr   r    r    r    r!   r   Y  s   r   c                       s    e Zd ZdZ fddZ  ZS )IntegerFieldFloatRoundingz
    Allow floats to work as query values for IntegerField. Without this, the
    decimal portion of the float would always be discarded.
    c                    s$   t | jtrt| j| _t  S r#   )r   r   floatmathceilri   r   r:   rn   r    r!   r   c  s    z)IntegerFieldFloatRounding.get_prep_lookup)r^   r_   r`   rf   r   rv   r    r    rn   r!   r   ^  s   r   c                   @   s   e Zd ZdS )IntegerGreaterThanOrEqualNr^   r_   r`   r    r    r    r!   r   i  s   r   c                   @   s   e Zd ZdS )IntegerLessThanNr   r    r    r    r!   r   n  s   r   c                       s<   e Zd ZdZ fddZdd Z fddZdd	 Z  ZS )
Ininc                    s   t | jdd }|d k	r(||jkr(td|  rzt| j}|d  W n$ tk
rl   dd | jD }Y nX |svt| 	|||\}}dd
| d }||fS t | jdd	s| j  | jd
g t ||S d S )N_dbzvSubqueries aren't allowed across different databases. Force the inner query to be evaluated using `list(inner_query)`.c                 S   s   g | ]}|d k	r|qS r#   r    )r{   rr    r    r!   r~     s      z"In.process_rhs.<locals>.<listcomp>(, )r   Tr   )rT   r   rH   r   r9   r   discard	TypeErrorr   r8   joinr   r   ri   rC   )r   r0   r1   Zdb_rhsr   r2   r3   placeholderrn   r    r!   rC   w  s(    

zIn.process_rhsc                 C   s   d| S )NzIN %sr    ru   r    r    r!   rp     s    zIn.get_rhs_opc                    s>   |j  }|  r0|r0t| j|kr0| ||S t ||S r#   )rk   max_in_list_sizer9   r/   r   split_parameter_list_as_sqlri   rB   )r   r0   r1   r   rn   r    r!   rB     s    
z	In.as_sqlc                 C   s   |j  }| ||\}}| ||\}}dg}g }	tdt||D ]v}
|
dkrZ|d |d|  |	| ||
|
|  }||
|
|  }d|}|| |d |	| qD|d d||	fS )Nr   r   z OR z%s IN (r   r    )	rk   r   rA   r8   ranger/   r,   r-   r   )r   r0   r1   r   r   r   r   rr   Zin_clause_elementsr7   offsetr2   r3   Zparam_groupr    r    r!   r     s$    






zIn.split_parameter_list_as_sql)	r^   r_   r`   ra   rC   rp   rB   r   rv   r    r    rn   r!   r   s  s
   r   c                       s0   e Zd ZdZdZ fddZ fddZ  ZS )PatternLookupz%%%s%%Fc                    sB   t | jds| jr0|j| j |j}||S t ||S d S rD   )	r   r   r   Zpattern_opsra   formatZpattern_escri   rp   )r   r1   r   patternrn   r    r!   rp     s    

zPatternLookup.get_rhs_opc                    sF   t  ||\}}|  r>|r>| js>| j|j|d  |d< ||fS rd   )ri   rC   r9   r   param_patternrk   Zprep_for_like_queryr   rn   r    r!   rC     s    zPatternLookup.process_rhs)r^   r_   r`   r   r@   rp   rC   rv   r    r    rn   r!   r     s   r   c                   @   s   e Zd ZdZdS )ContainscontainsNr   r    r    r    r!   r     s   r   c                   @   s   e Zd ZdZdS )	IContainsZ	icontainsNr   r    r    r    r!   r     s   r   c                   @   s   e Zd ZdZdZdS )
StartsWith
startswithz%s%%Nr^   r_   r`   ra   r   r    r    r    r!   r     s   r   c                   @   s   e Zd ZdZdS )IStartsWithZistartswithNr   r    r    r    r!   r     s   r   c                   @   s   e Zd ZdZdZdS )EndsWithendswithz%%%sNr   r    r    r    r!   r     s   r   c                   @   s   e Zd ZdZdS )	IEndsWithZ	iendswithNr   r    r    r    r!   r     s   r   c                   @   s   e Zd ZdZdd ZdS )Ranger   c                 C   s   d|d |d f S )NBETWEEN %s AND %sr   r<   r    ru   r    r    r!   rp     s    zRange.get_rhs_opN)r^   r_   r`   ra   rp   r    r    r    r!   r     s   r   c                   @   s   e Zd ZdZdZdd ZdS )IsNullZisnullFc                 C   sJ   t | jtstdt || j\}}| jr:d| |fS d| |fS d S )NzXUsing a non-boolean value for an isnull lookup is deprecated, use True or False instead.z
%s IS NULLz%s IS NOT NULL)r   r   r   warningswarnr   r+   r   )r   r0   r1   r5   r7   r    r    r!   rB     s    zIsNull.as_sqlN)r^   r_   r`   ra   r@   rB   r    r    r    r!   r     s   r   c                       s$   e Zd ZdZdZ fddZ  ZS )RegexregexFc                    s`   | j |jkrt ||S | ||\}}| ||\}}|j| j }|||f || fS d S r#   )ra   rt   ri   rB   rA   rC   rk   Zregex_lookup)r   r0   r1   r   r   r   rr   Zsql_templatern   r    r!   rB     s    zRegex.as_sql)r^   r_   r`   ra   r@   rB   rv   r    r    rn   r!   r     s   r   c                   @   s   e Zd ZdZdS )IRegexZiregexNr   r    r    r    r!   r     s   r   c                       s4   e Zd Zdd Z fddZdd Zdd Z  ZS )	
YearLookupc                 C   s2   | j j j}t|tr"|j|}n|j|}|S r#   )r   r'   r   r
   rk   Z%year_lookup_bounds_for_datetime_fieldZ!year_lookup_bounds_for_date_field)r   r1   yearr'   Zboundsr    r    r!   year_lookup_bounds%  s
    

zYearLookup.year_lookup_boundsc           	         s|   |   rn| ||| jj\}}| ||\}}| ||}| || j\}}|| || d||f |fS t	 
||S ro   )r9   rA   r   rC   get_direct_rhs_sqlr   r   r-   get_bound_paramsri   rB   )	r   r0   r1   rm   r7   rq   r6   startfinishrn   r    r!   rB   -  s    zYearLookup.as_sqlc                 C   s   |j | j | S r#   rs   ru   r    r    r!   r   ;  s    zYearLookup.get_direct_rhs_sqlc                 C   s   t dd S )NzAsubclasses of YearLookup must provide a get_bound_params() methodrJ   r   r   r   r    r    r!   r   >  s    zYearLookup.get_bound_params)r^   r_   r`   r   rB   r   r   rv   r    r    rn   r!   r   $  s   r   c                   @   s   e Zd Zdd Zdd ZdS )	YearExactc                 C   s   dS )Nr   r    ru   r    r    r!   r   E  s    zYearExact.get_direct_rhs_sqlc                 C   s   ||fS r#   r    r   r    r    r!   r   H  s    zYearExact.get_bound_paramsN)r^   r_   r`   r   r   r    r    r    r!   r   D  s   r   c                   @   s   e Zd Zdd ZdS )YearGtc                 C   s   |fS r#   r    r   r    r    r!   r   M  s    zYearGt.get_bound_paramsNr^   r_   r`   r   r    r    r    r!   r   L  s   r   c                   @   s   e Zd Zdd ZdS )YearGtec                 C   s   |fS r#   r    r   r    r    r!   r   R  s    zYearGte.get_bound_paramsNr   r    r    r    r!   r   Q  s   r   c                   @   s   e Zd Zdd ZdS )YearLtc                 C   s   |fS r#   r    r   r    r    r!   r   W  s    zYearLt.get_bound_paramsNr   r    r    r    r!   r   V  s   r   c                   @   s   e Zd Zdd ZdS )YearLtec                 C   s   |fS r#   r    r   r    r    r!   r   \  s    zYearLte.get_bound_paramsNr   r    r    r    r!   r   [  s   r   c                       s    e Zd ZdZ fddZ  ZS )UUIDTextMixinzs
    Strip hyphens from a value when filtering a UUIDField on backends without
    a native datatype for UUID.
    c                    s`   |j jsFddlm} |  r(t| j| _|| jtdtdt d| _t 	||\}}||fS )Nr   )Replace-r   r&   )
featuresZhas_native_uuid_fieldZdjango.db.models.functionsr   r9   r   r   r	   ri   rC   )r   r   r1   r   r   r7   rn   r    r!   rC   e  s    zUUIDTextMixin.process_rhs)r^   r_   r`   rf   rC   rv   r    r    rn   r!   r   `  s   r   c                   @   s   e Zd ZdS )
UUIDIExactNr   r    r    r    r!   r   o  s   r   c                   @   s   e Zd ZdS )UUIDContainsNr   r    r    r    r!   r   t  s   r   c                   @   s   e Zd ZdS )UUIDIContainsNr   r    r    r    r!   r   y  s   r   c                   @   s   e Zd ZdS )UUIDStartsWithNr   r    r    r    r!   r   ~  s   r   c                   @   s   e Zd ZdS )UUIDIStartsWithNr   r    r    r    r!   r     s   r   c                   @   s   e Zd ZdS )UUIDEndsWithNr   r    r    r    r!   r     s   r   c                   @   s   e Zd ZdS )UUIDIEndsWithNr   r    r    r    r!   r     s   r   )Fr   r   r   r   Zdjango.core.exceptionsr   Zdjango.db.models.expressionsr   r   r   r   r   Zdjango.db.models.fieldsr	   r
   r   r   r   Zdjango.db.models.query_utilsr   Zdjango.utils.datastructuresr   Zdjango.utils.deprecationr   Zdjango.utils.functionalr   Zdjango.utils.hashabler   r   rc   rh   rw   r   r   Zregister_lookupr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r    r    r!   <module>   s    3 A 