o h:@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|jdgkrdS|j|jvS)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_fieldsname)relation altered_fieldrrP/var/www/html/env/lib/python3.10/site-packages/django/db/backends/base/schema.py_is_relevant_relations  rcCs"t|jjdddddtddS)NFT)forwardreverseinclude_hiddeninclude_parentsr)key)sorted_meta _get_fieldsoperator attrgettermodelrrr_all_related_fields'sr(c#sdtfddtjDfddtjD}|D]\}}||fVt|j|jEdHqdS)Nc3|] }t|r|VqdSNr.0obj) old_fieldrr 8 z+_related_non_m2m_objects..c3r)r*r+r,) new_fieldrrr0=r1)zipr(r'_related_non_m2m_objects remote_field)r/r2related_fieldsold_relnew_relr)r2r/rr44s     r4c @s4eZdZdZdZdZdZdZdZdZ 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]ZE dd^d_ZFd`daZGddbdcZHdddeZIdfdgZJdhdiZKddkdlZLddmdnZMdodpZNdqdrZOdddjdjdd'dd'dddds dtduZPddvdwZQdxdyZRdzd{ZSd|d}ZTd~dZUddZVddZWddZXddZYddZZddZ[ddZ\ddZ]ddZ^     dddZ_      dddZ`dddZa     dddZbddZcddZdddZeddZf        dddZgdddZhddZiddZjddZkdddZldS)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,||_||_|jr g|_|jjjo||_dSr*) connection collect_sql collected_sqlfeaturescan_rollback_ddlatomic_migration)selfr:r;rrrr__init__s z!BaseDatabaseSchemaEditor.__init__cCs(g|_|jrt|jj|_|j|Sr*) deferred_sqlr?rr:alias __enter__r@rrrrDs  z"BaseDatabaseSchemaEditor.__enter__cCs<|dur|jD]}||q|jr|j|||dSdSr*)rBexecuter?r__exit__)r@exc_type exc_value tracebacksqlrrrrGs   z!BaseDatabaseSchemaEditor.__exit__rcCs|js|jjr|jjjstdt|}tjd||||dd|jrN| dr,dnd}|durD|j |t t|j||dS|j ||dS|j}|||WddS1sewYdS)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))paramsrK)extra;N)r;r:in_atomic_blockr=r>rstrloggerdebugrstripendswithr<appendtuplemap quote_valuecursorrF)r@rKrLendingrZrrrrFs. "z BaseDatabaseSchemaEditor.executecCs|jj|Sr*)r:ops quote_name)r@rrrrr]z#BaseDatabaseSchemaEditor.quote_namecsjjD]}fdd|D}j|qg}g}jjD]}|\}}|dur1q"|jjd} | drE|dj | 7}|j jd} | rT|d| 7}| ||j r|j r|j jjj} |j jj|j jj} jr|dj| | d7}njjjrj|d |d |j|f|d vrjjjj|j} | rj | q"fd djjD}jjjd ddg||RDd}jjrjjjj}|r|d|7}||fS)z-Take a model and return its table definition.cg|]}j|qSrr" get_fieldr-rr&rr z6BaseDatabaseSchemaEditor.table_sql..Nr:check z %s)to_table to_column_fk_%(to_table)s_%(to_column)sz%s %s) AutoField BigAutoFieldSmallAutoFieldcsg|]}|qSr)constraint_sqlr- constraintr'r@rrrc s , css|]}|r|VqdSr*rrorrrr0s  z5BaseDatabaseSchemaEditor.table_sql..)table definition) r"unique_togetherrBrV_create_unique_sql local_fields column_sql db_parametersr:sql_check_constraintdb_type_suffixextendr5 db_constraintr'db_tablera field_namecolumnsql_create_inline_fkr]r=supports_foreign_keys_create_fk_sqlget_internal_typer\ autoinc_sql constraintssql_create_tablejoin db_tablespacetablespace_sql)r@r' field_namesfields column_sqlsrLrrt extra_params db_paramscol_type_suffixrhrirrrKrrrqr table_sqls                 z"BaseDatabaseSchemaEditor.table_sqlc cs.|V|d}r||V|j}|o#|| o#|o"|| }|rL||} | durLd||} |jjj rD| | | Vn| V| | |j rY|j sY|jjjrYd}|s_dVn|jjjsgdV|j rndVn|jrtdV|jpz|jj} | r|jjjr|jr|jjj| ddVdSdSdSdS) N collationzDEFAULT TzNOT NULLNULLz PRIMARY KEYUNIQUE)inline)get _collate_sqlnull skip_defaultskip_default_on_altereffective_default_column_default_sqlr:r=requires_literal_defaultsprepare_defaultrVempty_strings_allowedr!interprets_empty_strings_as_nullsimplied_column_nulluniquerr"supports_tablespacesr\r) r@column_db_typerLr'rfield_db_paramsinclude_defaultrr default_valuecolumn_default tablespacerrr_iter_column_sqlsT       z)BaseDatabaseSchemaEditor._iter_column_sqlc CsD|j|jd}|d}|durdSg}d||||||||fS)z Return the column definition for a field. The field must already have had set_attributes_from_name() called. retypeN)NNrg)ryr:rr)r@r'rrrrrLrrrrxTs" z#BaseDatabaseSchemaEditor.column_sqlcCdS)z Some backends don't accept default values for certain columns types (i.e. MySQL longtext and longblob). Frr@rrrrroz%BaseDatabaseSchemaEditor.skip_defaultcCr)z Some backends don't accept default values for certain columns types (i.e. MySQL longtext and longblob) in the ALTER COLUMN statement. Frrrrrrvrz.BaseDatabaseSchemaEditor.skip_default_on_altercCstd)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() methodNotImplementedErrorr@valuerrrr}sz(BaseDatabaseSchemaEditor.prepare_defaultcCr)z Return the SQL to use in a DEFAULT clause. The resulting string should contain a '%s' placeholder for a default value. %srrrrrrrz,BaseDatabaseSchemaEditor._column_default_sqlcCs|r |}|S|js!|jr!|jr!|dkrd}|Sd}|St|dds-t|ddrS|}|dkr;t}|St }|dkrI| }|S|d krQ| }|Sd}|S) N BinaryFieldrOauto_nowF auto_now_add DateTimeField DateField TimeField) has_default get_defaultrblankrrgetattrrnowrdatetime)rdefault internal_typerrr_effective_defaults.   z+BaseDatabaseSchemaEditor._effective_defaultcCs||||jS)z2Return a field's effective database default value.)get_db_prep_saverr:rrrrrsz*BaseDatabaseSchemaEditor.effective_defaultcCst)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). rrrrrrYsz$BaseDatabaseSchemaEditor.quote_valuecCs\||\}}|||p d|j|||jjD]}|jjjj r+| |jjqdS)zr Create a table and any accompanying indexes or unique constraints for the given `model`. N) rrFrBr|_model_indexes_sqlr"local_many_to_manyr5through auto_created create_model)r@r'rKrLrrrrrs  z%BaseDatabaseSchemaEditor.create_modelcCs~|jjD]}|jjjjr||jjq||jd||jj it |j D]}t |t r<||jj r<|j |q(dS)z!Delete a model from the database.rsN)r"rr5rr delete_modelrFsql_delete_tabler]r~listrB isinstancerreferences_tableremove)r@r'rrKrrrrs"   z%BaseDatabaseSchemaEditor.delete_modelcCs.|jr |jjjs dS|j|||dddS)zAdd an index on a model.NrL)contains_expressionsr:r=supports_expression_indexesrF create_sqlr@r'indexrrr add_indexs z"BaseDatabaseSchemaEditor.add_indexcCs*|jr |jjjs dS||||dS)zRemove an index from a model.N)rr:r=rrF remove_sqlrrrr remove_indexs z%BaseDatabaseSchemaEditor.remove_indexcCsF|jjjr|j|||j|jdddS||||||dS)Nr)r:r=can_rename_indexrF_rename_index_sqlrrr)r@r' old_index new_indexrrr rename_indexs  z%BaseDatabaseSchemaEditor.rename_indexcCs&|||}|r|j|dddSdS)zAdd a constraint to a model.Nr)rrFr@r'rprKrrradd_constraints z'BaseDatabaseSchemaEditor.add_constraintcCs"|||}|r||dSdS)z!Remove a constraint from a model.N)rrFrrrrremove_constraints z*BaseDatabaseSchemaEditor.remove_constraintcs|dd|D}dd|D}||D]}||ddd|jq||D]}fdd|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. cSh|]}t|qSrrWr-rrrr zABaseDatabaseSchemaEditor.alter_unique_together..cSrrrrrrrrrTF)rrcr_rr`rbr&rrrcrdzBBaseDatabaseSchemaEditor.alter_unique_together..N) difference_delete_composed_indexsql_delete_uniquerFrv)r@r'old_unique_togethernew_unique_togetheroldsnewsrrrr&ralter_unique_together sz.BaseDatabaseSchemaEditor.alter_unique_togethercsdd|D}dd|D}||D]}||ddd|jq||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. cSrrrrrrrr$rz@BaseDatabaseSchemaEditor.alter_index_together..cSrrrrrrrr%rTF)rrcr_rr`rbr&rrrc0rdzABaseDatabaseSchemaEditor.alter_index_together.._idxrsuffixN)rrsql_delete_indexrF_create_index_sql)r@r'old_index_togethernew_index_togetherrrrrrr&ralter_index_togethersz-BaseDatabaseSchemaEditor.alter_index_togetherc sddjjD}ddjjD}fdd|D}|j|fd||Bi|}|ddurJ|rJ|jjjrJt|j jj |d d } | |vrJ| g}t |d kr`t d t |jj d |f||||ddS)NcSh|]}|jqSrrrorrrr4zBBaseDatabaseSchemaEditor._delete_composed_index..cSrrrrorrrr7csg|] }j|jqSr)r"rarrbr&rrrc8szCBaseDatabaseSchemaEditor._delete_composed_index..excluderTFquotez1Found wrong number (%s) of constraints for %s(%s)rrr)r"rindexes_constraint_namesrr:r=*allows_multiple_constraints_on_same_fieldsrQ_unique_constraint_namer~len ValueErrorrrF_delete_constraint_sql) r@r'rconstraint_kwargsrKmeta_constraint_namesmeta_index_namescolumnsconstraint_names default_namerr&rr3sB z/BaseDatabaseSchemaEditor._delete_composed_indexcCsn||ks|jjjr||krdS||j||||d|jD] }t|t r4| ||q'dS)z#Rename the table a model points to.N) old_table new_table) r:r=ignores_table_name_caselowerrFsql_rename_tabler]rBrrrename_table_references)r@r' old_db_table new_db_tablerKrrralter_db_tableUs"   z'BaseDatabaseSchemaEditor.alter_db_tablecCs2||j||jj||||ddS)z)Move a model's table between tablespaces.)rsold_tablespacenew_tablespaceN)rFsql_retablespace_tabler]r"r~)r@r'old_db_tablespacenew_db_tablespacerrralter_db_tablespacehs z,BaseDatabaseSchemaEditor.alter_db_tablespacec Cs|jr|jjjjr||jjS|j||dd\}}|dur dS|j|jd}|dr4|d|j |7}|jr|jj j r|j rd}|j r|jjjj}|jjj|jjj}t|jj\} } |d|j ||||| rpd|| nd ||j|||||jjd 7}n |j|||||j||jj||j|d } || |||s||dur|j|d|dd \} }|j ||jj| d } || ||j!|"|||jj j#r|j$dSdS)z Create a field on a model. Usually involves adding a column, but may involve adding a table instead (for M2M fields). T)rNrerfrgrjz%s.rO)r namespacerrhri deferrable)rsrrtdroprschanges)%rr5rr"rrrxryr:rzr=rr}sql_create_column_inline_fkr'r~rarrr _fk_constraint_namer]r\deferrable_sqlrBrVrsql_create_columnrFrr_alter_column_default_sqlsql_alter_columnr|_field_indexes_sqlconnection_persists_old_columnsclose) r@r'rrtrLrconstraint_suffixrhrir_rK changes_sqlrrr add_fieldssr             z"BaseDatabaseSchemaEditor.add_fieldcCs|jr|jjjjr||jjS|j|jdddurdS|jr8|j||j gdd}|D] }| | ||q,|j | |jj| |j d}| ||jjjrX|jt|jD]}t|trs||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. rerNT foreign_key)rsr)rr5rr"rrryr:rrrF_delete_fk_sqlsql_delete_columnr]r~r=r(r)rrBrrreferences_columnr)r@r'rfk_namesfk_namerKrrr remove_fields,       z%BaseDatabaseSchemaEditor.remove_fieldc Cs0|||sdS|j|jd}|d}|j|jd}|d}|dur'|jdus0|dur8|jdur8td||f|dur\|dur\|jjr\|jjr\|jjjjr\|jjjjr\|||||S|durz|durz|jjrz|jjrz|jjjjsz|jjjjszdS|dus|durtd||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. NrerzqCannot 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_alteredryr:r5rrr"r_alter_many_to_many _alter_field) r@r'r/r2strict old_db_paramsold_type new_db_paramsnew_typerrr alter_fieldsf      z$BaseDatabaseSchemaEditor.alter_fieldc /Cst} |jjjrC|jrC|jrC|j||jgdd} |r.t| dkr.t dt| |j j |jf| D]} | |jf| ||| q0|jr|jrO|||rdd|j jD} |j||jgdd| d} |rzt| dkrzt d t| |j j |jf| D] }| |||q||d }|d }|jjjo|jr|jp|jo|jo||kp||k}|rt||D]\}}|j|j|jjgdd}|D] } | ||j| qq|jr|js|jr|jrd d|j jD}|j||jgdtj|d }|D] }| |||q|d |d krN|d rNdd|j jD} |j||jgd| d} |r?t| dkr?t dt| |j j |jf| D] }| |||qA|j|jkrz| ||j j ||||jD]}t |t!rx|"|j j |j|jqeg}g}g}|j#|jd}|j#|jd}||kr|$||||}|%|n||f||fkr|&||||\}}|%||'|d} |j(r|j(s|)|}!|)|}"|*|s|!|"kr|"durd} |%|+||||j(|j(kr|,|||}|r|%||-o|j(o|j( }#|s|r|#s||}|jjj.r:|r:t/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|$qo|r|D] \}}$| ||$q|jr|js|6|||7||r| |8||g|jr|jr|jr|js| |j9||gdg}%|r|%'t|||||r| |:|||%'t|||%D]f\}&}|jj;|jd}'|'d}(|'d })|&jj;|jd}*|*d }+|+|)kr,|$|j|j|(|)}g}n |&|j|&j|j|(\}}| |j3|4|jj j |dd|d|D] \}}$| ||$qQq|jjjr|jr| st|jrt|js|jr| |<||d|r|%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@dSdS)z3Perform a "physical" (non-ManyToMany) field update.Tr.rz.F)rrrz7Found wrong number (%s) of unique constraints for %s.%srcSrrr)r-rrrrr|r)rtype_rrfcSrrrrorrrrr)rfrz6Found wrong number (%s) of check constraints for %s.%sreNrrrr)rsrrrrrrj_fk_checkrr)Asetr:r=rr5r}rrrrr"r~addrFr0r_field_became_primary_keyr_delete_unique_sqlrrr4 related_modelrdb_indexrr r_delete_index_sql_delete_check_sql_rename_field_sqlrBrrrename_column_referencesr{_alter_column_collation_sqlrV_alter_column_type_sqlr|rrrr%_alter_column_null_sqlrsupports_combined_altersrWr3rsumr&r]sql_update_with_default_delete_primary_key_unique_should_be_addedrvr_create_primary_key_sqlryr_create_index_name_create_check_sqlr(r))/r@r'r/r2r;r=r:r<r9 fks_droppedr3r4rr constraint_name old_collation new_collationdrop_foreign_keys_old_relr8 rel_fk_namesr index_names index_namerKactions null_actions post_actionsold_type_suffixnew_type_suffixfragment other_actionsneeds_database_default old_default new_defaultfour_way_default_alterationrLrels_to_updater7 rel_db_paramsrel_type rel_collationold_rel_db_paramsold_rel_collationr+relr,rrrr8s&                                             z%BaseDatabaseSchemaEditor._alter_fieldcCsR|jjjr |jr dS|j|jd}|jr|jn|j}|||j |ddgfS)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. Nrerrr) r:r=rrryrsql_alter_column_nullsql_alter_column_not_nullr]r)r@r'r/r2r<rKrrrrPts" z/BaseDatabaseSchemaEditor._alter_column_null_sqlc Cs||}||}|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. rer)rrr) rrr:r=rrryr sql_alter_column_no_default_nullsql_alter_column_no_defaultsql_alter_column_defaultr]r) r@r'r/r2rrkrrLr<rKrrrr%s,     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. rt)sql_alter_column_typer]r)r@r'r/r2r=rrrrOs z/BaseDatabaseSchemaEditor._alter_column_type_sqlcCs0|j||j||rd||nddgfS)NrgrO)rrr)sql_alter_column_collater]rr)r@r'r2r=r\rrrrNs 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) r5rr"r~rr>ram2m_reverse_field_namem2m_field_name)r@r'r/r2r9rrrr7s,       z,BaseDatabaseSchemaEditor._alter_many_to_manyrOc Cst|\}}dt|g|Rddi|f}|jjpd}d|d||f}t||kr/|St||dkr?|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%slengthz%s_%s_%sr+NrrzD%s)r r r:r\max_name_lengthrrisdigit) r@ table_name column_namesrr+hash_suffix_part max_lengthra other_lengthrrrrWs(   z+BaseDatabaseSchemaEditor._create_index_namecCsf|dur$t|dkr|djr|dj}ntjrtj}n|jjr$|jj}|dur1d|jj|SdS)NrrrgrO)rrrDEFAULT_INDEX_TABLESPACEr"r:r\r)r@r'rrrrr_get_index_tablespace_sqls z2BaseDatabaseSchemaEditor._get_index_tablespace_sqlcCs|rd|SdS)Nz WHERE rOr)r@ conditionrrr_index_condition_sql&sz-BaseDatabaseSchemaEditor._index_condition_sqlcCs,|r|jjjs dStdt|jj||jdS)NrOz INCLUDE (%(columns)s))r )r:r=supports_covering_indexesrrr"r~r])r@r'r rrr_index_include_sql+s z+BaseDatabaseSchemaEditor._index_include_sql) rrrusingr col_suffixesrK opclassesrinclude expressionsc  s|pg}| pg} t|ddjjd} 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_colsre)rcSg|]}|jqSrrrbrrrrcPrz>BaseDatabaseSchemaEditor._create_index_sql..cs"dur j|i|Sr*)rWr]argskwargsrr@rrcreate_index_nameTs zEBaseDatabaseSchemaEditor._create_index_sql..create_index_name)rsrrr rMrr)r get_compilerr:rsql_create_indexr"r~rr r]r_index_columnsrrYrr)r@r'rrrrrrrKrrrrcompilerrr rrsrrrrr3s0     z*BaseDatabaseSchemaEditor._create_index_sqlcCs&t|p|jt|jj|j||dSN)rsr)rrr r"r~r])r@r'rrKrrrrJis z*BaseDatabaseSchemaEditor._delete_index_sqlcCs*t|jt|jj|j||||dS)N)rsold_namenew_name)rsql_rename_indexr r"r~r])r@r'rrrrrrps z*BaseDatabaseSchemaEditor._rename_index_sqlcCst|||j|dS)N)r)rr])r@rsr rrrrrrxsz'BaseDatabaseSchemaEditor._index_columnscsjjr jjs jjrgSg}jjD] }|||qjjD]}fdd|D}||j |ddq$jj 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. cr_rr`rbr&rrrcrdz?BaseDatabaseSchemaEditor._model_indexes_sql..rr)r"managedproxyswappedrwr|r'index_togetherrVrrrr:r=rr)r@r'outputrrrrrr&rr{s    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_indexedrVr)r@r'rrrrrr's z+BaseDatabaseSchemaEditor._field_indexes_sqlc Cs||\}}}}|\}}}} |jD]} || dq|jD]} | | dq||j||jkp=|||f||| fkSr*) deconstruct non_db_attrspopr]r) r@r/r2r+old_pathold_args old_kwargsnew_pathnew_args new_kwargsattrrrrr6s  z1BaseDatabaseSchemaEditor._field_should_be_alteredcCs|jo|j Sr*)rIrr@r'rrrrrr^z1BaseDatabaseSchemaEditor._field_should_be_indexedcCs|j o|jSr*rr@r/r2rrrrFr^z2BaseDatabaseSchemaEditor._field_became_primary_keycCs|j o |jo |j p |jSr*)rrrrrrrUs  z0BaseDatabaseSchemaEditor._unique_should_be_addedcCs*|j||||j||j|dS)N)rs old_column new_columnr)sql_rename_columnr]r)r@rsr/r2r=rrrrLs   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)rsrrrhrir)r r"r~r]r"rr target_fieldr'r:r\r#r sql_create_fk) r@r'rrrsrrrhrirrrrrs&  z'BaseDatabaseSchemaEditor._create_fk_sqlcs<fdd}t|jj|jgt|jjjjd|jjg||S)Ncj|i|Sr*r]rWrrErrcreate_fk_namezDBaseDatabaseSchemaEditor._fk_constraint_name..create_fk_namer)rr"r~rr rr')r@r'rrrrrErr"s z,BaseDatabaseSchemaEditor._fk_constraint_namecC||j||Sr*)r sql_delete_fkr@r'rrrrr0z'BaseDatabaseSchemaEditor._delete_fk_sqlcCs,|durdS|tjkr dS|tjkrdSdS)NrOz DEFERRABLE INITIALLY DEFERREDz DEFERRABLE INITIALLY IMMEDIATE)r DEFERRED IMMEDIATE)r@rrrr_deferrable_constraint_sqls  z3BaseDatabaseSchemaEditor._deferrable_constraint_sqlc s|r jjjs dS|s|s|s|r'j|||||||d} | r%j| dSjdfdd|D|d} j || dS)N)rrrrrrrcsg|]}|jqSr)r]rrbrErrrc rdz8BaseDatabaseSchemaEditor._unique_sql..)r rrrp) r:r=&supports_deferrable_unique_constraintsrvrBrVsql_unique_constraintrrsql_constraintr]) r@r'rrrrrrrrKrprrEr _unique_sqls4  z$BaseDatabaseSchemaEditor._unique_sqlc Cs|r|jjjr|r|jjjr|r|jjjr|r|jjjsdSt|ddj|jd} |jj } dd|D} |durA|j | | dd}n| |}|sN|sN|sN|rR|j } n|j } | ra|j| | d|d } nt| || |j} t| t| |j || |||||||d S) NFrrecSrrrrbrrrrc0rz?BaseDatabaseSchemaEditor._create_unique_sql..Trr)rr)rsrr rrr)r:r=rsupports_partial_indexesrrrrr"r~rr]sql_create_unique_indexsql_create_uniquerrrYrr rrr) r@r'rrrrrrrrrsr rKrrrrvsT     z+BaseDatabaseSchemaEditor._create_unique_sqlcs&|r fdd}nj}t||d|S)Ncrr*rrrErrcreate_unique_nameLrzLBaseDatabaseSchemaEditor._unique_constraint_name..create_unique_name_uniq)rWr)r@rsr rrrrErrIsz0BaseDatabaseSchemaEditor._unique_constraint_namec Csh|r|jjjr|r|jjjr|r|jjjr|r|jjjsdS|s&|s&|s&|r*|j}n|j}||||Sr*) r:r=rrrrrrr) r@r'rrrrrrrKrrrrGTs*  z+BaseDatabaseSchemaEditor._delete_unique_sqlcCs |j|||jd|idS)Nrfr)rr]rz)r@rrfrrr _check_sqlps z#BaseDatabaseSchemaEditor._check_sqlcCs$t|jt|jj|j|||dS)N)rsrrf)rsql_create_checkr r"r~r])r@r'rrfrrrrXvs z*BaseDatabaseSchemaEditor._create_check_sqlcCrr*)rsql_delete_checkrrrrrK~rz*BaseDatabaseSchemaEditor._delete_check_sqlcCs t|t|jj|j||dSr)rr r"r~r])r@templater'rrrrrs z/BaseDatabaseSchemaEditor._delete_constraint_sqlc s|dur fdd|D}j} jj| |jj} Wdn1s'wYg} | D]Y\} }|dus@||dkr|durK|d|krKq2|durV|d|krVq2|dura|d|kraq2|durl|d|krlq2|duru|d suq2|dur|d |krq2| r| | vr| | q2| S) z@Return all constraint names matching the columns and conditions.Ncsg|] }jj|qSr)r: introspectionidentifier_converter)r-rrErrrcs z>BaseDatabaseSchemaEditor._constraint_names..r rrrrfr/r)r:rZrget_constraintsr"r~itemsrV)r@r'rrrrr/rfr?rrZrresultrinfodictrrErrs8    z*BaseDatabaseSchemaEditor._constraint_namescCsV|j|dd}|rt|dkrtdt||jjf|D] }||||qdS)NTrrz0Found wrong number (%s) of PK constraints for %s)rrrr"r~rF_delete_primary_key_sql)r@r'r9r rZrrrrTsz,BaseDatabaseSchemaEditor._delete_primary_keyc CsJt|jt|jj|j||j|jj|jgddt|jj|jg|jdS)N_pkrC)rsrr ) r sql_create_pkr r"r~r]rWrrrrrrrVsz0BaseDatabaseSchemaEditor._create_primary_key_sqlcCrr*)r sql_delete_pkrrrrrrz0BaseDatabaseSchemaEditor._delete_primary_key_sqlcCsd||S)NzCOLLATE )r])r@rrrrrr^z%BaseDatabaseSchemaEditor._collate_sqlcCs*|j||d|d}||dS)N,) procedure param_types)sql_delete_procedurer]rrF)r@procedure_namerrKrrrremove_procedures z)BaseDatabaseSchemaEditor.remove_procedure)FT)r)F)rOr*)NNNNN)NNNNNN)T)NNNNNNNN)m__name__ __module__ __qualname____doc__rrrrr$r&rzrurvryrxrwr{r1rrSrrzsql_delete_constraintrrrrrrrr!rrrrrrrrrArDrGrFr]rrrxrrrr staticmethodrrrYrrrrrrrrrrrrr-r5r>r8rPr%rOrNr7rWrrrrrJrrrr'r6rrFrUrLrr"r0rrrvrrGrrXrKrrrTrVrrrrrrrr9Ks0   Q 7      " H I [ &    6    , 5   )  r9)loggingr$r django.confr!django.db.backends.ddl_referencesrrrrrr django.db.backends.utilsr r django.db.modelsr r django.db.models.sqlrdjango.db.transactionrr django.utilsr getLoggerrRrr(r4r9rrrrs