OwlCyberSecurity - MANAGER
Edit File: schema.cpython-39.pyc
a kH�fd � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZmZmZm Z m Z m Z d dl m Z mZmZ d dlmZmZ d dlmZ d dlmZmZ d d lmZ e �d �Zd d � Z d d� Z dd� Z G dd� d�Z dS )� N)�datetime)�settings)�Columns� Expressions�ForeignKeyName� IndexName� Statement�Table)� names_digest�split_identifier� truncate_name)� Deferrable�Index)�Query)�TransactionManagementError�atomic)�timezonezdjango.db.backends.schemac C s2 | j }|jrdS |jr&|jdgkr&dS |j|jv S )zz When altering the given field, must constraints on its model from the given relation be temporarily dropped? FNT)�field� many_to_many� primary_keyZ to_fields�name)ZrelationZ altered_fieldr � r �`/home/gouroczh/virtualenv/test/3.9/lib/python3.9/site-packages/django/db/backends/base/schema.py�_is_relevant_relation s r c C s" t | jjddddd�t�d�d�S )NFT)�forward�reverseZinclude_hiddenZinclude_parentsr )�key)�sorted�_metaZ _get_fields�operator� attrgetter��modelr r r �_all_related_fields' s ��r# c # sb t �fdd�t�j�D �� fdd�t� j�D ��}|D ]&\}}||fV t|j|j�E d H q6d S )Nc 3 s | ]}t |� �r|V qd S �N�r ��.0�obj)� old_fieldr r � <genexpr>8 s �z+_related_non_m2m_objects.<locals>.<genexpr>c 3 s | ]}t |� �r|V qd S r$ r% r& )� new_fieldr r r* = s �)�zipr# r" �_related_non_m2m_objects� remote_field)r) r+ Zrelated_fields�old_rel�new_relr )r+ r) r r- 4 s � �� �r- c @ s e Zd ZdZdZdZdZdZdZdZ dZ d Z d Z d Z d ZeZd ZdZdZdZdZdZdZdZeZdZeZdZdZ dZ eZ dZ dZ dZ!dZ"d Z#eZ$d Z%d Z&d Z'd�d"d#�Z(d$d%� Z)d&d'� Z*d�d)d*�Z+d+d,� Z,d-d.� Z-d/d0� Z.d�d1d2�Z/d3d4� Z0d5d6� Z1d7d8� Z2d9d:� Z3e4d;d<� �Z5d=d>� Z6d?d@� Z7dAdB� Z8dCdD� Z9dEdF� Z:dGdH� Z;dIdJ� Z<dKdL� Z=dMdN� Z>dOdP� Z?dQdR� Z@dSdT� ZAdUdV� ZBdWdX� ZCdYdZ� ZDd[d\� ZEd]d^� ZFd�d_d`�ZGd�dadb�ZHdcdd� ZId�dedf�ZJdgdh� ZKdidj� ZLdkdl� ZMdmdn� ZNd�dpdq�ZOd�drds�ZPdtdu� ZQdvdw� ZRdddododd(dd(ddddx� dydz�ZSd�d{d|�ZTd}d~� ZUdd�� ZVd�d�� ZWd�d�� ZXd�d�d��ZYd�d�� ZZd�d�� Z[d�d�� Z\d�d�� Z]d�d�� Z^d�d�� Z_d�d�� Z`d�d�� Zad�d�d��Zbd�d�d��Zcd�d�d��Zdd�d�d��Zed�d�� Zfd�d�� Zgd�d�� Zhd�d�� Zid�d�d��Zjd�d�d��Zkd�d�� Zld�d�� Zmd�d�d��Znd�d�d��ZodS )��BaseDatabaseSchemaEditorz� This class and its subclasses are responsible for emitting schema-changing statements to the databases - model creation/removal/alteration, field renaming, index fiddling, and so on. z'CREATE TABLE %(table)s (%(definition)s)z1ALTER TABLE %(old_table)s RENAME TO %(new_table)sz7ALTER TABLE %(table)s SET TABLESPACE %(new_tablespace)sz DROP TABLE %(table)s CASCADEz:ALTER TABLE %(table)s ADD COLUMN %(column)s %(definition)sz!ALTER TABLE %(table)s %(changes)sz2ALTER COLUMN %(column)s TYPE %(type)s%(collation)sz%ALTER COLUMN %(column)s DROP NOT NULLz$ALTER COLUMN %(column)s SET NOT NULLz/ALTER COLUMN %(column)s SET DEFAULT %(default)sz$ALTER COLUMN %(column)s DROP DEFAULTz4ALTER TABLE %(table)s DROP COLUMN %(column)s CASCADEzDALTER TABLE %(table)s RENAME COLUMN %(old_column)s TO %(new_column)szFUPDATE %(table)s SET %(column)s = %(default)s WHERE %(column)s IS NULLz"UNIQUE (%(columns)s)%(deferrable)szCHECK (%(check)s)z.ALTER TABLE %(table)s DROP CONSTRAINT %(name)sz"CONSTRAINT %(name)s %(constraint)sz?ALTER TABLE %(table)s ADD CONSTRAINT %(name)s CHECK (%(check)s)zPALTER TABLE %(table)s ADD CONSTRAINT %(name)s UNIQUE (%(columns)s)%(deferrable)sz|ALTER TABLE %(table)s ADD CONSTRAINT %(name)s FOREIGN KEY (%(column)s) REFERENCES %(to_table)s (%(to_column)s)%(deferrable)sNzQCREATE INDEX %(name)s ON %(table)s (%(columns)s)%(include)s%(extra)s%(condition)szOCREATE UNIQUE INDEX %(name)s ON %(table)s (%(columns)s)%(include)s%(condition)sz/ALTER INDEX %(old_name)s RENAME TO %(new_name)szDROP INDEX %(name)szGALTER TABLE %(table)s ADD CONSTRAINT %(name)s PRIMARY KEY (%(columns)s)z DROP PROCEDURE %(procedure)sz)COMMENT ON TABLE %(table)s IS %(comment)sz5COMMENT ON COLUMN %(table)s.%(column)s IS %(comment)sFTc C s, || _ || _| jrg | _| j jjo$|| _d S r$ )� connection� collect_sql� collected_sql�features�can_rollback_ddl�atomic_migration)�selfr2 r3 r r r r �__init__� s z!BaseDatabaseSchemaEditor.__init__c C s( g | _ | jr$t| jj�| _| j�� | S r$ )� deferred_sqlr7 r r2 �alias� __enter__�r8 r r r r< � s z"BaseDatabaseSchemaEditor.__enter__c C s8 |d u r | j D ]}| �|� q| jr4| j�|||� d S r$ )r: �executer7 r �__exit__)r8 �exc_type� exc_value� traceback�sqlr r r r? � s z!BaseDatabaseSchemaEditor.__exit__r c C s� | j s | jjr | jjjs td��t|�}tjd||||d�d� | j r�|� � � d�rXdnd}|dur�| j � |t t| j|�� | � q�| j � || � n6| j�� � }|�||� W d � n1 s�0 Y dS )z:Execute the given SQL statement, with optional parameters.ziExecuting DDL statements while in a transaction on databases that can't perform a rollback is prohibited.z%s; (params %r))�paramsrC )�extra�;� N)r3 r2 Zin_atomic_blockr5 r6 r �str�logger�debug�rstrip�endswithr4 �append�tuple�map� quote_value�cursorr> )r8 rC rD �endingrQ r r r r> � s, ������ z BaseDatabaseSchemaEditor.executec C s | j j�|�S r$ )r2 �ops� quote_name)r8 r r r r rT � s z#BaseDatabaseSchemaEditor.quote_namec s � j jD ]*}� fdd�|D �}�j���� |�� qg }g }� j jD �]<}��� |�\}}|du rdqD|j�jd�} | d r�|d�j | 7 }|j �jd�} | r�|d| 7 }|� |� |j �r6|j �r6|j jj j} |j jj �|j j�j} �j�r|d�j��| ���| �d� 7 }n"�jjj�r6�j���� |d �� |�d ��|j�|f � |�� d v rD�jj�� j j|j�} | rD�j� | � qD� �fd d�� j jD �}�j ��� j j�d � dd� g |�|�R D ��d� }� j j �r��jj� � j j �}|�r�|d| 7 }||fS )z-Take a model and return its table definition.c s g | ]}� j �|��qS r �r � get_field�r' r r! r r � <listcomp>� � z6BaseDatabaseSchemaEditor.table_sql.<locals>.<listcomp>N�r2 �check� z %s)�to_table� to_column� _fk_%(to_table)s_%(to_column)sz%s %s)Z AutoFieldZ BigAutoFieldZSmallAutoFieldc s g | ]}|� � ���qS r )Zconstraint_sql�r' � constraint�r" r8 r r rX s ��, c s s | ]}|rt |�V qd S r$ )rH r` r r r r* s �z5BaseDatabaseSchemaEditor.table_sql.<locals>.<genexpr>)�table� definition) r Zunique_togetherr: rM �_create_unique_sql� local_fields� column_sql� db_parametersr2 �sql_check_constraint�db_type_suffix�extendr. � db_constraintr" �db_tablerV � field_name�column�sql_create_inline_fkrT r5 �supports_foreign_keys�_create_fk_sql�get_internal_typerS � autoinc_sql� constraints�sql_create_table�join� db_tablespace�tablespace_sql)r8 r" � field_names�fieldsZ column_sqlsrD r re Z extra_params� db_params�col_type_suffixr] r^ ru rv rC rz r rb r � table_sql� sz �� �� ��� � � �� � z"BaseDatabaseSchemaEditor.table_sqlc c sH |V |� d� }r | �|�V | jjjr>|jr>| �|j�V |j}|ob| �|� ob|o`| � |� }|r�| � |�} | d ur�d| � |� } | jjj r�| | � | � V n| V |�| � |jr�|js�| jjjr�d}|s�dV n| jjjs�dV |jr�dV n|j�rdV |j�p|jj} | �rD| jjj�rD|j�rD| jjj| dd�V d S ) N� collationzDEFAULT TzNOT NULLZNULLz PRIMARY KEYZUNIQUE)�inline)�get� _collate_sqlr2 r5 �supports_comments_inline� db_comment� _comment_sql�null� skip_default�skip_default_on_alter�effective_default�_column_default_sql�requires_literal_defaults�prepare_defaultrM �empty_strings_allowedr �!interprets_empty_strings_as_nullsZimplied_column_null�uniquery r Zsupports_tablespacesrS rz ) r8 �column_db_typerD r" r �field_db_params�include_defaultr� r� � default_valueZcolumn_defaultZ tablespacer r r �_iter_column_sql! sT �� ��� ���z)BaseDatabaseSchemaEditor._iter_column_sqlc C sD |j | jd�}|d }|du r"dS g }d�| �||||||��|fS )z� Return the column definition for a field. The field must already have had set_attributes_from_name() called. rZ �typeN)NNr\ )ri r2 rx r� )r8 r" r r� r� r� rD r r r rh Z s" �� �z#BaseDatabaseSchemaEditor.column_sqlc C s dS )z� Some backends don't accept default values for certain columns types (i.e. MySQL longtext and longblob). Fr �r8 r r r r r� u s z%BaseDatabaseSchemaEditor.skip_defaultc C s dS )z� Some backends don't accept default values for certain columns types (i.e. MySQL longtext and longblob) in the ALTER COLUMN statement. Fr r� r r r r� | s z.BaseDatabaseSchemaEditor.skip_default_on_alterc C s t d��dS )zU Only used for backends which have requires_literal_defaults feature z�subclasses of BaseDatabaseSchemaEditor for backends which have requires_literal_defaults must provide a prepare_default() methodN��NotImplementedError�r8 �valuer r r r� � s �z(BaseDatabaseSchemaEditor.prepare_defaultc C s dS )z� Return the SQL to use in a DEFAULT clause. The resulting string should contain a '%s' placeholder for a default value. �%sr r� r r r r� � s z,BaseDatabaseSchemaEditor._column_default_sqlc C s� | � � r| �� }n�| js<| jr<| jr<| �� dkr6d}q�d}nbt| dd�sTt| dd�r�| �� }|dkrnt�� }q�t �� }|dkr�|� � }q�|d kr�|� � }nd }|S ) NZ BinaryFieldrY rG Zauto_nowFZ auto_now_addZ DateTimeFieldZ DateFieldZ TimeField) � has_default� get_defaultr� �blankr� rt �getattrr �nowr �date�time)r �defaultZ internal_typer r r �_effective_default� s" z+BaseDatabaseSchemaEditor._effective_defaultc C s |� | �|�| j�S )z2Return a field's effective database default value.)Zget_db_prep_saver� r2 r� r r r r� � s z*BaseDatabaseSchemaEditor.effective_defaultc C s t � �dS )aX Return a quoted version of the value so it's safe to use in an SQL string. This is not safe against injection from user code; it is intended only for use in making SQL scripts or preparing default values for particularly tricky backends (defaults are not user-defined, though, so this is safe). Nr� r� r r r rP � s z$BaseDatabaseSchemaEditor.quote_valuec C s� | � |�\}}| �||pd� | jjjr�|jjrB| �|d|jj� | jjjs�|jj D ]8}|j rT|j | jd�}|d }| j| � ||||j �� qT| j �| �|�� |jjD ] }|jjjjr�| �|jj� q�dS )zr Create a table and any accompanying indexes or unique constraints for the given `model`. NrZ r� )r r> r2 r5 �supports_commentsr Zdb_table_comment�alter_db_table_commentr� rg r� ri �_alter_column_comment_sqlr: rl �_model_indexes_sql�local_many_to_manyr. �through� auto_created� create_model)r8 r" rC rD r r� � field_typer r r r� � s* � �� z%BaseDatabaseSchemaEditor.create_modelc C s~ |j jD ] }|jjj jr| �|jj� q| �| jd| �|j j �i � t | j �D ](}t |t �rP|�|j j �rP| j �|� qPdS )z!Delete a model from the database.rd N)r r� r. r� r� � delete_modelr> �sql_delete_tablerT rn �listr: � isinstancer Zreferences_table�remove)r8 r" r rC r r r r� � s ����z%BaseDatabaseSchemaEditor.delete_modelc C s. |j r| jjjsdS | j|�|| �dd� dS )zAdd an index on a model.N�rD )�contains_expressionsr2 r5 �supports_expression_indexesr> � create_sql�r8 r" �indexr r r � add_index� s ��z"BaseDatabaseSchemaEditor.add_indexc C s* |j r| jjjsdS | �|�|| �� dS )z Remove an index from a model.N)r� r2 r5 r� r> � remove_sqlr� r r r � remove_index� s ��z%BaseDatabaseSchemaEditor.remove_indexc C sD | j jjr(| j| �||j|j�d d� n| �||� | �||� d S )Nr� )r2 r5 Zcan_rename_indexr> �_rename_index_sqlr r� r� )r8 r" Z old_indexZ new_indexr r r � rename_index s � z%BaseDatabaseSchemaEditor.rename_indexc C s" |� || �}|r | j|dd� dS )z Add a constraint to a model.Nr� )r� r> �r8 r" ra rC r r r �add_constraint s z'BaseDatabaseSchemaEditor.add_constraintc C s |� || �}|r| �|� dS )z!Remove a constraint from a model.N)r� r> r� r r r �remove_constraint s z*BaseDatabaseSchemaEditor.remove_constraintc s| dd� |D �}dd� |D �}|� |�D ] }| �� |ddd�| j� q&|� |�D ](}� fdd�|D �}| �| �� |�� qNd S ) z� Deal with a model changing its unique_together. The input unique_togethers must be doubly-nested, not the single-nested ["foo", "bar"] format. c S s h | ] }t |��qS r �rN �r' r| r r r � <setcomp>& rY zABaseDatabaseSchemaEditor.alter_unique_together.<locals>.<setcomp>c S s h | ] }t |��qS r r� r� r r r r� ' rY TF)r� r c s g | ]}� j �|��qS r rU rW r! r r rX 2 rY zBBaseDatabaseSchemaEditor.alter_unique_together.<locals>.<listcomp>N)� difference�_delete_composed_index�sql_delete_uniquer> rf )r8 r" Zold_unique_togetherZnew_unique_together�olds�newsr| r{ r r! r �alter_unique_together s �z.BaseDatabaseSchemaEditor.alter_unique_togetherc s� dd� |D �}dd� |D �}|� |�D ] }| �� |ddd�| j� q&|� |�D ],}� fdd�|D �}| �| j� |d d �� qNd S ) z� Deal with a model changing its index_together. The input index_togethers must be doubly-nested, not the single-nested ["foo", "bar"] format. c S s h | ] }t |��qS r r� r� r r r r� ; rY z@BaseDatabaseSchemaEditor.alter_index_together.<locals>.<setcomp>c S s h | ] }t |��qS r r� r� r r r r� < rY TF)r� r� c s g | ]}� j �|��qS r rU rW r! r r rX G rY zABaseDatabaseSchemaEditor.alter_index_together.<locals>.<listcomp>�_idx�r| �suffixN)r� r� �sql_delete_indexr> �_create_index_sql)r8 r" Zold_index_togetherZnew_index_togetherr� r� r| r{ r r! r �alter_index_together5 s �z-BaseDatabaseSchemaEditor.alter_index_togetherc s� dd� � j jD �}dd� � j jD �}� fdd�|D �}| j� |fd||B i|��}|�d�du r�|r�| jjjr�t| j � j j |d d ��} | |v r�| g}t |�d kr�t d t |�� j j d � |�f ��| �| �|� |d �� d S )Nc S s h | ] }|j �qS r �r r` r r r r� K s zBBaseDatabaseSchemaEditor._delete_composed_index.<locals>.<setcomp>c S s h | ] }|j �qS r r� r` r r r r� N rY c s g | ]}� j �|�j�qS r )r rV rp rW r! r r rX O rY zCBaseDatabaseSchemaEditor._delete_composed_index.<locals>.<listcomp>�excluder� TF��quote� z1Found wrong number (%s) of constraints for %s(%s)rc r )r rv �indexes�_constraint_namesr� r2 r5 Z*allows_multiple_constraints_on_same_fieldsrH �_unique_constraint_namern �len� ValueErrorrx r> �_delete_constraint_sql) r8 r" r| Zconstraint_kwargsrC �meta_constraint_names�meta_index_names�columns�constraint_names� default_namer r! r r� J sD ���� ���� ���z/BaseDatabaseSchemaEditor._delete_composed_indexc C sn ||ks"| j jjr&|�� |�� kr&dS | �| j| �|�| �|�d� � | jD ]}t|t �rN|� ||� qNdS )z#Rename the table a model points to.N)Z old_tableZ new_table) r2 r5 Zignores_table_name_case�lowerr> �sql_rename_tablerT r: r� r Zrename_table_references)r8 r" Z old_db_tableZ new_db_tablerC r r r �alter_db_tablel s ����� z'BaseDatabaseSchemaEditor.alter_db_tablec C s> | j r:| jjjr:| �| j | �|jj�| �|p.d�d� � d S )NrG )rd �comment) �sql_alter_table_commentr2 r5 r� r> rT r rn rP )r8 r" Zold_db_table_commentZnew_db_table_commentr r r r� s ���z/BaseDatabaseSchemaEditor.alter_db_table_commentc C s2 | � | j| �|jj�| �|�| �|�d� � dS )z)Move a model's table between tablespaces.)rd Zold_tablespaceZnew_tablespaceN)r> �sql_retablespace_tablerT r rn )r8 r" Zold_db_tablespaceZnew_db_tablespacer r r �alter_db_tablespace� s ���z,BaseDatabaseSchemaEditor.alter_db_tablespacec C sJ |j r |jjjjr | �|jj�S | j||dd�\}}|du r@dS |j| jd� }r`|d|� �7 }|j | jd�}|d r�|d| j | 7 }|j�rT| jj j �rT|j �rTd}| j�r>|jjjj}|jjj�|jj�j} t|jj�\} } |d| j| �|||�| �r d| �| � nd | �|j�| �|�| �| �| jj�� d � 7 }n| j�| �|||�� | j | �|jj�| �|j�|d � } | � | |� | � |��s�| � |�du�r�| j |d|dd �\} }| j!| �|jj�| d � } | � | |� |j"�r | jj j#�r | jj j$�s |d }| j | �%||||j"�� | j�&| �'||�� | jj j(�rF| j�)� dS )z� Create a field on a model. Usually involves adding a column, but may involve adding a table instead (for M2M fields). T)r� NrZ r\ r[ r_ z%s.rG )r � namespacerp r] r^ � deferrable)rd rp re ��drop�rd �changesr� )*r r. r� r r� r� rh rk r2 ri rj r5 rr rm �sql_create_column_inline_fkr" rn rV ro rp r �_fk_constraint_namerT rS �deferrable_sqlr: rM rs �sql_create_columnr> r� r� �_alter_column_default_sql�sql_alter_columnr� r� r� r� rl �_field_indexes_sql� connection_persists_old_columns�close)r8 r" r re rD r~ r} Zconstraint_suffixr] r^ r� �_rC � changes_sqlr� r r r � add_field� s� ��� � � � � � � �� � ��� �� z"BaseDatabaseSchemaEditor.add_fieldc C s� |j r |jjjjr | �|jj�S |j| jd�d du r:dS |jrp| j||j gdd�}|D ]}| � | � ||�� qX| j | � |jj�| � |j �d� }| � |� | jjjr�| j�� t| j�D ],}t|t�r�|�|jj|j �r�| j�|� q�dS )z� Remove a field from a model. Usually involves deleting a column, but for M2Ms may involve deleting a table. rZ r� NT�� foreign_key)rd rp )r r. r� r r� r� ri r2 r� rp r> �_delete_fk_sql�sql_delete_columnrT rn r5 r� r� r� r: r� r Zreferences_columnr� )r8 r" r �fk_names�fk_namerC r r r � remove_field� s( � �z%BaseDatabaseSchemaEditor.remove_fieldc C s6 | � ||�sdS |j| jd�}|d }|j| jd�}|d }|du rN|jdu s`|du rr|jdu rrtd||f ��n�|du r�|du r�|jjr�|jjr�|jjjjr�|jjjjr�| �||||�S |du r�|du r�|jjr�|jjr�|jjjjs�|jjjjs�dS |du �s |du �rtd||f ��| � ||||||||� dS )a' Allow a field's type, uniqueness, nullability, default, column, constraints, etc. to be modified. `old_field` is required to compute the necessary changes. If `strict` is True, raise errors if the old column does not match `old_field` precisely. NrZ r� zqCannot alter field %s into %s - they do not properly define db_type (are you using a badly-written custom field?)z�Cannot alter field %s into %s - they are not compatible types (you cannot alter to or from M2M fields, or add or remove through= on M2M fields)) �_field_should_be_alteredri r2 r. r� r� r r� �_alter_many_to_many� _alter_field) r8 r" r) r+ �strict� old_db_params�old_type� new_db_params�new_typer r r � alter_field st �������� � � ���� � � ���z$BaseDatabaseSchemaEditor.alter_fieldc / C s� t � } | jjjr�|jr�|jr�| j||dhd�r�| j||jgdd�} |rnt | �dkrnt dt | �|j j |jf ��| D ]$} | � |jf� | �| �|| �� qr|j�r,|jr�| �||��r,dd� |j jD �} | j||jgdd | d �} |�rt | �dk�rt d t | �|j j |jf ��| D ]}| �| �||�� �q|�d �}|�d �}| jjj�o||j�r\|j�sl|j�o||j�o|||k�p|||k}|�r�t||�D ]B\}}| j|j|jjgdd�}|D ]} | �| �|j| �� �q��q�|j�r<|j�s<|j�r�|j�r<d d� |j jD �}| j||jgdtj |d�}|D ]}| �| � ||�� �q"|d |d k�r�|d �r�dd� |j jD �} | j||jgd| d�} |�r�t | �dk�r�t dt | �|j j |jf ��| D ]}| �| � ||�� �q�|j|jk�r&| �| � |j j |||�� | j D ](}t!|t"��r�|�#|j j |j|j� �q�g }g }g }|j$| jd�} |j$| jd�} ||k�s�| | k�s�||k�s�| jjj%�r�|j&|j&k�r�| �'||||||�\} } |�(| � |�)| � d } |j*�r|j*�s| �+|�}!| �+|�}"| �,|��s|!|"k�r|"du�rd} |�(| �-|||�� |j*|j*k�r>| �.|||�} | �r>|�(| � |�/� �oV|j*�oV|j* }#|�sd|�rB|#�sr||7 }| jjj0�r�|�r�t1t2|� �\}}$d�3|�t4|$g �fg}|D ],\}}$| �| j5| �6|j j �|d� |$� �q�|#�rB| �| j7| �6|j j �| �6|j�dd� |"g� |D ],\}}$| �| j5| �6|j j �|d� |$� �q|�rd|D ]\}}$| �||$� �qL|j�r�|j�s�| �8||� | �9||��r�| �| �:||g�� |j�r�|j�r�|j�r�|j�s�| �| j;||gd�� g }%|�r�|%�)t||�� | �||��r"| �| �<||�� |%�)t||�� |%D ]�\}&}|jj=| jd�}'|'d }(|'�d �})|&jj=| jd�}*|*�d �}+| �'|j|&j|j|(|+|)�\} } | �| j5| �6|jj j �| d d� | d � | D ]\}}$| �||$� �q��q&| jjj�r|j�r| �s�|j�r�|j�s|j�r| �| �>||d �� |�rR|%D ],\},}-|-jj�r$| �| �>|-j|-jd �� �q$|d |d k�r�|d �r�| j?|j j |jgd d �}| �| �@|||d �� | �r�| j-|||dd �\}.}$| j5| �6|j j �|.d� }| �||$� | jjjA�r�| j�B� dS )!z3Perform a "physical" (non-ManyToMany) field update.r� )�ignoreTr� r� z<Found wrong number (%s) of foreign key constraints for %s.%sc S s h | ] }|j �qS r r� r` r r r r� v s z8BaseDatabaseSchemaEditor._alter_field.<locals>.<setcomp>F)r� r r� z7Found wrong number (%s) of unique constraints for %s.%sr� c S s h | ] }|j �qS r r� )r' r� r r r r� � rY )r� �type_r� r[ c S s h | ] }|j �qS r r� r` r r r r� � s )r[ r� z6Found wrong number (%s) of check constraints for %s.%srZ Nrc r� r� )rd rp r� �r| r� r r_ Z_fk�_check�r� r� )C�setr2 r5 rr r. rm r r� rp r� r� r rn �addr> r� r� �_field_became_primary_keyrv �_delete_unique_sqlr� r r- Z related_modelr �db_indexr� r r� �_delete_index_sql�_delete_check_sql�_rename_field_sqlr: r� r Zrename_column_referencesrk r� r� �_alter_column_type_sqlrM rl r� r� r� r� �_alter_column_null_sqlr� Zsupports_combined_altersrN r, rx �sumr� rT �sql_update_with_default�_delete_primary_key�_unique_should_be_addedrf r� �_create_primary_key_sqlri rs �_create_index_name�_create_check_sqlr� r� )/r8 r" r) r+ r r r r r Z fks_droppedr� r� r� r� �constraint_name� old_collation� new_collationZdrop_foreign_keysZ_old_relr0 Z rel_fk_namesr� Z index_names� index_namerC �actionsZ null_actionsZ post_actionsZold_type_suffixZnew_type_suffix�fragment� other_actionsZneeds_database_defaultZ old_default� new_defaultZfour_way_default_alterationrD Zrels_to_updater/ Z rel_db_paramsZrel_typeZ rel_collationZold_rel_db_paramsZold_rel_collationr� �relr� r r r r J s. ����� ����� ������ ��� ����� ����� �� ����� � � ��� � � ��� ��� ��� ���� ���� ������ � � ��� � z%BaseDatabaseSchemaEditor._alter_fieldc C sV | j jjr|jrdS |j| j d�}|jr.| jn| j}|| �|j �|d d� g fS dS )z� Hook to specialize column null alteration. Return a (sql, params) fragment to set a column to null or non-null as required by new_field, or None if no changes are required. NrZ r� )rp r� ) r2 r5 r� r� ri r� �sql_alter_column_null�sql_alter_column_not_nullrT rp )r8 r" r) r+ r rC r r r r � s" ���� ���z/BaseDatabaseSchemaEditor._alter_column_null_sqlc C s� | � |�}| �|�}|g}|r$g }n| jjjr<| �|�}g }|j| jd�}|rd|jr\| j} qj| j } n| j } | | � |j �|d |d� |fS )z� Hook to specialize column default alteration. Return a (sql, params) fragment to add or drop (depending on the drop argument) a default to new_field's column. rZ r� )rp r� r� ) r� r� r2 r5 r� r� ri r� � sql_alter_column_no_default_null�sql_alter_column_no_default�sql_alter_column_defaultrT rp ) r8 r" r) r+ r� r' r� rD r rC r r r r� � s, ���z2BaseDatabaseSchemaEditor._alter_column_default_sqlc C s� g }| � |||jj� }r&d|� �}nd}d} | jjjr�|js�|j|jkrr| �||||j�\} } | rr|� | | f� |jr�| � |j�} | j | � |j �||| d� g f|fS )a� Hook to specialize column type alteration for different backends, for cases when a creation type is different to an alteration type (e.g. SERIAL in PostgreSQL, PostGIS fields). Return a two-tuple of: an SQL fragment of (sql, params) to insert into an ALTER TABLE statement and a list of extra (sql, params) tuples to run once the field is altered. r\ rG )rp r� r� r� )r� r rn r2 r5 r� r r� r� rM r� �sql_alter_column_typerT rp ) r8 r" r) r+ r r! r" r& Z collate_sqlZ comment_sqlrC rD r r r r � s6 � � ��� �z/BaseDatabaseSchemaEditor._alter_column_type_sqlc C s. | j | �|jj�| �|j�| �|�d� g fS )N)rd rp r� )�sql_alter_column_commentrT r rn rp r� )r8 r" r+ r Znew_db_commentr r r r� s ���z2BaseDatabaseSchemaEditor._alter_column_comment_sqlc C s | � |p d�S )NrG )rP )r8 r� r r r r� s z%BaseDatabaseSchemaEditor._comment_sqlc C s� |j jjj|j jjjkr:| �|j j|j jjj|j jjj� | �|j j|j jj�|�� �|j jj�|�� �� | �|j j|j jj�|�� �|j jj�|�� �� dS )z*Alter M2Ms to repoint their to= endpoints.N) r. r� r rn r� r rV Zm2m_reverse_field_nameZm2m_field_name)r8 r" r) r+ r r r r r # s. �� � � �� �z,BaseDatabaseSchemaEditor._alter_many_to_manyrG c C s� t |�\}}dt|g|�R ddi�|f }| jj�� p8d}d|d�|�|f }t|�|kr^|S t|�|d kr~|d|d � }|t|� d d }d|d|� d�|�d|� |f }|d dks�|d �� r�d |dd � }|S )z� Generate a unique name for an index/unique constraint. The name is divided into 3 parts: the table name, the column names, and a unique digest and suffix. z%s%s�length� �� z%s_%s_%sr� � N� r� r zD%s���)r r r2 rS �max_name_lengthrx r� �isdigit) r8 � table_name� column_namesr� r� Zhash_suffix_part� max_lengthr# Z other_lengthr r r r C s( � �z+BaseDatabaseSchemaEditor._create_index_namec C sf |d u rHt |�dkr*|d jr*|d j}n tjr8tj}n|jjrH|jj}|d urbd| jj�|� S dS )Nr� r r\ rG )r� ry r ZDEFAULT_INDEX_TABLESPACEr r2 rS rz )r8 r" r| ry r r r �_get_index_tablespace_sqlc s z2BaseDatabaseSchemaEditor._get_index_tablespace_sqlc C s |r d| S dS )Nz WHERE rG r )r8 � conditionr r r �_index_condition_sqlo s z-BaseDatabaseSchemaEditor._index_condition_sqlc C s, |r| j jjsdS tdt|jj|| j�d�S )NrG z INCLUDE (%(columns)s))r� )r2 r5 �supports_covering_indexesr r r rn rT )r8 r" r� r r r �_index_include_sqlt s �z+BaseDatabaseSchemaEditor._index_include_sql) r| r r� �usingry � col_suffixesrC � opclassesr< �include� expressionsc s� |pg }| pg } t |dd�j�jd�} �j|||d�}dd� |D �}|pL�j}|jj}� �fdd�}t|t|�j �t ||||�||r��� |||| �nt || | �j �|��| ���|| �d �S ) z� Return the SQL statement to create the index for one or several fields or expressions. `sql` can be specified if the syntax differs from the standard (GIS indexes, ...). F�Z alias_colsrZ )ry c S s g | ] }|j �qS r �rp rW r r r rX � rY z>BaseDatabaseSchemaEditor._create_index_sql.<locals>.<listcomp>c s" � d u r�j | i |��� ��� �S r$ )r rT ��args�kwargs�r r8 r r �create_index_name� s zEBaseDatabaseSchemaEditor._create_index_sql.<locals>.create_index_name)rd r r@ r� rE r< rC )r � get_compilerr2 r; �sql_create_indexr rn r r rT r �_index_columnsr rP r= r? )r8 r" r| r r� r@ ry rA rC rB r< rC rD �compilerrz r� rM rd rK r rJ r r� | s0 �� � �z*BaseDatabaseSchemaEditor._create_index_sqlc C s& t |p | jt|jj| j�| �|�d�S �N)rd r )r r� r r rn rT )r8 r" r rC r r r r � s �z*BaseDatabaseSchemaEditor._delete_index_sqlc C s* t | jt|jj| j�| �|�| �|�d�S )N)rd �old_name�new_name)r �sql_rename_indexr r rn rT )r8 r" rQ rR r r r r� � s �z*BaseDatabaseSchemaEditor._rename_index_sqlc C s t ||| j|d�S )N)rA )r rT )r8 rd r� rA rB r r r rN � s z'BaseDatabaseSchemaEditor._index_columnsc s� � j jr� j js� j jr g S g }� j jD ]}|�| �� |�� q(� j jD ],}� fdd�|D �}|�| j � |dd�� qH� j j D ]&}|j r�| j j jr~|�|�� | �� q~|S )z� Return a list of all index SQL statements (field indexes, index_together, Meta.indexes) for the specified model. c s g | ]}� j �|��qS r rU rW r! r r rX � rY z?BaseDatabaseSchemaEditor._model_indexes_sql.<locals>.<listcomp>r� r� )r Zmanaged�proxyZswappedrg rl r� Zindex_togetherrM r� r� r� r2 r5 r� r� )r8 r" �outputr r{ r| r� r r! r r� � s ��z+BaseDatabaseSchemaEditor._model_indexes_sqlc C s* g }| � ||�r&|�| j||gd�� |S )zT Return a list of all index SQL statements for the specified field. r )�_field_should_be_indexedrM r� )r8 r" r rU r r r r� � s z+BaseDatabaseSchemaEditor._field_indexes_sqlc C s� |pt � }|�� \}}}}|�� \}}} } |�|j�D ]} |�| d � q6|�|j�D ]} | �| d � qT| �|j�| �|j�kp�|||f|| | fkS r$ )r Z deconstruct�unionZ non_db_attrs�poprT rp ) r8 r) r+ r r� �old_pathZold_argsZ old_kwargs�new_path�new_argsZ new_kwargs�attrr r r r � s ��z1BaseDatabaseSchemaEditor._field_should_be_alteredc C s |j o |j S r$ )r r� �r8 r" r r r r rV � s z1BaseDatabaseSchemaEditor._field_should_be_indexedc C s |j o |j S r$ �r �r8 r) r+ r r r r � s z2BaseDatabaseSchemaEditor._field_became_primary_keyc C s |j o|jo|j p|j S r$ )r r� r_ r r r r � s � �z0BaseDatabaseSchemaEditor._unique_should_be_addedc C s* | j | �|�| �|j�| �|j�|d� S )N)rd Z old_columnZ new_columnr� )�sql_rename_columnrT rp )r8 rd r) r+ r r r r r s �z*BaseDatabaseSchemaEditor._rename_field_sqlc C s� t |jj| j�}| �|||�}t|jj|jg| j�}t |jjjj| j�}t|jjjj|jjg| j�}| j j � � } t | j |||||| d�S )N)rd r rp r] r^ r� )r r rn rT r� r rp � target_fieldr" r2 rS r� r � sql_create_fk) r8 r" r r� rd r rp r] r^ r� r r r rs s& � �z'BaseDatabaseSchemaEditor._create_fk_sqlc s< � fdd�}t |jj|jgt|jjjj�d |jjg||�S )Nc s � � � j| i |���S r$ �rT r rG r= r r �create_fk_name s zDBaseDatabaseSchemaEditor._fk_constraint_name.<locals>.create_fk_namer� )r r rn rp r ra r" )r8 r" r r� rd r r= r r� s �z,BaseDatabaseSchemaEditor._fk_constraint_namec C s | � | j||�S r$ )r� � sql_delete_fk�r8 r" r r r r r� , s z'BaseDatabaseSchemaEditor._delete_fk_sqlc C s, |d u r dS |t jkrdS |t jkr(dS d S )NrG z DEFERRABLE INITIALLY DEFERREDz DEFERRABLE INITIALLY IMMEDIATE)r ZDEFERREDZ IMMEDIATE)r8 r� r r r �_deferrable_constraint_sql/ s z3BaseDatabaseSchemaEditor._deferrable_constraint_sqlc s� |r� j jjsd S |s"|s"|s"|rN� j|||||||d�} | rJ� j�| � d S � jd�� fdd�|D ��� �|�d� } � j � � |�| d� S )N)r r< rC rB rD rc c s g | ]}� � |j��qS r )rT rp rW r= r r rX W rY z8BaseDatabaseSchemaEditor._unique_sql.<locals>.<listcomp>)r� r� �r ra ) r2 r5 �&supports_deferrable_unique_constraintsrf r: rM �sql_unique_constraintrx rg �sql_constraintrT ) r8 r" r| r r< r� rC rB rD rC ra r r= r � _unique_sql7 s4 ��� ��z$BaseDatabaseSchemaEditor._unique_sqlc C s |r| j jjr8|r | j jjr8|r*| j jjr8|r<| j jjs<d S t|dd�j| j d�} |jj } dd� |D �} |d u r�| j | | dd�}n | � |�}|s�|s�|s�|r�| j } n| j } | r�| j| | d|d �} nt| || | j�} t| t| | j �|| | �|�| �|�| �||�d �S ) NFrE rZ c S s g | ] }|j �qS r rF rW r r r rX { rY z?BaseDatabaseSchemaEditor._create_unique_sql.<locals>.<listcomp>Tr� r )rA rB )rd r r� r< r� rC )r2 r5 ri �supports_partial_indexesr> r� r rL r rn r� rT �sql_create_unique_index�sql_create_uniquerN r rP r r r= rg r? ) r8 r" r| r r< r� rC rB rD rO rd r� rC r r r rf _ sT �������� � � �z+BaseDatabaseSchemaEditor._create_unique_sqlc s& |r� fdd�}n� j }t||d|�S )Nc s � � � j| i |���S r$ rc rG r= r r �create_unique_name� s zLBaseDatabaseSchemaEditor._unique_constraint_name.<locals>.create_unique_name�_uniq)r r )r8 rd r� r� rp r r= r r� � s z0BaseDatabaseSchemaEditor._unique_constraint_namec C sh |r| j jjr8|r | j jjr8|r*| j jjr8|r<| j jjs<d S |sL|sL|sL|rT| j}n| j}| �|||�S r$ ) r2 r5 ri rm r> r� r� r� r� ) r8 r" r r< r� rC rB rD rC r r r r � s* �������� z+BaseDatabaseSchemaEditor._delete_unique_sqlc C s | j | �|�| jd|i d� S )Nr[ rh )rk rT rj )r8 r r[ r r r � _check_sql� s �z#BaseDatabaseSchemaEditor._check_sqlc C s$ t | jt|jj| j�| �|�|d�S )N)rd r r[ )r �sql_create_checkr r rn rT )r8 r" r r[ r r r r � s �z*BaseDatabaseSchemaEditor._create_check_sqlc C s | j jjsd S | �| j||�S r$ )r2 r5 Z supports_table_check_constraintsr� �sql_delete_checkrf r r r r � s z*BaseDatabaseSchemaEditor._delete_check_sqlc C s t |t|jj| j�| �|�d�S rP )r r r rn rT )r8 �templater" r r r r r� � s �z/BaseDatabaseSchemaEditor._delete_constraint_sqlc s" |dur� fdd�|D �}� j �� �$} � j j�| |jj�} W d � n1 sN0 Y g } | �� D ]�\} }|du s�||d krd|dur�|d |kr�qd|dur�|d |kr�qd|dur�|d |kr�qd|dur�|d |kr�qd|dur�|d s�qd|du�r|d |k�rqd| �r| | vrd| �| � qd| S ) z@Return all constraint names matching the columns and conditions.Nc s@ g | ]8}� j jjr.� j j�t|� j j�� ��n � j j�|��qS r )r2 r5 Ztruncates_names� introspectionZidentifier_converterr rS r6 )r' r r= r r rX � s �� �z>BaseDatabaseSchemaEditor._constraint_names.<locals>.<listcomp>r� r� r r� r[ r� r� )r2 rQ rv Zget_constraintsr rn �itemsrM )r8 r" r9 r� r r� r� r[ r r� rQ rv �resultr Zinfodictr r= r r� � s4 � �" z*BaseDatabaseSchemaEditor._constraint_namesc C sV | j |dd�}|r6t|�dkr6tdt|�|jjf ��|D ]}| �| �||�� q:d S )NTr^ r� z0Found wrong number (%s) of PK constraints for %s)r� r� r� r rn r> �_delete_primary_key_sql)r8 r" r r� r r r r r s ���z,BaseDatabaseSchemaEditor._delete_primary_keyc C sJ t | jt|jj| j�| �| j|jj|jgdd��t|jj|jg| j�d�S )NZ_pkr )rd r r� ) r � sql_create_pkr r rn rT r rp r r] r r r r s ���z0BaseDatabaseSchemaEditor._create_primary_key_sqlc C s | � | j||�S r$ )r� � sql_delete_pkrf r r r ry s z0BaseDatabaseSchemaEditor._delete_primary_key_sqlc C s |rd| � |� S dS )NzCOLLATE rG )rT )r8 r� r! r8 r r r r� s z%BaseDatabaseSchemaEditor._collate_sqlc C s* | j | �|�d�|�d� }| �|� d S )N�,)Z procedure� param_types)�sql_delete_procedurerT rx r> )r8 Zprocedure_namer} rC r r r �remove_procedure! s �z)BaseDatabaseSchemaEditor.remove_procedure)FT)r )F)F)F)F)rG )N)N)N)NNNNN)NNNNNN)T)NNNNN)NNNNNNNN)F)NN)r )p�__name__� __module__� __qualname__�__doc__rw r� r� r� r� r� r. r) r* r- r, r+ r� r` r rj rj Zsql_delete_constraintrk rs rt ro r� rb rq r� re rM rn rS r� rz r{ r~ r� r/ r9 r< r? r> rT r r� rh r� r� r� r� � staticmethodr� r� rP r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� r r� r� r r r; r= r? r� r r� rN r� r� r rV r r r rs r� r� rg rl rf r� r rr r r r� r� r r ry r� r r r r r r1 K s4 ������� S9 $ " V I � \ &. � 6 � , � 5 � � - r1 ) �loggingr r Z django.confr Z!django.db.backends.ddl_referencesr r r r r r Zdjango.db.backends.utilsr r r Zdjango.db.modelsr r Zdjango.db.models.sqlr Zdjango.db.transactionr r Z django.utilsr � getLoggerrI r r# r- r1 r r r r �<module> s