v^@sddlZddlmZddlmZmZmZmZmZddlm Z m Z ddl m Z ddl mZmZddlmZejdZd d Zd d Zd dZGdddZdS)N)datetime)ColumnsForeignKeyName IndexName StatementTable) names_digestsplit_identifier)Index)TransactionManagementErroratomic)timezonezdjango.db.backends.schemacCsE|j}|jrdS|jr5|jdgkr5dS|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_fieldrr@/tmp/pip-build-8lau8j11/django/django/db/backends/base/schema.py_is_relevant_relations   rcCs|jjddddddS)NZforwardFreverseTZinclude_hidden)_metaZ _get_fields)modelrrr_all_related_fieldssrcsEtfddtjDfddtjDS)Nc3s$|]}t|r|VqdS)N)r).0obj) old_fieldrr 'sz+_related_non_m2m_objects..c3s$|]}t|r|VqdS)N)r)rr) new_fieldrrr(s)ziprr)rrr)rrr_related_non_m2m_objects#src@s eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZeZdZeZdZdZdZeZdZdZdZ dZ!eZ"dZ#dddd Z$d!d"Z%d#d$Z&fd%d&Z'd'd(Z(d)d*Z)dd+d,Z*d-d.Z+d/d0Z,d1d2Z-e.d3d4Z/d5d6Z0d7d8Z1d9d:Z2d;d<Z3d=d>Z4d?d@Z5dAdBZ6dCdDZ7dEdFZ8dGdHZ9dIdJZ:dKdLZ;dMdNZ<dOdPZ=dQdRZ>ddSdTZ?ddUdVZ@dWdXZAddYdZZBd[d\ZCd]d^ZDd_d`daZEddbdcZFdddded_dfd_dgddhfdiddjfdkddldmZGddndoZHdpdqZIdrdsZJdtduZKdvdwZLdxdyZMdzd{ZNd|d}ZOd~dZPddZQddZRdddZSddddZTdddZUddZVddZWddZXddZYddddddddddZZdddZ[ddZ\ddZ]fddZ^dS)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 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 NULLzUNIQUE (%(columns)s)zCHECK (%(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)zBALTER TABLE %(table)s ADD CONSTRAINT %(name)s UNIQUE (%(columns)s)z|ALTER TABLE %(table)s ADD CONSTRAINT %(name)s FOREIGN KEY (%(column)s) REFERENCES %(to_table)s (%(to_column)s)%(deferrable)sNzFCREATE INDEX %(name)s ON %(table)s (%(columns)s)%(extra)s%(condition)szDCREATE UNIQUE INDEX %(name)s ON %(table)s (%(columns)s)%(condition)szDROP INDEX %(name)szGALTER TABLE %(table)s ADD CONSTRAINT %(name)s PRIMARY KEY (%(columns)s)zDROP PROCEDURE %(procedure)sFTcCs@||_||_|jr$g|_|jjjo6||_dS)N) connection collect_sql collected_sqlfeaturescan_rollback_ddlatomic_migration)selfr!r"r rrr__init__`s     z!BaseDatabaseSchemaEditor.__init__cCs8g|_|jr4t|jj|_|jj|S)N) deferred_sqlr&r r!alias __enter__)r'rrrr+is    z"BaseDatabaseSchemaEditor.__enter__cCsP|dkr-x|jD]}|j|qW|jrL|jj|||dS)N)r)executer&r __exit__)r'exc_type exc_value tracebacksqlrrrr-ps   z!BaseDatabaseSchemaEditor.__exit__c Cs|j r2|jjr2|jjj r2tdt|}tjd||dd|d|i|jr|j drdnd}|dk r|j j |t t |j||q|j j ||n)|jj}|j||WdQRXdS) 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)extraparamsr1;N)r"r!Zin_atomic_blockr$r%r strloggerdebugendswithr#appendtuplemap quote_valuecursorr,)r'r1r3Zendingr>rrrr,ys&  %  -z BaseDatabaseSchemaEditor.executecCs|jjj|S)N)r!ops quote_name)r'rrrrr@sz#BaseDatabaseSchemaEditor.quote_namec sxIjjD];}fdd|D}jjj|q Wg}g}xjjD]}j|\}}|dkrqe|jdj} | dr|dj | 7}|j dj} | r|d| 7}|j ||j r|j r|j jjj} |j jjj|j jj} jr|djdj| d j| i7}n.jjjrjjj|d |jd j|j|f|jdkrejjjjj|j} | rejj | qeWfddjjD}jdjjjddjdd||Di}jjrjjjjj}|r|d|7}||fS)z-Take a model and return its table definition.cs%g|]}jj|jqSr)r get_fieldcolumn)rr)rrr s z6BaseDatabaseSchemaEditor.table_sql..Nr!check z %sto_table to_columnz_fk_%(to_table)s_%(to_column)sz%s %s AutoField BigAutoFieldSmallAutoFieldcs"g|]}|jqSr)Zconstraint_sql)r constraint)rr'rrrCs table definitionz, css|]}|r|VqdS)Nr)rrKrrrrsz5BaseDatabaseSchemaEditor.table_sql..)rHrIrJ) rZunique_togetherr)r:_create_unique_sql local_fields column_sql db_parametersr!sql_check_constraintZdb_type_suffixextend remote_field db_constraintrdb_tablerA field_namerBsql_create_inline_fkr@r$supports_foreign_keys_create_fk_sqlget_internal_typer? autoinc_sql constraintssql_create_tablejoin db_tablespacetablespace_sql)r'rfieldscolumnsZ column_sqlsr3rrMZ extra_params db_paramsZcol_type_suffixrFrGr\r]r1rar)rr'r table_sqlsR    !    !") z"BaseDatabaseSchemaEditor.table_sqlc Cs|jd|j}|d}g}|dkr5d S|j}|oQ|j| }|r|j|}d|j|} |dk r|jjjr|| |j|7}n|| 7}||g7}|j r|j r|jjj rd}|r|jjj r|d7}n|s#|d7}|j r9|d7}n|j rL|d 7}|jp^|jj} | r|jjjr|j r|d |jjj| d d7}||fS) z Take a field and return its column definition. The field must already have had set_attributes_from_name() called. r!typeNz DEFAULT Tz NULLz NOT NULLz PRIMARY KEYz UNIQUEz %sinline)NN)rQr!null skip_defaulteffective_default_column_default_sqlr$requires_literal_defaultsprepare_defaultempty_strings_allowedr!interprets_empty_strings_as_nullsZimplied_column_nulluniquer`rZsupports_tablespacesr?ra) r'rrinclude_defaultrdr1r3rh default_valueZcolumn_defaultZ tablespacerrrrPs<            #z#BaseDatabaseSchemaEditor.column_sqlcCsdS)z Some backends don't accept default values for certain columns types (i.e. MySQL longtext and longblob). Fr)r'rrrrrisz%BaseDatabaseSchemaEditor.skip_defaultcCstddS)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() methodN)NotImplementedError)r'valuerrrrmsz(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. z%sr)r'rrrrrksz,BaseDatabaseSchemaEditor._column_default_sqlcCs|jr|j}n|j ra|jra|jra|jdkrUt}qt}nt|ddst|ddrt j }|j}|dkr|j }q|dkr|j }q|dkrt j }nd}|S)NZ BinaryFieldZauto_nowFZ auto_now_addZ DateFieldZ TimeFieldZ DateTimeField) has_defaultZ get_defaultrhZblankrnr[bytesr6getattrrnowdatetimer )rdefaultZ internal_typerrr_effective_defaults"   $     z+BaseDatabaseSchemaEditor._effective_defaultcCs|j|j||jS)z2Return a field's effective database default value.)Zget_db_prep_saver|r!)r'rrrrrj-sz*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). N)rs)r'rtrrrr=1sz$BaseDatabaseSchemaEditor.quote_valuecCs|j|\}}|j||p'd|jj|j|x9|jjD]+}|jjjj rQ|j |jjqQWdS)zr Create a table and any accompanying indexes or unique constraints for the given `model`. N) rer,r)rS_model_indexes_sqlrlocal_many_to_manyrTthrough auto_created create_model)r'rr1r3rrrrr=s z%BaseDatabaseSchemaEditor.create_modelcCsx9|jjD]+}|jjjjr |j|jjq W|j|jd|j|jj ixKt |j D]:}t |t ru|j|jj ru|j j|quWdS)z!Delete a model from the database.rLN)rr~rTrr delete_modelr,sql_delete_tabler@rVlistr) isinstancerZreferences_tableremove)r'rrr1rrrrNs $z%BaseDatabaseSchemaEditor.delete_modelcCs#|j|j||dddS)zAdd an index on a model.r3N)r, create_sql)r'rindexrrr add_index^sz"BaseDatabaseSchemaEditor.add_indexcCs|j|j||dS)zRemove an index from a model.N)r, remove_sql)r'rrrrr remove_indexbsz%BaseDatabaseSchemaEditor.remove_indexcCs)|j||}|r%|j|dS)zAdd a constraint to a model.N)rr,)r'rrKr1rrradd_constraintfsz'BaseDatabaseSchemaEditor.add_constraintcCs)|j||}|r%|j|dS)z!Remove a constraint from a model.N)rr,)r'rrKr1rrrremove_constraintlsz*BaseDatabaseSchemaEditor.remove_constraintcsdd|D}dd|D}x6|j|D]%}|j|ddi|jq6WxI|j|D]8}fdd|D}|j|j|qoWdS) 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|qSr)r;)rrbrrr xs zABaseDatabaseSchemaEditor.alter_unique_together..cSsh|]}t|qSr)r;)rrbrrrrys rpTcs%g|]}jj|jqSr)rrArB)rr)rrrrCs zBBaseDatabaseSchemaEditor.alter_unique_together..N) difference_delete_composed_indexsql_delete_uniquer,rN)r'rZold_unique_togetherZnew_unique_togetheroldsnewsrbrcr)rralter_unique_togetherrs#z.BaseDatabaseSchemaEditor.alter_unique_togethercsdd|D}dd|D}x6|j|D]%}|j|ddi|jq6WxO|j|D]>}fdd|D}|j|j|dd qoWd 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|qSr)r;)rrbrrrrs z@BaseDatabaseSchemaEditor.alter_index_together..cSsh|]}t|qSr)r;)rrbrrrrs rTcs"g|]}jj|qSr)rrA)rr)rrrrCs zABaseDatabaseSchemaEditor.alter_index_together..suffix_idxN)rrsql_delete_indexr,_create_index_sql)r'rZold_index_togetherZnew_index_togetherrrrb field_namesr)rralter_index_togethers#z-BaseDatabaseSchemaEditor.alter_index_togetherc sddjjD}ddjjD}fdd|D}|j|d||B|}t|dkrtdt|jjd j|f|j|j ||d dS) NcSsh|]}|jqSr)r)rrKrrrrs zBBaseDatabaseSchemaEditor._delete_composed_index..cSsh|]}|jqSr)r)rrKrrrrs cs%g|]}jj|jqSr)rrArB)rr)rrrrCs zCBaseDatabaseSchemaEditor._delete_composed_index..excludez1Found wrong number (%s) of constraints for %s(%s)z, r) rr]indexes_constraint_nameslen ValueErrorrVr_r,_delete_constraint_sql) r'rrbZconstraint_kwargsr1meta_constraint_namesmeta_index_namesrcconstraint_namesr)rrrs   z/BaseDatabaseSchemaEditor._delete_composed_indexcCs||ks3|jjjr7|j|jkr7dS|j|jd|j|d|j|ix0|jD]%}t|t rs|j ||qsWdS)z#Rename the table a model points to.NZ old_tableZ new_table) r!r$Zignores_table_name_caselowerr,sql_rename_tabler@r)rrZrename_table_references)r'rZ old_db_tableZ new_db_tabler1rrralter_db_tables  z'BaseDatabaseSchemaEditor.alter_db_tablec CsK|j|jd|j|jjd|j|d|j|idS)z)Move a model's table between tablespaces.rLZold_tablespaceZnew_tablespaceN)r,sql_retablespace_tabler@rrV)r'rZold_db_tablespaceZnew_db_tablespacerrralter_db_tablespaces z,BaseDatabaseSchemaEditor.alter_db_tablespacec Cs|jr.|jjjjr.|j|jjS|j||dd\}}|dkr\dS|jd|j}|dr|d|j |7}|jr|jj j r|j rd}|j rb|jjjj}|jjjj|jjj}|d|j d|j|||d |j|jd |j|d |j|d |jjji7}n|jj|j||||jd |j|jjd |j|jd|i} |j| ||j| rI|j|dk rI|j|d|dd\} }|jd |j|jjd| i} |j| ||jj |j!|||jj j"r|jj#dS)z Create a field on a model. Usually involves adding a column, but may involve adding a table instead (for M2M fields). rqTNr!rDrEz_fk_%(to_table)s_%(to_column)srrBrFrG deferrablerLrMdropchanges)$rrTrrrrrPrQr!rRr$rYrUsql_create_column_inline_fkrrVrArWrB_fk_constraint_namer@r?deferrable_sqlr)r:rZsql_create_columnr,rirj_alter_column_default_sqlsql_alter_columnrS_field_indexes_sqlconnection_persists_old_columnsclose) r'rrrMr3rdZconstraint_suffixrFrGr1 changes_sqlrrr add_fieldsD  ! !  %! z"BaseDatabaseSchemaEditor.add_fieldcCsW|jr.|jjjjr.|j|jjS|jd|jddkrQdS|jr|j||j gdd}x'|D]}|j |j ||qW|j d|j |jjd|j |j i}|j ||jjjr|jjxQt|jD]@}t|tr|j|jj|j r|jj|qWdS)z Remove a field from a model. Usually involves deleting a column, but for M2Ms may involve deleting a table. r!rfN foreign_keyTrLrB)rrTrrrrrQr!rrBr,_delete_fk_sqlsql_delete_columnr@rVr$rrrr)rrZreferences_columnr)r'rrfk_namesfk_namer1rrr remove_fields"    *z%BaseDatabaseSchemaEditor.remove_fieldc Cs|jd|j}|d}|jd|j}|d}|dkrY|jdkst|dkr|jdkrtd||fn|dkr|dkr|jjr|jjr|jjjjr|jjjjr|j||||S|dkrQ|dkrQ|jjrQ|jjrQ|jjjj rQ|jjjj rQdS|dksi|dkrtd||f|j||||||||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. r!rfNzqCannot 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)) rQr!rTrrrr_alter_many_to_many _alter_field) r'rrrstrict old_db_paramsold_type new_db_paramsnew_typerrr alter_fields2   z$BaseDatabaseSchemaEditor.alter_fieldc ' Cs t} |jr|jr|j||jgdd} |ryt| dkrytdt| |jj|jfx:| D]2} | j |jf|j |j || qW|j r|j s|j ||rdd|jjD} |j||jgdddd d | } |r^t| dkr^td t| |jj|jfx'| D]}|j |j||qeW|jr|js|j o|j o||k}|r/xnt||D]]\}}|j|j|jjgdd}x*|D]"} |j |j |j| qWqW|jr|j r|j sU|j rd d|jjD}|j||jgd ddtjd |}x'|D]}|j |j||qW|d|dkr|drdd|jjD} |j||jgddd | } |r`t| dkr`tdt| |jj|jfx'| D]}|j |j||qgW|j|jkr|j |j|jj|||x?|jD]4}t|tr|j|jj|j|jqWg}g}g}||krY|j||||\}}|j ||j!||j"|}|j"|}|j#o|j# o||ko|dk o|j$| }|r|j |j%||||j#|j#kr|j&|||}|r|j ||j'o*|j#o*|j# }|s9|rv|sI||}|j(j)j*r|rt+t,|\}} dj-|t.| gfg}xF|D]>\}} |j |j/d|j0|jjd|i| qW|rv|j |j1d|j0|jjd|j0|jddi|gxF|D]>\}} |j |j/d|j0|jjd|i| q4W|rx$|D]\}} |j || qW|jr|j r|j2|||j3||r|j |j4||jg|j s |j r9|jr9|j r9|j |j5||gg}!|r[|!j!t|||j ||r|j |j6|||!j!t||x|!D]\}"}|jj7d|j(}#|#d}$|j|j|"j|j|$\}}|j |j/d|j0|jjjd|di|dx$|D]\}} |j || q<WqW|jr| s|j s|j r|jr|j |j8||d|r xT|j9jj:D]C}%t;|%|r|%jjr|j |j8|%j|%jdqW|d|dkrg |drg |j<|jj|jgdd}|j |j=|||d|r |j%|||d d\}&} |j/d|j0|jjd|&i}|j || |j(j)j>r |j(j?dS)!z3Perform a "physical" (non-ManyToMany) field update.rTrz.rprFrz7Found wrong number (%s) of unique constraints for %s.%scSsh|]}|jqSr)r)rrrrrrts rtype_rDcSsh|]}|jqSr)r)rrKrrrrs z6Found wrong number (%s) of check constraints for %s.%sNz, rLrrBr{z%sr!rfrz_fk_%(to_table)s_%(to_column)sZ_fkr_checkr)@setrTrUrrBrrrrVaddr,rrp_field_became_primary_keyr]_delete_unique_sqlrrZ related_modelrdb_indexrr r_delete_index_sql_delete_check_sql_rename_field_sqlr)rrZrename_column_references_alter_column_type_sqlr:rSrjrhrir_alter_column_null_sqlrur!r$Zsupports_combined_altersr;rr_sumrr@sql_update_with_default_delete_primary_key_unique_should_be_addedrNr_create_primary_key_sqlrQrZrZrelated_objectsr_create_index_name_create_check_sqlrr)'r'rrrrrrrrZ fks_droppedrrrrconstraint_nameZdrop_foreign_keysZ_old_relZnew_relZ rel_fk_namesrZ index_names index_namer1ZactionsZ null_actionsZ post_actionsfragmentZ other_actionsZ old_default new_defaultZneeds_database_defaultZfour_way_default_alterationr3Zrels_to_updateZold_relZ rel_db_paramsZrel_typerelrrrrr7s*    %      $ &      %#           !       & !   &$ ! z%BaseDatabaseSchemaEditor._alter_fieldcCs|jjjr%|jdkr%dS|jd|j}|jrL|jn|j}|d|j|j d|digfSdS)z Hook to specialize column null alteration. Return a (sql, params) fragment to set a column to null or non-null as required by new_field, or None if no changes are required. CharField TextFieldNr!rBrf)rr) r!r$ror[rQrhsql_alter_column_nullsql_alter_column_not_nullr@rB)r'rrrrr1rrrr+sz/BaseDatabaseSchemaEditor._alter_column_null_sqlc Cs|j|}|j|}|g}|r6g}n$|jjjrZ|j|}g}|jd|j}|r~|jn|j} | d|j |j d|dd|i|fS)z Hook to specialize column default alteration. Return a (sql, params) fragment to add or drop (depending on the drop argument) a default to new_field's column. r!rBrfr{) rjrkr!r$rlrmrQsql_alter_column_no_defaultsql_alter_column_defaultr@rB) r'rrrrrr{r3rr1rrrrAs    z2BaseDatabaseSchemaEditor._alter_column_default_sqlcCs/|jd|j|jd|igfgfS)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. rBrf)sql_alter_column_typer@rB)r'rrrrrrrr`s  z/BaseDatabaseSchemaEditor._alter_column_type_sqlcCs|jjjj|jjjjkrU|j|jj|jjjj|jjjj|j|jj|jjjj|j|jjjj|j|j|jj|jjjj|j|jjjj|jdS)z*Alter M2Ms to repoint their to= endpoints.N) rTrrrVrrrAZm2m_reverse_field_nameZm2m_field_name)r'rrrrrrrrus$  z,BaseDatabaseSchemaEditor._alter_many_to_manyr5c Cs%t|\}}dt||dd|f}|jjjpFd}d|dj||f}t||kr{|St||dkr|d|d}|t|d d }d|d|dj|d||f}|d dks |d jr!d |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_%s_NrrzD%s)r rr!r?Zmax_name_lengthr_risdigit) r'Z table_name column_namesrrZhash_suffix_part max_lengthrZ other_lengthrrrrs    z+BaseDatabaseSchemaEditor._create_index_namecCsz|dkrSt|dkr;|djr;|dj}n|jjrS|jj}|dk rvd|jjj|SdS)NrrrEr5)rr`rr!r?ra)r'rrbr`rrr_get_index_tablespace_sqls    z2BaseDatabaseSchemaEditor._get_index_tablespace_sqlrrusingr` col_suffixesr1 opclasses conditioncsj||d|} dd|D} |p7j} |jj}fdd}t| dt|jdt|| ||d|d j|| || d | d | rd | nd S)z Return the SQL statement to create the index for one or several fields. `sql` can be specified if the syntax differs from the standard (GIS indexes, ...). r`cSsg|]}|jqSr)rB)rrrrrrCs z>BaseDatabaseSchemaEditor._create_index_sql..cs+dkrj||jS)N)rr@)argskwargs)rr'rrcreate_index_names zEBaseDatabaseSchemaEditor._create_index_sql..create_index_namerLrrrcr2rz WHERE r5) rsql_create_indexrrVrrr@r_index_columns)r'rrbrrrr`rr1rrrarcrrLrr)rr'rrs z*BaseDatabaseSchemaEditor._create_index_sqlcCs:t|p|jdt|jj|jd|j|S)NrLr)rrrrrVr@)r'rrr1rrrrsz*BaseDatabaseSchemaEditor._delete_index_sqlcCst|||jd|S)Nr)rr@)r'rLrcrrrrrrsz'BaseDatabaseSchemaEditor._index_columnscsjj s%jjs%jjr)gSg}x-jjD]}|j|j|q<WxLjjD]>}fdd|D}|j|j |ddqlWx-jj D]}|j|j |qW|S)z Return a list of all index SQL statements (field indexes, index_together, Meta.indexes) for the specified model. cs"g|]}jj|qSr)rrA)rr)rrrrCs z?BaseDatabaseSchemaEditor._model_indexes_sql..rr) rZmanagedproxyZswappedrOrSrZindex_togetherr:rrr)r'routputrrrbrr)rrr}s%#z+BaseDatabaseSchemaEditor._model_indexes_sqlcCs8g}|j||r4|j|j||g|S)zT Return a list of all index SQL statements for the specified field. )_field_should_be_indexedr:r)r'rrrrrrrsz+BaseDatabaseSchemaEditor._field_indexes_sqlcCs|jo|j S)N)rrp)r'rrrrrrsz1BaseDatabaseSchemaEditor._field_should_be_indexedcCs|j o|jS)N)r)r'rrrrrrsz2BaseDatabaseSchemaEditor._field_became_primary_keycCs-|j r|jp,|jo,|j o,|jS)N)rpr)r'rrrrrrsz0BaseDatabaseSchemaEditor._unique_should_be_addedc CsD|jd|j|d|j|jd|j|jd|iS)NrLZ old_columnZ new_columnrf)sql_rename_columnr@rB)r'rLrrrrrrrs z*BaseDatabaseSchemaEditor._rename_field_sqlc Cst|jj|j}|j|||}t|jj|jg|j}t|jjjj|j}t|jjjj|jjg|j}|j j j } t |j d|d|d|d|d|d| S)NrLrrBrFrGr)rrrVr@rrrB target_fieldrr!r?rr sql_create_fk) r'rrrrLrrBrFrGrrrrrZs!* z'BaseDatabaseSchemaEditor._create_fk_sqlcsVfdd}t|jj|jgt|jjjjd|jjg||S)Ncsjj||S)N)r@r)rr)r'rrcreate_fk_nameszDBaseDatabaseSchemaEditor._fk_constraint_name..create_fk_namer)rrrVrBr rr)r'rrrrr)r'rrs   z,BaseDatabaseSchemaEditor._fk_constraint_namecCs|j|j||S)N)r sql_delete_fk)r'rrrrrr%sz'BaseDatabaseSchemaEditor._delete_fk_sqlcCs|r>|j||d|d|}|r:|jj|dS|jddjt|j|i}|jd|j|d|iS)Nrrrcz, rK)rNr)r:sql_unique_constraintr_r<r@sql_constraint)r'rrbrrr1rKrrr _unique_sql(s"z$BaseDatabaseSchemaEditor._unique_sqlc sfdd}t|jjj}|dkrTt|jj|d|}nj|}t||j}|rjjjrt j d|d|d|dd|SdSt j d|d|d|SdS) Ncsjj||S)N)r@r)rr)r'rrcreate_unique_name9szGBaseDatabaseSchemaEditor._create_unique_sql..create_unique_nameZ_uniqrLrrcrz WHERE ) rrrVr@rrr!r$supports_partial_indexesrsql_create_unique_indexsql_create_unique)r'rrcrrrrLr)r'rrN8s < z+BaseDatabaseSchemaEditor._create_unique_sqlcCsE|r/|jjjr+|j|j||SdS|j|j||S)N)r!r$rrrr)r'rrrrrrrRs)z+BaseDatabaseSchemaEditor._delete_unique_sqlcCs-|jd|j|d|jd|iiS)NrrKrD)rr@rR)r'rrDrrr _check_sqlZsz#BaseDatabaseSchemaEditor._check_sqlcCs:t|jdt|jj|jd|j|d|S)NrLrrD)rsql_create_checkrrrVr@)r'rrrDrrrr`s  z*BaseDatabaseSchemaEditor._create_check_sqlcCs|j|j||S)N)rsql_delete_check)r'rrrrrrhsz*BaseDatabaseSchemaEditor._delete_check_sqlcCs1t|dt|jj|jd|j|S)NrLr)rrrrVr@)r'templaterrrrrrksz/BaseDatabaseSchemaEditor._delete_constraint_sqlc  st|dk r%fdd|D}jj%} jjj| |jj} WdQRXg} x | jD]\} }|dks||dkro|dk r|d|krqo|dk r|d|krqo|dk r|d|krqo|dk r|d|krqo|dk r-|d  r-qo|dk rL|d |krLqo| s_| | kro| j| qoW| S) z@Return all constraint names matching the columns and conditions.Ncs%g|]}jjj|qSr)r! introspectionZidentifier_converter)rr)r'rrrCxs z>BaseDatabaseSchemaEditor._constraint_names..rcrprrrDrrf)r!r>r Zget_constraintsrrVitemsr:)r'rrrprrrrDrrr>r]resultrZinfodictr)r'rrrs.  %z*BaseDatabaseSchemaEditor._constraint_namescCs}|j|dd}|rOt|dkrOtdt||jjfx'|D]}|j|j||qVWdS)NrTrz0Found wrong number (%s) of PK constraints for %s)rrrrrVr,_delete_primary_key_sql)r'rrrrrrrrs  z,BaseDatabaseSchemaEditor._delete_primary_keyc Csst|jdt|jj|jd|j|j|jj|jgdddt|jj|jg|jS)NrLrrZ_pkrc) r sql_create_pkrrrVr@rrBr)r'rrrrrrs  'z0BaseDatabaseSchemaEditor._create_primary_key_sqlcCs|j|j||S)N)r sql_delete_pk)r'rrrrrrsz0BaseDatabaseSchemaEditor._delete_primary_key_sqlcCs<|jd|j|ddj|i}|j|dS)NZ procedure param_types,)sql_delete_procedurer@r_r,)r'Zprocedure_namerr1rrrremove_proceduresz)BaseDatabaseSchemaEditor.remove_procedure)___name__ __module__ __qualname____doc__r^rrrrrrrrrrrrrrrRZsql_delete_constraintrr r rrrrXrrrrrrrrr(r+r-r,r@rerPrirmrk staticmethodr|rjr=rrrrrrrrrrrrrrrrrrrrrrrrr}rrrrrrZrrrrNrrrrrrrrrrrrrrr ,s     ;1               7 *                  r )loggingrZ!django.db.backends.ddl_referencesrrrrrZdjango.db.backends.utilsrr Zdjango.db.modelsr Zdjango.db.transactionr r Z django.utilsr getLoggerr7rrrr rrrrs (