
    rhK                     X   d Z 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	 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  ej0                  d      Z edd      Zd Z G d dej8                        Z G d d      Z G d d      Z G d d      Z d Z!d Z"d Z# G d d      Z$y)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)nullcontext)
FieldError)DEFAULT_DB_ALIASDatabaseErrorconnectionsmodelstransaction)
LOOKUP_SEP)tree)cached_property)make_hashabledjango.db.modelsPathInfozGfrom_opts to_opts target_fields join_field m2m direct filtered_relationc              #   h   K   |  | j                         D ]  }t        |      E d {     y 7 wN)__subclasses__
subclassesclssubclasss     o/var/www/html/ai-insurance-compliance-backend/venv/lib/python3.12/site-packages/django/db/models/query_utils.pyr   r   !   s4     
I&&( (h'''('s   &202c                        e Zd ZdZdZdZdZeZdZddd fd	
Z	d
 Z
d Zd Zd Zd Z	 ddZd Zd ZefdZd Zed        Zd Zd Zed        Z xZS )Qze
    Encapsulate filters as objects that can then be combined logically (using
    `&` and `|`).
    ANDORXORTNF)
_connector_negatedc                `    t         |   g |t        |j                               ||       y )N)children	connectornegated)super__init__sorteditems)selfr   r   argskwargs	__class__s        r   r%   z
Q.__init__4   s3    5t5fV\\^45  	 	
    c                    t        |dd      du rt        |      | s|j                         S |s t        |t              r| j                         S | j                  |      }|j                  | |       |j                  ||       |S )NconditionalF)r"   )getattr	TypeErrorcopy
isinstancer   createadd)r(   otherconnobjs       r   _combinez
Q._combine;   sv    5-/58E""::<E1-99;kkDk)dt
r,   c                 :    | j                  || j                        S r   )r8   r   r(   r5   s     r   __or__zQ.__or__H   s    }}UDGG,,r,   c                 :    | j                  || j                        S r   )r8   r   r:   s     r   __and__z	Q.__and__K       }}UDHH--r,   c                 :    | j                  || j                        S r   )r8   r   r:   s     r   __xor__z	Q.__xor__N   r>   r,   c                 F    | j                         }|j                          |S r   )r1   negate)r(   r7   s     r   
__invert__zQ.__invert__Q   s    iik


r,   c                 \    |j                  | ||dd|      \  }}|j                  |       |S )NF)allow_joins
split_subqcheck_filterable	summarize)_add_qpromote_joins)r(   queryrE   reuserH   for_saveclausejoinss           r   resolve_expressionzQ.resolve_expressionV   sB    
 #" % 
 	E"r,   c                    |s| S | j                  | j                  | j                        }| j                  D ]  }|}t	        |t
              r|\  }}t        |v r|j                  t        d      \  }}n|}d }t        j                  |      }	|	j                  |      x}
|	ur}|d}|
j                  |      x}/|
j                  |      x}	  ||
      }
d}|
j                  |      }||dk(  r|
j                  d      }d}| ||
|      }n|j                  |      }|j                  j                  |        |S )N)r"   r#      exactisnullT)r3   r"   r#   r!   r2   tupler   rsplitr	   Freplace_expressions
get_lookupget_transformappend)r(   replacementsclonechildchild_replacementlhsrhspathlookupfieldfield_replacementlookup_classtransform_classs                r   rX   zQ.replace_expressionsf   sb   KdnndllK]]  	5E %%' S$#&::j!#<LD&D!F).)B)B<)PP% ~!((9(D(DV(LLU/@/N/Nv/VVO!%& 1@@Q0R-%,F+<+G+G+OL{v'8'8'C'CH'M"#/,89JC,P)$)$=$=l$K!NN!!"34A 	5B r,   c              #      K   |  | j                   D ]@  }t        |t              r|d   }t        |d      r|j	                         E d{    =| B y7 w)zg
        Recursively yield this Q object and all subexpressions, in depth-first
        order.
        rR   flattenN)r!   r2   rU   hasattrri   )r(   r^   s     r   ri   z	Q.flatten   sV     
 
]] 	E%'aui( ==?**	
 +s   AA
AAc           
         ddl m}m} ddlm} ddlm} ddlm}  |d      }|j                         D ]-  \  }	}
t        |
d      s ||
      }
|j                  |
|	d	       / |j                   |d
      d       t        |   }|j                  j                  r)|j                  t!         || d |                          n|j                  |        |j#                  |      }|j$                  rt'        j(                  |      n	t+               }	 |5  |j-                  |      ducddd       S # 1 sw Y   yxY w# t.        $ r!}t0        j3                  d| |       Y d}~yd}~ww xY w)z|
        Do a database query to check if the expressions of the Q instance
        matches against the expressions.
        r   )BooleanFieldValue)Coalesce)Query)SINGLENrP   F)selectrR   _checkT)output_field)usingz.Got a database error calling check() on %r: %s)django.db.modelsrl   rm   django.db.models.functionsrn   django.db.models.sqlro   django.db.models.sql.constantsrp   r'   rj   add_annotationr   featuressupports_comparing_boolean_expradd_qr   get_compilerin_atomic_blockr
   atomicr   execute_sqlr   loggerwarning)r(   againstrt   rl   rm   rn   ro   rp   rK   namevalue
connectioncompilercontext_manageres                  r   checkzQ.check   sB    	97.9d"==? 	<KD%5"67e  U ;	< 	U1Xx0 '
>>KK(4LNKLMKK%%E%2 )) U+ 	
	  @++F34?@ @ @ 	NNKTSTU	s6   D> D2(	D> 2D;7D> ;D> >	E(E##E(c                 P   | j                   j                  d| j                   j                  }|j                  d      r|j	                  dd      }t        | j                        }i }| j                  | j                  k7  r| j                  |d<   | j                  rd|d<   |||fS )N.zdjango.db.models.query_utilsr   r   Tr   )
r+   
__module____name__
startswithreplacerU   r!   r"   defaultr#   )r(   rb   r)   r*   s       r   deconstructzQ.deconstruct   s    ..33T^^5L5LM??9:<< >@RSDT]]#>>T\\)#'>>F< <<!%F:T6!!r,   c                     | j                         \  }}}|g|j                         }|D ]G  }t        |t              r$|\  }}t	        |      }|j                  ||f       7|j                  |       I t        |      S r   )r   r'   r2   rU   r   r[   )r(   rb   r)   r*   identityr^   argr   s           r   r   z
Q.identity   s~    !--/dF*6<<>* 	'E%'"
U%e,e-&	' Xr,   c                 `    t        |t              st        S |j                  | j                  k(  S r   )r2   r   NotImplementedr   r:   s     r   __eq__zQ.__eq__   s%    %#!!~~..r,   c                 ,    t        | j                        S r   )hashr   )r(   s    r   __hash__z
Q.__hash__   s    DMM""r,   c                     ddl m} |j                  |       D ch c]  }|j                  t        d      d    c}S c c}w )z
        Retrieve all base fields referenced directly or through F expressions
        excluding any fields referenced through joins.
        r   )rK   rR   )rw   rK   get_children_from_qsplitr   )r(   rK   r^   s      r   referenced_base_fieldszQ.referenced_base_fields   sB     	/ 8=7P7PQU7V
.3EKK
A&q)
 	
 
s    =)NTNFF)r   r   __qualname____doc__r   r   r   r   r.   r%   r8   r;   r=   r@   rC   rP   rX   ri   r   r   r   r   r   r   r   r   __classcell__)r+   s   @r   r   r   '   s     C	B
CGK)- 
-.. SX %N $4 "H
" 
 
/
# 

 

r,   r   c                   $    e Zd ZdZd ZddZd Zy)DeferredAttributez
    A wrapper for a deferred-loading field. When the value is read from this
    object the first time, the query is executed.
    c                     || _         y r   )rd   )r(   rd   s     r   r%   zDeferredAttribute.__init__   s	    
r,   Nc                 "   || S |j                   }| j                  j                  }||vra| j                  |      }|I|j	                         s!| j                  j
                  rt        d      |j                  |g       ||   S |||<   ||   S )zx
        Retrieve and caches the value from the datastore on the first lookup.
        Return the cached value.
        z4Cannot read a generated field from an unsaved model.)fields)__dict__rd   attname_check_parent_chain
_is_pk_set	generatedAttributeErrorrefresh_from_db)r(   instancer   data
field_namevals         r   __get__zDeferredAttribute.__get__   s    
 K  ZZ''
T! **84C{**,1E1E(N  (((= J $'Z Jr,   c                     |j                   }|j                  | j                  j                        }| j                  j                  r%| j                  |k7  rt        ||j                        S y)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_linkrd   modelprimary_keyr/   r   )r(   r   opts
link_fields       r   r   z%DeferredAttribute._check_parent_chain  sU     ~~++DJJ,<,<=
::!!djjJ&>8Z%7%788r,   r   )r   r   r   r   r%   r   r    r,   r   r   r      s    
 .
r,   r   c                       e Zd ZdZd Zd Zy)class_or_instance_methodz
    Hook used in RegisterLookupMixin to return partial functions depending on
    the caller type (instance or class of models.Field).
    c                      || _         || _        y r   )class_methodinstance_method)r(   r   r   s      r   r%   z!class_or_instance_method.__init__"  s    (.r,   c                     | t        j                  | j                  |      S t        j                  | j                  |      S r   )	functoolspartialr   r   )r(   r   owners      r   r   z class_or_instance_method.__get__&  s:    $$T%6%6>>  !5!5x@@r,   N)r   r   r   r   r%   r   r   r,   r   r   r     s    
/Ar,   r   c                       e Zd Zd Zej
                  d        Zd Z eee      Z	 e
e      Zd Zd Zed        Ze
d        Zdd	Zdd
Z eee      Z e
e      ZddZddZ eee      Z e
e      Zy)RegisterLookupMixinc                 B    | j                         j                  |d       S r   )get_lookupsget)r(   lookup_names     r   _get_lookupzRegisterLookupMixin._get_lookup-  s    !%%k488r,   c                     t        j                  |       D cg c]  }|j                  j                  di         }}| j	                  |      S c c}w Nclass_lookups)inspectgetmror   r   merge_dicts)r   parentr   s      r   get_class_lookupsz%RegisterLookupMixin.get_class_lookups0  sQ     DK>>RUCV
9?FOO4
 
 }--
s   #Ac                 P    | j                         }t        | dd       x}ri ||S |S Ninstance_lookups)r   r/   )r(   r   r   s      r   get_instance_lookupsz(RegisterLookupMixin.get_instance_lookups7  s=    ..0&t-?FFF8m8'788r,   c                     ddl m} | j                  |      }|'t        | d      r| j                  j                  |      S |t        ||      sy |S )Nr   )Lookuprs   )django.db.models.lookupsr   r   rj   rs   rY   
issubclass)r(   r   r   founds       r   rY   zRegisterLookupMixin.get_lookup@  sQ    3  -=WT>:$$//<<Zv%>r,   c                     ddl m} | j                  |      }|'t        | d      r| j                  j                  |      S |t        ||      sy |S )Nr   )	Transformrs   )r   r   r   rj   rs   rZ   r   )r(   r   r   r   s       r   rZ   z!RegisterLookupMixin.get_transformJ  sQ    6  -=WT>:$$22;??Zy%Ar,   c                 L    i }t        |       D ]  }|j                  |        |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ds      r   r   zRegisterLookupMixin.merge_dictsT  s-     % 	AMM!	r,   c                 X    t        |       D ]  }|j                  j                           y r   )r   r   cache_clearr   s     r   _clear_cached_class_lookupsz/RegisterLookupMixin._clear_cached_class_lookups_  s'    "3 	5H&&224	5r,   Nc                     ||j                   }d| j                  vri | _        || j                  |<   | j                          |S r   )r   r   r   r   r   rc   r   s      r   register_class_lookupz)RegisterLookupMixin.register_class_lookupd  sH     ,,K#,,. "C)/+&'')r,   c                 j    ||j                   }d| j                  vri | _        || j                  |<   |S r   )r   r   r   r(   rc   r   s      r   register_instance_lookupz,RegisterLookupMixin.register_instance_lookupm  s<     ,,KT]]2$&D!-3k*r,   c                 Z    ||j                   }| j                  |= | j                          y)zn
        Remove given lookup from cls lookups. For use in tests only as it's
        not thread-safe.
        N)r   r   r   r   s      r   _unregister_class_lookupz,RegisterLookupMixin._unregister_class_lookupz  s/    
  ,,Kk*'')r,   c                 :    ||j                   }| j                  |= y)zs
        Remove given lookup from instance lookups. For use in tests only as
        it's not thread-safe.
        N)r   r   r   s      r   _unregister_instance_lookupz/RegisterLookupMixin._unregister_instance_lookup  s#    
  ,,K!!+.r,   r   )r   r   r   r   r   cacher   r   r   r   classmethodrY   rZ   staticmethodr   r   r   r   register_lookupr   r   _unregister_lookupr   r,   r   r   r   ,  s    9 __. . ++<>RSK#$56   5 5 /7O ((=>*/ 2 "=  ++CDr,   r   c                 
   | j                   syt        | j                   dd      ry|s| j                   S | j                  |vry|r>| |vr:t	        d| j
                  j                  j                   d| j                   d      y)a  
    Return whether `field` should be used to descend deeper for
    `select_related()` purposes.

    Arguments:
     * `field` - the field to be checked. Can be either a `Field` or
       `ForeignObjectRel` instance.
     * `restricted` - a boolean field, indicating if the field list has been
       manually restricted using a select_related() clause.
     * `requested` - the select_related() dictionary.
     * `select_mask` - the dictionary of selected fields.
    Fparent_linkzField r   zM cannot be both deferred and traversed using select_related at the same time.T)remote_fieldr/   nullr   r   r   r   object_name)rd   
restricted	requestedselect_masks       r   select_related_descendr     s      u!!=%8 ::~ zz" uK/U[[&&2231UZZL AL L
 	
 r,   c                     t        dt        |       dz         D ]4  }t        j                  | d|       }|j	                  |      s-|| |d fc S  y)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.
    rR   r   N)Nr   )rangelenr   joinr   )lookup_partsannotationsnlevel_n_lookups       r   refs_expressionr    s\     1c,'!+, 4#a):;??>*!<#3334 r,   c                 x      fd} ||      xs+ t        |dd      xr  ||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                     j                   j                  | j                  k(  xs2 | j                  j                   j                  v xs | j                  v S r   )r   concrete_modelall_parents)r   r   s    r   r   z-check_rel_lookup_compatibility.<locals>.check  sP    KK&&$*=*== )""ekk&=&==)(((	
r,   r   F)r/   r   r   )r   target_optsrd   r   s   `   r   check_rel_lookup_compatibilityr    s<    
   }e,Iu{{7H7H1Ir,   c                   D    e Zd ZdZ e       ddZd Zd Zd Zd Z	d Z
y	)
FilteredRelationz7Specify custom filtering in the ON clause of SQL joins.	conditionc                    |st        d      || _        d | _        t        |t              st        d      || _        d | _        y )Nzrelation_name cannot be empty.z*condition argument must be a Q() instance.)
ValueErrorrelation_namealiasr2   r   r  resolved_condition)r(   r  r  s      r   r%   zFilteredRelation.__init__  sG    =>>*
)Q'IJJ #"&r,   c                     t        || j                        st        S | j                  |j                  k(  xr4 | j                  |j                  k(  xr | j
                  |j
                  k(  S r   )r2   r+   r   r  r  r  r:   s     r   r   zFilteredRelation.__eq__  sY    %0!!%"5"55 2

ekk)2%//1	
r,   c                     t        | j                  | j                        }| j                  |_        | j                  x}|j                         |_        |S )Nr  )r  r  r  r  r  r]   )r(   r]   r  s      r   r]   zFilteredRelation.clone  sK     !3!3t~~Njj"&"9"99F'9'?'?'AE$r,   c                 n    | j                         }|j                  x}r|j                  |      |_        |S r   )r]   r  relabeled_clone)r(   
change_mapr]   r  s       r   r  z FilteredRelation.relabeled_clone  s7    

!&!9!999'9'I'I*'UE$r,   c                 v    | j                         }|j                  | j                  |ddd      d   |_        |S )NTF)	can_reuserE   rF   update_join_typesr   )r]   build_filterr  r  )r(   rK   rL   r)   r*   r]   s         r   rP   z#FilteredRelation.resolve_expression	  sI    

#(#5#5NN# $6 $
 $  r,   c                 8    |j                  | j                        S r   )compiler  )r(   r   r   s      r   as_sqlzFilteredRelation.as_sql  s     7 788r,   N)r   r   r   r   r   r%   r   r]   r  rP   r#  r   r,   r   r  r    s(    A343 '
	9r,   r  )%r   r   r   loggingcollectionsr   
contextlibr   django.core.exceptionsr   	django.dbr   r   r   r	   r
   django.db.models.constantsr   django.utilsr   django.utils.functionalr   django.utils.hashabler   	getLoggerr   r   r   Noder   r   r   r   r   r  r  r  r   r,   r   <module>r/     s       " " - W W 1  3 /			-	.
 M(E
		 E
P* *ZA A dE dEN#L
:29 29r,   