OwlCyberSecurity - MANAGER
Edit File: constraints.cpython-39.pyc
a kH�f < � @ s� d dl mZ d dlmZmZ d dlmZ d dlmZm Z m Z m Z d dl m Z d dlmZ d dlmZ d dlmZ d d lmZ d d lmZ g d �ZG d d � d �ZG dd� de�ZG dd� de�Z G dd� de�Z dS )� )�Enum)� FieldError�ValidationError)� connections)�Exists�ExpressionList�F�OrderBy)�IndexExpression)�Exact)�Q)�Query)�DEFAULT_DB_ALIAS)� gettext_lazy)�BaseConstraint�CheckConstraint� Deferrable�UniqueConstraintc @ sl e Zd Zed�ZdZddd�Zedd� �Zdd� Z d d � Z d d � Z de fd d�Z dd� Zdd� Zdd� ZdS )r u& Constraint “%(name)s” is violated.Nc C s" || _ |d ur|| _n| j| _d S �N)�name�violation_error_message� default_violation_error_message)�selfr r � r �^/home/gouroczh/virtualenv/test/3.9/lib/python3.9/site-packages/django/db/models/constraints.py�__init__ s zBaseConstraint.__init__c C s dS )NFr �r r r r �contains_expressions s z#BaseConstraint.contains_expressionsc C s t d��d S �Nz.This method must be implemented by a subclass.��NotImplementedError�r �model� schema_editorr r r �constraint_sql s z BaseConstraint.constraint_sqlc C s t d��d S r r r! r r r � create_sql" s zBaseConstraint.create_sqlc C s t d��d S r r r! r r r � remove_sql% s zBaseConstraint.remove_sqlc C s t d��d S r r )r r" �instance�exclude�usingr r r �validate( s zBaseConstraint.validatec C s | j d| ji S )Nr )r r r r r r �get_violation_error_message+ s z*BaseConstraint.get_violation_error_messagec C sT d| j j| j jf }|�dd�}d| ji}| jd urJ| j| jkrJ| j|d< |d|fS )Nz%s.%sz django.db.models.constraintszdjango.db.modelsr r r )� __class__� __module__�__name__�replacer r r )r �path�kwargsr r r � deconstruct. s � � zBaseConstraint.deconstructc C s | � � \}}}| j|i |��S r )r2 r, )r �_�argsr1 r r r �clone9 s zBaseConstraint.clone)N)r. r- � __qualname__r3 r r r �propertyr r$ r% r& r r* r+ r2 r5 r r r r r s r c sp e Zd Zdd�� fdd� Zdd� Zdd� Zd d � Zd d � Zdefd d�Z dd� Z � fdd�Z � fdd�Z � Z S )r N�r c s. || _ t|dd�std��t� j||d� d S )NZ conditionalFzACheckConstraint.check must be a Q instance or boolean expression.r8 )�check�getattr� TypeError�superr )r r9 r r �r, r r r ? s �zCheckConstraint.__init__c sR t |dd�}|�| j�}|j� jd�}|�|� j�\}}|t� fdd�|D �� S )NF�r" � alias_cols�� connectionc 3 s | ]}� � |�V qd S r �Z quote_value��.0�p�r# r r � <genexpr>L � z1CheckConstraint._get_check_sql.<locals>.<genexpr>)r � build_wherer9 � get_compilerrA �as_sql�tuple�r r" r# �query�where�compiler�sql�paramsr rF r �_get_check_sqlG s z CheckConstraint._get_check_sqlc C s | � ||�}|�| j|�S r )rS Z _check_sqlr �r r" r# r9 r r r r$ N s z CheckConstraint.constraint_sqlc C s | � ||�}|�|| j|�S r )rS Z_create_check_sqlr rT r r r r% R s zCheckConstraint.create_sqlc C s |� || j�S r )Z_delete_check_sqlr r! r r r r& V s zCheckConstraint.remove_sqlc C sL |j |j|d�}z$t| j�j||d�s2t| �� ��W n tyF Y n0 d S )N��metar( �r) )�_get_field_value_map�_metar r9 r r+ r )r r" r' r( r) �againstr r r r* Y s zCheckConstraint.validatec C s d| j j| jt| j�f S )Nz<%s: check=%s name=%s>)r, r6 r9 �reprr r r r r �__repr__a s �zCheckConstraint.__repr__c s: t |t�r.| j|jko,| j|jko,| j|jkS t� �|�S r )� isinstancer r r9 r r< �__eq__�r �otherr= r r r^ h s � �zCheckConstraint.__eq__c s$ t � �� \}}}| j|d< |||fS )Nr9 )r< r2 r9 �r r0 r4 r1 r= r r r2 q s zCheckConstraint.deconstruct)r. r- r6 r rS r$ r% r& r r* r\ r^ r2 � __classcell__r r r= r r > s r c @ s e Zd ZdZdZdd� ZdS )r �deferredZ immediatec C s | j j� d| j� �S )N�.)r, r6 �_name_r r r r r\ | s zDeferrable.__repr__N)r. r- r6 ZDEFERREDZ IMMEDIATEr\ r r r r r w s r c s� e Zd Zdddddddd�� fdd� Zedd� �Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z � fdd�Z � fdd�Z defdd�Z� ZS )r r N)�fieldsr � condition� deferrable�include� opclassesr c sV |s t d��|s |s t d��|r,|r,t d��t|td �tf�sFt d��|rV|rVt d��|rf|rft d��|rv|rvt d��|r�|r�t d��|r�|r�t d ��t|td �tf�s�t d ��t|td �ttf�s�t d ��t|ttf�s�t d ��|r�t|�t|�kr�t d ��t|�| _|| _ || _ |�r"t|�nd| _ || _ tdd� |D ��| _ t� j||d� d S )Nz"A unique constraint must be named.zKAt least one field or expression is required to define a unique constraint.z?UniqueConstraint.fields and expressions are mutually exclusive.z0UniqueConstraint.condition must be a Q instance.z4UniqueConstraint with conditions cannot be deferred.z8UniqueConstraint with include fields cannot be deferred.z3UniqueConstraint with opclasses cannot be deferred.z5UniqueConstraint with expressions cannot be deferred.zrUniqueConstraint.opclasses cannot be used with expressions. Use django.contrib.postgres.indexes.OpClass() instead.z:UniqueConstraint.deferrable must be a Deferrable instance.z1UniqueConstraint.include must be a list or tuple.z3UniqueConstraint.opclasses must be a list or tuple.z]UniqueConstraint.fields and UniqueConstraint.opclasses must have the same number of elements.r c s s$ | ] }t |t�rt|�n|V qd S r )r] �strr )rD � expressionr r r rG � s �z,UniqueConstraint.__init__.<locals>.<genexpr>r8 )� ValueErrorr] �typer r �listrL �lenrf rg rh ri rj � expressionsr< r ) r rf r rg rh ri rj r rq r= r r r � sZ ����� � zUniqueConstraint.__init__c C s t | j�S r )�boolrq r r r r r � s z%UniqueConstraint.contains_expressionsc s` | j d u rd S t|dd�}|�| j �}|j� jd�}|�|� j�\}}|t� fdd�|D �� S )NFr> r@ c 3 s | ]}� � |�V qd S r rB rC rF r r rG � rH z6UniqueConstraint._get_condition_sql.<locals>.<genexpr>)rg r rI rJ rA rK rL rM r rF r �_get_condition_sql� s z#UniqueConstraint._get_condition_sqlc C sN | j s d S g }| j D ]"}t|�}|�|j� |�|� qt|� �t|dd��S )NF)r? )rq r Zset_wrapper_classesrA �appendr Zresolve_expressionr )r r" r# Zindex_expressionsrl Zindex_expressionr r r �_get_index_expressions� s �z'UniqueConstraint._get_index_expressionsc s` � fdd�| j D �}� fdd�| jD �}| �� |�}| �� |�}|j� || j|| j|| j|d�S )Nc s g | ]}� j �|��qS r �rY � get_field�rD � field_name�r" r r � <listcomp>� rH z3UniqueConstraint.constraint_sql.<locals>.<listcomp>c s g | ]}� j �|�j�qS r �rY rw �columnrx rz r r r{ � s �rg rh ri rj rq )rf ri rs ru Z _unique_sqlr rh rj �r r" r# rf ri rg rq r rz r r$ � s � �z UniqueConstraint.constraint_sqlc s` � fdd�| j D �}� fdd�| jD �}| �� |�}| �� |�}|j� || j|| j|| j|d�S )Nc s g | ]}� j �|��qS r rv rx rz r r r{ � rH z/UniqueConstraint.create_sql.<locals>.<listcomp>c s g | ]}� j �|�j�qS r r| rx rz r r r{ � s r~ )rf ri rs ru Z_create_unique_sqlr rh rj r r rz r r% � s � �zUniqueConstraint.create_sqlc sJ | � � |�}� fdd�| jD �}| �� |�}|j� | j|| j|| j|d�S )Nc s g | ]}� j �|�j�qS r r| rx rz r r r{ � s z/UniqueConstraint.remove_sql.<locals>.<listcomp>r~ )rs ri ru Z_delete_unique_sqlr rh rj )r r" r# rg ri rq r rz r r& � s � �zUniqueConstraint.remove_sqlc C s� d| j j| jsdn dt| j� | js(dn dt| j� dt| j� | jd u rNdnd| j | jd u rddnd| j | jsvdn dt| j� | j s�dn d t| j � f S ) Nz<%s:%s%s%s%s%s%s%s>� z fields=%sz expressions=%sz name=%sz condition=%sz deferrable=%rz include=%sz opclasses=%s) r, r6 rf r[ rq r rg rh ri rj r r r r r\ s �zUniqueConstraint.__repr__c sv t |t�rj| j|jkoh| j|jkoh| j|jkoh| j|jkoh| j|jkoh| j|jkoh| j|jkoh| j |j kS t � � |�S r ) r] r r rf rg rh ri rj rq r r< r^ r_ r= r r r^ s" � � � � � � � zUniqueConstraint.__eq__c sl t � �� \}}}| jr | j|d< | jr0| j|d< | jr@| j|d< | jrP| j|d< | jr`| j|d< || j|fS )Nrf rg rh ri rj )r< r2 rf rg rh ri rj rq ra r= r r r2 % s z UniqueConstraint.deconstructc C s8 |j �|�}| jr�i }| jD ]Z}|r2||v r2 d S |j�|�}t||j�} | d u sf| dkrlt| jj rl d S | ||j < q |j f i |��}n�|r�| j D ]T} t | d�r�| �� D ] } t| t�r�| j |v r� d S q�q�t| t�r�| j |v r� d S q�dd� |j|j|d��� D �} g } | j D ].} t| t��r(| j} | �t| | �| ��� �q|j | � }|�|j�}|jj�sx|d u�rx|j|d�}| j �s�|� � �r4| j �r�t | � � ��|� � D ]2\}}|D ]"}|| u �r�t |�!|| j����q��q�nX|j|j|d�}z2| j t"|� | j ��@ j#||d��r t | � � ��W n t$�y2 Y n0 d S )Nr� �flattenc S s i | ]\}}t |�|�qS r )r )rD �field�valuer r r � <dictcomp>O s �z-UniqueConstraint.validate.<locals>.<dictcomp>rU )�pkrW )%Z_default_managerr) rf rY rw r: Zattnamer �featuresZ!interprets_empty_strings_as_nullsr �filterrq �hasattrr� r] r rX �itemsr rl rt r Zreplace_expressionsZ _get_pk_val�_stateZaddingr( rg �existsr r+ Zget_constraintsZunique_error_messager r9 r )r r" r' r( r) ZquerysetZ lookup_kwargsry r� Z lookup_valuerl �exprZ replacementsrq Zmodel_class_pk� constraints� constraintrZ r r r r* 3 sp � � �� ��zUniqueConstraint.validate)r. r- r6 r r7 r rs ru r$ r% r& r\ r^ r2 r r* rb r r r= r r � s& �= r N) �enumr Zdjango.core.exceptionsr r Z django.dbr Z django.db.models.expressionsr r r r Zdjango.db.models.indexesr Zdjango.db.models.lookupsr Z django.db.models.query_utilsr Zdjango.db.models.sql.queryr Zdjango.db.utilsr Zdjango.utils.translationr r3 �__all__r r r r r r r r �<module> s .9