U d@sddlZddlZddlmZddlmZddlmZmZmZm Z m Z m Z ddl m Z mZddlmZmZddlmZddlmZmZdd lmZed Zd d Zd dZddZGdddZdS)N)datetime)settings)Columns ExpressionsForeignKeyName IndexName StatementTable) names_digestsplit_identifier) DeferrableIndex)Query)TransactionManagementErroratomic)timezonezdjango.db.backends.schemacCs2|j}|jrdS|jr&|jdgkr&dS|j|jkS)zz When altering the given field, must constraints on its model from the given relation be temporarily dropped? FNT)field many_to_many primary_keyZ to_fieldsname)ZrelationZ altered_fieldrrB/tmp/pip-unpacked-wheel-7vj2equ7/django/db/backends/base/schema.py_is_relevant_relations rcCs"t|jjdddddtddS)NFT)ZforwardreverseZinclude_hiddenZinclude_parentsr)key)sorted_metaZ _get_fieldsoperator attrgettermodelrrr_all_related_fields'sr!c#sbtfddtjDfddtjD}|D]&\}}||fVt|j|jEdHq6dS)Nc3s|]}t|r|VqdSNr.0obj) old_fieldrr 8s z+_related_non_m2m_objects..c3s|]}t|r|VqdSr"r#r$) new_fieldrrr(=s )zipr!r _related_non_m2m_objects remote_field)r'r)Zrelated_fieldsold_relnew_relr)r)r'rr+4s    r+c @seZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d ZeZd ZdZdZdZdZdZdZdZdZeZdZeZdZdZdZeZdZ dZ!dZ"dZ#dZ$eZ%dZ&dd!d"Z'd#d$Z(d%d&Z)dd(d)Z*d*d+Z+d,d-Z,d.d/Z-dd0d1Z.d2d3Z/d4d5Z0d6d7Z1d8d9Z2e3d:d;Z4dd?Z6d@dAZ7dBdCZ8dDdEZ9dFdGZ:dHdIZ;dJdKZdPdQZ?dRdSZ@dTdUZAdVdWZBdXdYZCdZd[ZDdd\d]ZEdd^d_ZFd`daZGddbdcZHdddeZIdfdgZJdhdiZKddkdlZLddmdnZMdodpZNdqdrZOdddjdjdd'dd'dddds dtduZPddvdwZQdxdyZRdzd{ZSd|d}ZTd~dZUddZVddZWddZXddZYddZZddZ[ddZ\ddZ]ddZ^dddZ_dddZ`dddZadddZbddZcddZdddZeddZfdddZgdddZhddZiddZjddZkdddZldS)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)szDROP 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)sz/ALTER INDEX %(old_name)s RENAME TO %(new_name)szDROP INDEX %(name)szGALTER TABLE %(table)s ADD CONSTRAINT %(name)s PRIMARY KEY (%(columns)s)zDROP PROCEDURE %(procedure)sFTcCs,||_||_|jrg|_|jjjo$||_dSr") connection collect_sql collected_sqlfeaturescan_rollback_ddlatomic_migration)selfr0r1rrrr__init__s z!BaseDatabaseSchemaEditor.__init__cCs(g|_|jr$t|jj|_|j|Sr") deferred_sqlr5rr0alias __enter__r6rrrr:s  z"BaseDatabaseSchemaEditor.__enter__cCs8|dkr|jD]}||q|jr4|j|||dSr")r8executer5r__exit__)r6exc_type exc_value tracebacksqlrrrr=s   z!BaseDatabaseSchemaEditor.__exit__rc Cs|js |jjr |jjjs tdt|}tjd||||dd|jr| drXdnd}|dk r|j |t t|j||q|j ||n"|j}|||W5QRXdS)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))paramsrA)extra;N)r1r0Zin_atomic_blockr3r4rstrloggerdebugrstripendswithr2appendtuplemap quote_valuecursorr<)r6rArBZendingrOrrrr<s2 z BaseDatabaseSchemaEditor.executecCs|jj|Sr")r0ops quote_name)r6rrrrrQsz#BaseDatabaseSchemaEditor.quote_namecsjjD]*}fdd|D}j|qg}g}jjD]<}|\}}|dkrdqD|jjd} | dr|dj | 7}|j jd} | r|d| 7}| ||j r6|j r6|j jjj} |j jj|j jj} jr|dj| | d7}n"jjjr6j|d |d |j|f|d krDjjjj|j} | rDj | qDfd djjD}jjjd dd||Dd}jjrjjjj}|r|d|7}||fS)z-Take a model and return its table definition.csg|]}j|qSrr get_fieldr%rrrr sz6BaseDatabaseSchemaEditor.table_sql..Nr0check z %s)to_table to_column_fk_%(to_table)s_%(to_column)sz%s %s)Z AutoFieldZ BigAutoFieldZSmallAutoFieldcsg|]}|qSr)Zconstraint_sqlr% constraintr r6rrrU s, css|]}|rt|VqdSr")rFr\rrrr(sz5BaseDatabaseSchemaEditor.table_sql..)table definition) rZunique_togetherr8rK_create_unique_sql local_fields column_sql db_parametersr0sql_check_constraintdb_type_suffixextendr, db_constraintr db_tablerS field_namecolumnsql_create_inline_fkrQr3supports_foreign_keys_create_fk_sqlget_internal_typerP autoinc_sql constraintssql_create_tablejoin db_tablespacetablespace_sql)r6r field_namesfieldsZ column_sqlsrBrraZ extra_params db_paramscol_type_suffixrYrZrqrrrArvrr^r table_sqls             z"BaseDatabaseSchemaEditor.table_sqlc cs&|V|d}r ||V|j}|oD|| oD|oB|| }|r||} | dk rd||} |jjj r| | | Vn| V| | |j r|j s|jjjrd}|sdVn|jjjsdV|j rdVn |jrdV|jp|jj} | r"|jjjr"|jr"|jjj| ddVdS) N collationzDEFAULT TzNOT NULLZNULLz PRIMARY KEYZUNIQUE)inline)get _collate_sqlnull skip_defaultskip_default_on_altereffective_default_column_default_sqlr0r3requires_literal_defaultsprepare_defaultrKempty_strings_allowedr!interprets_empty_strings_as_nullsZimplied_column_nulluniquerurZsupports_tablespacesrPrv) r6column_db_typerBr rfield_db_paramsinclude_defaultr|r default_valueZcolumn_defaultZ tablespacerrr_iter_column_sqlsP       z)BaseDatabaseSchemaEditor._iter_column_sqlc CsD|j|jd}|d}|dkr"dSg}d||||||||fS)z Return the column definition for a field. The field must already have had set_attributes_from_name() called. rVtypeN)NNrX)rer0rtr)r6r rrrrrBrrrrdVs" z#BaseDatabaseSchemaEditor.column_sqlcCsdS)z Some backends don't accept default values for certain columns types (i.e. MySQL longtext and longblob). Frr6rrrrrqsz%BaseDatabaseSchemaEditor.skip_defaultcCsdS)z Some backends don't accept default values for certain columns types (i.e. MySQL longtext and longblob) in the ALTER COLUMN statement. Frrrrrrxsz.BaseDatabaseSchemaEditor.skip_default_on_altercCs tddS)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() methodNNotImplementedErrorr6valuerrrrsz(BaseDatabaseSchemaEditor.prepare_defaultcCsdS)z Return the SQL to use in a DEFAULT clause. The resulting string should contain a '%s' placeholder for a default value. %srrrrrrsz,BaseDatabaseSchemaEditor._column_default_sqlcCs|r|}n|js<|jr<|jr<|dkr6d}qd}nbt|ddsTt|ddr|}|dkrnt}qt }|dkr| }q|d kr| }nd}|S) NZ BinaryFieldrEZauto_nowFZ auto_now_addZ DateTimeFieldZ DateFieldZ TimeField) has_default get_defaultrZblankrrpgetattrrnowrdatetime)rdefaultZ internal_typerrr_effective_defaults"     z+BaseDatabaseSchemaEditor._effective_defaultcCs||||jS)z2Return a field's effective database default value.)Zget_db_prep_saverr0rrrrrsz*BaseDatabaseSchemaEditor.effective_defaultcCs tdS)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). NrrrrrrNsz$BaseDatabaseSchemaEditor.quote_valuecCs\||\}}|||pd|j|||jjD]}|jjjj r8| |jjq8dS)zr Create a table and any accompanying indexes or unique constraints for the given `model`. N) r{r<r8rh_model_indexes_sqlrlocal_many_to_manyr,through auto_created create_model)r6r rArBrrrrrs   z%BaseDatabaseSchemaEditor.create_modelcCs~|jjD]}|jjjjr||jjq||jd||jj it |j D](}t |t rP||jj rP|j |qPdS)z!Delete a model from the database.r`N)rrr,rr delete_modelr<sql_delete_tablerQrjlistr8 isinstancerZreferences_tableremove)r6r rrArrrrs   z%BaseDatabaseSchemaEditor.delete_modelcCs.|jr|jjjsdS|j|||dddS)zAdd an index on a model.NrB)contains_expressionsr0r3supports_expression_indexesr< create_sqlr6r indexrrr add_indexs z"BaseDatabaseSchemaEditor.add_indexcCs*|jr|jjjsdS||||dS)zRemove an index from a model.N)rr0r3rr< remove_sqlrrrr remove_indexs z%BaseDatabaseSchemaEditor.remove_indexcCsD|jjjr(|j|||j|jddn||||||dS)Nr)r0r3Zcan_rename_indexr<_rename_index_sqlrrr)r6r Z old_indexZ new_indexrrr rename_indexs  z%BaseDatabaseSchemaEditor.rename_indexcCs"|||}|r|j|dddS)zAdd a constraint to a model.Nr)rr<r6r r]rArrradd_constraints z'BaseDatabaseSchemaEditor.add_constraintcCs|||}|r||dS)z!Remove a constraint from a model.N)rr<rrrrremove_constraints z*BaseDatabaseSchemaEditor.remove_constraintcs|dd|D}dd|D}||D]}||ddd|jq&||D](}fdd|D}|||qNd S) z Deal with a model changing its unique_together. The input unique_togethers must be doubly-nested, not the single-nested ["foo", "bar"] format. cSsh|] }t|qSrrLr%rxrrr szABaseDatabaseSchemaEditor.alter_unique_together..cSsh|] }t|qSrrrrrrrsTF)rrcsg|]}j|qSrrRrTrrrrUszBBaseDatabaseSchemaEditor.alter_unique_together..N) difference_delete_composed_indexsql_delete_uniquer<rb)r6r Zold_unique_togetherZnew_unique_togetheroldsnewsrxrwrrralter_unique_together sz.BaseDatabaseSchemaEditor.alter_unique_togethercsdd|D}dd|D}||D]}||ddd|jq&||D],}fdd|D}||j|d d qNd S) z Deal with a model changing its index_together. The input index_togethers must be doubly-nested, not the single-nested ["foo", "bar"] format. cSsh|] }t|qSrrrrrrr&sz@BaseDatabaseSchemaEditor.alter_index_together..cSsh|] }t|qSrrrrrrr'sTF)rrcsg|]}j|qSrrRrTrrrrU2szABaseDatabaseSchemaEditor.alter_index_together.._idxrxsuffixN)rrsql_delete_indexr<_create_index_sql)r6r Zold_index_togetherZnew_index_togetherrrrxrwrrralter_index_together sz-BaseDatabaseSchemaEditor.alter_index_togetherc sddjjD}ddjjD}fdd|D}|j|fd||Bi|}|ddkr|r|jjjrt|j jj |d d } | |kr| g}t |d krt d t |jj d |f||||ddS)NcSsh|] }|jqSrrr\rrrr6szBBaseDatabaseSchemaEditor._delete_composed_index..cSsh|] }|jqSrrr\rrrr9scsg|]}j|jqSr)rrSrlrTrrrrU:szCBaseDatabaseSchemaEditor._delete_composed_index..excluderTFquotez1Found wrong number (%s) of constraints for %s(%s)r_r)rrrindexes_constraint_namesr~r0r3Z*allows_multiple_constraints_on_same_fieldsrF_unique_constraint_namerjlen ValueErrorrtr<_delete_constraint_sql) r6r rxZconstraint_kwargsrAmeta_constraint_namesmeta_index_namescolumnsconstraint_names default_namerrrr5sD  z/BaseDatabaseSchemaEditor._delete_composed_indexcCsn||ks"|jjjr&||kr&dS||j||||d|jD]}t|t rN| ||qNdS)z#Rename the table a model points to.N)Z old_tableZ new_table) r0r3Zignores_table_name_caselowerr<sql_rename_tablerQr8rrZrename_table_references)r6r Z old_db_tableZ new_db_tablerArrralter_db_tableWs   z'BaseDatabaseSchemaEditor.alter_db_tablecCs2||j||jj||||ddS)z)Move a model's table between tablespaces.)r`Zold_tablespaceZnew_tablespaceN)r<sql_retablespace_tablerQrrj)r6r Zold_db_tablespaceZnew_db_tablespacerrralter_db_tablespacejs z,BaseDatabaseSchemaEditor.alter_db_tablespacec Cs |jr |jjjjr ||jjS|j||dd\}}|dkr@dS|j|jd}r`|d|7}|j |jd}|dr|d|j |7}|jrT|jj j rT|j rTd}|jr>|jjjj}|jjj|jjj} t|jj\} } |d|j||||| r d|| nd ||j|||| |jjd 7}n|j|||||j||jj||j|d } || |||s||dk r|j |d|dd \} }|j!||jj| d } || ||j"|#|||jj j$r|j%dS)z Create a field on a model. Usually involves adding a column, but may involve adding a table instead (for M2M fields). T)rNrVrXrWr[z%s.rE)r namespacerlrYrZ deferrable)r`rlradropr`changes)&rr,rrrrrdrgr0rerfr3rnrisql_create_column_inline_fkr rjrSrkrlr _fk_constraint_namerQrPdeferrable_sqlr8rKrosql_create_columnr<rr_alter_column_default_sqlsql_alter_columnrh_field_indexes_sqlconnection_persists_old_columnsclose)r6r rrarBrzryZconstraint_suffixrYrZr_rA changes_sqlrrr add_fieldusx               z"BaseDatabaseSchemaEditor.add_fieldcCs|jr |jjjjr ||jjS|j|jdddkr:dS|jrp|j||j gdd}|D]}| | ||qX|j | |jj| |j d}| ||jjjr|jt|jD],}t|tr||jj|j r|j|qdS)z Remove a field from a model. Usually involves deleting a column, but for M2Ms may involve deleting a table. rVrNT foreign_key)r`rl)rr,rrrrrer0rrlr<_delete_fk_sqlsql_delete_columnrQrjr3rrrr8rrZreferences_columnr)r6r rfk_namesfk_namerArrr remove_fields*     z%BaseDatabaseSchemaEditor.remove_fieldc Cs6|||sdS|j|jd}|d}|j|jd}|d}|dkrN|jdks`|dkrr|jdkrrtd||fn|dkr|dkr|jjr|jjr|jjjjr|jjjjr|||||S|dkr|dkr|jjr|jjr|jjjjs|jjjjsdS|dks |dkrtd||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. NrVrzqCannot 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)) _field_should_be_alteredrer0r,rrrr_alter_many_to_many _alter_field) r6r r'r)strict old_db_paramsold_type new_db_paramsnew_typerrr alter_fieldst       z$BaseDatabaseSchemaEditor.alter_fieldc /Cst} |jjjr|jr|jr|j||jgdd} |r\t| dkr\t dt| |j j |jf| D]$} | |jf| ||| q`|jr|jr|||rdd|j jD} |j||jgdd| d} |rt| dkrt d t| |j j |jf| D]}| |||q|d }|d }|jjjod|jrD|jsT|jod|jod||kpd||k}|rt||D]B\}}|j|j|jjgdd}|D]} | ||j| qqv|jr$|js$|jr|jr$d d|j jD}|j||jgdtj|d }|D]}| |||q |d |d kr|d rdd|j jD} |j||jgd| d} |rt| dkrt dt| |j j |jf| D]}| |||q|j|jkr| ||j j ||||jD](}t |t!r|"|j j |j|jqg}g}g}|j#|jd}|j#|jd}||kr\|$||||}|%|n:||f||fkr|&||||\}}|%||'|d} |j(r|j(s|)|}!|)|}"|*|s|!|"kr|"dk rd} |%|+||||j(|j(kr"|,|||}|r"|%||-o:|j(o:|j( }#|sH|r&|#sV||}|jjj.r|rt/t0|\}}$d1|t2|$gfg}|D],\}}$| |j3|4|j j |d|$q|#r&| |j5|4|j j |4|jdd|"g|D],\}}$| |j3|4|j j |d|$q|rH|D]\}}$| ||$q0|jrd|jsd|6|||7||r| |8||g|jr|jr|jr|js| |j9||gdg}%|r|%'t|||||r| |:|||%'t|||%D]\}&}|jj;|jd}'|'d}(|'d })|&jj;|jd}*|*d }+|+|)krr|$|j|j|(|)}g}n|&|j|&j|j|(\}}| |j3|4|jj j |dd|d|D]\}}$| ||$qq |jjjr|jr| s|jr|js|jr| |<||d|rV|%D],\},}-|-jjr(| |<|-j|-jdq(|d |d kr|d r|j=|j j |jgdd}| |>|||d | r|j+|||dd\}.}$|j3|4|j j |.d}| ||$|jjj?r|j@dS)z3Perform a "physical" (non-ManyToMany) field update.Trrz.F)rrrz7Found wrong number (%s) of unique constraints for %s.%sr|cSsh|] }|jqSrr)r%rrrrrs)rtype_rrWcSsh|] }|jqSrrr\rrrrs)rWrz6Found wrong number (%s) of check constraints for %s.%srVNr_rr)r`rlrrxrrr[Z_fk_checkrr)Asetr0r3rnr,rirrlrrrrjaddr<rr_field_became_primary_keyrr_delete_unique_sqlr~rr+Z related_modelrdb_indexrr r_delete_index_sql_delete_check_sql_rename_field_sqlr8rrZrename_column_referencesrg_alter_column_collation_sqlrK_alter_column_type_sqlrhrrrr_alter_column_null_sqlrZsupports_combined_altersrLr*rtsumrrQsql_update_with_default_delete_primary_key_unique_should_be_addedrbr_create_primary_key_sqlrero_create_index_name_create_check_sqlrr)/r6r r'r)rrrrrZ fks_droppedrrrrconstraint_nameZ old_collation new_collationZdrop_foreign_keysZ_old_relr.Z rel_fk_namesrZ index_names index_namerAactionsZ null_actionsZ post_actionsZold_type_suffixZnew_type_suffixfragmentZ other_actionsZneeds_database_defaultZ old_default new_defaultZfour_way_default_alterationrBZrels_to_updater-Z rel_db_paramsZrel_typeZ rel_collationZold_rel_db_paramsZold_rel_collationrrelrrrrrsN                                        z%BaseDatabaseSchemaEditor._alter_fieldcCsV|jjjr|jrdS|j|jd}|jr.|jn|j}|||j |ddgfSdS)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. NrVrrlr) r0r3rrrersql_alter_column_nullsql_alter_column_not_nullrQrl)r6r r'r)rrArrrr xs" z/BaseDatabaseSchemaEditor._alter_column_null_sqlc Cs||}||}|g}|r$g}n|jjjr<||}g}|j|jd}|rd|jr\|j} qj|j } n|j } | | |j |d|d|fS)z Hook to specialize column default alteration. Return a (sql, params) fragment to add or drop (depending on the drop argument) a default to new_field's column. rVr)rlrr) rrr0r3rrrer sql_alter_column_no_default_nullsql_alter_column_no_defaultsql_alter_column_defaultrQrl) r6r r'r)rrrrBrrArrrrs,     z2BaseDatabaseSchemaEditor._alter_column_default_sqlcCs |j||j|dgfgfS)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_typerQrl)r6r r'r)rrrrr s  z/BaseDatabaseSchemaEditor._alter_column_type_sqlcCs0|j||j||r"d||nddgfS)NrXrE)rlrr|)sql_alter_column_collaterQrlr)r6r r)rrrrrr s z4BaseDatabaseSchemaEditor._alter_column_collation_sqlcCs|jjjj|jjjjkr:||jj|jjjj|jjjj||jj|jjj||jjj|||jj|jjj||jjj|dS)z*Alter M2Ms to repoint their to= endpoints.N) r,rrrjrrrSZm2m_reverse_field_nameZm2m_field_name)r6r r'r)rrrrrs.       z,BaseDatabaseSchemaEditor._alter_many_to_manyrEc Cst|\}}dt|f|ddi|f}|jjp6d}d|d||f}t||kr\|St||dkr||d|d}|t|d d }d|d|d|d||f}|d dks|d rd |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%slengthz%s_%s_%srNrrzD%s)r r r0rPZmax_name_lengthrtrisdigit) r6Z table_name column_namesrrZhash_suffix_part max_lengthrZ other_lengthrrrrs(   z+BaseDatabaseSchemaEditor._create_index_namecCsf|dkrHt|dkr*|djr*|dj}ntjr8tj}n|jjrH|jj}|dk rbd|jj|SdS)NrrrXrE)rrurZDEFAULT_INDEX_TABLESPACErr0rPrv)r6r rxrurrr_get_index_tablespace_sqls z2BaseDatabaseSchemaEditor._get_index_tablespace_sqlcCs|r d|SdS)Nz WHERE rEr)r6 conditionrrr_index_condition_sql*sz-BaseDatabaseSchemaEditor._index_condition_sqlcCs,|r|jjjsdStdt|jj||jdS)NrEz INCLUDE (%(columns)s))r)r0r3supports_covering_indexesrrrrjrQ)r6r rrrr_index_include_sql/s z+BaseDatabaseSchemaEditor._index_include_sql) rxrrusingru col_suffixesrA opclassesr,include expressionsc  s|pg}| pg} t|ddjjd} j|||d}dd|D}|pLj}|jj}fdd}t|t|j t ||||||r |||| nt || | j || || d S) z Return the SQL statement to create the index for one or several fields or expressions. `sql` can be specified if the syntax differs from the standard (GIS indexes, ...). FZ alias_colsrV)rucSsg|] }|jqSrrlrTrrrrUTsz>BaseDatabaseSchemaEditor._create_index_sql..csdkrj||Sr")rrQargskwargsrr6rrcreate_index_nameXs zEBaseDatabaseSchemaEditor._create_index_sql..create_index_name)r`rr0rrCr,r3)r get_compilerr0r+sql_create_indexrrjrr rQr_index_columnsrrNr-r/)r6r rxrrr0rur1rAr2r,r3r4compilerrvrr=r`r;rr:rr7s4     z*BaseDatabaseSchemaEditor._create_index_sqlcCs&t|p |jt|jj|j||dSN)r`r)rrr rrjrQ)r6r rrArrrrms z*BaseDatabaseSchemaEditor._delete_index_sqlcCs*t|jt|jj|j||||dS)N)r`old_namenew_name)rsql_rename_indexr rrjrQ)r6r rArBrrrrts z*BaseDatabaseSchemaEditor._rename_index_sqlcCst|||j|dS)N)r1)rrQ)r6r`rr1r2rrrr>|sz'BaseDatabaseSchemaEditor._index_columnscsjjrjjsjjrgSg}jjD]}|||q(jjD],}fdd|D}||j |ddqHjj D]&}|j r|j j jr~|||q~|S)z Return a list of all index SQL statements (field indexes, index_together, Meta.indexes) for the specified model. csg|]}j|qSrrRrTrrrrUsz?BaseDatabaseSchemaEditor._model_indexes_sql..rr)rZmanagedproxyZswappedrcrhrZindex_togetherrKrrrr0r3rr)r6r outputrrwrxrrrrrs   z+BaseDatabaseSchemaEditor._model_indexes_sqlcCs*g}|||r&||j||gd|S)zT Return a list of all index SQL statements for the specified field. r)_field_should_be_indexedrKr)r6r rrErrrrs z+BaseDatabaseSchemaEditor._field_indexes_sqlc Cs||\}}}}|\}}}} |jD]} || dq&|jD]} | | dq>||j||jkpz|||f||| fkSr")Z deconstructZ non_db_attrspoprQrl) r6r'r)rold_pathZold_argsZ old_kwargsnew_pathnew_argsZ new_kwargsattrrrrrs  z1BaseDatabaseSchemaEditor._field_should_be_alteredcCs|jo |j Sr")rrr6r rrrrrFsz1BaseDatabaseSchemaEditor._field_should_be_indexedcCs|j o |jSr"rr6r'r)rrrrsz2BaseDatabaseSchemaEditor._field_became_primary_keycCs|j o|jo|j p|jSr")rrrNrrrrs  z0BaseDatabaseSchemaEditor._unique_should_be_addedcCs*|j||||j||j|dS)N)r`Z old_columnZ new_columnr)sql_rename_columnrQrl)r6r`r'r)rrrrrs   z*BaseDatabaseSchemaEditor._rename_field_sqlc Cst|jj|j}||||}t|jj|jg|j}t|jjjj|j}t|jjjj|jjg|j}|j j } t |j |||||| dS)N)r`rrlrYrZr)r rrjrQrrrl target_fieldr r0rPrr sql_create_fk) r6r rrr`rrlrYrZrrrrros&  z'BaseDatabaseSchemaEditor._create_fk_sqlcs<fdd}t|jj|jgt|jjjjd|jjg||S)Ncsj||Sr"rQrr7r;rrcreate_fk_nameszDBaseDatabaseSchemaEditor._fk_constraint_name..create_fk_namer)rrrjrlr rPr )r6r rrrSrr;rrs z,BaseDatabaseSchemaEditor._fk_constraint_namecCs||j||Sr")r sql_delete_fkr6r rrrrrsz'BaseDatabaseSchemaEditor._delete_fk_sqlcCs,|dkr dS|tjkrdS|tjkr(dSdS)NrEz DEFERRABLE INITIALLY DEFERREDz DEFERRABLE INITIALLY IMMEDIATE)r ZDEFERREDZ IMMEDIATE)r6rrrr_deferrable_constraint_sqls   z3BaseDatabaseSchemaEditor._deferrable_constraint_sqlc s|rjjjsdS|s"|s"|s"|rNj|||||||d} | rJj| dSjdfdd|D|d} j || dS)N)rr,r3r2r4r_csg|]}|jqSr)rQrlrTr;rrrUsz8BaseDatabaseSchemaEditor._unique_sql..)rrrr]) r0r3&supports_deferrable_unique_constraintsrbr8rKsql_unique_constraintrtrVsql_constraintrQ) r6r rxrr,rr3r2r4rAr]rr;r _unique_sqls4   z$BaseDatabaseSchemaEditor._unique_sqlc Cs|r|jjjr8|r|jjjr8|r*|jjjr8|r<|jjjs.Trr)r1r2)r`rrr,rr3)r0r3rXsupports_partial_indexesr.rrr<rrjrrQsql_create_unique_indexsql_create_uniquer>rrNrr r-rVr/) r6r rxrr,rr3r2r4r?r`rrArrrrbsZ      z+BaseDatabaseSchemaEditor._create_unique_sqlcs&|rfdd}nj}t||d|S)Ncsj||Sr"rRr7r;rrcreate_unique_namePszLBaseDatabaseSchemaEditor._unique_constraint_name..create_unique_name_uniq)rr)r6r`rrr_rr;rrMsz0BaseDatabaseSchemaEditor._unique_constraint_namec Csh|r|jjjr8|r|jjjr8|r*|jjjr8|r<|jjjsBaseDatabaseSchemaEditor._constraint_names..rrrrrWrr)r0rOreZget_constraintsrrjitemsrK)r6r r)rrrrrWrrrOrrresultrZinfodictrr;rrs6     z*BaseDatabaseSchemaEditor._constraint_namescCsV|j|dd}|r6t|dkr6tdt||jjf|D]}||||q:dS)NTrMrz0Found wrong number (%s) of PK constraints for %s)rrrrrjr<_delete_primary_key_sql)r6r rrrrrrrsz,BaseDatabaseSchemaEditor._delete_primary_keyc CsJt|jt|jj|j||j|jj|jgddt|jj|jg|jdS)NZ_pkr)r`rr) r sql_create_pkr rrjrQrrlrrLrrrrsz0BaseDatabaseSchemaEditor._create_primary_key_sqlcCs||j||Sr")r sql_delete_pkrUrrrrhsz0BaseDatabaseSchemaEditor._delete_primary_key_sqlcCsd||S)NzCOLLATE )rQ)r6r|rrrrsz%BaseDatabaseSchemaEditor._collate_sqlcCs*|j||d|d}||dS)N,)Z procedure param_types)sql_delete_procedurerQrtr<)r6Zprocedure_namerlrArrrremove_procedures z)BaseDatabaseSchemaEditor.remove_procedure)FT)r)F)F)F)F)rE)N)N)NNNNN)NNNNNN)T)NNNNN)NNNNNNNN)F)r)m__name__ __module__ __qualname____doc__rsrrrrrr rrrrrr!rrOr rYrfZsql_delete_constraintrZrbrcr^rrQrmrrTr=r]rCrrirjrmr7r:r=r<rQr{rrdrrrr staticmethodrrrNrrrrrrrrrrrrrrrrr rr r rrr+r-r/rrrr>rrrrFrrrrorrrVr[rbrrrarrrrrrrhrrnrrrrr/Ks.  S7       " J I [ &    6     , 5    )  r/)loggingrrZ django.confrZ!django.db.backends.ddl_referencesrrrrrr Zdjango.db.backends.utilsr r Zdjango.db.modelsr r Zdjango.db.models.sqlrZdjango.db.transactionrrZ django.utilsr getLoggerrGrr!r+r/rrrrs