B 0H[@sddlZddlZddlmZddlmZmZmZmZmZddl m Z ddl m Z ddl mZmZddlmZddlmZed Zd d Zd d ZGdddZdS)N)datetime)ColumnsForeignKeyName IndexName StatementTable)split_identifier)Index)TransactionManagementErroratomic)timezone) force_byteszdjango.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_fieldrrbC:\Users\HIRONO~1\AppData\Local\Temp\pip-install-6bm3nxem\django\django\db\backends\base\schema.py_is_relevant_relations rcs2tfddjjjDfddjjjDS)Nc3s|]}t|r|VqdS)N)r).0obj) old_fieldrr %sz+_related_non_m2m_objects..c3s|]}t|r|VqdS)N)r)rr) new_fieldrrr&s)zipmodel_metarelated_objects)rrr)rrr_related_non_m2m_objects!src@seZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdpddZddZdd Z dqd"d#Z!d$d%Z"e#d&d'Z$drd(d)Z%d*d+Z&d,d-Z'd.d/Z(d0d1Z)d2d3Z*d4d5Z+d6d7Z,d8d9Z-d:d;Z.dd?Z0d@dAZ1dBdCZ2dDdEZ3dFdGZ4dsdHdIZ5dtdJdKZ6dLdMZ7dudNdOZ8dPdQZ9dRdSZ:dvdUdVZ;dwdWdXZd^d_Z?d`daZ@dbdcZAdddeZBdfdgZCdhdiZDdxdjdkZEdydldmZFdzdndoZGdS){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 NULLz?ALTER TABLE %(table)s ADD CONSTRAINT %(name)s CHECK (%(check)s)z.ALTER TABLE %(table)s DROP CONSTRAINT %(name)szBALTER 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)sNz9CREATE INDEX %(name)s ON %(table)s (%(columns)s)%(extra)szDROP INDEX %(name)szGALTER TABLE %(table)s ADD CONSTRAINT %(name)s PRIMARY KEY (%(columns)s)zDROP PROCEDURE %(procedure)sFTcCs,||_||_|jrg|_|jjjo$||_dS)N) connection collect_sql collected_sqlfeaturescan_rollback_ddlatomic_migration)selfr r!r rrr__init__Ws z!BaseDatabaseSchemaEditor.__init__cCs(g|_|jr$t|jj|_|j|S)N) deferred_sqlr%r r alias __enter__)r&rrrr*`s  z"BaseDatabaseSchemaEditor.__enter__cCs<|dkr"x|jD]}||qW|jr8|j|||dS)N)r(executer%r __exit__)r&exc_type exc_value tracebacksqlrrrr,gs  z!BaseDatabaseSchemaEditor.__exit__rc Cs|js |jjr |jjjs tdt|}tjd||||dd|jr| drTdnd}|dk r|j |t t |j||q|j ||n"|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))paramsr0)extra;N)r!r Zin_atomic_blockr#r$r strloggerdebugendswithr"appendtuplemap quote_valuecursorr+)r&r0r1Zendingr=rrrr+ps" z BaseDatabaseSchemaEditor.executecCs|jj|S)N)r ops quote_name)r&rrrrr?sz#BaseDatabaseSchemaEditor.quote_namecGs4t}x|D]}|t|qW|ddS)zw Generate a 32-bit digest of a set of arguments that can be used to shorten identifying names. N)hashlibmd5updater hexdigest)clsargshargrrr_digests z BaseDatabaseSchemaEditor._digestc Cs(|j|jd}|d}g}|dkr&dS|j}|o:|| }|r||}|dk r|jjjrp|d||7}n|d7}||g7}|jr|j s|jjj rd}|r|jjj s|d7}n |s|d7}|j r|d 7}n|j r|d 7}|j p|jj } | r |jjjr |j r |d |jjj| dd 7}||fS) z Take a field and return its column definition. The field must already have had set_attributes_from_name() called. )r typeN)NNz DEFAULT %sTz NULLz NOT NULLz PRIMARY KEYz UNIQUEz %s)inline) db_parametersr null skip_defaulteffective_defaultr#requires_literal_defaultsprepare_defaultempty_strings_allowedr!interprets_empty_strings_as_nullsZimplied_column_nullunique db_tablespacerZsupports_tablespacesr>tablespace_sql) r&rrinclude_default db_paramsr0r1rM default_valueZ tablespacerrr column_sqls:       z#BaseDatabaseSchemaEditor.column_sqlcCsdS)z Some backends don't accept default values for certain columns types (i.e. MySQL longtext and longblob). Fr)r&rrrrrNsz%BaseDatabaseSchemaEditor.skip_defaultcCs 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() methodN)NotImplementedError)r&valuerrrrQsz(BaseDatabaseSchemaEditor.prepare_defaultcCs|r|}n|js@|jr@|jr@|dkr8t}qt}n\t|ddsXt|ddrt }|}|dkrx|j }q|dkr|j }q|dkrt j }nd}t|r|}|||j}|S) z2Return a field's effective database default value.Z BinaryFieldZauto_nowFZ auto_now_addZ DateFieldZ TimeFieldZ DateTimeFieldN) has_defaultZ get_defaultrMZblankrRget_internal_typebytesr5getattrrnowdatetimer callableZget_db_prep_saver )r&rdefaultZ internal_typerrrrOs(  z*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)r[)r&r\rrrr<sz$BaseDatabaseSchemaEditor.quote_valuecs,g}g}x@jjD]2}||\}}|dkr4q|j|jd}|drZ|d|d7}|j|jd}|rx|d|7}|||jr|jr|jj jj } |jj j |jj j } |jr|d|j|| || d7}n |jjjr|j||d|d ||j |f|d kr|jjjj |j } | r|j| qWx8jjD],} fd d | D} |j|| qVW|j|jj d |d}jjr|jjjj}|r|d|7}|||pd|j|x.jj D]"}|jj!jj"r|#|jj!qWdS)zr Create a table and any accompanying indexes or unique constraints for the given `model`. N)r checkz CHECK (%s)z %s )to_table to_columnz_fk_%(to_table)s_%(to_column)sz%s %s)Z AutoFieldZ BigAutoFieldcsg|]}j|jqSr)r get_fieldcolumn)rr)rrr .sz9BaseDatabaseSchemaEditor.create_model..z, )table definition)$r local_fieldsrZrLr Zdb_type_suffixextend remote_field db_constraintrdb_tablerj field_namerksql_create_inline_fkr?r#supports_foreign_keysr(r9_create_fk_sqlr^r> autoinc_sqlZunique_together_create_unique_sqlsql_create_tablejoinrUrVr+_model_indexes_sqllocal_many_to_manythrough auto_created create_model)r&rZ column_sqlsr1rrnZ extra_paramsrXZcol_type_suffixrhrirxfieldscolumnsr0rVr)rrrsX           z%BaseDatabaseSchemaEditor.create_modelcCsx*|jjD]}|jjjjr ||jjq W||jd||jj ix6t |j D](}t |t rV||jj rV|j |qVWdS)z!Delete a model from the database.rmN)rr}rqr~r delete_modelr+sql_delete_tabler?rslistr( isinstancerZreferences_tableremove)r&rrr0rrrrDs z%BaseDatabaseSchemaEditor.delete_modelcCs||||dS)zAdd an index on a model.N)r+ create_sql)r&rindexrrr add_indexTsz"BaseDatabaseSchemaEditor.add_indexcCs||||dS)zRemove an index from a model.N)r+Z remove_sql)r&rrrrr remove_indexXsz%BaseDatabaseSchemaEditor.remove_indexcsdd|D}dd|D}x(||D]}||ddi|jq(Wx6||D](}fdd|D}|||qRWdS) 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:)rrrrr bszABaseDatabaseSchemaEditor.alter_unique_together..cSsh|] }t|qSr)r:)rrrrrrcsrTTcsg|]}j|jqSr)rrjrk)rr)rrrrliszBBaseDatabaseSchemaEditor.alter_unique_together..N) difference_delete_composed_indexsql_delete_uniquer+ry)r&rZold_unique_togetherZnew_unique_togetheroldsnewsrrr)rralter_unique_together\sz.BaseDatabaseSchemaEditor.alter_unique_togethercsdd|D}dd|D}x(||D]}||ddi|jq(Wx:||D],}fdd|D}||j|dd qRWd 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:)rrrrrrrsz@BaseDatabaseSchemaEditor.alter_index_together..cSsh|] }t|qSr)r:)rrrrrrssrTcsg|]}j|qSr)rrj)rr)rrrrlyszABaseDatabaseSchemaEditor.alter_index_together.._idx)suffixN)rrsql_delete_indexr+_create_index_sql)r&rZold_index_togetherZnew_index_togetherrrr field_namesr)rralter_index_togetherlsz-BaseDatabaseSchemaEditor.alter_index_togethercsjfdd|D}|j|f|}t|dkrNtdt|jjd|f||||ddS)Ncsg|]}j|jqSr)rrjrk)rr)rrrrl}szCBaseDatabaseSchemaEditor._delete_composed_index..z1Found wrong number (%s) of constraints for %s(%s)z, r)_constraint_nameslen ValueErrorrrsr{r+_delete_constraint_sql)r&rrZconstraint_kwargsr0rconstraint_namesr)rrr|s z/BaseDatabaseSchemaEditor._delete_composed_indexcCsr||ks"|jjjr&||kr&dS||j||||dx$|jD]}t|t rP| ||qPWdS)z#Rename the table a model points to.N)Z old_tableZ new_table) r r#Zignores_table_name_caselowerr+sql_rename_tabler?r(rrZrename_table_references)r&rZ old_db_tableZ new_db_tabler0rrralter_db_tables   z'BaseDatabaseSchemaEditor.alter_db_tablecCs2||j||jj||||ddS)z)Move a model's table between tablespaces.)rmZold_tablespaceZnew_tablespaceN)r+sql_retablespace_tabler?rrs)r&rZold_db_tablespaceZnew_db_tablespacerrralter_db_tablespaces z,BaseDatabaseSchemaEditor.alter_db_tablespacecCsJ|jr |jjjjr ||jjS|j||dd\}}|dkr@dS|j|jd}|drf|d|d7}|j | |jj | |j |d}| ||||s||dk r|j|d|dd\}}|j| |jj |d }| |||j||||jr0|jjjr0|jr0|j|||d |jjjrF|jdS) z Create a field on a model. Usually involves adding a column, but may involve adding a table instead (for M2M fields). T)rWN)r rfz CHECK (%s))rmrkrn)drop)rmchangesz_fk_%(to_table)s_%(to_column)s)rrqr~rrrrZrLr sql_create_columnr?rsrkr+rNrO_alter_column_default_sqlsql_alter_columnr(rp_field_indexes_sqlr#rvrrr9rwconnection_persists_old_columnsclose)r&rrrnr1rXr0 changes_sqlrrr add_fields0        z"BaseDatabaseSchemaEditor.add_fieldcCs|jr |jjjjr ||jjS|j|jdddkr:dS|jrx|j||j gdd}x"|D]}| | |j ||qZW|j ||jj||j d}| ||jjjr|jx:t|jD],}t|tr||jj|j r|j|qWdS)z Remove a field from a model. Usually involves deleting a column, but for M2Ms may involve deleting a table. )r rJNT) foreign_key)rmrk)rrqr~rrrrLr rrkr+r sql_delete_fksql_delete_columnr?rsr#rrrr(rrZreferences_columnr)r&rrfk_namesfk_namer0rrr remove_fields"     z%BaseDatabaseSchemaEditor.remove_fieldc Cs$|j|jd}|d}|j|jd}|d}|dkr>|jdksP|dkrb|jdkrbtd||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. )r rJNzqCannot 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)) rLr rqrr~rr_alter_many_to_many _alter_field) r&rrrstrict old_db_paramsold_type new_db_paramsnew_typerrr alter_fields2      z$BaseDatabaseSchemaEditor.alter_fieldc & Cst} |jr|jr|j||jgdd} |rRt| dkrRtdt| |jj|jfx0| D](} | |jf| | |j || qXW|j r|j r|js|jr|j||jgdd} |rt| dkrtdt| |jj|jfx"| D]} | | |j|| qW|jr|js&|j o,|j o,||k}|rxXt||D]J\}}|j|j|jjgdd}x&|D]} | | |j |j| qfWq@W|jr |j s |jr|j r dd|jjD}|j||jgdtjd }x2|D]*}||krq| | |j||qW|d |d kr|d r|j||jgdd } |rlt| dkrltd t| |jj|jfx$| D]} | | |j|| qrW|j|jkr| ||jj|||x2|jD](}t|tr||jj|j|jqWg}g}g}||kr,|||||\}}||| ||!|}|!|}|j"op|j" op||kop|d k op|#| }|r||$||||j"|j"kr|%|||}|r|||&o|j"o|j" }|s|r|s||}|j'j(j)r$|r$t*t+|\}}d,|t-|gfg}x4|D],\}}| |j.|/|jj|d|q*W|r| |j0|/|jj|/|jdd|gx4|D],\}}| |j.|/|jj|d|qW|rx|D]\}}| ||qW|j s|j s|jr(|js(|j r(| |1||jg|jr8|j r\|jr\|j s\| |2||gg} |jr|jr||kr| t|||js|jr|3||| |j4|/|jj|/|j5|jj|jgdd|/|jd| t||x| D]\}!}|jj6|j'd}"|"d}#||j|!j|j|#\}}| |j.|/|jjj|dd|dx|D]\}}| ||qlWqW|jr| s|jr|js|jr| |7||d|rxB|j8jj9D]4}$t:|$|r|$jjr| |7|$j|$jdqW|d |d krv|d rv| |j;|/|jj|/|j5|jj|jgdd|/|j|d d|r|j$|||dd\}%}|j.|/|jj|%d}| |||j'j(j<r|j'=d S)z3Perform a "physical" (non-ManyToMany) field update.T)rrz.)rtype_rf)rfz6Found wrong number (%s) of check constraints for %s.%sNz, )rmrz%s)rmrkreZ_pk)r)rmrr)r rJrz_fk_%(to_table)s_%(to_column)sZ_fk_check)rmrrkrf)r)>setrqrrrrkrrrrsaddr+rrrTrrrZ related_modelrdb_indexindexesr rrsql_delete_check_rename_field_sqlr(rrZrename_column_references_alter_column_type_sqlr9rprOrMrNr_alter_column_null_sqlr]r r#Zsupports_combined_altersr:rr{sumrr?sql_update_with_defaultryr_delete_primary_key sql_create_pk_create_index_namerLrwrrrsql_create_checkrr)&r&rrrrrrrrZ fks_droppedrrrconstraint_nameZdrop_foreign_keysZ_old_relZnew_relZ rel_fk_namesZmeta_index_namesZ index_names index_namer0ZactionsZ null_actionsZ post_actionsfragmentZ other_actionsZ old_default new_defaultZneeds_database_defaultZfour_way_default_alterationr1Zrels_to_updateZold_relZ rel_db_paramsZrel_typerelrrrrrs2      $                                    z%BaseDatabaseSchemaEditor._alter_fieldcCs\|jjjr|dkrdS|j|jd}|jr4|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. )Z CharFieldZ TextFieldN)r rJ)rkrJ) r r#rSr^rLrMsql_alter_column_nullsql_alter_column_not_nullr?rk)r&rrrrr0rrrr s    z/BaseDatabaseSchemaEditor._alter_column_null_sqlc Cst||}d}|g}|rg}n|jjjr6||}g}|j|jd}|rN|jn|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. z%s)r rJ)rkrJre) rOr r#rPrQrLsql_alter_column_no_defaultsql_alter_column_defaultr?rk) r&rrrrrrer1rr0rrrr#s    z2BaseDatabaseSchemaEditor._alter_column_default_sqlcCs |j||j|dgfgfS)a Hook to specialize column type alteration for different backends, for cases when a creation type is different to an alteration type (e.g. SERIAL in PostgreSQL, PostGIS fields). Return a two-tuple of: an SQL fragment of (sql, params) to insert into an ALTER TABLE statement and a list of extra (sql, params) tuples to run once the field is altered. )rkrJ)sql_alter_column_typer?rk)r&rrrrrrrrBs  z/BaseDatabaseSchemaEditor._alter_column_type_sqlcCs|jjjj|jjjjkr:||jj|jjjj|jjjj||jj|jjj||jjj|||jj|jjj||jjj|dS)z*Alter M2Ms to repoint their to= endpoints.N) rqr~rrsrrrjZm2m_reverse_field_nameZm2m_field_name)r&rrrrrrrrWsz,BaseDatabaseSchemaEditor._alter_many_to_manyr4c Cst|\}}|gt|}d|j||f}|jjp:d}d|d||f}t||kr`|St||dkr|d|d}|t|dd} 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%sz%s_%s_%s_NrrzD%s) rrrIr r>Zmax_name_lengthr{risdigit) r&Z table_name column_namesrrZ hash_dataZhash_suffix_part max_lengthrZ other_lengthrrrrls"   z+BaseDatabaseSchemaEditor._create_index_namecCsX|dkr:t|dkr*|djr*|dj}n|jjr:|jj}|dk rTd|jj|SdS)Nrrrgr4)rrUrr r>rV)r&rrrUrrr_get_index_tablespace_sqls z2BaseDatabaseSchemaEditor._get_index_tablespace_sql)rrusingrU col_suffixesr0c srj|||d} dd|D} |p&j} |jj} fdd} t| t| jt| | || |t| | j|d| dS)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, ...). )rUcSsg|] }|jqSr)rk)rrrrrrlsz>BaseDatabaseSchemaEditor._create_index_sql..csdkrj||S)N)rr?)rFkwargs)rr&rrcreate_index_names zEBaseDatabaseSchemaEditor._create_index_sql..create_index_name)r)rmrrrr2) rsql_create_indexrrsrrr?rr)r&rrrrrrUrr0rVrrrmrr)rr&rrs   z*BaseDatabaseSchemaEditor._create_index_sqlcsjjrjjsjjrgSg}x"jjD]}|||q*Wx8jjD],}fdd|D}||j |ddqNWx"jj D]}|| |qW|S)z Return a list of all index SQL statements (field indexes, index_together, Meta.indexes) for the specified model. csg|]}j|qSr)rrj)rr)rrrrlsz?BaseDatabaseSchemaEditor._model_indexes_sql..r)r) rZmanagedproxyZswappedrorprZindex_togetherr9rrr)r&routputrrrrr)rrr|sz+BaseDatabaseSchemaEditor._model_indexes_sqlcCs(g}|||r$||||g|S)zT Return a list of all index SQL statements for the specified field. )_field_should_be_indexedr9r)r&rrrrrrrs z+BaseDatabaseSchemaEditor._field_indexes_sqlcCs|jo |j S)N)rrT)r&rrrrrrsz1BaseDatabaseSchemaEditor._field_should_be_indexedcCs*|j||||j||j|dS)N)rmZ old_columnZ new_columnrJ)sql_rename_columnr?rk)r&rmrrrrrrrs   z*BaseDatabaseSchemaEditor._rename_field_sqlc s|jj}|j}t|jjjj\}}|jj}fdd} tjt|j t ||g||g|| t ||gj t|jjjjj t |jjjj|gj j j dS)Ncsj||S)N)r?r)rFr)r&rrcreate_fk_namesz?BaseDatabaseSchemaEditor._create_fk_sql..create_fk_name)rmrrkrhriZ deferrable)rrsrkrZ target_fieldrr sql_create_fkrr?rrr r>Zdeferrable_sql) r&rrrZ from_tableZ from_columnrrhrirr)r&rrws  z'BaseDatabaseSchemaEditor._create_fk_sqlcCs8|jj}t|jt||jt||d|jt|||jdS)N_uniq)rmrr) rrsrsql_create_uniquerr?rrr)r&rrrmrrrrys  z+BaseDatabaseSchemaEditor._create_unique_sqlcCs|||jj||dS)N)rmr)r?rrs)r&templaterrrrrrs z/BaseDatabaseSchemaEditor._delete_constraint_sqlc  s|dk rfdd|D}j} jj| |jj} WdQRXg} x| D]\} } |dksn|| dkrR|dk r| d|krqR|dk r| d|krqR|dk r| d|krqR|dk r| d|krqR|dk r| d sqR|dk r| d |krqR| | qRW| S) z@Return all constraint names matching the columns and conditions.Ncsg|]}jj|qSr)r introspectionZcolumn_name_converter)rr)r&rrrlsz>BaseDatabaseSchemaEditor._constraint_names..rrTrrrfrrJ)r r=rZget_constraintsrrsitemsr9)r&rrrTrrrrfrr= constraintsresultrZinfodictr)r&rrs,  z*BaseDatabaseSchemaEditor._constraint_namescCs^|j|dd}|r6t|dkr6tdt||jjfx"|D]}|||j||qs