o
    {,hA                    @   s   d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlmZm	Z	m
Z
mZmZmZ d dlmZmZmZ d dlmZmZ d dlmZ d d	lmZ d d
lmZmZ d dlmZ e dZdd Z dd Z!dd Z"G dd dZ#dS )    N)datetime)settings)
FieldError)ColumnsExpressionsForeignKeyName	IndexName	StatementTable)names_digestsplit_identifiertruncate_name)
DeferrableIndex)CompositePrimaryKey)Query)TransactionManagementErroratomic)timezonezdjango.db.backends.schemac                 C   s2   | j }|jrdS |jr|jdgkrdS |j|jv 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   T/var/www/html/rh/venv/lib/python3.10/site-packages/django/db/backends/base/schema.py_is_relevant_relation   s   r   c                 C   s"   t | jjdddddtddS )NFT)forwardreverseinclude_hiddeninclude_parentsr   )key)sorted_meta_get_fieldsoperator
attrgettermodelr   r   r   _all_related_fields)   s   r+   c                 #   sd    t fddtjD  fddt jD }|D ]\}}||fV  t|j|jE d H  qd S )Nc                 3       | ]
}t | r|V  qd 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   r3   ?   r4   )zipr+   r*   _related_non_m2m_objectsremote_field)r2   r5   related_fieldsold_relnew_relr   )r5   r2   r   r7   6   s    


r7   c                   @   s  e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZeZdZdZdZdZdZdZdZdZdZeZdZeZdZdZdZeZdZ dZ!dZ"dZ#dZ$eZ%dZ&dZ'd Z(dd#d$Z)d%d& Z*d'd( Z+dd*d+Z,d,d- Z-d.d/ Z.d0d1 Z/dd2d3Z0d4d5 Z1d6d7 Z2d8d9 Z3d:d; Z4d<d= Z5d>d? Z6e7d@dA Z8dBdC Z9dDdE Z:dFdG Z;dHdI Z<dJdK Z=dLdM Z>dNdO Z?dPdQ Z@dRdS ZAdTdU ZBdVdW ZCdXdY ZDdZd[ ZEd\d] ZFd^d_ ZGd`da ZHdbdc ZIddddeZJdfdg ZK	!ddhdiZLdjdk ZMddldmZN	!ddndoZOdpdq ZPdrds ZQdtdu ZRdvdw ZSddydzZTdd{d|ZUd}d~ ZVdd ZWdddxdxdd)dd)ddddddZXdddZYdd ZZdd Z[dd Z\dd Z]dddZ^dd Z_dd Z`dd Zadd Zbdd Zcdd Zddd Zedd Zfdd Zg					dddZh						dddZi							dddZjdddZk						dddZldd Zmdd Zndd Zodd Zp								dddZqdd ZrdddZsdd Ztdd ZudddZvdddZwdS )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)szCHECK (%(check)s)z.ALTER TABLE %(table)s DROP CONSTRAINT %(name)sz"CONSTRAINT %(name)s %(constraint)szPRIMARY KEY (%(columns)s)z?ALTER TABLE %(table)s ADD CONSTRAINT %(name)s CHECK (%(check)s)zbALTER TABLE %(table)s ADD CONSTRAINT %(name)s UNIQUE%(nulls_distinct)s (%(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)szaCREATE UNIQUE INDEX %(name)s ON %(table)s (%(columns)s)%(include)s%(nulls_distinct)s%(condition)sz/ALTER INDEX %(old_name)s RENAME TO %(new_name)szDROP 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,   || _ || _| jrg | _| j jjo|| _d S r-   )
connectioncollect_sqlcollected_sqlfeaturescan_rollback_ddlatomic_migration)selfr=   r>   r   r   r   r   __init__   s
   z!BaseDatabaseSchemaEditor.__init__c                 C   s(   g | _ | jrt| jj| _| j  | S r-   )deferred_sqlrB   r   r=   alias	__enter__rC   r   r   r   rG      s
   
z"BaseDatabaseSchemaEditor.__enter__c                 C   s>   |d u r| j D ]}| |d  q| jr| j||| d S d S r-   )rE   executerB   r   __exit__)rC   exc_type	exc_value	tracebacksqlr   r   r   rJ      s   
z!BaseDatabaseSchemaEditor.__exit__r   c                 C   s   | j s| jjr| jjjstdt|}tjd||||dd | j rN|	 
dr,dnd}|durD| j|tt| j| |  dS | j||  dS | j }||| W d   dS 1 sew   Y  dS )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))paramsrN   )extra; N)r>   r=   in_atomic_blockr@   rA   r   strloggerdebugrstripendswithr?   appendtuplemapquote_valuecursorrI   )rC   rN   rO   endingr]   r   r   r   rI      s.   "z BaseDatabaseSchemaEditor.executec                 C   s   | j j|S r-   )r=   ops
quote_name)rC   r   r   r   r   r`         z#BaseDatabaseSchemaEditor.quote_namec                    sR   j jD ]} fdd|D }j | qg }g } j jD ]} |\}}|du r1q"|jjd}	|	d rE|dj	|	  7 }|j
jd}
|
rT|d|
 7 }|| |jr|jr|jjj j}|jjj |jjj}jr|dj||d  7 }njjjrj |d	 |d
|j|f  | dv rjj j j|j}|rj| q"g }|rӈ j jD ]}j|  qn| fdd j jD   j j}t|tr| |j! j" j jd#dd g ||R D d } j j$r%jj% j j$}|r%|d| 7 }||fS )z-Take a model and return its table definition.c                       g | ]} j |qS r   r%   	get_fieldr0   r   r)   r   r   
<listcomp>       z6BaseDatabaseSchemaEditor.table_sql.<locals>.<listcomp>Nr=   check z %s)to_table	to_column_fk_%(to_table)s_%(to_column)sz%s %s)	AutoFieldBigAutoFieldSmallAutoFieldc                 3   s    | ]	}|  V  qd S r-   )constraint_sqlr0   
constraintr*   rC   r   r   r3     s
    

z5BaseDatabaseSchemaEditor.table_sql.<locals>.<genexpr>, c                 s   s    | ]	}|rt |V  qd S r-   )rT   )r0   	statementr   r   r   r3   &  s    
)table
definition)&r%   unique_togetherrE   rY   _create_unique_sqllocal_fields
column_sqldb_parametersr=   sql_check_constraintdb_type_suffixextendr8   db_constraintr*   db_tablerd   
field_namecolumnsql_create_inline_fkr`   r@   supports_foreign_keys_create_fk_sqlget_internal_typer_   autoinc_sqlconstraints
create_sqlpk
isinstancer   _pk_constraint_sqlcolumnssql_create_tablejoindb_tablespacetablespace_sql)rC   r*   field_namesfieldscolumn_sqlsrO   r   rx   extra_params	db_paramscol_type_suffixrk   rl   r   constraint_sqlsrs   r   rN   r   r   rt   r   	table_sql   s   



	



