File: //lib/python3/dist-packages/django/db/backends/base/__pycache__/schema.cpython-310.pyc
o
3�a�� � @ s� d dl Z d dlmZ d dlmZmZmZmZmZmZ d dl m
Z
m Z d dl m
Z
mZ d dlmZ d dlmZmZ d dlmZ e �d �Zd
d � Zd d
� Zdd� ZG dd� d�ZdS )� N)�datetime)�Columns� Expressions�ForeignKeyName� IndexName� Statement�Table)� names_digest�split_identifier)�
Deferrable�Index)�Query)�TransactionManagementError�atomic)�timezonezdjango.db.backends.schemac C s2 | j }|jrdS |jr|jdgkrdS |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_key� to_fields�name)�relation�
altered_fieldr � r �@/usr/lib/python3/dist-packages/django/db/backends/base/schema.py�_is_relevant_relation s r c C s | j jdddd�S )NFT)�forward�reverse�include_hidden)�_meta� _get_fields��modelr r r �_all_related_fields � r" c s2 t �fdd�t�j�D �� fdd�t� j�D ��S )Nc 3 � � | ]
}t |� �r|V qd S �N�r ��.0�obj)� old_fieldr r � <genexpr>( � � z+_related_non_m2m_objects.<locals>.<genexpr>c 3 r$ r% r& r' )� new_fieldr r r+ ) r, )�zipr" r! )r* r- r )r- r* r �_related_non_m2m_objects$ s �r/ c @ s� e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d ZeZd
ZdZdZdZdZdZdZdZdZeZdZeZdZ dZ dZ eZ dZ dZ!dZ"d Z#eZ$d Z%d�d d!�Z&d"d#� Z'd$d%� Z(d�d'd(�Z)d)d*� Z*d+d,� Z+d�d-d.�Z,d/d0� Z-d1d2� Z.d3d4� Z/d5d6� Z0e1d7d8� �Z2d9d:� Z3d;d<� Z4d=d>� Z5d?d@� Z6dAdB� Z7dCdD� Z8dEdF� Z9dGdH� Z:dIdJ� Z;dKdL� Z<dMdN� Z=dOdP� Z>dQdR� Z?dSdT� Z@dUdV� ZAd�dWdX�ZB d�dYdZ�ZCd[d\� ZDd�d]d^�ZEd_d`� ZFdadb� ZGdcdd� ZHd�dfdg�ZId�dhdi�ZJdjdk� ZKdldm� ZLdddededd&dd&ddddn� dodp�ZMd�dqdr�ZNdsdt� ZOdudv� ZPdwdx� ZQdydz� ZRd{d|� ZSd}d~� ZTdd�� ZUd�d�� ZVd�d�� ZWd�d�� ZXd�d�� ZYd�d�� ZZ d�d�d��Z[ d�d�d��Z\ d�d�d��Z]d�d�� Z^d�d�� Z_d�d�� Z`d�d�� Za d�d�d��Zbd�d�d��Zcd�d�� Zdd�d�� Zed�d�� Zfd�d�d��ZgdS )��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)sz%ALTER COLUMN %(column)s TYPE %(type)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 DEFAULTz2ALTER COLUMN %(column)s TYPE %(type)s%(collation)sz4ALTER 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)szDROP INDEX %(name)szGALTER TABLE %(table)s ADD CONSTRAINT %(name)s PRIMARY KEY (%(columns)s)z DROP PROCEDURE %(procedure)sFTc C s, || _ || _| jr g | _| j jjo|| _d S r% )�
connection� collect_sql�
collected_sql�features�can_rollback_ddl�atomic_migration)�selfr1 r2 r r r r �__init__c s
z!BaseDatabaseSchemaEditor.__init__c C s( g | _ | jrt| jj�| _| j�� | S r% )� deferred_sqlr6 r r1 �alias� __enter__�r7 r r r r; l s
z"BaseDatabaseSchemaEditor.__enter__c C s< |d u r| j D ]}| �|� q| jr | j�|||� d S d S r% )r9 �executer6 r �__exit__)r7 �exc_type� exc_value� traceback�sqlr r r r> s s
�z!BaseDatabaseSchemaEditor.__exit__r c C s� | j s| jjr| jjjstd��t|�}tjd||||d�d� | j rL|� d�r*dnd}|durB| j
� |t t
| j|�� | � dS | j
� || � dS | j�� �}|�||� W d � dS 1 scw 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))�paramsrB )�extra�;� N)r2 r1 �in_atomic_blockr4 r5 r �str�logger�debug�endswithr3 �append�tuple�map� quote_value�cursorr= )r7 rB rC �endingrP r r r r= | s �$ "�z BaseDatabaseSchemaEditor.executec C s | j j�|�S r% )r1 �ops�
quote_name)r7 r r r r rS � � z#BaseDatabaseSchemaEditor.quote_namec s� � j jD ]}� fdd�|D �}�j���� |�� qg }g }� j jD ]�}��� |�\}}|du r1q"|j�jd�} | d rE|d�j | 7 }|j
�jd�}
|
rT|d|
7 }|� |� |j r�|j
r�|j jj j} |j jj �|j j�j} �jr�|d�j��| ���| �d� 7 }n�jjjr��j���� |d �� |�d
��|j�|f � |�� d v r��jj�� j j|j�}
|
r��j� |
� q"� �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 � g | ] }� j �|�j�qS r �r � get_field�column�r( r r r r �
<listcomp>� � z6BaseDatabaseSchemaEditor.table_sql.<locals>.<listcomp>N�r1 �check� � %s)�to_table� to_column� _fk_%(to_table)s_%(to_column)sz%s %s)� AutoField� BigAutoField�SmallAutoFieldc s g | ]}|� � ���qS r )�constraint_sql�r( �
constraint�r! r7 r r rZ � � �, c s s � | ]}|r|V qd S r% r rg r r r r+ � s � z5BaseDatabaseSchemaEditor.table_sql.<locals>.<genexpr>)�table�
definition) r �unique_togetherr9 rL �_create_unique_sql� local_fields�
column_sql�
db_parametersr1 �sql_check_constraint�db_type_suffix�extend� remote_field�
db_constraintr! �db_tablerW �
field_namerX �sql_create_inline_fkrS r4 �supports_foreign_keys�_create_fk_sql�get_internal_typerR � autoinc_sql� constraints�sql_create_table�join�
db_tablespace�tablespace_sql)r7 r! �fields�columns� column_sqlsrC r rm � extra_params� db_params�col_type_suffixr` ra r~ r rB r� r ri r � table_sql� sZ
�
� � � z"BaseDatabaseSchemaEditor.table_sqlc C s^ |j | jd�}|d }g }|du rdS t|dd�}|r"|| �|�7 }|j}|o4| �|� o4|o3| �|� }|r_| �|�} d| �|� }
| dur_| jj j
rV||
| � | � 7 }n ||
7 }|| g7 }|j rl|j
sl| jj jrld}|rx| jj jsx|d7 }n|s~|d 7 }|j
r�|d
7 }n|jr�|d 7 }|jp�|jj} | r�| jj jr�|jr�|d | jjj| dd
� 7 }||fS )z�
Take a field and return its column definition.
The field must already have had set_attributes_from_name() called.
r\ �typeN)NN� db_collationz DEFAULT Tz NULLz NOT NULLz PRIMARY KEYz UNIQUEr_ )�inline)rr r1 �getattr� _collate_sql�null� skip_default�skip_default_on_alter�effective_default�_column_default_sqlr4 �requires_literal_defaults�prepare_default�empty_strings_allowedr �!interprets_empty_strings_as_nulls�implied_column_null�uniquer� r �supports_tablespacesrR r� ) r7 r! r �include_defaultr� rB rC � collationr� �
default_value�column_default�
tablespacer r r rq � sL
��
�
z#BaseDatabaseSchemaEditor.column_sqlc C � dS )z�
Some backends don't accept default values for certain columns types
(i.e. MySQL longtext and longblob).
Fr �r7 r r r r r�
� z%BaseDatabaseSchemaEditor.skip_defaultc C r� )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� r� z.BaseDatabaseSchemaEditor.skip_default_on_alterc C s t d��)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() method��NotImplementedError�r7 �valuer r r r� s �z(BaseDatabaseSchemaEditor.prepare_defaultc C r� )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� $ r� z,BaseDatabaseSchemaEditor._column_default_sqlc C s� | � � r
| �� }|S | js!| jr!| jr!| �� dkr d}|S d}|S t| dd�s-t| dd�rSt�� }| �� }|dkr?|� � }|S |dkrI|�
� }|S |d krQt �� }|S d }|S )
N� BinaryField� rF �auto_nowF� auto_now_add� DateField� TimeField�
DateTimeField) � has_default� get_defaultr� �blankr� r} r� r �now�date�timer )r �default�
internal_typer r r �_effective_default+ s. � � ����z+BaseDatabaseSchemaEditor._effective_defaultc C s |� | �|�| j�S )z2Return a field's effective database default value.)�get_db_prep_saver� r1 r� r r r r� B s z*BaseDatabaseSchemaEditor.effective_defaultc C s t � �)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).
r� r� r r r rO F s z$BaseDatabaseSchemaEditor.quote_valuec C s\ | � |�\}}| �||p
d� | j�| �|�� |jjD ]}|jjjj r+| �
|jj� q dS )zr
Create a table and any accompanying indexes or unique constraints for
the given `model`.
N) r� r= r9 ru �_model_indexes_sqlr �local_many_to_manyrv �through� auto_created� create_model)r7 r! rB rC r r r 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
�r<|�|j j �r<| j �|� q(dS )z!Delete a model from the database.rl N)r r� rv r� r� � delete_modelr= �sql_delete_tablerS rx �listr9 �
isinstancer �references_table�remove)r7 r! r rB r r r r� c s �� ��z%BaseDatabaseSchemaEditor.delete_modelc C s. |j r
| jjjs
dS | j|�|| �dd� dS )zAdd an index on a model.N�rC )�contains_expressionsr1 r4 �supports_expression_indexesr= �
create_sql�r7 r! �indexr r r � add_indexs s ��z"BaseDatabaseSchemaEditor.add_indexc C s* |j r
| jjjs
dS | �|�|| �� dS )z Remove an index from a model.N)r� r1 r4 r� r= �
remove_sqlr� r r r � remove_index~ s ��z%BaseDatabaseSchemaEditor.remove_indexc C s& |� || �}|r| j|dd� dS dS )z Add a constraint to a model.Nr� )r� r= �r7 r! rh rB r r r �add_constraint� s �z'BaseDatabaseSchemaEditor.add_constraintc C s" |� || �}|r| �|� dS dS )z!Remove a constraint from a model.N)r� r= r� r r r �remove_constraint� s �z*BaseDatabaseSchemaEditor.remove_constraintc sz dd� |D �}dd� |D �}|� |�D ]
}| �� |ddi| j� q|� |�D ]}� fdd�|D �}| �| �� |�� q&dS ) 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 � h | ]}t |��qS r �rM �r( r� r r r � <setcomp>� � zABaseDatabaseSchemaEditor.alter_unique_together.<locals>.<setcomp>c S r� r r� r� r r r r� � r� r� Tc rU r rV rY r r r rZ � r[ zBBaseDatabaseSchemaEditor.alter_unique_together.<locals>.<listcomp>N)�
difference�_delete_composed_index�sql_delete_uniquer= ro )r7 r! �old_unique_together�new_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
�� q'd 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 r� r r� r� r r r r� � r� z@BaseDatabaseSchemaEditor.alter_index_together.<locals>.<setcomp>c S r� r r� r� r r r r� � r� TF)r� r� c � g | ]}� j �|��qS r �r rW rY r r r rZ � rj zABaseDatabaseSchemaEditor.alter_index_together.<locals>.<listcomp>�_idx�r� �suffixN)r� r� �sql_delete_indexr= �_create_index_sql)r7 r! �old_index_together�new_index_togetherr� r� r� � field_namesr r r �alter_index_together� s ��z-BaseDatabaseSchemaEditor.alter_index_togetherc s� dd� � j jD �}dd� � j jD �}� fdd�|D �}| j� |fd||B i|��}t|�dkr?tdt|�� j jd �|�f ��| �| � |� |d
�� d S ) Nc S � h | ]}|j �qS r �r rg r r r r� � � zBBaseDatabaseSchemaEditor._delete_composed_index.<locals>.<setcomp>c S r� r r� rg r r r r� � r� c rU r rV rY r r r rZ � r[ zCBaseDatabaseSchemaEditor._delete_composed_index.<locals>.<listcomp>�exclude� z1Found wrong number (%s) of constraints for %s(%s)rk r )
r r �indexes�_constraint_names�len�
ValueErrorrx r� r= �_delete_constraint_sql) r7 r! r� �constraint_kwargsrB �meta_constraint_names�meta_index_namesr� �constraint_namesr r r r� � s" ��� � z/BaseDatabaseSchemaEditor._delete_composed_indexc C sn ||ks| j jjr|�� |�� krdS | �| j| �|�| �|�d� � | jD ]
}t|t �r4|�
||� q'dS )z#Rename the table a model points to.N)� old_table� new_table) r1 r4 �ignores_table_name_case�lowerr= �sql_rename_tablerS r9 r� r �rename_table_references)r7 r! � old_db_table� new_db_tablerB r r r �alter_db_table� s �
�
��z'BaseDatabaseSchemaEditor.alter_db_tablec C s2 | � | j| �|jj�| �|�| �|�d� � dS )z)Move a model's table between tablespaces.)rl �old_tablespace�new_tablespaceN)r= �sql_retablespace_tablerS r rx )r7 r! �old_db_tablespace�new_db_tablespacer r r �alter_db_tablespace� s
�z,BaseDatabaseSchemaEditor.alter_db_tablespacec
C s� |j r|jjjjr| �|jj�S | j||dd�\}}|du r dS |j| jd�}|d r4|d| j | 7 }|jr�| jj
j r�|j r�d}| j
r�|jjjj}|jjj�|jj�j}t|jj�\} }
|d| j
| �|||�| rpd| �| � nd | �|j�| �|�| �|�| jj�� d
� 7 }n | j�| �|||�� | j| �|jj�| �|j�|d � } | � | |� | � |�s�| � |�dur�| j |d|dd �\} }| j | �|jj�| d
� } | � | |� | j�!| �"||�� | jj
j#r�| j�$� dS 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� Nr\ r] r^ rb z%s.rF )r � namespacerX r` ra �
deferrable)rl rX rm ��drop�rl �changes)%r rv r� r r� r� rq rr r1 rs r4 r{ rw �sql_create_column_inline_fkr! rx rW ry rX r
�_fk_constraint_namerS rR �deferrable_sqlr9 rL r| �sql_create_columnr= r� r� �_alter_column_default_sql�sql_alter_columnru �_field_indexes_sql� connection_persists_old_columns�close)
r7 r! r rm rC r� �constraint_suffixr` ra r �_rB � changes_sqlr r r � add_field� sP
�
� �
�z"BaseDatabaseSchemaEditor.add_fieldc C s� |j r|jjjjr| �|jj�S |j| jd�d du r dS |jr8| j||j gdd�}|D ] }| �
| � ||�� q,| j | �
|jj�| �
|j �d� }| �
|� | jjjrX| j�� t| j�D ]}t|t�rs|�|jj|j �rs| j�|� q]dS )z�
Remove a field from a model. Usually involves deleting a column,
but for M2Ms may involve deleting a table.
r\ r� NT�� foreign_key)rl rX )r rv r� r r� r� rr r1 r� rX r= �_delete_fk_sql�sql_delete_columnrS rx r4 r r r� r9 r� r �references_columnr� )r7 r! r �fk_names�fk_namerB r r r � remove_field s(
�
��z%BaseDatabaseSchemaEditor.remove_fieldc
C s0 | � ||�sdS |j| jd�}|d }|j| jd�}|d }|du r'|jdu s0|du r8|jdu r8td||f ��|du r\|du r\|jjr\|jjr\|jjjjr\|jjjjr\| �||||�S |du rz|du rz|jjrz|jjrz|jjjjsz|jjjjszdS |du s�|du r�td||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.
Nr\ 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_alteredrr r1 rv r� r� r r� �_alter_many_to_many� _alter_field) r7 r! r* r- �strict�
old_db_params�old_type�
new_db_params�new_typer r r � alter_field8 sT ����
�
���
�
����z$BaseDatabaseSchemaEditor.alter_fieldc ) C s� t � } | jjjrC|jrC|jrC| j||jgdd�}
|r.t|
�dkr.t dt|
�|j
j |jf ��|
D ]} | � |jf� | �
| �|| �� q0|jr�|jrO| �||�r�dd� |j
jD �} | j||jgdd| d�}
|rzt|
�dkrzt d t|
�|j
j |jf ��|
D ] }| �
| �||�� q|| jjjo�|jr�|jp�|jo�|jo�||k}|r�t||�D ] \}}| j|j|jjgdd�}|D ] } | �
| �|j| �� q�q�|jr�|js�|jr�|jr�d
d� |j
jD �}| j||jgdtj|d �}|D ] }| �
| �||�� q�|d |d k�r=|d �r=d
d� |j
jD �} | j||jgd| d�}
|�r.t|
�dk�r.t dt|
�|j
j |jf ��|
D ] }| �
| � ||�� �q0|j|jk�ri| �
| � |j
j |||�� | j D ]}t |t ��rg|�!|j
j |j|j� �qTg }g }g }t"|dd�}t"|dd�}||k�r�| �#||||�} |�$| � n||k�r�| �%||||�\} } |�$| � |�&| � d} |j'�r�|j'�s�| �(|�} | �(|�} | �)|��s�| | k�r�| du�r�d} |�$| �*|||�� |j'|j'k�r�| �+|||�} | �r�|�$| � |�,� �o�|j'�o�|j' }!|�s |�ro|!�s|| }| jjj-�r#|�r#t.t/|� �\}}"d�0|�t1|"g �fg}|D ]\}}"| �
| j2| �3|j
j �|d� |"� �q%|!�ro| �
| j4| �3|j
j �| �3|j�dd� | g� |D ]\}}"| �
| j2| �3|j
j �|d� |"� �qX|�r�|D ] \}}"| �
||"� �qt|j�r�|j�s�| �5||� | �6||��r�| �
| �7||jg�� |j�r�|j�r�|j�r�|j�s�| �
| j8||gd�� g }#|�r�|#�&t||�� | �||��r�| �
| �9||�� |#�&t||�� |#D ]B\}$}|jj:| jd�}%|%d }&| �%|j|$j|j|&�\} } | �
| j2| �3|jj
j �| d d� | d � | D ] \}}"| �
||"� �q�q�| jjj�rH|j�rH| �s:|j�r:|j�sH|j�rH| �
| �;||d�� |�rk|j<j
j=D ]}'t>|'|��ri|'jj�ri| �
| �;|'j|'jd�� �qP|d |d k�r�|d �r�| j?|j
j |jgd d �}| �
| �@|||d �� | �r�| j*|||dd �\}(}"| j2| �3|j
j �|(d� }| �
||"� | jjjA�r�| j�B� dS dS ) z3Perform a "physical" (non-ManyToMany) field update.Tr r� z<Found wrong number (%s) of foreign key constraints for %s.%sc S r� r r� rg r r r r� z r� z8BaseDatabaseSchemaEditor._alter_field.<locals>.<setcomp>F)r� r r� z7Found wrong number (%s) of unique constraints for %s.%sc S r� r r� )r( r� r r r r� � r� )r� �type_r� r] c S r� r r� rg r r r r� � r� )r] r� z6Found wrong number (%s) of check constraints for %s.%sr� Nrk r
r� )rl rX r� �r� r\ r� r rb �_fk�_check�r� r )C�setr1 r4 r{ rv rw r� rX r� r� r rx �addr= r r� �_field_became_primary_keyr �_delete_unique_sqlr r/ �
related_modelr �db_indexr� r r� �_delete_index_sql�_delete_check_sql�_rename_field_sqlr9 r� r �rename_column_referencesr� �_alter_column_collation_sqlrL �_alter_column_type_sqlru r� r� r� r �_alter_column_null_sqlr� �supports_combined_altersrM r. r� �sumr rS �sql_update_with_default�_delete_primary_key�_unique_should_be_addedro r� �_create_primary_key_sqlrr r| r! �related_objectsr �_create_index_name�_create_check_sqlr r ))r7 r! r* r- r& r( r% r' r$ � fks_droppedr r r� r� �constraint_name�drop_foreign_keys�_old_rel�new_rel� rel_fk_namesr� � index_names�
index_namerB �actions� null_actions� post_actions�
old_collation�
new_collation�fragment�
other_actions�needs_database_default� old_default� new_default�four_way_default_alterationrC �rels_to_update�old_rel�
rel_db_params�rel_type�relr r r r r# c s� ���� ��
���� �
��
�
�
� ��
��
��
��� ������ � �z%BaseDatabaseSchemaEditor._alter_fieldc C sX | j jjr
|�� dv r
dS |j| j d�}|jr| jn| j}|| �|j �|d d� g fS )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.
)� CharField� TextFieldNr\ r� �rX r� )
r1 r4 r� r} rr r� �sql_alter_column_null�sql_alter_column_not_nullrS rX )r7 r! r* r- r' rB r r r r; b s
��z/BaseDatabaseSchemaEditor._alter_column_null_sqlc
C s� | � |�}| �|�}|g}|rg }n | jjjr | �|�}g }|j| jd�}|r2|jr.| j} n| 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.
r\ r� )rX r� r� )
r� r� r1 r4 r� r� rr r� � sql_alter_column_no_default_null�sql_alter_column_no_default�sql_alter_column_defaultrS rX )
r7 r! r* r- r rV r� rC r' rB r r r r x s*
��z2BaseDatabaseSchemaEditor._alter_column_default_sqlc C s | j | �|j�|d� g fg 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_ )�sql_alter_column_typerS rX )r7 r! r* r- r( r r r r: � s
���z/BaseDatabaseSchemaEditor._alter_column_type_sqlc C s, | j | �|j�||r| �|�ndd� g fS )NrF )rX r� r� )�sql_alter_column_collaterS rX r� )r7 r! r- r( rQ r r r r9 � s
��z4BaseDatabaseSchemaEditor._alter_column_collation_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) rv r� r rx r� r) rW �m2m_reverse_field_name�m2m_field_name)r7 r! r* r- r$ r r r r" � s
���z,BaseDatabaseSchemaEditor._alter_many_to_manyrF c C s� t |�\}}dt|g|�R ddi�|f }| jj�� p d}d|d�|�|f }t|�|kr/|S t|�|d kr?|d|d � }|t|� d d
}d|d|� d�|�d|� |f }|d dksg|d �� rod |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 r1 rR �max_name_lengthr� r� �isdigit) r7 �
table_name� column_namesr� r �hash_suffix_part�
max_lengthrL � other_lengthr r r rC � s"
�z+BaseDatabaseSchemaEditor._create_index_namec C sX |d u r t |�dkr|d jr|d j}n|jjr |jj}|d ur*d| jj�|� S dS )Nr� r r^ rF )r� r� r r1 rR r� )r7 r! r� r� r r r �_get_index_tablespace_sql� s z2BaseDatabaseSchemaEditor._get_index_tablespace_sqlc C s |rd| S dS )Nz WHERE rF r )r7 � conditionr r r �_index_condition_sql� s z-BaseDatabaseSchemaEditor._index_condition_sqlc C s, |r| j jjs dS tdt|jj|| j�d�S )NrF z INCLUDE (%(columns)s))r� )r1 r4 �supports_covering_indexesr r r rx rS )r7 r! r� r r r �_index_include_sql� s �z+BaseDatabaseSchemaEditor._index_include_sql) r� r r� �usingr� � col_suffixesrB � opclassesrw �include� expressionsc s� |pg }| pg } t |dd�j�jd�}
�j|||d�}dd� |D �}|p&�j}|jj}� �fdd�}t|t|�j �t
||||�||rJ�� |||| �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)�
alias_colsr\ )r� c S s g | ]}|j �qS r )rX rY r r r rZ r� z>BaseDatabaseSchemaEditor._create_index_sql.<locals>.<listcomp>c s" � d u r �j | i |��� ��� �S r% )rC rS ��args�kwargs�r r7 r r �create_index_name s
zEBaseDatabaseSchemaEditor._create_index_sql.<locals>.create_index_name)rl r r{ r� rD rw r~ )r
� get_compilerr1 rv �sql_create_indexr rx r r rS r �_index_columnsr rO rx rz )r7 r! r� r r� r{ r� r| rB r} rw r~ r �compilerr� r� r� rl r� r r� r r� s, �
�
�z*BaseDatabaseSchemaEditor._create_index_sqlc C s& t |p| jt|jj| j�| �|�d�S �N)rl r )r r� r r rx rS )r7 r! r rB r r r r5 , s
�z*BaseDatabaseSchemaEditor._delete_index_sqlc C s t ||| j|d�S )N)r| )r rS )r7 rl r� r| r} r r r r� 3 r# z'BaseDatabaseSchemaEditor._index_columnsc s� � j jr � j js � j jrg S g }� j jD ] }|�| �� |�� q� j jD ]}� fdd�|D �}|�| j � |dd�� q$� j j
D ]}|j rI| j j
jrR|�|�� | �� q?|S )z�
Return a list of all index SQL statements (field indexes,
index_together, Meta.indexes) for the specified model.
c r� r r� rY r r r rZ B rj z?BaseDatabaseSchemaEditor._model_indexes_sql.<locals>.<listcomp>r� r� )r �managed�proxy�swappedrp ru r �index_togetherrL r� r� r� r1 r4 r� r� )r7 r! �outputr r� r� r� r r r r� 6 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_indexedrL r� )r7 r! r r� r r r r M s z+BaseDatabaseSchemaEditor._field_indexes_sqlc C sv |� � \}}}}|� � \}}}} g d�}
|
D ]} |�| d � | �| d � q| �|j�| �|j�kp:|||f||| fkS )N) r� � db_column�editable�error_messages� help_text�limit_choices_to� on_delete� related_name�related_query_name�
validators� verbose_name)� deconstruct�poprS rX ) r7 r* r- r �old_path�old_args�
old_kwargs�new_path�new_args�
new_kwargs�non_database_attrs�attrr r r r! V s �z1BaseDatabaseSchemaEditor._field_should_be_alteredc C s |j o|j S r% )r4 r� �r7 r! r r r r r� s rT z1BaseDatabaseSchemaEditor._field_should_be_indexedc C s |j o|j S r% �r �r7 r* r- r r r r1 v rT z2BaseDatabaseSchemaEditor._field_became_primary_keyc C s" |j r|j p|jo|j o|j S r% )r� r r� r r r r@ y s �z0BaseDatabaseSchemaEditor._unique_should_be_addedc C s* | j | �|�| �|j�| �|j�|d� S )N)rl �
old_column�
new_columnr� )�sql_rename_columnrS rX )r7 rl r* r- r( r r r r7 ~ 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)rl r rX r` ra r )r r rx rS r
r rX � target_fieldr! r1 rR r r �
sql_create_fk)
r7 r! r r� rl r rX r` ra r r r r r| � s �z'BaseDatabaseSchemaEditor._create_fk_sqlc s< � fdd�}t |jj|jgt|jjjj�d |jjg||�S )Nc � � � � j| i |���S r% �rS rC r� r< r r �create_fk_name� � zDBaseDatabaseSchemaEditor._fk_constraint_name.<locals>.create_fk_namer� )r r rx rX r
r� r! )r7 r! r r� r� r r< r r
� s �z,BaseDatabaseSchemaEditor._fk_constraint_namec C � | � | j||�S r% )r� �
sql_delete_fk�r7 r! r r r r r � � z'BaseDatabaseSchemaEditor._delete_fk_sqlc C s, |d u rdS |t jkr
dS |t jkrdS d S )NrF z DEFERRABLE INITIALLY DEFERREDz DEFERRABLE INITIALLY IMMEDIATE)r �DEFERRED� IMMEDIATE)r7 r r r r �_deferrable_constraint_sql� s
�z3BaseDatabaseSchemaEditor._deferrable_constraint_sqlc
C s� |r | j jjs d S |s|s|r$| j||||||d�}|r"| j�|� d S | jd�t| j |��| �
|�d� } | j | � |�| d� S )N)r rw r~ r} rk )r� r �r rh ) r1 r4 �&supports_deferrable_unique_constraintsro r9 rL �sql_unique_constraintr� rN rS r� �sql_constraint)
r7 r! r� r rw r r~ r} rB rh r r r � _unique_sql� s2 �� � ��z$BaseDatabaseSchemaEditor._unique_sqlc s� |r� j jjr|r� j jjr|r� j jjsd S � fdd�}t|jj� j�} |d u r3t |jj|d|�}n� �|�}� j
| |d|d�}|sG|sG|rK� j }
n� j }
t
|
| ||� �|�� �|�� �||�d�S )Nc r� r% r� r� r< r r �create_unique_name� r� zGBaseDatabaseSchemaEditor._create_unique_sql.<locals>.create_unique_name�_uniqr )r| r} )rl r r� rw r r~ )r1 r4 r� �supports_partial_indexesry r r rx rS r r� �sql_create_unique_index�sql_create_uniquer rx r� rz ) r7 r! r� r rw r r~ r} r� rl rB r r< r ro � s> ������
�z+BaseDatabaseSchemaEditor._create_unique_sqlc C sV |r| j jjr|r| j jjr|r| j jjsd S |s |s |r!| j}n| j}| �|||�S r% )r1 r4 r� r� ry r� r� r� )r7 r! r rw r r~ r} rB r r r r2 � s" ������ z+BaseDatabaseSchemaEditor._delete_unique_sqlc C s | j | �|�| jd|i d� S )Nr] r� )r� rS rs )r7 r r] r r r �
_check_sql s �z#BaseDatabaseSchemaEditor._check_sqlc C s$ t | jt|jj| j�| �|�|d�S )N)rl r r] )r �sql_create_checkr r rx rS )r7 r! r r] r r r rD
s �z*BaseDatabaseSchemaEditor._create_check_sqlc C r� r% )r� �sql_delete_checkr� r r r r6 r� z*BaseDatabaseSchemaEditor._delete_check_sqlc C s t |t|jj| j�| �|�d�S r� )r r r rx rS )r7 �templater! r r r r r� s
�z/BaseDatabaseSchemaEditor._delete_constraint_sqlc
s |dur
� fdd�|D �}� j �� �}
� j j�|
|jj�} W d � n1 s'w Y g } | �� D ]Y\}
}|du s@||d kr�|durK|d |krKq2|durV|d |krVq2|dura|d |kraq2|durl|d |krlq2|duru|d suq2|dur�|d
|kr�q2| r�|
| vr�| �|
� q2| S ) z@Return all constraint names matching the columns and conditions.Nc s g | ] }� j j�|��qS r )r1 �
introspection�identifier_converter)r( r r< r r rZ $ s ��z>BaseDatabaseSchemaEditor._constraint_names.<locals>.<listcomp>r� r� r r� r] r r� )r1 rP r� �get_constraintsr rx �itemsrL )r7 r! rr r� r r� r r] r* r� rP r �resultr �infodictr r< r r� s4
� �
�z*BaseDatabaseSchemaEditor._constraint_namesc C sV | j |dd�}|rt|�dkrtdt|�|jjf ��|D ] }| �| �||�� q d S )NTr� r� z0Found wrong number (%s) of PK constraints for %s)r� r� r� r rx r= �_delete_primary_key_sql)r7 r! r$ r� rF 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 )N�_pkr. )rl r r� ) r �
sql_create_pkr r rx rS rC rX r r� r r r rA G s ��z0BaseDatabaseSchemaEditor._create_primary_key_sqlc C r� r% )r� �
sql_delete_pkr� r r r r� Q r� z0BaseDatabaseSchemaEditor._delete_primary_key_sqlc C s d| � |� S )Nz COLLATE )rS )r7 r� r r r r� T rT z%BaseDatabaseSchemaEditor._collate_sqlc C s* | j | �|�d�|�d� }| �|� d S )N�,)� procedure� param_types)�sql_delete_procedurerS r� r= )r7 �procedure_namer� rB r r r �remove_procedureW s
�z)BaseDatabaseSchemaEditor.remove_procedure)FT)r )F)rF r% )NNNN)NNNNN)NNNNNNNN)h�__name__�
__module__� __qualname__�__doc__r� r� r r� r r re r` ra rd rc rb rf r r� r> r� rs �sql_delete_constraintr� r� r� r� r� r� rz r r� r� r� r� r� r� r� r8 r; r> r= rS r� rq r� r� r� r� � staticmethodr� r� rO r� r� r� r� r� r� r� r� r� r� r r r r) r# r; r r: r9 r" rC rv rx rz r� r5 r� r� r r! r� r1 r@ r7 r| r
r r� r� ro r2 r� rD r6 r� r� r? rA r� r� r� r r r r r0 - s� �
;<
9
,
�
%
�
'
�!
�&
�
�
r0 ) �loggingr �!django.db.backends.ddl_referencesr r r r r r �django.db.backends.utilsr r
�django.db.modelsr r �django.db.models.sqlr
�django.db.transactionr r � django.utilsr � getLoggerrI r r" r/ r0 r r r r �<module> s