o
    ^"<f0                     @   s   d Z ddlZddlZddlZddlZddlmZ ddlmZm	Z	 ddl
mZ ddlmZ ddlmZ edd	ZG d
d deZG dd deedZdd ZG dd dejZG dd dZG dd dZd ddZdd Zdd ZG dd dZdS )!z
Various data structures used in query construction.

Factored out from django.db.models.query to avoid making the main module very
large and/or so that they can be used by other modules without getting into
circular import difficulties.
    N)
namedtupleFieldDoesNotExist
FieldError)
LOOKUP_SEP)tree)RemovedInDjango40WarningPathInfozGfrom_opts to_opts target_fields join_field m2m direct filtered_relationc                       s<   e Zd Zedd Zdd Z fddZ fddZ  ZS )	InvalidQueryTypec                 C   s   t tfS Nr   self r   S/var/www/html/kck/venv/lib/python3.10/site-packages/django/db/models/query_utils.py_subclasses   s   zInvalidQueryType._subclassesc                 C   s   t jdtdd d S )Nz\The InvalidQuery exception class is deprecated. Use FieldDoesNotExist or FieldError instead.   )category
stacklevel)warningswarnr   r   r   r   r   __warn   s
   
zInvalidQueryType.__warnc                        |    t|| jpt |S r   )_InvalidQueryType__warn
isinstancer   super__instancecheck__)r   instance	__class__r   r   r   &      z"InvalidQueryType.__instancecheck__c                    r   r   )r   
issubclassr   r   __subclasscheck__)r   subclassr   r   r   r!   *   r   z"InvalidQueryType.__subclasscheck__)	__name__
__module____qualname__propertyr   r   r   r!   __classcell__r   r   r   r   r
      s    
r
   c                   @   s   e Zd ZdS )InvalidQueryN)r#   r$   r%   r   r   r   r   r(   /   s    r(   )	metaclassc                 c   s(    | V  |   D ]	}t|E d H  qd S r   )__subclasses__
subclassesclsr"   r   r   r   r+   3   s
   r+   c                       sj   e Zd ZdZdZdZeZdZddd fdd	
Zd
d Z	dd Z
dd Zdd ZdddZdd Z  ZS )Qze
    Encapsulate filters as objects that can then be combined logically (using
    `&` and `|`).
    ANDORTNF)
_connector_negatedc                   s&   t  jg |t| ||d d S )N)children	connectornegated)r   __init__sorteditems)r   r1   r2   argskwargsr   r   r   r6   D   s   &z
Q.__init__c                 C   s   t |tst|dddu st|| s!t|dr| S t|S t |tr8|s8|  \}}}t| |i |S t|  }||_|	| | |	|| |S )NconditionalFTcopy)
r   r.   getattr	TypeErrorhasattrr<   deconstructtyper4   add)r   otherconn_r9   r:   objr   r   r   _combineG   s   
z
Q._combinec                 C      |  || jS r   )rG   r0   r   rC   r   r   r   __or__W      zQ.__or__c                 C   rH   r   )rG   r/   rI   r   r   r   __and__Z   rK   z	Q.__and__c                 C   s$   t |  }|| | j |  |S r   )rA   rB   r/   negate)r   rF   r   r   r   
__invert__]   s   
zQ.__invert__c                 C   s&   |j | ||ddd\}}|| |S )NF)allow_joins
split_subqcheck_filterable)_add_qpromote_joins)r   queryrO   reuse	summarizefor_saveclausejoinsr   r   r   resolve_expressionc   s   

zQ.resolve_expressionc                 C   sf   d| j j| j jf }|dr|dd}t| j}i }| j| jkr'| j|d< | j	r.d|d< |||fS )Nz%s.%szdjango.db.models.query_utilszdjango.db.modelsr1   Tr2   )
r   r$   r#   
startswithreplacetupler3   r4   defaultr5   )r   pathr9   r:   r   r   r   r@   m   s   



zQ.deconstruct)NTNFF)r#   r$   r%   __doc__r/   r0   r^   r;   r6   rG   rJ   rL   rN   rZ   r@   r'   r   r   r   r   r.   9   s    

r.   c                   @   s*   e Zd ZdZdd Zd	ddZdd ZdS )
DeferredAttributez
    A wrapper for a deferred-loading field. When the value is read from this
    object the first time, the query is executed.
    c                 C   s
   || _ d S r   )field)r   rb   r   r   r   r6      s   
zDeferredAttribute.__init__Nc                 C   sZ   |du r| S |j }| jj}||vr)| |}|du r%|j|gd || S |||< || S )zx
        Retrieve and caches the value from the datastore on the first lookup.
        Return the cached value.
        N)fields)__dict__rb   attname_check_parent_chainrefresh_from_db)r   r   r-   data
field_namevalr   r   r   __get__   s   
zDeferredAttribute.__get__c                 C   s6   |j }|| jj}| jjr| j|krt||jS dS )z
        Check if the field value can be fetched from a parent field already
        loaded in the instance. This can be done if the to-be fetched
        field is a primary key field.
        N)_metaget_ancestor_linkrb   modelprimary_keyr=   re   )r   r   opts
link_fieldr   r   r   rf      s
   z%DeferredAttribute._check_parent_chainr   )r#   r$   r%   r`   r6   rk   rf   r   r   r   r   ra   z   s
    
ra   c                   @   st   e Zd Zedd Zeejdddd Zdd Zd	d
 Z	e
