OwlCyberSecurity - MANAGER
Edit File: where.cpython-39.pyc
a ���e�1 � @ s� d Z ddlZddlmZ ddlmZmZ ddlmZm Z ddl m Z ddl m Z ddlmZ d Zd Zd ZG d d � d e j�ZG dd� d�ZG dd� d�ZG dd� d�ZdS )zG Code to manage the creation and SQL rendering of 'where' constraints. � N)�reduce)�EmptyResultSet� FullResultSet��Case�When)�Exact)�tree)�cached_property�AND�OR�XORc @ s e Zd ZdZeZdZdZd5dd�Zdd� Z dd � Z d d � Z d d � Z dd� Z dd� Zdd� Zdd� Zdd� Zedd� �Zedd� �Zed d � �Zed d � �Zed d!� �Zed"d#� �Zed$d%� �Z d&d'� Z ed(d)� �Z ed*d+� �Z d,d-� Z d.d/� Z!d0d1� Z"d2d3� Z#d4S )6� WhereNodea� An SQL WHERE clause. The class is tied to the Query class that created it (in order to create the correct SQL). A child is usually an expression producing boolean values. Most likely the expression is a Lookup instance. However, a child could also be any class with as_sql() and either relabeled_clone() method or relabel_aliases() and clone() methods and contains_aggregate attribute. FTc C s� | j s| js| ddfS || jA }|r.| jtkpF| r>| jtkpF| jtk}|rb| j rb| jsbd| dfS g }g }g }| jD ]�}t|d�r�|� ||�\} } } | dur�|� | � | dur�|� | � | dur�|� | � qt|jr�|� |� qt|j r�|� |� qt|� |� qt|�r2|�r2|�r|�r$|�s$dd| fS |�r2t d��|�rJ| � || j| j�nd} |�rf| � || j| j�nd} |�r�| � || j| j�nd}| | |fS )a Return three possibly None nodes: one for those parts of self that should be included in the WHERE clause, one for those parts of self that must be included in the HAVING clause, and one for those parts that refer to window functions. N�split_having_qualifyzzHeterogeneous disjunctive predicates against window functions are not implemented when performing conditional aggregation.) �contains_aggregate�contains_over_clause�negated� connectorr r r �children�hasattrr �append�NotImplementedError�create)�selfr Z must_group_byZ in_negatedZmust_remain_connectedZ where_partsZ having_partsZ qualify_parts�cZ where_partZ having_partZ qualify_partZ where_nodeZ having_nodeZ qualify_node� r �[/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-packages/django/db/models/sql/where.pyr & sl ����� � �������z WhereNode.split_having_qualifyc C s� g }g }| j tkr$t| j�d }}ndt| j� }}| j tkr�|jjs�| �| jt�}t t j dd� | jD ��}t d|�} | �|| gt| j ��||�S | jD ]�} z|�| �\} } W n2 ty� |d8 }Y n> ty� |d8 }Y n&0 | � r�|�| � |�| � n|d8 }|dk�r"| j �r t�nt�|dkr�| j �r8t�q�t�q�d| j } | �|�}|�s\t�| j �rnd| }n t|�dk�s�| j�r�d| }||fS )a Return the SQL version of the where clause and the value to be substituted in. Return '', [] if this node matches everything, None, [] if this node is empty, and raise EmptyResultSet if this node can't match anything. � c s s" | ]}t t|d d�dd�V qdS )r )Zthenr )�defaultNr ��.0r r r r � <genexpr>� � z#WhereNode.as_sql.<locals>.<genexpr>r z %s zNOT (%s)�(%s))r r �lenr r �featuresZsupports_logical_xor� __class__r r �operator�addr r �as_sql�compiler r r �extend�join�resolved)r �compiler� connection�resultZ result_paramsZ full_neededZ empty_needed�lhsZrhs_sum�rhs�child�sql�params�connZ sql_stringr r r r) r sV � � zWhereNode.as_sqlc C s" g }| j D ]}|�|�� � q |S �N)r r+ �get_group_by_cols)r �colsr3 r r r r8 � s zWhereNode.get_group_by_colsc C s | j d d � S r7 )r �r r r r �get_source_expressions� s z WhereNode.get_source_expressionsc C s t |�t | j�ksJ �|| _d S r7 )r$ r )r r r r r �set_source_expressions� s z WhereNode.set_source_expressionsc C sH t | j�D ]8\}}t|d�r(|�|� q t|d�r |�|�| j|< q dS )z� Relabel the alias values of any children. 'change_map' is a dictionary mapping old (current) alias values to the new values. �relabel_aliases�relabeled_cloneN)� enumerater r r= r> )r � change_map�posr3 r r r r= � s zWhereNode.relabel_aliasesc C s@ | j | j| jd�}| jD ]"}t|d�r.|�� }|j�|� q|S )N�r r �clone)r r r r r rC r )r rC r3 r r r rC � s zWhereNode.clonec C s | � � }|�|� |S r7 )rC r= )r r@ rC r r r r> � s zWhereNode.relabeled_clonec C sF |� | � }r|S | j| j| jd�}| jD ]}|j�|�|�� q*|S )NrB )�getr r r r r �replace_expressions)r Z replacements� replacementrC r3 r r r rE � s z WhereNode.replace_expressionsc C s" t � }| jD ]}||�� O }q |S r7 )�setr �get_refs)r �refsr3 r r r rH � s zWhereNode.get_refsc s* t |tj�r$t� fdd�|jD ��S |jS )Nc 3 s | ]}� � |�V qd S r7 ��_contains_aggregater ��clsr r r! � r" z0WhereNode._contains_aggregate.<locals>.<genexpr>)� isinstancer �Node�anyr r �rM �objr rL r rK � s z WhereNode._contains_aggregatec C s | � | �S r7 rJ r: r r r r � s z WhereNode.contains_aggregatec s* t |tj�r$t� fdd�|jD ��S |jS )Nc 3 s | ]}� � |�V qd S r7 ��_contains_over_clauser rL r r r! � r" z2WhereNode._contains_over_clause.<locals>.<genexpr>)rN r rO rP r r rQ r rL r rT � s z WhereNode._contains_over_clausec C s | � | �S r7 rS r: r r r r � s z WhereNode.contains_over_clausec C s t dd� | jD ��S )Nc s s | ] }|j V qd S r7 )� is_summary)r r3 r r r r! r" z'WhereNode.is_summary.<locals>.<genexpr>)rP r r: r r r rU s zWhereNode.is_summaryc O s& t | d�r"| j|g|�R i |��} | S )N�resolve_expression)r rV )�expr�query�args�kwargsr r r � _resolve_leaf s zWhereNode._resolve_leafc O s� t |d�r0|jD ] }| j||g|�R i |�� qt |d�rX| j|j|g|�R i |��|_t |d�r�| j|j|g|�R i |��|_d S )Nr r1 r2 )r r � _resolve_noder[ r1 r2 )rM �noderX rY rZ r3 r r r r\ s zWhereNode._resolve_nodec O s* | � � }|j|g|�R i |�� d|_|S )NT)rC r\ r- )r rY rZ rC r r r rV s z WhereNode.resolve_expressionc C s ddl m} |� S )Nr )� BooleanField)Zdjango.db.modelsr^ )r r^ r r r � output_field s zWhereNode.output_fieldc C s | j S r7 )r_ r: r r r �_output_field_or_none s z WhereNode._output_field_or_nonec C s |j jjsd|� d�}||fS )Nz CASE WHEN z THEN 1 ELSE 0 END)r/ r% Z&supports_boolean_expr_in_select_clause)r r. r4 r5 r r r � select_format$ s zWhereNode.select_formatc C s | j �|�S r7 )r_ �get_db_converters)r r/ r r r rb , s zWhereNode.get_db_convertersc C s | j �|�S r7 )r_ � get_lookup)r �lookupr r r rc / s zWhereNode.get_lookupc c s0 | j D ]$}t|t�r$|�� E d H q|V qd S r7 )r rN r �leaves)r r3 r r r re 2 s zWhereNode.leavesN)FF)$�__name__� __module__� __qualname__�__doc__r r r- Z conditionalr r) r8 r; r< r= rC r> rE rH � classmethodrK r r rT r �propertyrU � staticmethodr[ r\ rV r_ r` ra rb rc re r r r r r sJ LF r c @ s" e Zd ZdZdZdZddd�ZdS )� NothingNodez A node that matches nothing.FNc C s t �d S r7 )r )r r. r/ r r r r) @ s zNothingNode.as_sql)NN)rf rg rh ri r r r) r r r r rm : s rm c @ s& e Zd ZdZdZdd� Zddd�ZdS )� ExtraWhereFc C s || _ || _d S r7 )�sqlsr5 )r ro r5 r r r �__init__I s zExtraWhere.__init__Nc C s( dd� | j D �}d�|�t| jp"d�fS )Nc S s g | ] }d | �qS )r# r )r r4 r r r � <listcomp>N r" z%ExtraWhere.as_sql.<locals>.<listcomp>z AND r )ro r, �listr5 )r r. r/ ro r r r r) M s zExtraWhere.as_sql)NN�rf rg rh r r rp r) r r r r rn D s rn c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�SubqueryConstraintFc C s( || _ || _|| _|jdd� || _d S )NT)Z clear_default)�alias�columns�targetsZclear_ordering� query_object)r ru rv rw rx r r r rp X s zSubqueryConstraint.__init__c C s0 | j }|�| j� |j|d�}|�| j| j|�S )N)r/ )rx Z set_valuesrw Z get_compilerZas_subquery_conditionru rv )r r. r/ rX Zquery_compilerr r r r) _ s zSubqueryConstraint.as_sqlNrs r r r r rt R s rt )ri r' � functoolsr Zdjango.core.exceptionsr r Z django.db.models.expressionsr r Zdjango.db.models.lookupsr Z django.utilsr Zdjango.utils.functionalr r r r rO r rm rn rt r r r r �<module> s )