Dashboard
PHP:
8.2.30
OS:
Linux
User:
dh_nv8b2m
/
/
usr
/
lib
/
python3
/
dist-packages
/
django
/
db
/
migrations
/
__pycache__
📤 Upload
📝 New File
📁 New Folder
Close
Editing: 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
Save
Cancel