U R+|jd@sddlZddlZddlmZddlmZddlmZmZmZm Z m Z m Z ddl m Z mZmZddlmZmZddlmZddlmZmZdd lmZed Zd d Zd dZddZGdddZdS)N)datetime)settings)Columns ExpressionsForeignKeyName IndexName StatementTable) names_digestsplit_identifier truncate_name) 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_fieldrrfc:\users\snmko\desktop\web_content\django_cbt\venv\Lib\site-packages\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 ZdZdZdZdZdZdZdZeZdZeZdZdZdZeZdZdZ dZ!dZ"dZ#eZ$dZ%dZ&dZ'dd"d#Z(d$d%Z)d&d'Z*dd)d*Z+d+d,Z,d-d.Z-d/d0Z.dd1d2Z/d3d4Z0d5d6Z1d7d8Z2d9d:Z3e4d;d<Z5d=d>Z6d?d@Z7dAdBZ8dCdDZ9dEdFZ:dGdHZ;dIdJZdOdPZ?dQdRZ@dSdTZAdUdVZBdWdXZCdYdZZDd[d\ZEd]d^ZFdd_d`ZGddadbZHdcddZIddedfZJdgdhZKdidjZLdkdlZMdmdnZNddpdqZOddrdsZPdtduZQdvdwZRdddododd(dd(ddddx dydzZSdd{d|ZTd}d~ZUddZVddZWddZXdddZYddZZddZ[ddZ\ddZ]ddZ^ddZ_ddZ`ddZadddZbdddZcdddZddddZeddZfddZgddZhddZidddZjdddZkddZlddZmdddZndddZodS)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)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)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)sz)COMMENT ON TABLE %(table)s IS %(comment)sz5COMMENT ON COLUMN %(table)s.%(column)s IS %(comment)sFTcCs,||_||_|jrg|_|jjjo$||_dSr#) connection collect_sql collected_sqlfeaturescan_rollback_ddlatomic_migration)selfr1r2rrrr__init__s z!BaseDatabaseSchemaEditor.__init__cCs(g|_|jr$t|jj|_|j|Sr#) deferred_sqlr6rr1alias __enter__r7rrrr;s  z"BaseDatabaseSchemaEditor.__enter__cCs8|dkr|jD]}||q|jr4|j|||dSr#)r9executer6r__exit__)r7exc_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))paramsrB)extra;N)r2r1Zin_atomic_blockr4r5rstrloggerdebugrstripendswithr3appendtuplemap quote_valuecursorr=)r7rBrCendingrPrrrr=s2 z BaseDatabaseSchemaEditor.executecCs|jj|Sr#)r1ops quote_name)r7rrrrrSsz#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&rr rr sz6BaseDatabaseSchemaEditor.table_sql..Nr1check z %s)to_table to_column_fk_%(to_table)s_%(to_column)sz%s %s)Z AutoFieldZ BigAutoFieldZSmallAutoFieldcsg|]}|qSr)Zconstraint_sqlr& constraintr!r7rrrW s, css|]}|rt|VqdSr#)rGr^rrrr)sz5BaseDatabaseSchemaEditor.table_sql..)table definition) rZunique_togetherr9rL_create_unique_sql local_fields column_sql db_parametersr1sql_check_constraintdb_type_suffixextendr- db_constraintr!db_tablerU field_namecolumnsql_create_inline_fkrSr4supports_foreign_keys_create_fk_sqlget_internal_typerR autoinc_sql constraintssql_create_tablejoin db_tablespacetablespace_sql)r7r! field_namesfieldsZ column_sqlsrCrrcZ extra_params db_paramscol_type_suffixr[r\rsrtrBrxrr`r table_sqls             z"BaseDatabaseSchemaEditor.table_sqlc csH|V|d}r ||V|jjjr>|jr>||jV|j}|ob|| ob|o`| | }|r| |} | dk rd| |} |jjj r| | | Vn| V|| |jr|js|jjjrd}|sdVn|jjjsdV|jrdVn|jrdV|jp|jj} | rD|jjjrD|jrD|jjj| ddVdS) N collationzDEFAULT TzNOT NULLNULLz PRIMARY KEYZUNIQUE)inline)get _collate_sqlr1r4supports_comments_inline db_comment _comment_sqlnull skip_defaultskip_default_on_altereffective_default_column_default_sqlrequires_literal_defaultsprepare_defaultrLempty_strings_allowedr!interprets_empty_strings_as_nullsZimplied_column_nulluniquerwrZsupports_tablespacesrRrx) r7column_db_typerCr!rfield_db_paramsinclude_defaultr~r default_valueZcolumn_defaultZ tablespacerrr_iter_column_sql!sT       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. rXtypeN)NNrZ)rgr1rvr)r7r!rrrrrCrrrrfZs" z#BaseDatabaseSchemaEditor.column_sqlcCsdS)z Some backends don't accept default values for certain columns types (i.e. MySQL longtext and longblob). Frr7rrrrrusz%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. Frrrrrr|sz.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() methodNNotImplementedErrorr7valuerrrrsz(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 BinaryFieldrFZauto_nowFZ auto_now_addZ DateTimeFieldZ DateFieldZ TimeField) has_defaultZ get_defaultrZblankrrrgetattrrnowrdatetime)rdefaultZ internal_typerrr_effective_defaults"     z+BaseDatabaseSchemaEditor._effective_defaultcCs||||jS)z2Return a field's effective database default value.)Zget_db_prep_saverr1rrrrrsz*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). NrrrrrrOsz$BaseDatabaseSchemaEditor.quote_valuecCs||\}}|||pd|jjjr|jjrB||d|jj|jjjs|jj D]8}|j rT|j |jd}|d}|j| ||||j qT|j |||jjD]}|jjjjr||jjqdS)zr Create a table and any accompanying indexes or unique constraints for the given `model`. NrXr)r}r=r1r4supports_commentsrZdb_table_commentalter_db_table_commentrrerrg_alter_column_comment_sqlr9rj_model_indexes_sqllocal_many_to_manyr-through auto_created create_model)r7r!rBrCrr field_typerrrrs0     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.rbN)rrr-rr delete_modelr=sql_delete_tablerSrllistr9 isinstancerZreferences_tableremove)r7r!rrBrrrrs   z%BaseDatabaseSchemaEditor.delete_modelcCs.|jr|jjjsdS|j|||dddS)zAdd an index on a model.NrC)contains_expressionsr1r4supports_expression_indexesr= create_sqlr7r!indexrrr add_indexs z"BaseDatabaseSchemaEditor.add_indexcCs*|jr|jjjsdS||||dS)zRemove an index from a model.N)rr1r4rr= remove_sqlrrrr remove_indexs z%BaseDatabaseSchemaEditor.remove_indexcCsD|jjjr(|j|||j|jddn||||||dS)Nr)r1r4Zcan_rename_indexr=_rename_index_sqlrrr)r7r!Z old_indexZ new_indexrrr rename_indexs  z%BaseDatabaseSchemaEditor.rename_indexcCs"|||}|r|j|dddS)zAdd a constraint to a model.Nr)rr=r7r!r_rBrrradd_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|qSrrMr&rzrrr &szABaseDatabaseSchemaEditor.alter_unique_together..cSsh|] }t|qSrrrrrrr'sTF)rrcsg|]}j|qSrrTrVr rrrW2szBBaseDatabaseSchemaEditor.alter_unique_together..N) difference_delete_composed_indexsql_delete_uniquer=rd)r7r!Zold_unique_togetherZnew_unique_togetheroldsnewsrzryrr ralter_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|qSrrTrVr rrrWGszABaseDatabaseSchemaEditor.alter_index_together.._idxrzsuffixN)rrsql_delete_indexr=_create_index_sql)r7r!Zold_index_togetherZnew_index_togetherrrrzryrr ralter_index_together5sz-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^rrrrKszBBaseDatabaseSchemaEditor._delete_composed_index..cSsh|] }|jqSrrr^rrrrNscsg|]}j|jqSr)rrUrnrVr rrrWOszCBaseDatabaseSchemaEditor._delete_composed_index..excluderTFquotez1Found wrong number (%s) of constraints for %s(%s)rar)rrtindexes_constraint_namesrr1r4Z*allows_multiple_constraints_on_same_fieldsrG_unique_constraint_namerllen ValueErrorrvr=_delete_constraint_sql) r7r!rzZconstraint_kwargsrBmeta_constraint_namesmeta_index_namescolumnsconstraint_names default_namerr rrJsD  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) r1r4Zignores_table_name_caselowerr=sql_rename_tablerSr9rrZrename_table_references)r7r!Z old_db_tableZ new_db_tablerBrrralter_db_tablels   z'BaseDatabaseSchemaEditor.alter_db_tablecCs>|jr:|jjjr:||j||jj||p.dddS)NrF)rbcomment) sql_alter_table_commentr1r4rr=rSrrlrO)r7r!Zold_db_table_commentZnew_db_table_commentrrrrs  z/BaseDatabaseSchemaEditor.alter_db_table_commentcCs2||j||jj||||ddS)z)Move a model's table between tablespaces.)rbZold_tablespaceZnew_tablespaceN)r=sql_retablespace_tablerSrrl)r7r!Zold_db_tablespaceZnew_db_tablespacerrralter_db_tablespaces z,BaseDatabaseSchemaEditor.alter_db_tablespacec CsJ|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"r|jj j#r|jj j$s|d}|j|%||||j"|j&|'|||jj j(rF|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)rNrXrZrYr]z%s.rF)r namespacernr[r\ deferrable)rbrnrcdroprbchangesr)*rr-rrrrrfrir1rgrhr4rprksql_create_column_inline_fkr!rlrUrmrnr _fk_constraint_namerSrRdeferrable_sqlr9rLrqsql_create_columnr=rr_alter_column_default_sqlsql_alter_columnrrrrrj_field_indexes_sqlconnection_persists_old_columnsclose)r7r!rrcrCr|r{Zconstraint_suffixr[r\r_rB changes_sqlrrrr add_fields               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. rXrNT foreign_key)rbrn)rr-rrrrrgr1rrnr=_delete_fk_sqlsql_delete_columnrSrlr4rrrr9rrZreferences_columnr)r7r!rfk_namesfk_namerBrrr 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. NrXrzqCannot 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_alteredrgr1r-rrrr_alter_many_to_many _alter_field) r7r!r(r*strict old_db_paramsold_type new_db_paramsnew_typerrr alter_field st       z$BaseDatabaseSchemaEditor.alter_fieldc / Cst} |jjjr|jr|jr|j||dhdr|j||jgdd} |rnt | dkrnt dt | |j j |jf| D]$} | |jf|||| qr|jr,|jr|||r,dd|j jD} |j||jgdd | d } |rt | dkrt d t | |j j |jf| D]}||||q|d }|d }|jjjo||jr\|jsl|jo||jo|||kp|||k}|rt||D]B\}}|j|j|jjgdd}|D]} |||j| qq|jr<|js<|jr|jr|.|||}|r>|(||/oV|j*oV|j* }#|sd|rB|#sr||7}|jjj0r|rt1t2|\}}$d3|t4|$gfg}|D],\}}$||j5|6|j j |d|$q|#rB||j7|6|j j |6|jdd|"g|D],\}}$||j5|6|j j |d|$q|rd|D]\}}$|||$qL|jr|js|8|||9||r||:||g|jr|jr|jr|js||j;||gdg}%|r|%)t|||||r"||<|||%)t|||%D]\}&}|jj=|jd}'|'d}(|'d })|&jj=|jd}*|*d }+|'|j|&j|j|(|+|)\}}||j5|6|jj j |dd|d|D]\}}$|||$qq&|jjjr|jr| s|jr|js|jr||>||d|rR|%D],\},}-|-jjr$||>|-j|-jdq$|d|dkr|dr|j?|j j |jgdd}||@|||d| r|j-|||dd \}.}$|j5|6|j j |.d}|||$|jjjAr|jBdS)!z3Perform a "physical" (non-ManyToMany) field update.r)ignoreTrrz.F)rrrz7Found wrong number (%s) of unique constraints for %s.%sr~cSsh|] }|jqSrr)r&rrrrrs)rtype_rrYcSsh|] }|jqSrrr^rrrrs)rYrz6Found wrong number (%s) of check constraints for %s.%srXNrarr)rbrnrrzrrr]Z_fk_checkrr)Csetr1r4rpr-rkrrrnrrrrladdr=rr_field_became_primary_keyrt_delete_unique_sqlrrr,Z related_modelrdb_indexrrr_delete_index_sql_delete_check_sql_rename_field_sqlr9rrZrename_column_referencesrirr_alter_column_type_sqlrLrjrrrr_alter_column_null_sqlrZsupports_combined_altersrMr+rvsumrrSsql_update_with_default_delete_primary_key_unique_should_be_addedrdr_create_primary_key_sqlrgrq_create_index_name_create_check_sqlrr)/r7r!r(r*rrrrrZ fks_droppedrrrrconstraint_name old_collation new_collationZdrop_foreign_keysZ_old_relr/Z rel_fk_namesrZ index_names index_namerBZactionsZ null_actionsZ post_actionsZold_type_suffixZnew_type_suffixfragment other_actionsZneeds_database_defaultZ old_default new_defaultZfour_way_default_alterationrCZrels_to_updater.Z rel_db_paramsZrel_typeZ rel_collationZold_rel_db_paramsZold_rel_collationrrelrrrrrJsT                                        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. NrXr)rnr) r1r4rrrgrsql_alter_column_nullsql_alter_column_not_nullrSrn)r7r!r(r*rrBrrrrs" 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. rXr)rnrr) rrr1r4rrrgr sql_alter_column_no_default_nullsql_alter_column_no_defaultsql_alter_column_defaultrSrn) r7r!r(r*rr$rrCrrBrrrrs,     z2BaseDatabaseSchemaEditor._alter_column_default_sqlc Csg}||||jj}r&d|}nd}d} |jjjr|js|j|jkrr|||||j\} } | rr| | | f|jr| |j} |j | |j ||| dgf|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. rZrF)rnrr~r)rrrlr1r4rrrrrLrsql_alter_column_typerSrn) r7r!r(r*rrr r#Z collate_sqlZ comment_sqlrBrCrrrrs@      z/BaseDatabaseSchemaEditor._alter_column_type_sqlcCs.|j||jj||j||dgfS)N)rbrnr)sql_alter_column_commentrSrrlrnr)r7r!r*rZnew_db_commentrrrrs  z2BaseDatabaseSchemaEditor._alter_column_comment_sqlcCs||p dS)NrF)rO)r7rrrrr sz%BaseDatabaseSchemaEditor._comment_sqlcCs|jjjj|jjjjkr:||jj|jjjj|jjjj||jj|jjj||jjj|||jj|jjj||jjj|dS)z*Alter M2Ms to repoint their to= endpoints.N) r-rrrlrrrUZm2m_reverse_field_nameZm2m_field_name)r7r!r(r*rrrrr#s.       z,BaseDatabaseSchemaEditor._alter_many_to_manyrFc 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 r1rRmax_name_lengthrvrisdigit) r7 table_name column_namesrrZhash_suffix_part max_lengthr!Z other_lengthrrrrCs(   z+BaseDatabaseSchemaEditor._create_index_namecCsf|dkrHt|dkr*|djr*|dj}ntjr8tj}n|jjrH|jj}|dk rbd|jj|SdS)NrrrZrF)rrwrZDEFAULT_INDEX_TABLESPACErr1rRrx)r7r!rzrwrrr_get_index_tablespace_sqlcs z2BaseDatabaseSchemaEditor._get_index_tablespace_sqlcCs|r d|SdS)Nz WHERE rFr)r7 conditionrrr_index_condition_sqlosz-BaseDatabaseSchemaEditor._index_condition_sqlcCs,|r|jjjsdStdt|jj||jdS)NrFz INCLUDE (%(columns)s))r)r1r4supports_covering_indexesrrrrlrS)r7r!rrrr_index_include_sqlts z+BaseDatabaseSchemaEditor._index_include_sql) rzrrusingrw col_suffixesrB opclassesr9include 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_colsrX)rwcSsg|] }|jqSrrnrVrrrrWsz>BaseDatabaseSchemaEditor._create_index_sql..csdkrj||Sr#)rrSargskwargsrr7rrcreate_index_names zEBaseDatabaseSchemaEditor._create_index_sql..create_index_name)rbrr=rrDr9r@)r get_compilerr1r8sql_create_indexrrlrr rSr_index_columnsrrOr:r<)r7r!rzrrr=rwr>rBr?r9r@rAcompilerrxrrJrbrHrrGrr|s4     z*BaseDatabaseSchemaEditor._create_index_sqlcCs&t|p |jt|jj|j||dSN)rbr)rrr rrlrS)r7r!rrBrrrrs z*BaseDatabaseSchemaEditor._delete_index_sqlcCs*t|jt|jj|j||||dS)N)rbold_namenew_name)rsql_rename_indexr rrlrS)r7r!rNrOrrrrs z*BaseDatabaseSchemaEditor._rename_index_sqlcCst|||j|dS)N)r>)rrS)r7rbrr>r?rrrrKsz'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|qSrrTrVr rrrWsz?BaseDatabaseSchemaEditor._model_indexes_sql..rr)rZmanagedproxyZswappedrerjrZindex_togetherrLrrrr1r4rr)r7r!outputrryrzrrr rrs   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_indexedrLr)r7r!rrRrrrrs z+BaseDatabaseSchemaEditor._field_indexes_sqlc Cs|pt}|\}}}}|\}}} } ||jD]} || dq6||jD]} | | dqT||j||jkp|||f|| | fkSr#)r Z deconstructunionZ non_db_attrspoprSrn) r7r(r*rrold_pathZold_argsZ old_kwargsnew_pathnew_argsZ new_kwargsattrrrrrs z1BaseDatabaseSchemaEditor._field_should_be_alteredcCs|jo |j Sr#)rrr7r!rrrrrSsz1BaseDatabaseSchemaEditor._field_should_be_indexedcCs|j o |jSr#rr7r(r*rrrrsz2BaseDatabaseSchemaEditor._field_became_primary_keycCs|j o|jo|j p|jSr#)rrr\rrrrs  z0BaseDatabaseSchemaEditor._unique_should_be_addedcCs*|j||||j||j|dS)N)rbZ old_columnZ new_columnr)sql_rename_columnrSrn)r7rbr(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)rbrrnr[r\r)r rrlrSrrrn target_fieldr!r1rRrr sql_create_fk) r7r!rrrbrrnr[r\rrrrrq s&  z'BaseDatabaseSchemaEditor._create_fk_sqlcs<fdd}t|jj|jgt|jjjjd|jjg||S)Ncsj||Sr#rSrrDr<rrcreate_fk_name szDBaseDatabaseSchemaEditor._fk_constraint_name..create_fk_namer)rrrlrnr r^r!)r7r!rrrarr<rrs z,BaseDatabaseSchemaEditor._fk_constraint_namecCs||j||Sr#)r sql_delete_fkr7r!rrrrr,sz'BaseDatabaseSchemaEditor._delete_fk_sqlcCs,|dkr dS|tjkrdS|tjkr(dSdS)NrFz DEFERRABLE INITIALLY DEFERREDz DEFERRABLE INITIALLY IMMEDIATE)r ZDEFERREDZ IMMEDIATE)r7rrrr_deferrable_constraint_sql/s   z3BaseDatabaseSchemaEditor._deferrable_constraint_sqlc s|rjjjsdS|s"|s"|s"|rNj|||||||d} | rJj| dSjdfdd|D|d} j || dS)N)rr9r@r?rAracsg|]}|jqSr)rSrnrVr<rrrWWsz8BaseDatabaseSchemaEditor._unique_sql..)rrrr_) r1r4&supports_deferrable_unique_constraintsrdr9rLsql_unique_constraintrvrdsql_constraintrS) r7r!rzrr9rr@r?rArBr_rr<r _unique_sql7s4   z$BaseDatabaseSchemaEditor._unique_sqlc Cs|r|jjjr8|r|jjjr8|r*|jjjr8|r<|jjjs.Trr)r>r?)rbrrr9rr@)r1r4rfsupports_partial_indexesr;rrrIrrlrrSsql_create_unique_indexsql_create_uniquerKrrOrr r:rdr<) r7r!rzrr9rr@r?rArLrbrrBrrrrd_sZ      z+BaseDatabaseSchemaEditor._create_unique_sqlcs&|rfdd}nj}t||d|S)Ncsj||Sr#r`rDr<rrcreate_unique_nameszLBaseDatabaseSchemaEditor._unique_constraint_name..create_unique_name_uniq)rr)r7rbrrrmrr<rrsz0BaseDatabaseSchemaEditor._unique_constraint_namec Csh|r|jjjr8|r|jjjr8|r*|jjjr8|r<|jjjsBaseDatabaseSchemaEditor._constraint_names..rrrrrYrr)r1rPrsZget_constraintsrrlitemsrL)r7r!r6rrrrrYr rrPrtresultrZinfodictrr<rrs6     z*BaseDatabaseSchemaEditor._constraint_namescCsV|j|dd}|r6t|dkr6tdt||jjf|D]}||||q:dS)NTr[rz0Found wrong number (%s) of PK constraints for %s)rrrrrlr=_delete_primary_key_sql)r7r!rrrrrrrsz,BaseDatabaseSchemaEditor._delete_primary_keyc CsJt|jt|jj|j||j|jj|jgddt|jj|jg|jdS)NZ_pkr )rbrr) r sql_create_pkr rrlrSrrnrrZrrrrsz0BaseDatabaseSchemaEditor._create_primary_key_sqlcCs||j||Sr#)r sql_delete_pkrcrrrrvsz0BaseDatabaseSchemaEditor._delete_primary_key_sqlcCs|rd||SdS)NzCOLLATE rF)rS)r7r~rr5rrrrsz%BaseDatabaseSchemaEditor._collate_sqlcCs*|j||d|d}||dS)N,)Z procedure param_types)sql_delete_procedurerSrvr=)r7Zprocedure_namerzrBrrrremove_procedure!s z)BaseDatabaseSchemaEditor.remove_procedure)FT)r)F)F)F)F)rF)N)N)N)NNNNN)NNNNNN)T)NNNNN)NNNNNNNN)F)NN)r)p__name__ __module__ __qualname____doc__rurrrrrr+r&r'r*r)r(rr]rrgrhZsql_delete_constraintrhrprqrlrr_rorrbrJrkrPrrwrxr{rr,r8r;r>r=rSr}rrfrrrr staticmethodrrrOrrrrrrrrrrrrrrrrrrrrrrrrr8r:r<rrrrKrrrrSrrrrqrrrdrirdrrrorrrrrrrvrr|rrrrr0Ks4  S9    $   "  V I \ &.    6     , 5    -   r0) loggingrrZ django.confrZ!django.db.backends.ddl_referencesrrrrrr Zdjango.db.backends.utilsr r r Zdjango.db.modelsr rZdjango.db.models.sqlrZdjango.db.transactionrrZ django.utilsr getLoggerrHrr"r,r0rrrrs