Dashboard
PHP:
8.2.30
OS:
Linux
User:
dh_nv8b2m
/
/
usr
/
lib
/
python3
/
dist-packages
/
django
/
db
/
backends
/
base
/
__pycache__
📤 Upload
📝 New File
📁 New Folder
Close
Editing: 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
Save
Cancel