z"BaseDatabaseSchemaEditor.table_sqlc                 c   s   |V  | d }r| |V  | jjjr |jr | |jV  |j}| r;| 	|\}	}
d|	 V  |
|
 d}|oJ| | oJ|oI| | }|rs| |}|d ursd| | }| jjjrk|| | V  n|V  || |jr|js| jjjrd}|jr| |\}}|
| |V  n|sdV  n| jjjsdV  |jrdV  n|jrdV  |jp|jj}|r| jjjr|jr| jjj|dd	V  d S d S d S d S )
N	collationzDEFAULT FTzNOT NULLNULLzPRIMARY KEYUNIQUE)inline)get_collate_sqlr=   r@   supports_comments_inline
db_comment_comment_sqlnullhas_db_defaultdb_default_sqlr   skip_defaultskip_default_on_altereffective_default_column_default_sqlrequires_literal_defaultsprepare_defaultrY   empty_strings_allowedr   !interprets_empty_strings_as_nulls	generated_column_generated_sqlimplied_column_nulluniquer   r%   supports_tablespacesr_   r   )rC   column_db_typerO   r*   r   field_db_paramsinclude_defaultr   r   default_sqldefault_paramsdefault_valuecolumn_defaultgenerated_sqlgenerated_params
tablespacer   r   r   _iter_column_sql6  sj   

	




z)BaseDatabaseSchemaEditor._iter_column_sqlc              
   C   sD   |j | jd}|d }|du rdS g }d| |||||||fS )z
        Return the column definition for a field. The field must already have
        had set_attributes_from_name() called.
        rh   typeNNNrj   )r}   r=   r   r   )rC   r*   r   r   r   r   rO   r   r   r   r|   y  s"   z#BaseDatabaseSchemaEditor.column_sqlc                 C      dS )z
        Some backends don't accept default values for certain columns types
        (i.e. MySQL longtext and longblob).
        Fr   rC   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
        zsubclasses of BaseDatabaseSchemaEditor for backends which have requires_literal_defaults must provide a prepare_default() methodNotImplementedErrorrC   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           	         s   ddl m} |j}t||r |nd}t|jd}|j jd}|	|\}} jj
jr=|t fdd|D ; }g }|| |fS )z;Return the sql and params for the field's database default.r   )Valuez(%s)r)   rh   c                 3       | ]}  |V  qd S r-   )r   r0   prH   r   r   r3         z:BaseDatabaseSchemaEditor.db_default_sql.<locals>.<genexpr>)django.db.models.expressionsr   _db_default_expressionr   r   r   r*   get_compilerr=   compiler@   r   rZ   )	rC   r   r   
db_defaultrN   querycompilerr   rO   r   rH   r   r     s   
z'BaseDatabaseSchemaEditor.db_default_sqlc                    sZ   |  j\}}|jrdnd} jjjr#|t fdd|D  }d}d| d| |fS )z3Return the SQL to use in a GENERATED ALWAYS clause.STOREDVIRTUALc                 3   r   r-   r\   r   rH   r   r   r3     r   zABaseDatabaseSchemaEditor._column_generated_sql.<locals>.<genexpr>r   zGENERATED ALWAYS AS (z) )r   r=   
db_persistr@   r   rZ   )rC   r   expression_sqlrO   persistency_sqlr   rH   r   r     s   
z.BaseDatabaseSchemaEditor._column_generated_sqlc                 C   s   |   r
|  }|S | jrd }|S | js(| jr(| jr(|  dkr$d}|S d}|S t| dds4t| ddrZ|  }|dkrBt	 }|S t
	 }|dkrP| }|S |d	krX| }|S d }|S )