dd Zedd ZedddZedddZdS )RegisterLookupMixinc                 C   s   |   |d S r   )get_lookupsget)r-   lookup_namer   r   r   _get_lookup   s   zRegisterLookupMixin._get_lookupN)maxsizec                 C   s   dd t | D }| |S )Nc                 S   s   g | ]	}|j d i qS )class_lookups)rd   rt   ).0parentr   r   r   
<listcomp>   s    z3RegisterLookupMixin.get_lookups.<locals>.<listcomp>)inspectgetmromerge_dicts)r-   rx   r   r   r   rs      s   
zRegisterLookupMixin.get_lookupsc                 C   N   ddl m} | |}|d u rt| dr| j|S |d ur%t||s%d S |S )Nr   )Lookupoutput_field)django.db.models.lookupsr   rv   r?   r   
get_lookupr    )r   ru   r   foundr   r   r   r         
zRegisterLookupMixin.get_lookupc                 C   r   )Nr   )	Transformr   )r   r   rv   r?   r   get_transformr    )r   ru   r   r   r   r   r   r      r   z!RegisterLookupMixin.get_transformc                 C   s    i }t | D ]}|| q|S )z
        Merge dicts in reverse to preference the order of the original list. e.g.,
        merge_dicts([a, b]) will preference the keys in 'a' over those in 'b'.
        )reversedupdate)dictsmergeddr   r   r   r~      s   zRegisterLookupMixin.merge_dictsc                 C   s   t | D ]}|j  qd S r   )r+   rs   cache_clearr,   r   r   r   _clear_cached_lookups   s   z)RegisterLookupMixin._clear_cached_lookupsc                 C   s4   |d u r|j }d| jvri | _|| j|< |   |S )Nrx   )ru   rd   rx   r   r-   lookupru   r   r   r   register_lookup   s   

z#RegisterLookupMixin.register_lookupc                 C   s   |du r|j }| j|= dS )zn
        Remove given lookup from cls lookups. For use in tests only as it's
        not thread-safe.
        N)ru   rx   r   r   r   r   _unregister_lookup   s   z&RegisterLookupMixin._unregister_lookupr   )r#   r$   r%   classmethodrv   	functools	lru_cachers   r   r   staticmethodr~   r   r   r   r   r   r   r   rr      s     

		


	rr   Fc                 C   s   | j sdS | j jr|sdS |r"|r|  |vrdS |s"| j|vr"dS |s)| jr)dS |rE| j|vrE|rE| j|v rEd| jjj| jf }t	|dS )au  
    Return True if this field should be used to descend deeper for
    select_related() purposes. Used by both the query construction code
    (sql.query.fill_related_selections()) and the model instance creation code
    (query.get_klass_info()).

    Arguments:
     * field - the field to be checked
     * restricted - a boolean field, indicating if the field list has been
       manually restricted using a requested clause)
     * requested - The select_related() dictionary.
     * load_fields - the set of fields to be loaded on this model
     * reverse - boolean, True if we are checking a reverse select related
    FzXField %s.%s cannot be both deferred and traversed using select_related at the same time.T)
remote_fieldparent_linkrelated_query_namenamenullre   rn   rl   object_namer   )rb   
restricted	requestedload_fieldsreversemsgr   r   r   select_related_descend   s&   

r   c                 C   sV   t dt| d D ]}t| d| }||v r(|| r(|| | |d f  S q	dS )z
    Check if the lookup_parts contains references to the given annotations set.
    Because the LOOKUP_SEP is contained in the default annotation names, check
    each prefix of the lookup_parts for a match.
       r   N)Fr   )rangelenr   join)lookup_partsannotationsnlevel_n_lookupr   r   r   refs_expression
  s   r   c                    s,    fdd}||pt |ddo||jjS )z
    Check that self.model is compatible with target_opts. Compatibility
    is OK if:
      1) model and opts match (where proxy inheritance is removed)
      2) model is parent of opts' model or the other way around
    c                    s*    j j| jkp| j j  v p |  v S r   )rl   concrete_modelget_parent_list)rp   rn   r   r   check  s
   
z-check_rel_lookup_compatibility.<locals>.checkro   F)r=   rn   rl   )rn   target_optsrb   r   r   r   r   check_rel_lookup_compatibility  s   r   c                   @   s@   e Zd ZdZe dddZdd Zdd Zd	d
 Zdd Z	dS )FilteredRelationz7Specify custom filtering in the ON clause of SQL joins.	conditionc                C   s:   |st d|| _d | _t|tst d|| _g | _d S )Nzrelation_name cannot be empty.z*condition argument must be a Q() instance.)
ValueErrorrelation_namealiasr   r.   r   r_   )r   r   r   r   r   r   r6   6  s   

zFilteredRelation.__init__c                 C   s4   t || jstS | j|jko| j|jko| j|jkS r   )r   r   NotImplementedr   r   r   rI   r   r   r   __eq__@  s   

zFilteredRelation.__eq__c                 C   s,   t | j| jd}| j|_| jd d  |_|S )Nr   )r   r   r   r   r_   )r   cloner   r   r   r   I  s   zFilteredRelation.clonec                 O   s   t d)zz
        QuerySet.annotate() only accepts expression-like arguments
        (with a resolve_expression() method).
        z0FilteredRelation.resolve_expression() is unused.)NotImplementedError)r   r9   r:   r   r   r   rZ   O  s   z#FilteredRelation.resolve_expressionc                 C   s&   |j }|j| jt| jd}||S )N)rU   )rT   build_filtered_relation_qr   setr_   compile)r   compiler
connectionrT   wherer   r   r   as_sqlV  s   
zFilteredRelation.as_sqlN)
r#   r$   r%   r`   r.   r6   r   r   rZ   r   r   r   r   r   r   3  s    
	r   )F)r`   r<   r   r|   r   collectionsr   django.core.exceptionsr   r   django.db.models.constantsr   django.utilsr   django.utils.deprecationr   r	   rA   r
   	Exceptionr(   r+   Noder.   ra   rr   r   r   r   r   r   r   r   r   <module>   s*    
A(
C%