3 gE@s^ddlmZddlmZmZddlmZddlm Z ddl m Z ddl m Z Gdd d Zd S) )apps) migrationsrouter)InvalidMigrationPlan)MigrationLoader)MigrationRecorder) ProjectStatec@steZdZdZdddZdddZddd Zdd d Zd d ZddZ dddZ ddZ dddZ ddZ ddZdS) MigrationExecutorzu End-to-end migration execution - load migrations and run them up or down to a specified set of targets. NcCs(||_t|j|_t|j|_||_dS)N) connectionrloaderrrecorderprogress_callback)selfr rr[/var/www/tester-filtro-web/env/lib/python3.6/site-packages/django/db/migrations/executor.py__init__s  zMigrationExecutor.__init__Fc srg}|ri}n t|jj}xP|D]Fddkrxf|jjjD]V}|ddkrBx@|jjj|D].}||krf|j|jjj|df|j|qfWqBWq"|krt fdd|jjj j D}x|D]F}x@|jjj|D].}||kr|j|jjj|df|j|qWqWq"xL|jjj D]:}||kr,|j|jjj|df|jjj|||<q,Wq"W|S)z\ Given a set of targets, return a list of (Migration instance, backwards?). rNrTc3s"|]}|ddkr|VqdS)rNr).0n)targetrr 0sz3MigrationExecutor.migration_plan..F) dictr applied_migrationsgraphZ root_nodesZbackwards_planappendnodespopsortedZnode_mapchildrenZ forwards_plan) rtargets clean_startplanappliedroot migrationZ next_in_appnoder)rrmigration_plans4      z MigrationExecutor.migration_plancsnttjjd}|rjjjjjdd}fddjjD}x&|D]\}}||krH|j|ddqHW|S)z Create a project state including all the applications without migrations and applied migrations if with_applied_migrations=True. )Z real_appsT)r cs(h|] }|jjjkrjjj|qSr)r rr)rkey)rrr Isz:MigrationExecutor._create_project_state..F)preserve) r listr Zunmigrated_appsr&r leaf_nodesr mutate_state)rwith_applied_migrationsstate full_planrr$_r)rr_create_project_state@s  z'MigrationExecutor._create_project_statec Cs|jj|dkr|j|}|j|jjjdd}tdd|D}tdd|D}|sp|dkr|jdd}nR||krtd|n>|r|dkr|jdd}|j |||||d }n|j |||d }|j |S) z Migrate the database up to the given targets. Django first needs to create all project states before a migration is (un)applied and in a second step run all the database operations. NT)r css|]\}}| VqdS)Nr)rmig backwardsrrrrbsz,MigrationExecutor.migrate..css|]\}}|VqdS)Nr)rr2r3rrrrcs)r-zMigration plans with both forwards and backwards migrations are not supported. Please split your migration process into separate plans of only forwards OR backwards migrations.)fake fake_initial)r4) r Z ensure_schemar&r rr+allr1r_migrate_all_forwards_migrate_all_backwardscheck_replacements) rrr!r.r4r5r/Z all_forwardsZ all_backwardsrrrmigrateRs(   zMigrationExecutor.migratec Cs~dd|D}xj|D]b\}}|s"P||krd|jkrZ|jrD|jd|j|jrZ|jd|j||||d}|j|qW|S)z Take a list of 2-tuples of the form (migration instance, False) and apply them in the order they occur in the full_plan. cSsh|] }|dqS)rr)rmrrrr(sz:MigrationExecutor._migrate_all_forwards..r render_startrender_success)r4r5)__dict__rrapply_migrationremove) rr.r!r/r4r5migrations_to_runr$r0rrrr7s   z'MigrationExecutor._migrate_all_forwardsc s\dd|D}i}j}fddjjD}jr@jdxf|D]^\}} |sTP||krd|jkrl|j|||<|j|dd}|j|qF||krF|j|ddqFWjrjd x.|D]&\}} j||||d |j|qW|dd } || }xZt |D]N\} \}} || krx2|| d D]"\}} ||kr*|j|ddq*WPqW|S)a Take a list of 2-tuples of the form (migration instance, True) and unapply them in reverse order they occur in the full_plan. Since unapplying a migration requires the project state prior to that migration, Django will compute the migration states before each of them in a first run over the plan and then unapply them in a second run over the plan. cSsh|] }|dqS)rr)rr;rrrr(sz;MigrationExecutor._migrate_all_backwards..cs(h|] }|jjjkrjjj|qSr)r rr)rr')rrrr(sr<rT)r)Fr=)r4rrN) r1r rrr>rr,r@unapply_migration enumerate) rr!r/r4rAZstatesr.rr$r0Zlast_unapplied_migrationindexr)rrr8s@          z(MigrationExecutor._migrate_all_backwardsc Csd}|jr|jd|||sx|r8|j||\}}|r8d}|sx|jj|jd&}|j||}|jsn|j|d}WdQRX|s|j||jr|jd|||S)zRun a migration forwards.FZ apply_startT)atomicNZ apply_success)rdetect_soft_appliedr schema_editorrFapplyZ deferred_sqlrecord_migration)rr.r$r4r5Zmigration_recordedr"rHrrrr?s&   z!MigrationExecutor.apply_migrationcCs@|jr*x4|jD]\}}|jj||qWn|jj|j|jdS)N)replacesr record_applied app_labelname)rr$rMrNrrrrJsz"MigrationExecutor.record_migrationc Cs|jr|jd|||s@|jj|jd}|j||}WdQRX|jrjx4|jD]\}}|jj||qNWn|jj|j|j |jr|jd|||S)zRun a migration backwards.Z unapply_start)rFNZunapply_success) rr rHrFZunapplyrKr Zrecord_unappliedrMrN)rr.r$r4rHrMrNrrrrCsz#MigrationExecutor.unapply_migrationcsX|jjxH|jjjD]8\}}tfdd|jD}|r|kr|jj|qWdS)a Mark replacement migrations applied if their replaced set all are. Do this unconditionally on every migrate, rather than just when migrations are applied or unapplied, to correctly handle the case when a new squash migration is pushed to a deployment that already had all its replaced migrations applied. In this case no new migration will be applied, but the applied state of the squashed migration must be maintained. c3s|]}|kVqdS)Nr)rr;)r"rrrsz7MigrationExecutor.check_replacements..N)r rr replacementsitemsr6rKrL)rr'r$Z all_appliedr)r"rr9s  z$MigrationExecutor.check_replacementscs^fdd}jdkr8tfddjDrJd|fSnjdkrJd|fS|dkrljjjjfdd}n j|}|j}d}d}j j j }j j *} t j jj| } |rd d | D} WdQRXxjD]x} t| tjrJ|jj| j} | jjrtj| jj} || rq| jj} |r2| j} | | krDd|fSd}qt| tjr|jj| j} | jjr~tj| jj} || rq| jj}| jj| j}|jr|jjjj}|r|j}|| krd|fSd}q҈j j } j jj | |}WdQRXxH|D]8}|j!}|j}|r.|j}|j}||krd}PqWd|fSqW|pX||fS) z Test whether a migration has been implicitly applied - that the tables or columns it would create exist. This is intended only for use on initial migrations (as it only looks for CreateModel and AddField). cs.|jjp,|jj p,tjjj|j|jjd S)z No need to detect tables for proxy models, unmanaged models, or models that can't be migrated on the current database. ) model_name) _metaproxyZmanagedrZ allow_migrater aliasrMrQ)r$model)rrrshould_skip_detecting_models zJMigrationExecutor.detect_soft_applied..should_skip_detecting_modelNc3s|]\}}|jkVqdS)N)rM)rZapprN)r$rrr.sz8MigrationExecutor.detect_soft_applied..FT)Zat_endcSsh|] }|jqSr)casefold)rrNrrrr(?sz8MigrationExecutor.detect_soft_applied..)"initialany dependenciesr project_staterMrNr,rr featuresZignores_table_name_casecursorsetZ introspectionZ table_names operations isinstancerZ CreateModelZ get_modelrRZswapped global_appsdb_tablerWZAddFieldrQ get_fieldZ many_to_manyZ remote_fieldZthroughZget_table_descriptioncolumn)rr[r$rVZ after_staterZfound_create_model_migrationZfound_add_field_migrationZfold_identifier_caser]Zexisting_table_namesZ operationrUrbtablefieldZthrough_db_tablecolumnsrdZ field_columnZ column_namer)r$rrrGsv                  z%MigrationExecutor.detect_soft_applied)N)F)F)NNFF)FF)F)__name__ __module__ __qualname____doc__rr&r1r:r7r8r?rJrCr9rGrrrrr s  *  -=  r N)Zdjango.apps.registryrraZ django.dbrr exceptionsrr rr rr.r r rrrrs