NBinaryField    rR   auto_nowFauto_now_addDateTimeField	DateField	TimeField)has_defaultget_defaultr   r   blankr   r   getattrr   nowr   datetime)r   defaultinternal_typer   r   r   _effective_default  s4   	z+BaseDatabaseSchemaEditor._effective_defaultc                 C   s   | | || jS )z2Return a field's effective database default value.)get_db_prep_saver   r=   r   r   r   r   r     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   r\     s   z$BaseDatabaseSchemaEditor.quote_valuec                 C   s   |  |\}}| ||pd | jjjrG|jjr!| |d|jj | jjjsG|jj	D ]}|j
rF|j| jd}|d }| j| ||||j
  q*| j| | |jjD ]}|jjjjrc| |jj qTdS )zr
        Create a table and any accompanying indexes or unique constraints for
        the given `model`.
        Nrh   r   )r   rI   r=   r@   supports_commentsr%   db_table_commentalter_db_table_commentr   r{   r   r}   _alter_column_comment_sqlrE   r   _model_indexes_sqllocal_many_to_manyr8   throughauto_createdcreate_model)rC   r*   rN   rO   r   r   
field_typer   r   r   r     s0   


z%BaseDatabaseSchemaEditor.create_modelc                 C   s~   |j jD ]}|jjj jr| |jj q| | jd| |j j	i  t
| jD ]}t|tr<||j j	r<| j| q(dS )z!Delete a model from the database.rw   N)r%   r   r8   r   r   delete_modelrI   sql_delete_tabler`   r   listrE   r   r	   references_tableremove)rC   r*   r   rN   r   r   r   r     s"   z%BaseDatabaseSchemaEditor.delete_modelc                 C   s.   |j r
| jjjs
dS | j||| dd dS )zAdd an index on a model.NrO   )contains_expressionsr=   r@   supports_expression_indexesrI   r   rC   r*   indexr   r   r   	add_index0  s   z"BaseDatabaseSchemaEditor.add_indexc                 C   s*   |j r
| jjjs
dS | |||  dS )zRemove an index from a model.N)r   r=   r@   r   rI   
remove_sqlr   r   r   r   remove_index;  s   z%BaseDatabaseSchemaEditor.remove_indexc                 C   sF   | j jjr| j| ||j|jd d d S | || | || d S )Nr   )r=   r@   can_rename_indexrI   _rename_index_sqlr   r  r  )rC   r*   	old_index	new_indexr   r   r   rename_indexD  s   

z%BaseDatabaseSchemaEditor.rename_indexc                 C   s&   | || }|r| j|dd dS dS )zAdd a constraint to a model.Nr   )r   rI   rC   r*   rs   rN   r   r   r   add_constraintN  s   z'BaseDatabaseSchemaEditor.add_constraintc                 C   s"   | || }|r| | dS dS )z!Remove a constraint from a model.N)r  rI   r
  r   r   r   remove_constraintV  s   z*BaseDatabaseSchemaEditor.remove_constraintc                    s|   dd |D }dd |D }| |D ]}|  |ddd| j q| |D ]} fdd|D }| |  | q'd	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      h | ]}t |qS r   rZ   r0   r   r   r   r   	<setcomp>b      zABaseDatabaseSchemaEditor.alter_unique_together.<locals>.<setcomp>c                 S   r  r   r  r  r   r   r   r  c  r  TF)r   r   c                    rb   r   rc   re   r)   r   r   rf   n  rg   zBBaseDatabaseSchemaEditor.alter_unique_together.<locals>.<listcomp>N)
difference_delete_composed_indexsql_delete_uniquerI   rz   )rC   r*   old_unique_togethernew_unique_togetheroldsnewsr   r   r   r)   r   alter_unique_together\  s   z.BaseDatabaseSchemaEditor.alter_unique_togetherc                    s   dd |D }dd |D }| |D ]}|  |ddd| j q| |D ]} fdd|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  w  r  z@BaseDatabaseSchemaEditor.alter_index_together.<locals>.<setcomp>c                 S   r  r   r  r  r   r   r   r  x  r  TF)r  r   c                    rb   r   rc   re   r)   r   r   rf     rg   zABaseDatabaseSchemaEditor.alter_index_together.<locals>.<listcomp>_idx)r   suffixN)r  r  sql_delete_indexrI   _create_index_sql)rC   r*   old_index_togethernew_index_togetherr  r  r   r   r   r)   r   alter_index_togetherq  s   z-BaseDatabaseSchemaEditor.alter_index_togetherc           
         s   dd  j jD }dd  j jD } fdd|D }| j |fd||B i|}|ddu rJ|rJ| jjjrJt| j	 j j
|d	d
}	|	|v rJ|	g}t|dkr`tdt| j j
d|f | | | |d  d S )Nc                 S      h | ]}|j qS r   r   rr   r   r   r   r        zBBaseDatabaseSchemaEditor._delete_composed_index.<locals>.<setcomp>c                 S   r!  r   r"  rr   r   r   r   r        c                    s   g | ]	} j |jqS r   )r%   rd   r   re   r)   r   r   rf     s    zCBaseDatabaseSchemaEditor._delete_composed_index.<locals>.<listcomp>excluder   TFquote   z1Found wrong number (%s) of constraints for %s(%s)ru   r   )r%   r   indexes_constraint_namesr   r=   r@   *allows_multiple_constraints_on_same_fieldsrT   _unique_constraint_namer   len
ValueErrorr   rI   _delete_constraint_sql)
rC   r*   r   constraint_kwargsrN   meta_constraint_namesmeta_index_namesr   constraint_namesdefault_namer   r)   r   r    sB   z/BaseDatabaseSchemaEditor._delete_composed_indexc                 C   sn   ||ks| j jjr| | krdS | | j| || |d  | jD ]}t|t	r4|
|| q'dS )z#Rename the table a model points to.N)	old_table	new_table)r=   r@   ignores_table_name_caselowerrI   sql_rename_tabler`   rE   r   r	   rename_table_references)rC   r*   old_db_tablenew_db_tablerN   r   r   r   alter_db_table  s"   

