OwlCyberSecurity - MANAGER
Edit File: query_utils.cpython-39.pyc
a ���e�; � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z m Z m Z ddl m Z ddlmZ e�d�Zed d �Zd d � ZG d d� dej�ZG dd� d�ZG dd� d�ZG dd� d�Zd dd�Zdd� Zdd� ZG d d � d �Z dS ) 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)� FieldError)�DEFAULT_DB_ALIAS� DatabaseError� connections)� LOOKUP_SEP)�tree�django.db.models�PathInfozGfrom_opts to_opts target_fields join_field m2m direct filtered_relationc c s& | V | � � D ]}t|�E d H qd S �N)�__subclasses__� subclasses��cls�subclass� r �]/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-packages/django/db/models/query_utils.pyr s r c s� e Zd ZdZdZdZdZeZdZddd�� fd d � Z d d � Z d d� Z dd� Z dd� Z dd� Zd dd�Zdd� Zefdd�Zdd � Z� ZS ) �Qze Encapsulate filters as objects that can then be combined logically (using `&` and `|`). �AND�OR�XORTNF)� _connector�_negatedc s& t � jg |�t|�� ��||d� d S )N)�children� connector�negated)�super�__init__�sorted�items)�selfr r �args�kwargs�� __class__r r r 0 s �z Q.__init__c C sb t |dd�du rt|��| s$|�� S |s:t|t�r:| �� S | j|d�}|�| |� |�||� |S )N� conditionalF)r )�getattr� TypeError�copy� isinstancer �create�add)r �other�conn�objr r r �_combine7 s z Q._combinec C s | � || j�S r )r/ r �r r, r r r �__or__D s zQ.__or__c C s | � || j�S r )r/ r r0 r r r �__and__G s z Q.__and__c C s | � || j�S r )r/ r r0 r r r �__xor__J s z Q.__xor__c C s | � � }|�� |S r )r( �negate)r r. r r r � __invert__M s z Q.__invert__c C s( |j | ||dd|d�\}}|�|� |S )NF)� allow_joinsZ split_subqZcheck_filterable� summarize)Z_add_qZ promote_joins)r �queryr6 �reuser7 Zfor_saveZclauseZjoinsr r r �resolve_expressionR s � zQ.resolve_expressionc c sH | V | j D ]6}t|t�r"|d }t|d�r<|�� E dH q |V q dS )zg Recursively yield this Q object and all subexpressions, in depth-first order. � �flattenN)r r) �tuple�hasattrr< )r �childr r r r< b s z Q.flattenc C s ddl m}m} ddlm} ddlm} ddlm} |d�}|� � D ]*\} } t | d�s^|| �} |j | | dd � qD|� |d �d � t | j jr�|�t|| d |� d ��� n |�| � |j|d�} z| �|�duW S t�y } z t�d| | � W Y d} ~ d S d} ~ 0 0 dS )z| Do a database query to check if the expressions of the Q instance matches against the expressions. r )� BooleanField�Value)�Coalesce)�Query)�SINGLENr: F)�selectr; �_checkT)� output_field)�usingz.Got a database error calling check() on %r: %s)Zdjango.db.modelsr@ rA Zdjango.db.models.functionsrB Zdjango.db.models.sqlrC Z django.db.models.sql.constantsrD r r> Zadd_annotationr �featuresZ supports_comparing_boolean_exprZadd_qr Z get_compilerZ execute_sqlr �logger�warning) r ZagainstrH r@ rA rB rC rD r8 �name�value�compiler�er r r �checkq s& zQ.checkc C sf d| j j| j jf }|�d�r*|�dd�}t| j�}i }| j| jkrN| j|d< | j r\d|d< |||fS )Nz%s.%sz django.db.models.query_utilsr r Tr ) r$ � __module__�__name__� startswith�replacer= r r �defaultr )r �pathr! r" r r r � deconstruct� s z Q.deconstruct)NTNFF)rR rQ � __qualname__�__doc__r r r rU r% r r/ r1 r2 r3 r5 r: r< r rP rW � __classcell__r r r# r r # s" � r c @ s* e Zd ZdZdd� Zd dd�Zdd� ZdS ) �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 r\ r r r r � s zDeferredAttribute.__init__Nc C sT |du r | S |j }| jj}||vrL| �|�}|du rD|j|gd� n|||< || S )zx Retrieve and caches the value from the datastore on the first lookup. Return the cached value. N)�fields)�__dict__r\ �attname�_check_parent_chainZrefresh_from_db)r �instancer �data� field_name�valr r r �__get__� s zDeferredAttribute.__get__c C s6 |j }|�| jj�}| jjr2| j|kr2t||j�S dS )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)�_metaZget_ancestor_linkr\ �model� primary_keyr& r_ )r ra �optsZ link_fieldr r r r` � s z%DeferredAttribute._check_parent_chain)N)rR rQ rX rY r re r` r r r r r[ � s r[ c @ s e Zd ZdZdd� Zdd� ZdS )�class_or_instance_methodz� Hook used in RegisterLookupMixin to return partial functions depending on the caller type (instance or class of models.Field). c C s || _ || _d S r )� class_method�instance_method)r rk rl r r r r � s z!class_or_instance_method.__init__c C s$ |d u rt �| j|�S t �| j|�S r )� functools�partialrk rl )r ra �ownerr r r re � s z class_or_instance_method.__get__N)rR rQ rX rY r re r r r r rj � s rj c @ s� e Zd Zdd� Zejdd�dd� �Zdd� Zeee�Z e e�Zd d � Z d d � Z e d d� �Ze dd� �Zddd�Zddd�Zeee�Ze e�Zddd�Zd dd�Zeee�Ze e�ZdS ) �RegisterLookupMixinc C s | � � �|d �S r )� get_lookups�get)r � lookup_namer r r � _get_lookup� s z RegisterLookupMixin._get_lookupN)�maxsizec C s dd� t �| �D �}| �|�S )Nc S s g | ]}|j �d i ��qS )� class_lookups)r^ rr )�.0�parentr r r � <listcomp>� s z9RegisterLookupMixin.get_class_lookups.<locals>.<listcomp>)�inspect�getmro� merge_dicts)r rv r r r �get_class_lookups� s �z%RegisterLookupMixin.get_class_lookupsc C s( | � � }t| dd � }r$i |�|�S |S �N�instance_lookups)r} r& )r rv r r r r �get_instance_lookups� s z(RegisterLookupMixin.get_instance_lookupsc C sN ddl m} | �|�}|d u r4t| d�r4| j�|�S |d urJt||�sJd S |S )Nr )�LookuprG )�django.db.models.lookupsr� rt r> rG � get_lookup� issubclass)r rs r� �foundr r r r� � s z RegisterLookupMixin.get_lookupc C sN ddl m} | �|�}|d u r4t| d�r4| j�|�S |d urJt||�sJd S |S )Nr )� TransformrG )r� r� rt r> rG � get_transformr� )r rs r� r� r r r r� � s 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)Zdicts�merged�dr r r r| � s z RegisterLookupMixin.merge_dictsc C s t | �D ]}|j�� qd S r )r r} � cache_clearr r r r �_clear_cached_class_lookups s z/RegisterLookupMixin._clear_cached_class_lookupsc C s4 |d u r|j }d| jvr i | _|| j|< | �� |S )Nrv )rs r^ rv r� �r �lookuprs r r r �register_class_lookup s z)RegisterLookupMixin.register_class_lookupc C s, |d u r|j }d| jvr i | _|| j|< |S r~ )rs r^ r �r r� rs r r r �register_instance_lookup s z,RegisterLookupMixin.register_instance_lookupc C s" |du r|j }| j|= | �� dS )zn Remove given lookup from cls lookups. For use in tests only as it's not thread-safe. N)rs rv r� r� r r r �_unregister_class_lookup" s z,RegisterLookupMixin._unregister_class_lookupc C s |du r|j }| j|= dS )zs Remove given lookup from instance lookups. For use in tests only as it's not thread-safe. N)rs r r� r r r �_unregister_instance_lookup, s z/RegisterLookupMixin._unregister_instance_lookup)N)N)N)N)rR rQ rX rt rm � lru_cacher} r� rj rq � classmethodr� r� � staticmethodr| r� r� r� Zregister_lookupr� r� Z_unregister_lookupr r r r rp � s0 � �rp Fc C s� | j s dS | j jr|sdS |rD|r2| �� |vr2dS |sD| j|vrDdS |sR| jrRdS |r�|r�| j|v r�| |vr�td| jjj� d| j� d���dS )af Return True if this field should be used to descend deeper for select_related() purposes. Used by both the query construction code (compiler.get_related_selections()) and the model instance creation code (compiler.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. * select_mask - the dictionary of selected fields. * reverse - boolean, True if we are checking a reverse select related FzField �.zM cannot be both deferred and traversed using select_related at the same time.T) Z remote_fieldZ parent_linkZrelated_query_namerL �nullr rg rf Z object_name)r\ Z restricted� requestedZ select_mask�reverser r r �select_related_descend; s. �����r� c C sL t dt| �d �D ]4}t�| d|� �}|�|�r|| |d� f S qdS )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; r N)Nr )�range�lenr �joinrr )Z lookup_parts� annotations�nZlevel_n_lookupr r r �refs_expressionb s r� c s, � fdd�}||�p*t |dd�o*||jj�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| jkp(| j� j �� v p(� | �� v S r )rf Zconcrete_modelZget_parent_list)ri �rg r r rP w s � �z-check_rel_lookup_compatibility.<locals>.checkrh F)r& rg rf )rg Z target_optsr\ rP r r� r � check_rel_lookup_compatibilityo s �r� c @ s@ e Zd ZdZe� d�dd�Zdd� Zdd� Zd d � Zd 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� rV )r r� r� r r r r � s zFilteredRelation.__init__c C s4 t || j�stS | j|jko2| j|jko2| j|jkS r )r) r$ �NotImplementedr� r� r� r0 r r r �__eq__� s � �zFilteredRelation.__eq__c C s, t | j| jd�}| j|_| jd d � |_|S )Nr� )r� r� r� r� rV )r �cloner r r r� � s zFilteredRelation.clonec O s t d��dS )zz QuerySet.annotate() only accepts expression-like arguments (with a resolve_expression() method). z0FilteredRelation.resolve_expression() is unused.N)�NotImplementedError)r r! r" r r r r: � s z#FilteredRelation.resolve_expressionc C s& |j }|j| jt| j�d�}|�|�S )N)r9 )r8 Zbuild_filtered_relation_qr� �setrV �compile)r rN � connectionr8 �wherer r r �as_sql� s zFilteredRelation.as_sqlN) rR rQ rX rY r r r� r� r: r� r r r r r� � s r� )F) rY rm rz �logging� collectionsr Zdjango.core.exceptionsr Z django.dbr r r Zdjango.db.models.constantsr Z django.utilsr � getLoggerrJ r r �Noder r[ rj rp r� r� r� r� r r r r �<module> s, �x)g '