o [hA@sddlZddlZddlmZddlmZddlmZddlmZm Z m Z m Z m Z m Z ddlmZmZmZddlmZmZddlmZdd lmZdd lmZmZdd lmZed Zd dZ ddZ!ddZ"GdddZ#dS)N)datetime)settings) FieldError)Columns ExpressionsForeignKeyName IndexName StatementTable) names_digestsplit_identifier truncate_name) DeferrableIndex)CompositePrimaryKey)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_keyZ to_fieldsname)ZrelationZ altered_fieldrr/home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/django/db/backends/base/schema.py_is_relevant_relations  rcCs"t|jjdddddtddS)NFT)forwardreverseZinclude_hiddenZinclude_parentsr)key)sorted_metaZ _get_fieldsoperator attrgettermodelrrr_all_related_fields)sr%c#sdtfddtjDfddtjD}|D]\}}||fVt|j|jEdHqdS)Nc3|] }t|r|VqdSNr.0obj) old_fieldrr : z+_related_non_m2m_objects..c3r&r'r(r)) new_fieldrrr-?r.)zipr%r$_related_non_m2m_objects remote_field)r,r/Zrelated_fieldsold_relnew_relr)r/r,rr16s     r1c @seZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d ZeZd ZdZdZdZdZdZdZdZdZeZdZeZdZdZdZeZdZ dZ!dZ"dZ#dZ$eZ%dZ&dZ'd Z(dd#d$Z)d%d&Z*d'd(Z+dd*d+Z,d,d-Z-d.d/Z.d0d1Z/dd2d3Z0d4d5Z1d6d7Z2d8d9Z3d:d;Z4dd?Z6e7d@dAZ8dBdCZ9dDdEZ:dFdGZ;dHdIZdNdOZ?dPdQZ@dRdSZAdTdUZBdVdWZCdXdYZDdZd[ZEd\d]ZFd^d_ZGd`daZHdbdcZIddddeZJdfdgZK !ddhdiZLdjdkZMddldmZN !ddndoZOdpdqZPdrdsZQdtduZRdvdwZSddydzZTdd{d|ZUd}d~ZVddZWdddxdxdd)dd)dddd ddZXdddZYddZZddZ[ddZ\ddZ]dddZ^ddZ_ddZ`ddZaddZbddZcddZdddZeddZfddZg     dddZh      dddZi       dddZjdddZk      dddZlddZmddZnddZoddZp        dddZqddZrdddZsddZtddZudddZvdddZwdS)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)szPRIMARY 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)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,||_||_|jr g|_|jjjo||_dSr') connection collect_sql collected_sqlfeaturescan_rollback_ddlatomic_migration)selfr6r7rrrr__init__s z!BaseDatabaseSchemaEditor.__init__cCs(g|_|jrt|jj|_|j|Sr') deferred_sqlr;rr6alias __enter__r<rrrr@s  z"BaseDatabaseSchemaEditor.__enter__cCs>|dur|jD]}||dq|jr|j|||dSdSr')r>executer;r__exit__)r<exc_type exc_value tracebacksqlrrrrCs  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))paramsrG)extra;N)r7r6Zin_atomic_blockr9r:rstrloggerdebugrstripendswithr8appendtuplemap quote_valuecursorrB)r<rGrHendingrUrrrrBs. "z BaseDatabaseSchemaEditor.executecCs|jj|Sr')r6ops quote_name)r<rrrrrXz#BaseDatabaseSchemaEditor.quote_namecsRjjD]}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"g}|rӈjjD] }j|qn| fd d jjDjj}t|tr| |j!j"jjd#dd g||RDd}jj$r%jj%jj$}|r%|d|7}||fS)z-Take a model and return its table definition.cg|]}j|qSrr get_fieldr*rr#rr z6BaseDatabaseSchemaEditor.table_sql..Nr6check z %s)to_table to_column_fk_%(to_table)s_%(to_column)sz%s %s)Z AutoFieldZ BigAutoFieldZSmallAutoFieldc3s|] }|VqdSr')Zconstraint_sqlr* constraintr$r<rrr-s  z5BaseDatabaseSchemaEditor.table_sql.., css|] }|rt|VqdSr')rL)r* statementrrrr-&s )table definition)&r Zunique_togetherr>rQ_create_unique_sql local_fields column_sql db_parametersr6sql_check_constraintdb_type_suffixextendr2 db_constraintr$db_tabler\ field_namecolumnsql_create_inline_fkrXr9supports_foreign_keys_create_fk_sqlget_internal_typerW autoinc_sql constraints create_sqlpk isinstancer_pk_constraint_sqlcolumnssql_create_tablejoin db_tablespacetablespace_sql)r<r$ field_namesfieldsZ column_sqlsrHrrlZ extra_params db_paramscol_type_suffixrcrdr|Zconstraint_sqlsrgrrGrrrhr table_sqls                   z"BaseDatabaseSchemaEditor.table_sqlccs|V|d}r||V|jjjr |jr ||jV|j}|r;| |\} } d| V| | d}|oJ| | oJ|oI| | }|rs| |} | dursd||} |jjjrk| || Vn| V|| |jr|js|jjjrd}|jr||\} }| || Vn|sdVn|jjjsdV|jrdVn|jrdV|jp|jj}|r|jjjr|jr|jjj|dd VdSdSdSdS) N collationzDEFAULT FTzNOT NULLZNULLz PRIMARY KEYZUNIQUE)inline)get _collate_sqlr6r9supports_comments_inline db_comment _comment_sqlnullhas_db_defaultdb_default_sqlrs skip_defaultskip_default_on_altereffective_default_column_default_sqlrequires_literal_defaultsprepare_defaultrQempty_strings_allowedr!interprets_empty_strings_as_nulls generated_column_generated_sqlZimplied_column_nulluniquerr Zsupports_tablespacesrWr)r<column_db_typerHr$rfield_db_paramsinclude_defaultrr default_sqlZdefault_params default_valueZcolumn_default generated_sqlZgenerated_paramsZ tablespacerrr_iter_column_sql6sj          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. r`typeNNNrb)rpr6rr)r<r$rrrrrHrrrroys" z#BaseDatabaseSchemaEditor.column_sqlcCdS)z Some backends don't accept default values for certain columns types (i.e. MySQL longtext and longblob). Frr<rrrrrz%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. Frrrrrrrz.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<valuerrrrsz(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_sqlc sddlm}|j}t||r|nd}t|jd}|jjd}| |\}}jj j r=|t fdd|D;}g}|||fS)z;Return the sql and params for the field's database default.r)Valuez(%s)r#r`c3|]}|VqdSr')rr*prArrr-z:BaseDatabaseSchemaEditor.db_default_sql..) Zdjango.db.models.expressionsrZ_db_default_expressionrrrr$ get_compilerr6compiler9rrR) r<rr db_defaultrGquerycompilerrrHrrArrs    z'BaseDatabaseSchemaEditor.db_default_sqlcsZ|j\}}|jr dnd}jjjr#|tfdd|D}d}d|d||fS)z3Return the SQL to use in a GENERATED ALWAYS clause.ZSTOREDZVIRTUALc3rr'rTrrArrr-rzABaseDatabaseSchemaEditor._column_generated_sql..rzGENERATED ALWAYS AS (z) )rr6 db_persistr9rrR)r<rZexpression_sqlrHZpersistency_sqlrrArrs  z.BaseDatabaseSchemaEditor._column_generated_sqlcCs|r |}|S|jrd}|S|js(|jr(|jr(|dkr$d}|Sd}|St|dds4t|ddrZ|}|dkrBt }|St }|dkrP| }|S|d krX| }|Sd}|S) NZ BinaryFieldrKZauto_nowFZ auto_now_addZ DateTimeFieldZ DateFieldZ TimeField) has_default get_defaultrrblankrr{getattrrnowrdatetime)rdefaultZ internal_typerrr_effective_defaults4   z+BaseDatabaseSchemaEditor._effective_defaultcCs||||jS)z2Return a field's effective database default value.)Zget_db_prep_saverr6rrrrrsz*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). rrrrrrTsz$BaseDatabaseSchemaEditor.quote_valuecCs||\}}|||p d|jjjrG|jjr!||d|jj|jjjsG|jj D]}|j rF|j |jd}|d}|j| ||||j q*|j |||jjD]}|jjjjrc||jjqTdS)zr Create a table and any accompanying indexes or unique constraints for the given `model`. Nr`r)rrBr6r9supports_commentsr Zdb_table_commentalter_db_table_commentrrnrrp_alter_column_comment_sqlr>rs_model_indexes_sqllocal_many_to_manyr2through auto_created create_model)r<r$rGrHrr field_typerrrrs0      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.rkN)r rr2rr delete_modelrBsql_delete_tablerXrulistr>rr Zreferences_tableremove)r<r$rrGrrrrs"   z%BaseDatabaseSchemaEditor.delete_modelcCs.|jr |jjjs dS|j|||dddS)zAdd an index on a model.NrH)contains_expressionsr6r9supports_expression_indexesrBr~r<r$indexrrr add_index0s z"BaseDatabaseSchemaEditor.add_indexcCs*|jr |jjjs dS||||dS)zRemove an index from a model.N)rr6r9rrB remove_sqlrrrr remove_index;s z%BaseDatabaseSchemaEditor.remove_indexcCsF|jjjr|j|||j|jdddS||||||dS)Nr)r6r9Zcan_rename_indexrB_rename_index_sqlrrr)r<r$Z old_indexZ new_indexrrr rename_indexDs  z%BaseDatabaseSchemaEditor.rename_indexcCs&|||}|r|j|dddSdS)zAdd a constraint to a model.Nr)r~rBr<r$rgrGrrradd_constraintNs z'BaseDatabaseSchemaEditor.add_constraintcCs"|||}|r||dSdS)z!Remove a constraint from a model.N)rrBrrrrremove_constraintVs 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|qSrrRr*rrrr bzABaseDatabaseSchemaEditor.alter_unique_together..cSrrrrrrrrcrTF)rrcrZrr[r]r#rrr^nr_zBBaseDatabaseSchemaEditor.alter_unique_together..N) difference_delete_composed_indexsql_delete_uniquerBrm)r<r$Zold_unique_togetherZnew_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. cSrrrrrrrrwrz@BaseDatabaseSchemaEditor.alter_index_together..cSrrrrrrrrxrTF)rrcrZrr[r]r#rrr^r_zABaseDatabaseSchemaEditor.alter_index_together..Z_idx)rsuffixN)rrsql_delete_indexrB_create_index_sql)r<r$Zold_index_togetherZnew_index_togetherrrrrrr#ralter_index_togetherqsz-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|]}|jqSrrrfrrrrzBBaseDatabaseSchemaEditor._delete_composed_index..cSrrrrfrrrrcsg|] }j|jqSr)r r\rwr]r#rrr^szCBaseDatabaseSchemaEditor._delete_composed_index..excluderTFquotez1Found wrong number (%s) of constraints for %s(%s)rir)r r}indexes_constraint_namesrr6r9Z*allows_multiple_constraints_on_same_fieldsrL_unique_constraint_namerulen ValueErrorrrB_delete_constraint_sql) r<r$rZconstraint_kwargsrGmeta_constraint_namesmeta_index_namesrconstraint_names default_namerr#rrsB 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)Z old_tableZ new_table) r6r9Zignores_table_name_caselowerrBsql_rename_tablerXr>rr Zrename_table_references)r<r$Z old_db_tableZ new_db_tablerGrrralter_db_tables"   z'BaseDatabaseSchemaEditor.alter_db_tablecCsF|jr|jjjr!||j||jj||pdddSdSdS)NrK)rkcomment) sql_alter_table_commentr6r9rrBrXr rurT)r<r$Zold_db_table_commentZnew_db_table_commentrrrrs  z/BaseDatabaseSchemaEditor.alter_db_table_commentcCs2||j||jj||||ddS)z)Move a model's table between tablespaces.)rkZold_tablespaceZnew_tablespaceN)rBsql_retablespace_tablerXr ru)r<r$Zold_db_tablespaceZnew_db_tablespacerrralter_db_tablespaces z,BaseDatabaseSchemaEditor.alter_db_tablespacec CsL|jr|jjjjr||jjS|j||dd\}}|dur dS|j|jd}r0|d|7}|j |jd}|drD|d|j |7}|jr|jj j r|j rd}|jr|jjjj}|jjj|jjj} t|jj\} } |d|j||||| rd|| nd ||j|||| |jjd 7}n |j|||||j||jj||j|d } || |pd|s||s| |dur|j!|d|dd \} }|j"||jj| d } || ||j#r |jj j$r |jj j%s |d}|j|&||||j#|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)rNr`rbrarez%s.rK)r namespacerwrcrd deferrable)rkrwrldroprkchangesr)+rr2rr rrrorrr6rprqr9ryrtsql_create_column_inline_fkr$rur\rvrwr _fk_constraint_namerXrWdeferrable_sqlr>rQrzsql_create_columnrBrrr_alter_column_default_sqlsql_alter_columnrrrrrs_field_indexes_sqlconnection_persists_old_columnsclose)r<r$rrlrHrrZconstraint_suffixrcrdr _rG changes_sqlrrrr add_fields             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. r`rNT foreign_key)rkrw)rr2rr rrrpr6rrwrB_delete_fk_sqlsql_delete_columnrXrur9rrrr>rr Zreferences_columnr)r<r$rfk_namesfk_namerGrrr remove_field(s,       z%BaseDatabaseSchemaEditor.remove_fieldc Cs|||sdS|j|jd}|d}|j|jd}|d}d} |dur)|jdus2|dur:|jdur:td||f|dur^|dur^|jjr^|jjr^|jjjjr^|jjjjr^|||||S|dur||dur||jjr||jjr||jjjjs||jjjjs|dS|dus|durtd||f|j |j ks|j r|j |j krd} n!|j rz| |j} Wn t yd} Yn w| |j} | | k} | rtd|d | ||||||||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. Nr`rFzqCannot 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_alteredrpr6r2rrr r_alter_many_to_manyrrrr _alter_field) r<r$r,r/strict old_db_paramsold_type new_db_paramsnew_typeZmodifying_generated_fieldZ old_field_sqlZ new_field_sqlrrr alter_fieldHs           z$BaseDatabaseSchemaEditor.alter_fieldcCsD|jj}||j}d|d<z |||WSty!YdSw)NZ__column_name__rw)r6Zdata_type_check_constraintsZdb_type_parametersr{KeyError)r<rrZcheck_constraintsdatarrr_field_db_checks  z(BaseDatabaseSchemaEditor._field_db_checkc 2 Cspt} |jjjrL|jrL|jrL|j||dhdrL|j||jgdd} |r7t | dkr7t dt | |j j |jf| D]} | |jf|||| q9|jr|jrX|||rdd|j jD} |j||jgdd | d } |rt | dkrt 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|||}|||}||kr^|r^dd|j jD} |j||jgd| d} |rOt | dkrOt dt | |j j |jf| D] }||||qQ|j|jkr|| |j j ||||j!D]}t"|t#r|$|j j |j|jqug}g}g}|j%|jd}|j%|jd}||ks||ks||ks|jjj&r|j'|j'kr|(||||||\} }!|)| |*|!|+r|+r|j,|j,kr|)|-|||n|+r|)|j-|||ddd }"|j.r2|j.s2|+s2|/|}#|/|}$|0|s2|#|$kr2|$dur2d}"|)|1||||j.|j.krH|2|||} | rH|)| |3pP|+oY|j.oY|j. }%|s`|r|%sg||7}|jjj4r|rt5t6|\}}&d7|t8|&gfg}|D]\}}&||j9|:|j j |d|&q|%r|+sd}'|$g}&n|;|\}'}&||j<|:|j j |:|j|'d|&|D]\}}&||j9|:|j j |d|&q|r|D] \}}&|||&q|jr|js|=|||>||r||?||g|jr|jr+|jr+|js+||j@||gdg}(|r8|(*t|||||rP||A|||(*t|||(D]V\})}|jjB|jd}*|*d}+|*d },|)jjB|jd}-|-d }.|(|j|)j|j|+|.|,\} }!||j9|:|jj j | dd| d|!D] \}}&|||&qqR|jjjr|jr| s|jr|js|jr||C||d|r|(D]\}/}0|0jjr||C|0j|0jdq||kr|r|jD|j j |jgdd}||E|||d |"r)|j1|||dd\}1}&|j9|:|j j |1d}|||&|jjjFr6|jGdSdS)!z3Perform a "physical" (non-ManyToMany) field update.r)ignoreTrrz.F)rrrz7Found wrong number (%s) of unique constraints for %s.%srcSrrr)r*rrrrr r)rtype_rcSrrrrfrrrrr)rarz6Found wrong number (%s) of check constraints for %s.%sr`r Nrirr)rkrwrrrrreZ_fk_checkrra)Hsetr6r9ryr2rtr#rrwrrr ruaddrBrr_field_became_primary_keyr}_delete_unique_sqlrrr1Z related_modelrdb_indexrrr_delete_index_sqlr._delete_check_sql_rename_field_sqlr>rr Zrename_column_referencesrrrr_alter_column_type_sqlrQrsrr"_alter_column_database_default_sqlrrrr_alter_column_null_sqlrZsupports_combined_altersrRr0rsumrrXrsql_update_with_default_delete_primary_key_unique_should_be_addedrmr_create_primary_key_sqlrprz_create_index_name_create_check_sqlrr)2r<r$r,r/r(r*r'r)r&Z fks_droppedr r!rrconstraint_name old_collation new_collationZdrop_foreign_keysZ_old_relr4Z rel_fk_namesrZ index_names index_nameZ old_db_checkZ new_db_checkrGactionsZ null_actionsZ post_actionsZold_type_suffixZnew_type_suffixfragment other_actionsZneeds_database_defaultZ old_default new_defaultZfour_way_default_alterationrHrZrels_to_updater3Z rel_db_paramsZrel_typeZ rel_collationZold_rel_db_paramsZold_rel_collationrrelrrrrr%sV                                                    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. Nr`r)rwr) r6r9rrrprsql_alter_column_nullsql_alter_column_not_nullrXrw)r<r$r,r/r)rGrrrr>s" 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. r`rrwrr) rrr6r9rrrpr sql_alter_column_no_default_nullsql_alter_column_no_defaultsql_alter_column_defaultrXrw) r<r$r,r/r rMrrHr)rGrrrr3s,     z2BaseDatabaseSchemaEditor._alter_column_default_sqlc CsV|r |j}d}g}n |j}||\}}|j|jd}|||j|d|d|fS)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. rKr`rrQ)rSrTrrpr6rXrw) r<r$r,r/r rGrrHr)rrrr=Ys  z;BaseDatabaseSchemaEditor._alter_column_database_default_sqlc Csg}||||jj}rd|}nd}d} |jjjrB|jsB|j|jkr9|||||j\} } | r9| | | f|jrB| |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 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. rbrK)rwrrr)rr rur6r9rrrrrQrsql_alter_column_typerXrw) r<r$r,r/r*rGrHrLZ collate_sqlZ comment_sqlrGrHrrrr<us6        z/BaseDatabaseSchemaEditor._alter_column_type_sqlcCs.|j||jj||j||dgfS)N)rkrwr)sql_alter_column_commentrXr rurwr)r<r$r/r*Znew_db_commentrrrrs  z2BaseDatabaseSchemaEditor._alter_column_comment_sqlcCs||pdS)NrKr)r<rrrrrrYz%BaseDatabaseSchemaEditor._comment_sqlcCs|jjjj|jjjjkr||jj|jjjj|jjjj||jj|jjj||jjj|||jj|jjj||jjj|dS)z*Alter M2Ms to repoint their to= endpoints.N) r2rr rurr+r\Zm2m_reverse_field_nameZm2m_field_name)r<r$r,r/r&rrrr$s,       z,BaseDatabaseSchemaEditor._alter_many_to_manyrKc 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_%srNrrzD%s)r r r6rWmax_name_lengthrrisdigit) r< table_name column_namesrrZhash_suffix_part max_lengthrIZ other_lengthrrrrDs(   z+BaseDatabaseSchemaEditor._create_index_namecCsf|dur$t|dkr|djr|dj}ntjrtj}n|jjr$|jj}|dur1d|jj|SdS)NrrrbrK)rrrZDEFAULT_INDEX_TABLESPACEr r6rWr)r<r$rrrrr_get_index_tablespace_sqls z2BaseDatabaseSchemaEditor._get_index_tablespace_sqlcCs|rd|SdS)Nz WHERE rKr)r< conditionrrr_index_condition_sqlsz-BaseDatabaseSchemaEditor._index_condition_sqlcCs,|r|jjjs dStdt|jj||jdS)NrKz INCLUDE (%(columns)s))r)r6r9supports_covering_indexesr rr rurX)r<r$rrrr_index_include_sqls z+BaseDatabaseSchemaEditor._index_include_sql) rrrusingr col_suffixesrG opclassesrcinclude 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, ...). FZ alias_colsr`)rcSg|]}|jqSrrwr]rrrr^'rz>BaseDatabaseSchemaEditor._create_index_sql..cs"dur j|i|Sr')rDrXargskwargsrr<rrcreate_index_name+s zEBaseDatabaseSchemaEditor._create_index_sql..create_index_name)rkrrgrrIrcrj)rrr6rbsql_create_indexr rur r rXr_index_columnsrrTrdrf)r<r$rrrrgrrhrGrircrjrkrrrrtrkrsrrrrr s0     z*BaseDatabaseSchemaEditor._create_index_sqlcCsrt|p|jt|jj|j||d}|jdj}|jd}t|j D]}t |tr6| ||r6|j |q#|S)Nrkrrkr) r rr r rurXpartsrkrr>rZreferences_indexr)r<r$rrGrjr_rIrrrr9@s   z*BaseDatabaseSchemaEditor._delete_index_sqlcCs*t|jt|jj|j||||dS)N)rkold_namenew_name)r sql_rename_indexr r rurX)r<r$rxryrrrrRs z*BaseDatabaseSchemaEditor._rename_index_sqlcCst|||j|dS)N)rh)rrX)r<rkrrhrirrrruZsz'BaseDatabaseSchemaEditor._index_columnscCst|jjr |jjs |jjrgSg}|jjD] }||||q|jjD]}|jr.|j j j r7| | ||q$|S)zz Return a list of all index SQL statements (field indexes, Meta.indexes) for the specified model. )r ZmanagedproxyZswappedrnrsrrrr6r9rrQr~)r<r$outputrrrrrr]s  z+BaseDatabaseSchemaEditor._model_indexes_sqlcCs*g}|||r||j||gd|S)zT Return a list of all index SQL statements for the specified field. r1)_field_should_be_indexedrQr)r<r$rr|rrrrps z+BaseDatabaseSchemaEditor._field_indexes_sqlc Cs |js|jsdS|p t}|\}}}}|\}}} } ||jD]} || dq#||jD]} | | dq2|js\|jr\|jr\|jjj j |jjj j kr\|dd| dd| drz| drz| || |krz|d| d| |j| |jkp|||f|| | fkS)NFtor)Zconcreter4Z deconstructunionZ non_db_attrspoprr2r$r rurrrXrw) r<r,r/r/rold_pathZold_argsZ old_kwargsnew_pathnew_argsZ new_kwargsattrrrrr#ys@        z1BaseDatabaseSchemaEditor._field_should_be_alteredcCs|jo|j Sr')r8rr<r$rrrrr}rYz1BaseDatabaseSchemaEditor._field_should_be_indexedcCs|j o|jSr'rr<r,r/rrrr6rYz2BaseDatabaseSchemaEditor._field_became_primary_keycCs|j o |jo |j p |jSr')rrrrrrrBs  z0BaseDatabaseSchemaEditor._unique_should_be_addedcCs*|j||||j||j|dS)N)rkZ old_columnZ new_columnr)sql_rename_columnrXrw)r<rkr,r/r*rrrr;s   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)rkrrwrcrdr )r r rurXrrrw target_fieldr$r6rWrr sql_create_fk) r<r$rrrkrrwrcrdr rrrrzs&  z'BaseDatabaseSchemaEditor._create_fk_sqlcs<fdd}t|jj|jgt|jjjjd|jjg||S)Ncj|i|Sr'rXrDrorArrcreate_fk_namezDBaseDatabaseSchemaEditor._fk_constraint_name..create_fk_namer)rr rurwr rr$)r<r$rrrrrArrs z,BaseDatabaseSchemaEditor._fk_constraint_namecC||j||Sr')r sql_delete_fkr<r$rrrrrz'BaseDatabaseSchemaEditor._delete_fk_sqlcCs,|durdS|tjkr dS|tjkrdSdS)NrKz DEFERRABLE INITIALLY DEFERREDz DEFERRABLE INITIALLY IMMEDIATE)rZDEFERREDZ IMMEDIATE)r<r rrr_deferrable_constraint_sqls  z3BaseDatabaseSchemaEditor._deferrable_constraint_sqlcCs|durdS|dur dSdS)NFz NULLS NOT DISTINCTTz NULLS DISTINCTrKr)r<nulls_distinctrrr _unique_index_nulls_distinct_sqls z9BaseDatabaseSchemaEditor._unique_index_nulls_distinct_sqlcCsR| s|jjjo(| p|jjjo(| p|jjjo(| p|jjjo(|dup(|jjjSr')r6r9Zsupports_partial_indexesZ&supports_deferrable_unique_constraintsrerZ*supports_nulls_distinct_unique_constraints)r<rcr rjrkrrrr_unique_supporteds  z*BaseDatabaseSchemaEditor._unique_supportedc sj||||| ds dS|s|s|s|s| dur/j|||||||| d} | r-j| dSjdfdd|D|d} j|| dS)Nrcr rjrkr)rrcrjrirkrricsg|]}|jqSr)rXrwr]rArrr^+r_z8BaseDatabaseSchemaEditor._unique_sql..)rr rrg) rrmr>rQsql_unique_constraintrrsql_constraintrX) r<r$rrrcr rjrirkrrGrgrrAr _unique_sqlsL  z$BaseDatabaseSchemaEditor._unique_sqlc  Cs|j||||| ds dSt|ddj|jd} |jj} dd|D} |dur/|j| | dd}n||}|s<|s<|s<|r@|j} n|j } | rO|j | | d |d } nt | || |j } t | t| |j|| ||||||||| d S) NrFrlr`cSrmrrnr]rrrr^Lrz?BaseDatabaseSchemaEditor._create_unique_sql..Trr)rhri)rkrrrcr rjr)rrrr6r rurrXsql_create_unique_indexsql_create_uniquerurrTr r rdrrfr)r<r$rrrcr rjrirkrrrkrrGrrrrm3sD     z+BaseDatabaseSchemaEditor._create_unique_sqlcs&|r fdd}nj}t||d|S)Ncrr'rrorArrcreate_unique_nameirzLBaseDatabaseSchemaEditor._unique_constraint_name..create_unique_name_uniq)rDr)r<rkrrrrrArrfsz0BaseDatabaseSchemaEditor._unique_constraint_namec CsD|j|||||ds dS|s|s|s|r|j} n|j} || ||S)Nr)rrrr) r<r$rrcr rjrirkrrGrrrr7qs z+BaseDatabaseSchemaEditor._delete_unique_sqlcCs |j|||jd|idS)Nrar)rrXrq)r<rrarrr _check_sqls z#BaseDatabaseSchemaEditor._check_sqlcCs2|jjjsdSt|jt|jj|j|||dS)N)rkrra) r6r9 supports_table_check_constraintsr sql_create_checkr r rurX)r<r$rrarrrrEs z*BaseDatabaseSchemaEditor._create_check_sqlcCs|jjjsdS||j||Sr')r6r9rrsql_delete_checkrrrrr:s z*BaseDatabaseSchemaEditor._delete_check_sqlcCs t|t|jj|j||dS)Nrv)r r r rurX)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.Ncs@g|]}jjjrjjt|jjnjj|qSr)r6r9Ztruncates_names introspectionZidentifier_converterr rWr])r*rrArrr^s  z>BaseDatabaseSchemaEditor._constraint_names..rrrrrarr)r6rUrZget_constraintsr ruitemsrQ)r<r$r`rrrrrar0rrUr}resultrZinfodictrrArrs8    z*BaseDatabaseSchemaEditor._constraint_namescs"jddfdd|DiS)Nrric3rr'rX)r*rwrArrr-rz>BaseDatabaseSchemaEditor._pk_constraint_sql..)sql_pk_constraintr)r<rrrArrsz+BaseDatabaseSchemaEditor._pk_constraint_sqlcCsV|j|dd}|rt|dkrtdt||jjf|D] }||||qdS)NTrrz0Found wrong number (%s) of PK constraints for %s)rrrr rurB_delete_primary_key_sql)r<r$r&rrFrrrrAsz,BaseDatabaseSchemaEditor._delete_primary_keyc CsJt|jt|jj|j||j|jj|jgddt|jj|jg|jdS)NZ_pkr3)rkrr) r sql_create_pkr r rurXrDrwrrrrrrCsz0BaseDatabaseSchemaEditor._create_primary_key_sqlcCrr')r sql_delete_pkrrrrrrz0BaseDatabaseSchemaEditor._delete_primary_key_sqlcCs|r d||SdS)NzCOLLATE rKr)r<rrGr_rrrrrz%BaseDatabaseSchemaEditor._collate_sqlcCs*|j||d|d}||dS)N,)Z procedure param_types)sql_delete_procedurerXrrB)r<Zprocedure_namerrGrrrremove_procedures z)BaseDatabaseSchemaEditor.remove_procedure)FT)r)F)rKr')NNNNN)NNNNNN)NNNNNNN)T)NNNNNNNNr)x__name__ __module__ __qualname____doc__rrrrrrrUrOrPrTrSrRrrr@rrqZsql_delete_constraintrrrrrrrrxrrrtrrzrrrrrrVr=r@rCrBrXrrrorrrrrr staticmethodrrrTrrrrrrrrrrrrr rr"r+r.r%r>rr=r<rrr$rDrbrdrfrr9rrurrr#r}r6rBr;rzrrrrrrrmrr7rrEr:rrrrArCrrrrrrrr5MsX   e C   #   "  X S r ' .     6 %    8 3    /   r5)$loggingr!rZ django.confrZdjango.core.exceptionsrZ!django.db.backends.ddl_referencesrrrrr r Zdjango.db.backends.utilsr r r Zdjango.db.modelsrrZ!django.db.models.fields.compositerZdjango.db.models.sqlrZdjango.db.transactionrrZ django.utilsr getLoggerrMrr%r1r5rrrrs"