z'BaseDatabaseSchemaEditor.alter_db_tablec                 C   sF   | j r| jjjr!| | j | |jj| |pdd  d S d S d S )NrR   )rw   comment)	sql_alter_table_commentr=   r@   r   rI   r`   r%   r   r\   )rC   r*   old_db_table_commentnew_db_table_commentr   r   r   r     s   z/BaseDatabaseSchemaEditor.alter_db_table_commentc                 C   s2   |  | j| |jj| || |d  dS )z)Move a model's table between tablespaces.)rw   old_tablespacenew_tablespaceN)rI   sql_retablespace_tabler`   r%   r   )rC   r*   old_db_tablespacenew_db_tablespacer   r   r   alter_db_tablespace  s   z,BaseDatabaseSchemaEditor.alter_db_tablespacec              
   C   sL  |j r|jjjjr| |jjS | j||dd\}}|du r dS |j| jd }r0|d| 7 }|j	| jd}|d rD|d| j
|  7 }|jr| jjjr|jrd}| jr|jjjj}|jjj|jjj}	t|jj\}
}|d| j| ||||
rd| |
 nd	| |j| || |	| jj d
  7 }n| j| ||| | j| |jj| |j|d }| ||pd | s| |s|  |dur| j!|d|dd\}}| j"| |jj|d }| || |j#r| jjj$r| jjj%s|d }| j| &||||j#  | j'| (|| | jjj)r$| j*  dS dS )z
        Create a field on a model. Usually involves adding a column, but may
        involve adding a table instead (for M2M fields).
        T)r   Nrh   rj   ri   rm   z%s.rR   )r   	namespacer   rk   rl   
deferrable)rw   r   rx   droprw   changesr   )+r   r8   r   r%   r   r   r|   r   r=   r}   r~   r@   r   r   sql_create_column_inline_fkr*   r   rd   r   r   r   _fk_constraint_namer`   r_   deferrable_sqlrE   rY   r   sql_create_columnrI   r   r   r   _alter_column_default_sqlsql_alter_columnr   r   r   r   r   _field_indexes_sqlconnection_persists_old_columnsclose)rC   r*   r   rx   rO   r   r   constraint_suffixrk   rl   rH  _rN   changes_sqlr   r   r   r   	add_field  s   





