OwlCyberSecurity - MANAGER
Edit File: constraints.cpython-39.pyc
a jH�f' � @ s� d dl Z d dlmZ d dlmZ d dlmZmZ d dlm Z m Z m Z d dl m Z mZmZmZ d dlmZmZ d dlmZ d d lmZ d d lmZ d d lmZ d gZ G d d� de�Z G dd � d e �Z dS )� N)�OpClass)�ValidationError)�DEFAULT_DB_ALIAS�NotSupportedError)� Expressions� Statement�Table)�BaseConstraint� Deferrable�F�Q)�Exists�ExpressionList)�IndexExpression)�PostgresOperatorLookup)�Query)�RemovedInDjango50Warning�ExclusionConstraintc @ s e Zd ZdZdS )� ExclusionConstraintExpressionz!%(expressions)s WITH %(operator)sN)�__name__� __module__� __qualname__�template� r r �e/home/gouroczh/virtualenv/test/3.9/lib/python3.9/site-packages/django/contrib/postgres/constraints.pyr s r c s� e Zd ZdZddddddd�� fdd� Zdd� Zd d � Zd d � Zd d� Zdd� Z dd� Z � fdd�Z � fdd�Z dd� Z defdd�Z� ZS )r zdCONSTRAINT %(name)s EXCLUDE USING %(index_type)s (%(expressions)s)%(include)s%(where)s%(deferrable)sNr )� index_type� condition� deferrable�include� opclasses�violation_error_messagec s |r|� � dvrtd��|s$td��tdd� |D ��s>td��t|td �tf�sXtd��t|td �tf�srtd��t|td �ttf�s�td ��t|ttf�s�td ��|r�t |�t |�kr�td ��|| _ |p�d | _ || _ || _ |r�t|�nd | _|| _| j�r tjdtdd� t� j||d� d S )N> �gist�spgistz;Exclusion constraints only support GiST or SP-GiST indexes.zFAt least one expression is required to define an exclusion constraint.c s s( | ] }t |ttf�o t|�d kV qdS )� N)� isinstance�list�tuple�len)�.0�exprr r r � <genexpr>0 s z/ExclusionConstraint.__init__.<locals>.<genexpr>z+The expressions must be a list of 2-tuples.z3ExclusionConstraint.condition must be a Q instance.z=ExclusionConstraint.deferrable must be a Deferrable instance.z4ExclusionConstraint.include must be a list or tuple.z6ExclusionConstraint.opclasses must be a list or tuple.zhExclusionConstraint.expressions and ExclusionConstraint.opclasses must have the same number of elements.ZGISTr z�The opclasses argument is deprecated in favor of using django.contrib.postgres.indexes.OpClass in ExclusionConstraint.expressions.r# )�category� stacklevel)�namer )�lower� ValueError�allr$ �typer r r% r& r' � expressionsr r r r r �warnings�warnr �super�__init__) �selfr- r2 r r r r r r �� __class__r r r6 sN ����� �z ExclusionConstraint.__init__c C s� g }t | j�D ]h\}\}}t|t�r,t|�}zt|| j| �}W n tyR Y n0 t||d�}|� |j � |� |� qt |� � |�S )N)�operator)� enumerater2 r$ �strr r r � IndexErrorr Zset_wrapper_classes� connection�appendr Zresolve_expression)r7 � schema_editor�queryr2 �idx� expressionr: r r r �_get_expressionsT s z$ExclusionConstraint._get_expressionsc sF | j d u rd S |�| j �}|�|� j�\}}|t� fdd�|D �� S )Nc 3 s | ]}� � |�V qd S �N)� quote_value)r( �p�r@ r r r* g � z9ExclusionConstraint._get_condition_sql.<locals>.<genexpr>)r Z build_whereZas_sqlr> r& )r7 �compilerr@ rA �where�sql�paramsr rH r �_get_condition_sqlb s z&ExclusionConstraint._get_condition_sqlc s� t � dd�}|j|jd�}| �||�}� jj}| �|||�}� fdd�| jD �}t| j t ||j �|� | j �| j t||||j�|r�d| nd|�� |�|�| j�d�S ) NF)Z alias_cols)r> c s g | ]}� j �|�j�qS r )�_meta� get_field�column)r( � field_name��modelr r � <listcomp>o s z6ExclusionConstraint.constraint_sql.<locals>.<listcomp>z WHERE (%s)� )�tabler- r r2 rK r r )r Z get_compilerr> rD rO �db_tablerN r r r r � quote_namer- r r rF Z_index_include_sqlZ_deferrable_constraint_sqlr ) r7 rT r@ rA rJ r2 rW r r r rS r �constraint_sqli s( � � �z"ExclusionConstraint.constraint_sqlc C s, | � |� tdt|jj|j�| �||�d�S )Nz(ALTER TABLE %(table)s ADD %(constraint)s)rW � constraint)�check_supportedr r rO rX rY rZ �r7 rT r@ r r r � create_sql s �z ExclusionConstraint.create_sqlc C s |� |j||�| j��S rE )Z_delete_constraint_sqlZsql_delete_checkrY r- r] r r r � remove_sql� s �z ExclusionConstraint.remove_sqlc C s* | j r&| j�� dkr&|jjjs&td��d S )Nr" zMCovering exclusion constraints using an SP-GiST index require PostgreSQL 14+.)r r r. r> �featuresZ supports_covering_spgist_indexesr )r7 r@ r r r r\ � s � ���z#ExclusionConstraint.check_supportedc s� t � �� \}}}| j|d< | jd ur.| j|d< | j�� dkrF| j|d< | jrV| j|d< | jrf| j|d< | jrv| j|d< |||fS )Nr2 r r! r r r r ) r5 � deconstructr2 r r r. r r r )r7 �path�args�kwargsr8 r r ra � s z ExclusionConstraint.deconstructc sx t || j�rl| j|jkoj| j|jkoj| j|jkoj| j|jkoj| j|jkoj| j|jkoj| j|jkoj| j |j kS t � � |�S rE ) r$ r9 r- r r2 r r r r r r5 �__eq__)r7 �otherr8 r r re � s" � � � � � � � zExclusionConstraint.__eq__c C s~ d| j jt| j�t| j�t| j�| jd u r.dnd| j | jd u rDdnd| j | jsVdn dt| j� | j sldn dt| j � f S )Nz2<%s: index_type=%s expressions=%s name=%s%s%s%s%s>rV z condition=%sz deferrable=%rz include=%sz opclasses=%s) r9 r �reprr r2 r- r r r r )r7 r r r �__repr__� s �z ExclusionConstraint.__repr__c C s� |j �|�}|j|j|d�}dd� |�� D �}g }t| j�D ]�\} \} } t| t�rZt | �} |r�t| t �rz| j |v r� d S n*| � � D ] } t| t �r�| j |v r� d S q�| � |�} t| t �r�| �� d } t| t �r�| �� d } t| | d�}| |_|�|� q<|j|� }|�|j�}|jj�s.|d u�r.|j|d�}| j�sN|�� �r|t| �� ��n.| jt|�| j��@ j ||d��r|t| �� ��d S )N)�meta�excludec S s i | ]\}}t |�|�qS r )r )r( �field�valuer r r � <dictcomp>� rI z0ExclusionConstraint.validate.<locals>.<dictcomp>r )�lhs�rhs)�pk)�using) Z_default_managerrq Z_get_field_value_maprO �itemsr; r2 r$ r< r r- �flattenZreplace_expressionsr Zget_source_expressionsr Zpostgres_operatorr? �filterZ _get_pk_val�_stateZaddingrj r �existsr Zget_violation_error_messager �check)r7 rT �instancerj rq ZquerysetZreplacement_mapZ replacements�lookupsrB rC r: r) Zrhs_expression�lookupZmodel_class_pkr r r �validate� sF � �z ExclusionConstraint.validate)r r r r r6 rD rN rZ r^ r_ r\ ra re rh r r{ � __classcell__r r r8 r r s& � �9 ) r3 Z django.contrib.postgres.indexesr Zdjango.core.exceptionsr Z django.dbr r Z!django.db.backends.ddl_referencesr r r Zdjango.db.modelsr r r r Z django.db.models.expressionsr r Zdjango.db.models.indexesr Zdjango.db.models.lookupsr Zdjango.db.models.sqlr Zdjango.utils.deprecationr �__all__r r r r r r �<module> s