File: //lib/python3/dist-packages/django/db/migrations/__pycache__/executor.cpython-310.pyc
o
3�a�E � @ s^ d dl mZ d dlmZmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
G dd � d �Zd
S ) � )�apps)�
migrations�router� )�InvalidMigrationPlan)�MigrationLoader)�MigrationRecorder)� ProjectStatec @ st e Zd ZdZddd�Zddd�Zddd �Zd d
d �Zd d
� Zdd� Z d dd�Z
dd� Z ddd�Z dd� Z
dd� ZdS ) �MigrationExecutorzu
End-to-end migration execution - load migrations and run them up or down
to a specified set of targets.
Nc C s( || _ t| j �| _t| j �| _|| _d S �N)�
connectionr �loaderr �recorder�progress_callback)�selfr r � r �?/usr/lib/python3/dist-packages/django/db/migrations/executor.py�__init__ s
zMigrationExecutor.__init__Fc sP g }|ri }nt | jj�}|D ]�� � d du rH| jj�� D ])}|d � d krF| jj�|�D ]}||v rE|�| jjj| df� |�|� q.q q� |v r�t � fdd�| jjj
� j D ��}|D ]!}| jj�|�D ]}||v r|�| jjj| df� |�|� qhq_q| jj� � �D ]}||vr�|�| jjj| df� | jjj| ||< q�q|S )z\
Given a set of targets, return a list of (Migration instance, backwards?).
r Nr Tc 3 s$ � | ]
}|d � d kr|V qdS )r Nr )�.0�n��targetr r � <genexpr>/ s � ��z3MigrationExecutor.migration_plan.<locals>.<genexpr>F)
�dictr
�applied_migrations�graph�
root_nodes�backwards_plan�append�nodes�pop�sorted�node_map�children�
forwards_plan) r �targets� clean_start�plan�applied�root� migration� next_in_app�noder r r �migration_plan sB
�� �
�����z MigrationExecutor.migration_planc sj t t� jj�d�}|r3� j� jj�� dd�}� fdd�� jjD �}|D ]\}}||v r2|j|dd� q#|S )z�
Create a project state including all the applications without
migrations and applied migrations if with_applied_migrations=True.
)� real_appsT�r& c �( h | ]}|� j jjv r� j jj| �qS r �r
r r �r �key�r r r � <setcomp>I �
�z:MigrationExecutor._create_project_state.<locals>.<setcomp>F��preserve) r �listr
�unmigrated_appsr- r �
leaf_nodesr � mutate_state)r �with_applied_migrations�state� full_planr r* �_r r4 r �_create_project_state@ s
� �z'MigrationExecutor._create_project_statec C s� | j �� |du r| �|�}| j| jj�� dd�}tdd� |D ��}tdd� |D ��}|s8|du r7| jdd�}n(||krAtd|��|rX|du rM| jdd�}| j |||||d �}n| j
|||d
�}| � � |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.
NTr/ c s s � | ]\}}| V qd S r r �r �mig� backwardsr r r r b s � z,MigrationExecutor.migrate.<locals>.<genexpr>c s s � | ]\}}|V qd S r r rB r r r r c � � )r= z�Migration 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�rG ) r �
ensure_schemar- r
r r; �allrA r �_migrate_all_forwards�_migrate_all_backwards�check_replacements) r r% r' r> rG rH r? � all_forwards�
all_backwardsr r r �migrateR s,
�� zMigrationExecutor.migratec C s~ dd� |D �}|D ]3\}}|s |S ||v r<d|j vr.| jr#| �d� |j | jr.| �d� | j||||d�}|�|� q |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.
c S � h | ]}|d �qS �r r �r �mr r r r5 � � z:MigrationExecutor._migrate_all_forwards.<locals>.<setcomp>r � render_start�render_successrF )�__dict__r r �apply_migration�remove) r r> r' r? rG rH �migrations_to_runr* r@ r r r rL s �
�z'MigrationExecutor._migrate_all_forwardsc sJ dd� |D �}i }� � � }� fdd�� jjD �}� jr � �d� |D ]0\}} |s* n)||v rGd|jvr6|j |||< |j|dd�}|�|� q"||v rR|j|dd� q"� jr[� �d � |D ]\}} � j|| ||d
� |�|� q]|d d }
||
}t |�D ]#\} \}} ||
kr�|| d
� D ]\}} ||v r�|j|dd� q� |S q|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.
c S rR rS r rT r r r r5 � rV z;MigrationExecutor._migrate_all_backwards.<locals>.<setcomp>c r0 r r1 r2 r4 r r r5 � r6 rW r Tr7 FrX rI ���r N)
rA r
r r rY r r< r[ �unapply_migration� enumerate) r r' r? rG r\ �statesr> r r* r@ �last_unapplied_migration�indexr r4 r rM � sJ
�
�
��z(MigrationExecutor._migrate_all_backwardsc C s� d}| j r | � d||� |sF|r | �||�\}}|r d}|sF| jj|jd��}|�||�}|js7| �|� d}W d � n1 sAw Y |sM| �|� | j rW| � d||� |S )zRun a migration forwards.F� apply_startT��atomicN�
apply_success)r �detect_soft_appliedr �
schema_editorre �apply� deferred_sql�record_migration)r r> r* rG rH �migration_recordedr( rh r r r rZ � s*
� �
z!MigrationExecutor.apply_migrationc C s> |j r|j D ] \}}| j�||� qd S | j�|j|j� d S r )�replacesr �record_applied� app_label�name)r r* ro rp r r r rk � s
�z"MigrationExecutor.record_migrationc C s� | j r
| � d||� |s*| jj|jd��}|�||�}W d � n1 s%w Y |jr=|jD ] \}}| j�||� q0n | j�|j|j � | j rP| � d||� |S )zRun a migration backwards.�
unapply_startrd N�unapply_success)
r r rh re �unapplyrm r �record_unappliedro rp )r r> r* rG rh ro rp r r r r^ � s ��z#MigrationExecutor.unapply_migrationc sT | j �� � | jj�� D ] \}}t� fdd�|jD ��}|r'|� vr'| j j|� q dS )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.
c 3 s � | ]}|� v V qd S r r rT �r( r r r rE z7MigrationExecutor.check_replacements.<locals>.<genexpr>N)r r r
� replacements�itemsrK rm rn )r r3 r* � all_appliedr ru r rN s
��z$MigrationExecutor.check_replacementsc sx �fdd�}� j du r t� fdd�� jD ��rd|fS n � j du r%d|fS |du r6�jj� j� jfdd�}n� �|�}|j}d}d}�j j
j }�j � � �} t
�j j�| ��}
|r_d d
� |
D �}
W d � n1 siw Y � jD ]�} t| tj�r�|�� j| j�} | jjr�t�| jj�} |� | �r�qq| jj}
|r�|
�� }
|
|
vr�d|f S d}qqt| tj��r4|�� j| j�} | jjr�t�| jj�} |� | �r�qq| jj}| j� | j�}|j r�|j j jj}|r�|�� }||
vr�d|f S d}qq�j � � �} �j j� | |�}W d � n 1 �s w Y |D ] }|j!}|j}|�r#|�� }|�� }||k�r,d} n�qd|f S qq|�p9||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).
c s. |j jp|j j ptj� jj| j|j jd� 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) �_meta�proxy�managedr �
allow_migrater �aliasro ry )r* �modelr4 r r �should_skip_detecting_model s
���zJMigrationExecutor.detect_soft_applied.<locals>.should_skip_detecting_modelNc 3 s � | ]
\}}|� j kV qd S r )ro )r �apprp )r* r r r . s � z8MigrationExecutor.detect_soft_applied.<locals>.<genexpr>FT)�at_endc S s h | ]}|� � �qS r )�casefold)r rp r r r r5 ? rV z8MigrationExecutor.detect_soft_applied.<locals>.<setcomp>)"�initial�any� dependenciesr
�
project_statero rp r< r r �features�ignores_table_name_case�cursor�set�
introspection� table_names�
operations�
isinstancer � CreateModel� get_modelrz �swapped� global_apps�db_tabler� �AddFieldry � get_field� many_to_many� remote_field�through�get_table_description�column)r r� r* r� � after_stater � found_create_model_migration�found_add_field_migration�fold_identifier_caser� �existing_table_names� operationr r� �table�field�through_db_table�columnsr� � field_column� column_namer )r* r r rg s�
�
� �
�
� �z%MigrationExecutor.detect_soft_appliedr )F)NNFF)FF)�__name__�
__module__� __qualname__�__doc__r r- rA rQ rL rM rZ rk r^ rN rg r r r r r
s
*
-
=
r
N)�django.apps.registryr r� � django.dbr r �
exceptionsr r
r r r r> r r
r r r r �<module> s