z"BaseDatabaseSchemaEditor.add_fieldc                 C   s   |j r|jjjjr| |jjS |j| jdd du rdS |jr8| j||j	gdd}|D ]}| 
| || q,| j| |jj| |j	d }| 
| | jjjrX| j  t| jD ]}t|trs||jj|j	rs| j| q]dS )z
        Remove a field from a model. Usually involves deleting a column,
        but for M2Ms may involve deleting a table.
        rh   r   NTforeign_key)rw   r   )r   r8   r   r%   r   r   r}   r=   r*  r   rI   _delete_fk_sqlsql_delete_columnr`   r   r@   rU  rV  r   rE   r   r	   references_columnr   )rC   r*   r   fk_namesfk_namerN   r   r   r   remove_field(  s,   




z%BaseDatabaseSchemaEditor.remove_fieldc              
   C   s  |  ||sdS |j| jd}|d }|j| jd}|d }d}	|du r)|jdu s2|du r:|jdu r:td||f |du r^|du r^|jjr^|jjr^|jjjjr^|jjjjr^| ||||S |du r||du r||jjr||jjr||jjjjs||jjjjs|dS |du s|du rtd||f |j	|j	ks|j	r|j
|j
krd}	n!|j	rz|| j}
W n ty   d}	Y nw || j}|
|k}	|	rtd| d	| |||||||| dS )
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.
        Nrh   r   FzqCannot alter field %s into %s - they do not properly define db_type (are you using a badly-written custom field?)zCannot 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)Tz7Modifying GeneratedFields is not supported - the field z6 must be removed and re-added with the new definition.)_field_should_be_alteredr}   r=   r8   r.  r   r%   r   _alter_many_to_manyr   r   r   r   _alter_field)rC   r*   r2   r5   strictold_db_paramsold_typenew_db_paramsnew_typemodifying_generated_fieldold_field_sqlnew_field_sqlr   r   r   alter_fieldH  s   



	
z$BaseDatabaseSchemaEditor.alter_fieldc                 C   sD   | j j}|| j }d|d< z	||  | W S  ty!   Y d S w )N__column_name__r   )r=   data_type_check_constraintsdb_type_parametersr   KeyError)rC   r   r   check_constraintsdatar   r   r   _field_db_check  s   z(BaseDatabaseSchemaEditor._field_db_checkc	           2   	   C   sp  t  }	| jjjrL|jrL|jrL| j||dhdrL| j||jgdd}
|r7t	|
dkr7t
dt	|
|jj|jf |
D ]}|	|jf | | || q9|jr|jrX| ||rdd |jjD }| j||jgdd	|d
}|rt	|dkrt
dt	||jj|jf |D ]}| | || q|d}|d}| jjjo|jr|jp|jo|jo||kp||k}|rt||D ]\}}| j|j|jjgdd}|D ]}| | |j| qq|jr|js|jr|jrdd |jjD }| j||jgdtj|d}|D ]}| | || q| ||}| ||}||kr^|r^dd |jjD }| j||jgd|d}|rOt	|dkrOt
dt	||jj|jf |D ]}| | || qQ|j|jkr| |  |jj||| | j!D ]}t"|t#r|$|jj|j|j qug }g }g }|j%| jd}|j%| jd}||ks||ks||ks| jjj&r|j'|j'kr| (||||||\} }!|)|  |*|! |+ r|+ r|j,|j,kr|)| -||| n|+ r|)| j-|||dd d	}"|j.r2|j.s2|+ s2| /|}#| /|}$| 0|s2|#|$kr2|$dur2d}"|)| 1||| |j.|j.krH| 2|||} | rH|)|  |3 pP|+ oY|j.oY|j. }%|s`|r|%sg||7 }| jjj4r|rt5t6| \}}&d7|t8|&g fg}|D ]\}}&| | j9| :|jj|d |& q|%r|+ sd}'|$g}&n| ;|\}'}&| | j<| :|jj| :|j|'d |& |D ]\}}&| | j9| :|jj|d |& q|r|D ]\}}&| ||& q|jr|js| =|| | >||r| | ?||g |jr|jr+|jr+|js+| | j@||gd g }(|r8|(*t|| | ||rP| | A|| |(*t|| |(D ]V\})}|jjB| jd}*|*d }+|*d},|)jjB| jd}-|-d}.| (|j|)j|j|+|.|,\} }!| | j9| :|jjj| d d | d  |!D ]\}}&| ||& qqR| jjjr|jr|	s|jr|js|jr| | C||d |r|(D ]\}/}0|0jjr| | C|0j|0jd q||kr|r| jD|jj|jgdd}| | E|||d   |"r)| j1|||dd\}1}&| j9| :|jj|1d }| ||& | jjjFr6| jG  dS dS )!z3Perform a "physical" (non-ManyToMany) field update.r   )ignoreTr[  r(  z<Found wrong number (%s) of foreign key constraints for %s.%sc                 S   r!  r   r"  rr   r   r   r   r    r#  z8BaseDatabaseSchemaEditor._alter_field.<locals>.<setcomp>F)r   r   r%  z7Found wrong number (%s) of unique constraints for %s.%sr   c                 S   r!  r   r"  )r0   r  r   r   r   r    r$  )r  type_r%  c                 S   r!  r   r"  rr   r   r   r   r    r#  )ri   r%  z6Found wrong number (%s) of check constraints for %s.%srh   rJ  Nru   rL  r   )rw   r   r   r   r   r   rm   _fk_checkr  ri   )Hsetr=   r@   r   r8   r   rc  r*  r   r-  r.  r%   r   addrI   r]  r   _field_became_primary_keyr   _delete_unique_sqlr   r   r7   related_modelr   db_indexr)  r   r  _delete_index_sqlru  _delete_check_sql_rename_field_sqlrE   r   r	   rename_column_referencesr   r   r   _alter_column_type_sqlrY   r   r   r   "_alter_column_database_default_sqlr   r   r   rR  _alter_column_null_sqlr   supports_combined_altersrZ   r6   r   sumrS  r`   r   sql_update_with_default_delete_primary_key_unique_should_be_addedrz   r  _create_primary_key_sqlr}   r   _create_index_name_create_check_sqlrU  rV  )2rC   r*   r2   r5   rh  rj  rg  ri  rf  fks_droppedr`  ra  r1  r3  constraint_nameold_collationnew_collationdrop_foreign_keys_old_relr;   rel_fk_namesr2  index_names
index_nameold_db_checknew_db_checkrN   actionsnull_actionspost_actionsold_type_suffixnew_type_suffixfragmentother_actionsneeds_database_defaultold_defaultnew_defaultfour_way_default_alterationrO   r   rels_to_updater:   rel_db_paramsrel_typerel_collationold_rel_db_paramsold_rel_collationrX  relrY  r   r   r   re    sV  

























z%BaseDatabaseSchemaEditor._alter_fieldc                 C   sR   | j jjr
|jr
dS |j| j d}|jr| jn| j}|| |j	|d d g fS )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.
        Nrh   r   )r   r   )
r=   r@   r   r   r}   r   sql_alter_column_nullsql_alter_column_not_nullr`   r   )rC   r*   r2   r5   ri  rN   r   r   r   r    s"   
z/BaseDatabaseSchemaEditor._alter_column_null_sqlc           
      C   s   |  |}| |}|g}|rg }n| jjjr| |}g }|j| jd}|r2|jr.| j}	n| j	}	n| j
}	|	| |j|d |d |fS )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.
        rh   r   r   r   r   )r   r   r=   r@   r   r   r}   r    sql_alter_column_no_default_nullsql_alter_column_no_defaultsql_alter_column_defaultr`   r   )
rC   r*   r2   r5   rK  r  r   rO   ri  rN   r   r   r   rR  3  s,   




z2BaseDatabaseSchemaEditor._alter_column_default_sqlc           	      C   sV   |r
| j }d}g }n
| j}| |\}}|j| jd}|| |j|d |d |fS )z
        Hook to specialize column database default alteration.

        Return a (sql, params) fragment to add or drop (depending on the drop
        argument) a default to new_field's column.
        rR   rh   r   r  )r  r  r   r}   r=   r`   r   )	rC   r*   r2   r5   rK  rN   r   rO   ri  r   r   r   r  Y  s   	
z;BaseDatabaseSchemaEditor._alter_column_database_default_sqlc                 C   s   g }|  |||jj }rd| }nd}d}	| jjjrB|jsB|j|jkr9| ||||j\}
}|
r9|	|
|f |jrB| 
|j}	| j| |j|||	d g f|fS )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 2-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.
        rj   rR   )r   r   r   r>  )r   r%   r   r=   r@   r   r   r   r   rY   r   sql_alter_column_typer`   r   )rC   r*   r2   r5   rj  r  r  r  collate_sqlcomment_sqlrN   rO   r   r   r   r  u  s6   



