o Fh<@sddlZddlZddlmZddlmZddlmZddlmZm Z m Z m Z m Z m Z ddlmZmZmZddlmZmZmZddlmZdd lmZmZdd lmZed Zd d ZddZ ddZ!GdddZ"dS)N)datetime)settings) FieldError)Columns ExpressionsForeignKeyName IndexName StatementTable) names_digestsplit_identifier truncate_name) NOT_PROVIDED 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_fieldrrW/var/www/html/loop/nvenv/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 9 z+_related_non_m2m_objects..c3r,r-r.r/) new_fieldrrr3>r4)zipr+r*_related_non_m2m_objects remote_field)r2r5related_fieldsold_relnew_relr)r5r2rr75s     r7c @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:Z3d;d<Z4d=d>Z5e6d?d@Z7dAdBZ8dCdDZ9dEdFZ:dGdHZ;dIdJZdOdPZ?dQdRZ@dSdTZAdUdVZBdWdXZCdYdZZDd[d\ZEd]d^ZFd_d`ZGdadbZHddcddZIdedfZJ ddgdhZKdidjZLddkdlZM ddmdnZNdodpZOdqdrZPdsdtZQdudvZRddxdyZSddzd{ZTd|d}ZUd~dZVdddwdwdd(dd(dddd ddZWdddZXddZYddZZddZ[ddZ\dddZ]ddZ^ddZ_ddZ`ddZaddZbddZcddZdddZeddZf     dddZg      dddZh       dddZidddZj      dddZkddZlddZmddZnddZo        dddZpdddZqddZrddZsdddZtdddZudS)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)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)selfr=r>rrrr__init__s z!BaseDatabaseSchemaEditor.__init__cCs(g|_|jrt|jj|_|j|Sr-) deferred_sqlrBrr=alias __enter__rCrrrrGs  z"BaseDatabaseSchemaEditor.__enter__cCs<|dur|jD]}||q|jr|j|||dSdSr-)rEexecuterBr__exit__)rCexc_type exc_value tracebacksqlrrrrJs   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))paramsrN)extra;N)r>r=in_atomic_blockr@rArstrloggerdebugrstripendswithr?appendtuplemap quote_valuecursorrI)rCrNrOendingr]rrrrIs. "z BaseDatabaseSchemaEditor.executecCs|jj|Sr-)r=ops quote_name)rCrrrrr`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_fieldr0rr)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_sqlr0 constraintr*rCrrrf s , css|] }|rt|VqdSr-)rTrrrrrr3s z5BaseDatabaseSchemaEditor.table_sql..)table definition) r%unique_togetherrErY_create_unique_sql local_fields column_sql db_parametersr=sql_check_constraintdb_type_suffixextendr8 db_constraintr*db_tablerd field_namecolumnsql_create_inline_fkr`r@supports_foreign_keys_create_fk_sqlget_internal_typer_ autoinc_sql constraintssql_create_tablejoin db_tablespacetablespace_sql)rCr* field_namesfields column_sqlsrOrrw extra_params db_paramscol_type_suffixrkrlrrrNrrrtr table_sqls                 z"BaseDatabaseSchemaEditor.table_sqlccs|V|d}r||V|jjjr |jr ||jV|j}|jt ur<| |\} } d| V| | d}|oK| | oK|oJ| | }|rt||} | durtd||} |jjjrl| || Vn| V|| |jr|js|jjjrd}|jr||\} }| || Vn|sdVn|jjjsdV|jrdVn|jrdV|jp|jj}|r|jjjr|jr|jjj|dd VdSdSdSdS) N collationzDEFAULT FTzNOT NULLNULLz PRIMARY KEYUNIQUE)inline)get _collate_sqlr=r@supports_comments_inline db_comment _comment_sqlnull db_defaultrdb_default_sqlr skip_defaultskip_default_on_altereffective_default_column_default_sqlrequires_literal_defaultsprepare_defaultrYempty_strings_allowedr!interprets_empty_strings_as_nulls generated_column_generated_sqlimplied_column_nulluniquerr%supports_tablespacesr_r)rCcolumn_db_typerOr*rfield_db_paramsinclude_defaultrr default_sqldefault_params default_valuecolumn_default generated_sqlgenerated_params tablespacerrr_iter_column_sql"sj           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. rhtypeNNNrj)r|r=rr)rCr*rrrrrOrrrr{es" z#BaseDatabaseSchemaEditor.column_sqlcCdS)z Some backends don't accept default values for certain columns types (i.e. MySQL longtext and longblob). FrrCrrrrrz%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() methodNotImplementedErrorrCvaluerrrrsz(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)rhc3|]}|VqdSr-)rr0prHrrr3z:BaseDatabaseSchemaEditor.db_default_sql..) django.db.models.expressionsr_db_default_expression isinstancerrr* get_compilerr=compiler@rrZ) rCrrrrNquerycompilerrrOrrHrrs    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.STOREDVIRTUALc3rr-r\rrHrrr3rzABaseDatabaseSchemaEditor._column_generated_sql..rzGENERATED ALWAYS AS (z) )rr= db_persistr@rrZ)rCrexpression_sqlrOpersistency_sqlrrHrrs  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) N BinaryFieldrRauto_nowF auto_now_add DateTimeField DateField TimeField) has_default get_defaultrrblankrrgetattrrnowrdatetime)rdefault internal_typerrr_effective_defaults4   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). rrrrrr\sz$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`. Nrhr)rrIr=r@supports_commentsr%db_table_commentalter_db_table_commentrrzrr|_alter_column_comment_sqlrEr_model_indexes_sqllocal_many_to_manyr8through auto_created create_model)rCr*rNrOrr 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.rvN)r%rr8rr delete_modelrIsql_delete_tabler`rlistrErr references_tableremove)rCr*rrNrrrrs"   z%BaseDatabaseSchemaEditor.delete_modelcCs.|jr |jjjs dS|j|||dddS)zAdd an index on a model.NrO)contains_expressionsr=r@supports_expression_indexesrI create_sqlrCr*indexrrr add_indexs z"BaseDatabaseSchemaEditor.add_indexcCs*|jr |jjjs dS||||dS)zRemove an index from a model.N)rr=r@rrI remove_sqlrrrr remove_index(s z%BaseDatabaseSchemaEditor.remove_indexcCsF|jjjr|j|||j|jdddS||||||dS)Nr)r=r@can_rename_indexrI_rename_index_sqlrrr)rCr* old_index new_indexrrr rename_index1s  z%BaseDatabaseSchemaEditor.rename_indexcCs&|||}|r|j|dddSdS)zAdd a constraint to a model.Nr)rrIrCr*rsrNrrradd_constraint;s z'BaseDatabaseSchemaEditor.add_constraintcCs"|||}|r||dSdS)z!Remove a constraint from a model.N)rrIrrrrremove_constraintCs 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|qSrrZr0rrrr OzABaseDatabaseSchemaEditor.alter_unique_together..cSrrrr rrrr Pr TF)rrcrbrrcrer)rrrf[rgzBBaseDatabaseSchemaEditor.alter_unique_together..N) difference_delete_composed_indexsql_delete_uniquerIry)rCr*old_unique_togethernew_unique_togetheroldsnewsrrrr)ralter_unique_togetherIsz.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. cSrrrr rrrr dr z@BaseDatabaseSchemaEditor.alter_index_together..cSrrrr rrrr er TF)rrcrbrrcrer)rrrfprgzABaseDatabaseSchemaEditor.alter_index_together.._idxrsuffixN)r r sql_delete_indexrI_create_index_sql)rCr*old_index_togethernew_index_togetherrrrrrr)ralter_index_together^sz-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|]}|jqSrrrrrrrr tzBBaseDatabaseSchemaEditor._delete_composed_index..cSrrrrrrrrr wcsg|] }j|jqSr)r%rdrrer)rrrfxszCBaseDatabaseSchemaEditor._delete_composed_index..excluderTFquotez1Found wrong number (%s) of constraints for %s(%s)rur)r%rindexes_constraint_namesrr=r@*allows_multiple_constraints_on_same_fieldsrT_unique_constraint_namerlen ValueErrorrrI_delete_constraint_sql) rCr*rconstraint_kwargsrNmeta_constraint_namesmeta_index_namescolumnsconstraint_names default_namerr)rr ssB 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_caselowerrIsql_rename_tabler`rErr rename_table_references)rCr* old_db_table new_db_tablerNrrralter_db_tables"   z'BaseDatabaseSchemaEditor.alter_db_tablecCsF|jr|jjjr!||j||jj||pdddSdSdS)NrR)rvcomment) sql_alter_table_commentr=r@rrIr`r%rr\)rCr*old_db_table_commentnew_db_table_commentrrrrs  z/BaseDatabaseSchemaEditor.alter_db_table_commentcCs2||j||jj||||ddS)z)Move a model's table between tablespaces.)rvold_tablespacenew_tablespaceN)rIsql_retablespace_tabler`r%r)rCr*old_db_tablespacenew_db_tablespacerrralter_db_tablespaces z,BaseDatabaseSchemaEditor.alter_db_tablespacec CsN|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|jtur| |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)rNrhrjrirmz%s.rR)r namespacerrkrl deferrable)rvrrwdroprvchangesr),rr8rr%rrr{r~r=r|r}r@rrsql_create_column_inline_fkr*rrdrrr _fk_constraint_namer`r_deferrable_sqlrErYrsql_create_columnrIrrrr_alter_column_default_sqlsql_alter_columnrrrrr_field_indexes_sqlconnection_persists_old_columnsclose)rCr*rrwrOrrconstraint_suffixrkrlrD_rN 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. rhrNT foreign_key)rvr)rr8rr%rrr|r=r%rrI_delete_fk_sqlsql_delete_columnr`rr@rQrRrrErr references_columnr)rCr*rfk_namesfk_namerNrrr remove_fields,       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. NrhrFzqCannot 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_alteredr|r=r8r)rr%r_alter_many_to_manyrrrr _alter_field) rCr*r2r5strict old_db_paramsold_type new_db_paramsnew_typemodifying_generated_field old_field_sql new_field_sqlrrr alter_field5s           z$BaseDatabaseSchemaEditor.alter_fieldcCsD|jj}||j}d|d<z |||WSty!YdSw)N__column_name__r)r=data_type_check_constraintsdb_type_parametersrKeyError)rCrrcheck_constraintsdatarrr_field_db_checks  z(BaseDatabaseSchemaEditor._field_db_checkc 2 Cs|t} |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|(||||||\} }!|)| |*|!|j+t,ur|j+t,us|j+|j+kr|)|-|||n|j+t,ur|)|j-|||ddd }"|j.r6|j.s6|j+t,ur6|/|}#|/|}$|0|s6|#|$kr6|$dur6d}"|)|1||||j.|j.krL|2|||} | rL|)| |3pU|j+t,uo^|j.o^|j. }%|se|r|%sl||7}|jjj4r|rt5t6|\}}&d7|t8|&gfg}|D]\}}&||j9|:|j j |d|&q|%r|j+t,urd}'|$g}&n|;|\}'}&||j<|:|j j |:|j|'d|&|D]\}}&||j9|:|j j |d|&q|r|D] \}}&|||&q|jr|js|=|||>||r||?||g|jr|jr1|jr1|js1||j@||gdg}(|r>|(*t|||||rV||A|||(*t|||(D]V\})}|jjB|jd}*|*d}+|*d },|)jjB|jd}-|-d }.|(|j|)j|j|+|.|,\} }!||j9|:|jj j | dd| d|!D] \}}&|||&qqX|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}|||&|jjjFr<|jGdSdS)!z3Perform a "physical" (non-ManyToMany) field update.r)ignoreTrWr#z.F)rrr z7Found wrong number (%s) of unique constraints for %s.%srcSrrr)r0rrrrr r)rtype_r cSrrrrrrrrr  r)rir z6Found wrong number (%s) of check constraints for %s.%srhrFNrurHr)rvrrrrrrm_fk_checkrri)Hsetr=r@rr8rr_r%rr(r)r%raddrIrYr_field_became_primary_keyr_delete_unique_sqlrrr7 related_modelrdb_indexr$rr_delete_index_sqlrq_delete_check_sql_rename_field_sqlrErr rename_column_referencesr~rr_alter_column_type_sqlrYrrr"_alter_column_database_default_sqlrrrrN_alter_column_null_sqlrsupports_combined_altersrZr6rsumrOr`rsql_update_with_default_delete_primary_key_unique_should_be_addedryr_create_primary_key_sqlr|r_create_index_name_create_check_sqlrQrR)2rCr*r2r5rdrfrcrerb fks_droppedr\r]r,r/constraint_name old_collation new_collationdrop_foreign_keys_old_relr; rel_fk_namesr- index_names index_name old_db_check new_db_checkrNactions null_actions post_actionsold_type_suffixnew_type_suffixfragment other_actionsneeds_database_default old_default new_defaultfour_way_default_alterationrOrrels_to_updater: rel_db_paramsrel_type rel_collationold_rel_db_paramsold_rel_collationrTrelrUrrrras\                                                      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. Nrhr)rr) r=r@rrr|rsql_alter_column_nullsql_alter_column_not_nullr`r)rCr*r2r5rerNrrrrs" 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. rhrrrr) rrr=r@rrr|r sql_alter_column_no_default_nullsql_alter_column_no_defaultsql_alter_column_defaultr`r) rCr*r2r5rGrrrOrerNrrrrN$s,     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. rRrhrr)rrrr|r=r`r) rCr*r2r5rGrNrrOrerrrrJs  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. rjrR)rrrr:)rr%rr=r@rrrrrYrsql_alter_column_typer`r) rCr*r2r5rfrrr collate_sql comment_sqlrNrOrrrrfs6        z/BaseDatabaseSchemaEditor._alter_column_type_sqlcCs.|j||jj||j||dgfS)N)rvrr:)sql_alter_column_commentr`r%rrr)rCr*r5rfnew_db_commentrrrrs  z2BaseDatabaseSchemaEditor._alter_column_comment_sqlcCs||pdS)NrRr)rCr:rrrrraz%BaseDatabaseSchemaEditor._comment_sqlcCs|jjjj|jjjjkr||jj|jjjj|jjjj||jj|jjj||jjj|||jj|jjj||jjj|dS)z*Alter M2Ms to repoint their to= endpoints.N) r8rr%rr9rjrdm2m_reverse_field_namem2m_field_name)rCr*r2r5rbrrrr`s,       z,BaseDatabaseSchemaEditor._alter_many_to_manyrRc 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_%srTNr#rzD%s)r r r=r_max_name_lengthrr(isdigit) rC table_name column_namesrrThash_suffix_part max_lengthr other_lengthrrrrs(   z+BaseDatabaseSchemaEditor._create_index_namecCsf|dur$t|dkr|djr|dj}ntjrtj}n|jjr$|jj}|dur1d|jj|SdS)Nr#rrjrR)r(rrDEFAULT_INDEX_TABLESPACEr%r=r_r)rCr*rrrrr_get_index_tablespace_sqls z2BaseDatabaseSchemaEditor._get_index_tablespace_sqlcCs|rd|SdS)Nz WHERE rRr)rC conditionrrr_index_condition_sqlsz-BaseDatabaseSchemaEditor._index_condition_sqlcCs,|r|jjjs dStdt|jj||jdS)NrRz INCLUDE (%(columns)s))r.)r=r@supports_covering_indexesr rr%rr`)rCr*r.rrr_index_include_sqls z+BaseDatabaseSchemaEditor._index_include_sql) rrrusingr col_suffixesrN 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_colsrh)rcSg|]}|jqSrrrerrrrfrz>BaseDatabaseSchemaEditor._create_index_sql..cs"dur j|i|Sr-)rr`argskwargsrrCrrcreate_index_names zEBaseDatabaseSchemaEditor._create_index_sql..create_index_name)rvrrr.rPrr)rrr=rsql_create_indexr%rr r r`r_index_columnsrr\rr)rCr*rrrrrrrNrrrrrrr.rrvrrrrrs0     z*BaseDatabaseSchemaEditor._create_index_sqlcCs&t|p|jt|jj|j||dSN)rvr)r rr r%rr`)rCr*rrNrrrr~1s z*BaseDatabaseSchemaEditor._delete_index_sqlcCs*t|jt|jj|j||||dS)N)rvold_namenew_name)r sql_rename_indexr r%rr`)rCr*rrrrrr8s z*BaseDatabaseSchemaEditor._rename_index_sqlcCst|||j|dS)N)r)rr`)rCrvr.rrrrrr@sz'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. crbrrcrer)rrrfPrgz?BaseDatabaseSchemaEditor._model_indexes_sql..rr)r%managedproxyswappedrzrrPindex_togetherrYrr$rr=r@rr)rCr*outputrrrrrr)rrCs    z+BaseDatabaseSchemaEditor._model_indexes_sqlcCs*g}|||r||j||gd|S)zT Return a list of all index SQL statements for the specified field. rt)_field_should_be_indexedrYr)rCr*rrrrrrP[s 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)concreterx deconstructunion non_db_attrspoprr8r*r%rrrr`r) rCr2r5rrrTold_pathold_args old_kwargsnew_pathnew_args new_kwargsattrrrrr_ds@        z1BaseDatabaseSchemaEditor._field_should_be_alteredcCs|jo|j Sr-)r}rrCr*rrrrrraz1BaseDatabaseSchemaEditor._field_should_be_indexedcCs|j o|jSr-rrCr2r5rrrrzraz2BaseDatabaseSchemaEditor._field_became_primary_keycCs|j o |jo |j p |jSr-)rrrrrrrs  z0BaseDatabaseSchemaEditor._unique_should_be_addedcCs*|j||||j||j|dS)N)rv old_column new_columnr)sql_rename_columnr`r)rCrvr2r5rfrrrrs   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)rvrrrkrlrE)r r%rr`rKrr target_fieldr*r=r_rLr sql_create_fk) rCr*rrrvrrrkrlrErrrrs&  z'BaseDatabaseSchemaEditor._create_fk_sqlcs<fdd}t|jj|jgt|jjjjd|jjg||S)Ncj|i|Sr-r`rrrHrrcreate_fk_namezDBaseDatabaseSchemaEditor._fk_constraint_name..create_fk_namer#)rr%rrr rr*)rCr*rrrrrHrrKs z,BaseDatabaseSchemaEditor._fk_constraint_namecC||j||Sr-)r* sql_delete_fkrCr*rrrrrYz'BaseDatabaseSchemaEditor._delete_fk_sqlcCs,|durdS|tjkr dS|tjkrdSdS)NrRz DEFERRABLE INITIALLY DEFERREDz DEFERRABLE INITIALLY IMMEDIATE)rDEFERRED IMMEDIATE)rCrErrr_deferrable_constraint_sqls  z3BaseDatabaseSchemaEditor._deferrable_constraint_sqlcCs|durdS|dur dSdS)NFz NULLS NOT DISTINCTTz NULLS DISTINCTrRr)rCnulls_distinctrrr _unique_index_nulls_distinct_sqls z9BaseDatabaseSchemaEditor._unique_index_nulls_distinct_sqlcCsR| s|jjjo(| p|jjjo(| p|jjjo(| p|jjjo(|dup(|jjjSr-)r=r@supports_partial_indexes&supports_deferrable_unique_constraintsrr*supports_nulls_distinct_unique_constraints)rCrrErrrrrr_unique_supporteds  z*BaseDatabaseSchemaEditor._unique_supportedc sj||||| ds dS|s|s|s|s| dur/j|||||||| d} | r-j| dSjdfdd|D|d} j|| dS)NrrErrr)rrrrrrrucsg|]}|jqSr)r`rrerHrrrfrgz8BaseDatabaseSchemaEditor._unique_sql..)r.rErrs) r ryrErYsql_unique_constraintrrsql_constraintr`) rCr*rrrrErrrrrNrsrrHr _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) Nr FrrhcSrrrrerrrrf7rz?BaseDatabaseSchemaEditor._create_unique_sql..Tr!r)rr)rvrr.rrErr)r rrr=r%rr'r`sql_create_unique_indexsql_create_uniquerrr\r r rrrr)rCr*rrrrErrrrrrvr.rNrrrrysD     z+BaseDatabaseSchemaEditor._create_unique_sqlcs&|r fdd}nj}t||d|S)Ncrr-rrrHrrcreate_unique_nameTrzLBaseDatabaseSchemaEditor._unique_constraint_name..create_unique_name_uniq)rr)rCrvr.r"rrrHrr'Qsz0BaseDatabaseSchemaEditor._unique_constraint_namec CsD|j|||||ds dS|s|s|s|r|j} n|j} || ||S)Nr )r rrr*) rCr*rrrErrrrrNrrrr{\s z+BaseDatabaseSchemaEditor._delete_unique_sqlcCs |j|||jd|idS)Nrir )r r`r})rCrrirrr _check_sqlus z#BaseDatabaseSchemaEditor._check_sqlcCs2|jjjsdSt|jt|jj|j|||dS)N)rvrri) r=r@ supports_table_check_constraintsr sql_create_checkr r%rr`)rCr*rrirrrr{s z*BaseDatabaseSchemaEditor._create_check_sqlcCs|jjjsdS||j||Sr-)r=r@rr*sql_delete_checkrrrrrs z*BaseDatabaseSchemaEditor._delete_check_sqlcCs t|t|jj|j||dSr)r r r%rr`)rCtemplater*rrrrr*s 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)r=r@truncates_names introspectionidentifier_converterr r_r)r0rrHrrrfs  z>BaseDatabaseSchemaEditor._constraint_names..r.rrrrirXr)r=r]rget_constraintsr%ritemsrY)rCr*rrrrrXrirsr r]rresultrinfodictrrHrr%s8    z*BaseDatabaseSchemaEditor._constraint_namescCsV|j|dd}|rt|dkrtdt||jjf|D] }||||qdS)NTrr#z0Found wrong number (%s) of PK constraints for %s)r%r(r)r%rrI_delete_primary_key_sql)rCr*rbr/rrrrrsz,BaseDatabaseSchemaEditor._delete_primary_keyc CsJt|jt|jj|j||j|jj|jgddt|jj|jg|jdS)N_pkrw)rvrr.) r sql_create_pkr r%rr`rrrrrrrrsz0BaseDatabaseSchemaEditor._create_primary_key_sqlcCrr-)r* sql_delete_pkrrrrrrz0BaseDatabaseSchemaEditor._delete_primary_key_sqlcCs|r d||SdS)NzCOLLATE rR)r`)rCrrrrrrrrz%BaseDatabaseSchemaEditor._collate_sqlcCs*|j||d|d}||dS)N,) procedure param_types)sql_delete_procedurer`rrI)rCprocedure_namer%rNrrrremove_procedures z)BaseDatabaseSchemaEditor.remove_procedure)FT)r)F)rRr-)NNNNN)NNNNNN)NNNNNNN)T)NNNNNNNNr)v__name__ __module__ __qualname____doc__rr5r@rrMrOrrrrrrrZrrr r}sql_delete_constraintr rrrrrrrJrrrrrr!r"r&r;rrDrGrJrIr`rrr{rrrrrr staticmethodrrr\rrrrrrrrrr r9rrCrVr^rjrqrarrNrrrrr`rrrrrr~rrrrPr_rrzrrrrKrYrrr rryr'r{rrrr*r%rrrrr(rrrrr<LsT   S C   $   "  X S v ' .     6 %    8 3    /  r<)#loggingr'r django.confrdjango.core.exceptionsr!django.db.backends.ddl_referencesrrrrr r django.db.backends.utilsr r r django.db.modelsrrrdjango.db.models.sqlrdjango.db.transactionrr django.utilsr getLoggerrUrr+r7r<rrrrs