z/BaseDatabaseSchemaEditor._alter_column_type_sqlc                 C   s.   | j | |jj| |j| |d g fS )N)rw   r   r>  )sql_alter_column_commentr`   r%   r   r   r   )rC   r*   r5   rj  new_db_commentr   r   r   r     s   
z2BaseDatabaseSchemaEditor._alter_column_comment_sqlc                 C   s   |  |pdS )NrR   r   )rC   r>  r   r   r   r     ra   z%BaseDatabaseSchemaEditor._comment_sqlc                 C   s   |j jjj|j jjjkr| |j j|j jjj|j jjj | |j j|j jj| |j jj|  | |j j|j jj| |j jj|  dS )z*Alter M2Ms to repoint their to= endpoints.N)	r8   r   r%   r   r=  rn  rd   m2m_reverse_field_namem2m_field_name)rC   r*   r2   r5   rf  r   r   r   rd    s,   





z,BaseDatabaseSchemaEditor._alter_many_to_manyrR   c           	      C   s   t |\}}dt|g|R ddi|f }| jj pd}d|d||f }t||kr/|S t||d kr?|d|d  }|t| d	 d
 }d|d| d|d| |f }|d dksg|d  rod|dd  }|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_%srX     N   r(  r   zD%s)r   r   r=   r_   max_name_lengthr   r-  isdigit)	rC   
table_namecolumn_namesr  rX  hash_suffix_part
max_lengthr  other_lengthr   r   r   r    s(   
z+BaseDatabaseSchemaEditor._create_index_namec                 C   sf   |d u r$t |dkr|d jr|d j}ntjrtj}n|jjr$|jj}|d ur1d| jj| S dS )Nr(  r   rj   rR   )r-  r   r   DEFAULT_INDEX_TABLESPACEr%   r=   r_   r   )rC   r*   r   r   r   r   r   _get_index_tablespace_sql  s   z2BaseDatabaseSchemaEditor._get_index_tablespace_sqlc                 C   s   |rd| S dS )Nz WHERE rR   r   )rC   	conditionr   r   r   _index_condition_sql  s   z-BaseDatabaseSchemaEditor._index_condition_sqlc                 C   s,   |r| j jjs	dS tdt|jj|| jdS )NrR   z INCLUDE (%(columns)s))r   )r=   r@   supports_covering_indexesr	   r   r%   r   r`   )rC   r*   r   r   r   r   _index_include_sql  s   z+BaseDatabaseSchemaEditor._index_include_sql)r   r   r  usingr   col_suffixesrN   	opclassesr  includeexpressionsc                   s   |pg }|pg }t |ddjjd}j|||d}dd |D }|p&j}|jj} fdd}t|t|j	t
||||||rJ||||	nt|||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_colsrh   )r   c                 S      g | ]}|j qS r   r   re   r   r   r   rf   '  r$  z>BaseDatabaseSchemaEditor._create_index_sql.<locals>.<listcomp>c                     s"    d u rj | i |  S r-   )r  r`   argskwargsr   rC   r   r   create_index_name+  s   
zEBaseDatabaseSchemaEditor._create_index_sql.<locals>.create_index_name)rw   r   r  r   rP   r  r  )r   r   r=   r  sql_create_indexr%   r   r	   r
   r`   r   _index_columnsr   r\   r  r  )rC   r*   r   r   r  r  r   r  rN   r  r  r  r  r   r   r   r  rw   r  r   r  r   r  
  s0   


z*BaseDatabaseSchemaEditor._create_index_sqlc                 C   sr   t |p| jt|jj| j| |d}|jd j}|jd }t| j	D ]}t
|t r6|||r6| j	| q#|S )Nrw   r   rw   r   )r	   r  r
   r%   r   r`   partsrw   r   rE   r   references_indexr   )rC   r*   r   rN   rv   r  r  r   r   r   r  @  s   
z*BaseDatabaseSchemaEditor._delete_index_sqlc                 C   s*   t | jt|jj| j| || |dS )N)rw   old_namenew_name)r	   sql_rename_indexr
   r%   r   r`   )rC   r*   r  r  r   r   r   r  R  s   z*BaseDatabaseSchemaEditor._rename_index_sqlc                 C   s   t ||| j|dS )N)r  )r   r`   )rC   rw   r   r  r  r   r   r   r  Z  s   z'BaseDatabaseSchemaEditor._index_columnsc                 C   st   |j jr|j js|j jrg S g }|j jD ]}|| || q|j jD ]}|jr.| j	j
jr7||||  q$|S )zz
        Return a list of all index SQL statements (field indexes, Meta.indexes)
        for the specified model.
        )r%   managedproxyswappedr{   r   rT  r)  r   r=   r@   r   rY   r   )rC   r*   outputr   r  r   r   r   r   ]  s   z+BaseDatabaseSchemaEditor._model_indexes_sqlc                 C   s*   g }|  ||r|| j||gd |S )zT
        Return a list of all index SQL statements for the specified field.
        rx  )_field_should_be_indexedrY   r  )rC   r*   r   r  r   r   r   rT  p  s   z+BaseDatabaseSchemaEditor._field_indexes_sqlc                 C   s   |j s|j sdS |pt }| \}}}}| \}}}	}
||jD ]}||d  q#||jD ]}|
|d  q2|js\|jr\|jr\|jjj	j
|jjj	j
kr\|dd  |
dd  |drz|
drz| || |krz|d |
d | |j| |jkp|||f||	|
fkS )NFtor   )concreter|  deconstructunionnon_db_attrspopr   r8   r*   r%   r   r   r   r`   r   )rC   r2   r5   rv  rX  old_pathold_args
old_kwargsnew_pathnew_args
new_kwargsattrr   r   r   rc  y  s@   




z1BaseDatabaseSchemaEditor._field_should_be_alteredc                 C   s   |j o|j S r-   )r  r   rC   r*   r   r   r   r   r    ra   z1BaseDatabaseSchemaEditor._field_should_be_indexedc                 C   s   |j  o|j S r-   r   rC   r2   r5   r   r   r   r~    ra   z2BaseDatabaseSchemaEditor._field_became_primary_keyc                 C   s   |j  o|jo|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)rw   
old_column
new_columnr   )sql_rename_columnr`   r   )rC   rw   r2   r5   rj  r   r   r   r    s   

z*BaseDatabaseSchemaEditor._rename_field_sqlc           
   	   C   s   t |jj| j}| |||}t|jj|jg| j}t |jjjj| j}t|jjjj|jjg| j}| j	j
 }	t| j||||||	dS )N)rw   r   r   rk   rl   rI  )r
   r%   r   r`   rO  r   r   target_fieldr*   r=   r_   rP  r	   sql_create_fk)
rC   r*   r   r  rw   r   r   rk   rl   rI  r   r   r   r     s&   
z'BaseDatabaseSchemaEditor._create_fk_sqlc                    s<    fdd}t |jj|jgt|jjjjd |jjg||S )Nc                           j| i |S r-   r`   r  r  rH   r   r   create_fk_name     zDBaseDatabaseSchemaEditor._fk_constraint_name.<locals>.create_fk_namer(  )r   r%   r   r   r   r  r*   )rC   r*   r   r  r   r   rH   r   rO    s   z,BaseDatabaseSchemaEditor._fk_constraint_namec                 C      |  | j||S r-   )r/  sql_delete_fkrC   r*   r   r   r   r   r]       z'BaseDatabaseSchemaEditor._delete_fk_sqlc                 C   s,   |d u rdS |t jkrdS |t jkrdS d S )NrR   z DEFERRABLE INITIALLY DEFERREDz DEFERRABLE INITIALLY IMMEDIATE)r   DEFERRED	IMMEDIATE)rC   rI  r   r   r   _deferrable_constraint_sql  s   

z3BaseDatabaseSchemaEditor._deferrable_constraint_sqlc                 C   s   |du rdS |du rdS dS )NFz NULLS NOT DISTINCTTz NULLS DISTINCTrR   r   )rC   nulls_distinctr   r   r    _unique_index_nulls_distinct_sql  s
   z9BaseDatabaseSchemaEditor._unique_index_nulls_distinct_sqlc                 C   sR   | s| j jjo(| p| j jjo(| p| j jjo(| p| j jjo(|d u p(| j jjS r-   )r=   r@   supports_partial_indexes&supports_deferrable_unique_constraintsr  r   *supports_nulls_distinct_unique_constraints)rC   r  rI  r  r  r	  r   r   r   _unique_supported  s   	
z*BaseDatabaseSchemaEditor._unique_supportedc
              
      s    j |||||	dsd S |s|s|s|s|	d ur/ j||||||||	d}
|
r- j|
 d S  jd fdd|D  |d } j ||d S )Nr  rI  r  r  r	  )r   r  r  r  r  r	  ru   c                    s   g | ]}  |jqS r   )r`   r   re   rH   r   r   rf   +  rg   z8BaseDatabaseSchemaEditor._unique_sql.<locals>.<listcomp>)r   rI  r   rs   )	r  rz   rE   rY   sql_unique_constraintr   r  sql_constraintr`   )rC   r*   r   r   r  rI  r  r  r  r	  rN   rs   r   rH   r   _unique_sql  sL   
z$BaseDatabaseSchemaEditor._unique_sqlc
                 C   s   | j |||||	dsd S t|ddj| jd}
|jj}dd |D }|d u r/| j||dd}n| |}|s<|s<|s<|r@| j}n| j	}|rO| j
||d	|d
}nt|||
| j}t|t|| j||| || || ||| |	dS )Nr  Fr  rh   c                 S   r  r   r  re   r   r   r   rf   L  r$  z?BaseDatabaseSchemaEditor._create_unique_sql.<locals>.<listcomp>Tr&  r   )r  r  )rw   r   r   r  rI  r  r	  )r  r   r   r=   r%   r   r,  r`   sql_create_unique_indexsql_create_uniquer  r   r\   r	   r
   r  r  r  r
  )rC   r*   r   r   r  rI  r  r  r  r	  r   rw   r   rN   r   r   r   rz   3  sD   


z+BaseDatabaseSchemaEditor._create_unique_sqlc                    s&   |r	 fdd}n j }t||d|S )Nc                     r  r-   r  r  rH   r   r   create_unique_namei  r  zLBaseDatabaseSchemaEditor._unique_constraint_name.<locals>.create_unique_name_uniq)r  r   )rC   rw   r   r'  r  r   rH   r   r,  f  s   z0BaseDatabaseSchemaEditor._unique_constraint_namec	           
      C   sD   | j |||||dsd S |s|s|s|r| j}	n| j}	| |	||S )Nr  )r  r  r  r/  )
rC   r*   r   r  rI  r  r  r  r	  rN   r   r   r   r  q  s   z+BaseDatabaseSchemaEditor._delete_unique_sqlc                 C   s    | j | || jd|i d S )Nri   r  )r  r`   r~   )rC   r   ri   r   r   r   
_check_sql  s   z#BaseDatabaseSchemaEditor._check_sqlc                 C   s2   | j jjsd S t| jt|jj| j| ||dS )N)rw   r   ri   )	r=   r@    supports_table_check_constraintsr	   sql_create_checkr
   r%   r   r`   )rC   r*   r   ri   r   r   r   r    s   
z*BaseDatabaseSchemaEditor._create_check_sqlc                 C   s   | j jjsd S | | j||S r-   )r=   r@   r  r/  sql_delete_checkr  r   r   r   r    s   
z*BaseDatabaseSchemaEditor._delete_check_sqlc                 C   s    t |t|jj| j| |dS )Nr  )r	   r
   r%   r   r`   )rC   templater*   r   r   r   r   r/    s
   z/BaseDatabaseSchemaEditor._delete_constraint_sqlc
                    s  |dur fdd|D } j  }
 j j|
|jj}W d   n1 s'w   Y  g }| D ]Y\}}|du s@||d kr|durK|d |krKq2|durV|d |krVq2|dura|d |kraq2|durl|d |krlq2|duru|d	 suq2|dur|d
 |krq2|	r||	vr|| q2|S )z@Return all constraint names matching the columns and conditions.Nc                    s@   g | ]} j jjr j jt| j j n j j|qS r   )r=   r@   truncates_namesintrospectionidentifier_converterr   r_   r  )r0   r   rH   r   r   rf     s    
z>BaseDatabaseSchemaEditor._constraint_names.<locals>.<listcomp>r   r   r   r  ri   r\  r   )r=   r]   r  get_constraintsr%   r   itemsrY   )rC   r*   r  r   r   r  r\  ri   rw  r%  r]   r   resultr   infodictr   rH   r   r*    s8   


z*BaseDatabaseSchemaEditor._constraint_namesc                    s"    j dd fdd|D i S )Nr   ru   c                 3   r   r-   r`   )r0   r   rH   r   r   r3     r   z>BaseDatabaseSchemaEditor._pk_constraint_sql.<locals>.<genexpr>)sql_pk_constraintr   )rC   r   r   rH   r   r     s   z+BaseDatabaseSchemaEditor._pk_constraint_sqlc                 C   sV   | j |dd}|rt|dkrtdt||jjf |D ]}| | || qd S )NTr  r(  z0Found wrong number (%s) of PK constraints for %s)r*  r-  r.  r%   r   rI   _delete_primary_key_sql)rC   r*   rf  r3  r  r   r   r   r    s   z,BaseDatabaseSchemaEditor._delete_primary_keyc              
   C   sJ   t | jt|jj| j| | j|jj|jgddt|jj|jg| jdS )N_pkr{  )rw   r   r   )	r	   sql_create_pkr
   r%   r   r`   r  r   r   r  r   r   r   r    s   z0BaseDatabaseSchemaEditor._create_primary_key_sqlc                 C   r  r-   )r/  sql_delete_pkr  r   r   r   r&    r  z0BaseDatabaseSchemaEditor._delete_primary_key_sqlc                 C   s   |r	d|  | S dS )NzCOLLATE rR   r$  )rC   r   r  r  r   r   r   r     r  z%BaseDatabaseSchemaEditor._collate_sqlc                 C   s*   | j | |d|d }| | d S )N,)	procedureparam_types)sql_delete_procedurer`   r   rI   )rC   procedure_namer,  rN   r   r   r   remove_procedure  s
   z)BaseDatabaseSchemaEditor.remove_procedure)FT)r   )F)rR   r-   )NNNNN)NNNNNN)NNNNNNN)T)NNNNNNNNr   )x__name__
__module____qualname____doc__r   r9  rD  r   rQ  rS  r  r  r  r  r  r  r^  r  r  r  r~   sql_delete_constraintr  r%  r  r  r  r  r  r   rN  r  r  r  r  r  r(  r)  r-  r?  r  rD   rG   rJ   rI   r`   r   r   r|   r   r   r   r   r   r   staticmethodr   r   r\   r   r   r  r  r	  r  r  r  r   r  r=  r   rG  rZ  rb  rn  ru  re  r  rR  r  r  r   r   rd  r  r  r  r  r  r  r  r  r   rT  rc  r  r~  r  r  r   rO  r]  r  r
  r  r  rz   r,  r  r  r  r  r/  r*  r   r  r  r&  r   r/  r   r   r   r   r<   M   sX   
	
	e
C		
#	
"
X
 S
  r
'
.
 
 
6
	%	

8

3



/

r<   )$loggingr'   r   django.confr   django.core.exceptionsr   !django.db.backends.ddl_referencesr   r   r   r   r	   r
   django.db.backends.utilsr   r   r   django.db.modelsr   r   !django.db.models.fields.compositer   django.db.models.sqlr   django.db.transactionr   r   django.utilsr   	getLoggerrU   r   r+   r7   r<   r   r   r   r   <module>   s"     
