OwlCyberSecurity - MANAGER
Edit File: executor.cpython-39.pyc
a ���e�I � @ 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�Zd dd �Zd d d �Zd d � Zdd� Z d dd�Z dd� Z d dd�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 �^/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-packages/django/db/migrations/executor.py�__init__ s zMigrationExecutor.__init__Fc s� g }|ri }n t | jj�}|D �]p� � d du r�| jj�� D ]R}|d � d kr<| jj�|�D ].}||v r^|�| jjj| df� |�|� q^q<q � |v �rF| jj rֈ | jjj vr�d| j_ | j� � | j ||d� S t � fdd�| jjj � jD ��}|D ]F}| jj�|�D ]2}||v �r|�| jjj| df� |�|� �qq�q | jj�� �D ]:}||v�rT|�| jjj| df� | jjj| ||< �qTq |S ) z\ Given a set of targets, return a list of (Migration instance, backwards?). r Nr TF�� clean_startc 3 s" | ]}|d � d kr|V qdS )r Nr )�.0�n��targetr r � <genexpr>8 s �z3MigrationExecutor.migration_plan.<locals>.<genexpr>)�dictr �applied_migrations�graphZ root_nodesZbackwards_plan�append�nodes�popZreplace_migrationsZnode_mapZ build_graph�migration_plan�sorted�childrenZ forwards_plan) r �targetsr �plan�applied�root� migrationZ next_in_app�noder r r r! sB � � � z MigrationExecutor.migration_planc sf t � jjd�}|rb� j� jj�� dd�}� fdd�� jjD �}|D ] \}}||v rB|j|dd� qB|S )z� Create a project state including all the applications without migrations and applied migrations if with_applied_migrations=True. )Z real_appsTr c s( h | ] }|� j jjv r� j jj| �qS r �r r r �r �key�r r r � <setcomp>T s �z:MigrationExecutor._create_project_state.<locals>.<setcomp>F�Zpreserve)r r Zunmigrated_appsr! r � leaf_nodesr � mutate_state)r �with_applied_migrations�state� full_planr r( �_r r- r �_create_project_stateI s � � z'MigrationExecutor._create_project_statec C s� |g kr | j �� s*| jdd�S n | j �� |du r<| �|�}| j| jj�� dd�}tdd� |D ��}tdd� |D ��}|s�|du r�| jdd�}nR||kr�t d |��n>|r�|du r�| 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. F)r2 NTr c s s | ]\}}| V qd S r r �r ZmigZ backwardsr r r r t � z,MigrationExecutor.migrate.<locals>.<genexpr>c s s | ]\}}|V qd S r r r7 r r r r u r8 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�r: ) r Z has_tabler6 Z ensure_schemar! r r r0 �allr �_migrate_all_forwards�_migrate_all_backwards�check_replacements) r r$ r% r3 r: r; r4 Z all_forwardsZ all_backwardsr r r �migrate^ s8 �� �zMigrationExecutor.migratec C s| dd� |D �}|D ]d\}}|s" qx||v rd|j vrZ| jrD| �d� |j | jrZ| �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 s h | ] }|d �qS �r r �r �mr r r r. � r8 z:MigrationExecutor._migrate_all_forwards.<locals>.<setcomp>r � render_start�render_successr9 )�__dict__r r �apply_migration�remove) r r3 r% r4 r: r; �migrations_to_runr( r5 r r r r> � s � z'MigrationExecutor._migrate_all_forwardsc sN dd� |D �}i }� � � }� fdd�� jjD �}� jr@� �d� |D ]`\}} |sT q�||v r�d|jvrl|j |||< |j|dd�}|�|� qD||v rD|j|dd� qD� jr�� �d � |D ]&\}} � j|| ||d � |�|� q�|d d } || }t |�D ]J\} \}} || kr�|| d � D ]"\}} ||v �r |j|dd� �q �qJq�|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 s h | ] }|d �qS rB r rC r r r r. � r8 z;MigrationExecutor._migrate_all_backwards.<locals>.<setcomp>c s( h | ] }|� j jjv r� j jj| �qS r r* r+ r- r r r. � s �rE r Tr/ FrF r< ���r N) r6 r r r rG r r1 rI �unapply_migration� enumerate) r r% r4 r: rJ Zstatesr3 r r( r5 Zlast_unapplied_migration�indexr r- r r? � sB � z(MigrationExecutor._migrate_all_backwardsc C s� d}| j r| � d||� |s�|r8| �||�\}}|r8d}|s�| jj|jd��0}|�||�}|jsn| �|� d}W d � n1 s�0 Y |s�| �|� | j r�| � d||� |S )zRun a migration forwards.FZ apply_startT��atomicNZ apply_success)r �detect_soft_appliedr � schema_editorrP �applyZ deferred_sql�record_migration)r r3 r( r: r; Zmigration_recordedr& rR r r r rH � s, � " z!MigrationExecutor.apply_migrationc C s< |j r&|j D ]\}}| j�||� q n| j�|j|j� d S r )�replacesr �record_applied� app_label�name)r r( rW rX r r r rT s z"MigrationExecutor.record_migrationc C s� | j r| � d||� |sT| jj|jd�� }|�||�}W d � n1 sJ0 Y |jrx|jD ]\}}| j�||� q`| j�|j|j � | j r�| � d||� |S )zRun a migration backwards.Z unapply_startrO NZunapply_success) r r rR rP ZunapplyrU r Zrecord_unappliedrW rX )r r3 r( r: rR rW rX r r r rL s �*z#MigrationExecutor.unapply_migrationc sT | j �� � | jj�� D ]8\}}t� fdd�|jD ��}|r|� vr| j j|� qdS )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 rC �r& r r r / r8 z7MigrationExecutor.check_replacements.<locals>.<genexpr>N)r r r Z replacements�itemsr= rU rV )r r, r( Z all_appliedr rY r r@ " s z$MigrationExecutor.check_replacementsc s� �fdd�}� j du r8t� fdd�� jD ��rJd|fS n� j du rJd|fS |du rl�jj� j� jfdd�}n � �|�}|j}d}d}�j j j }�j � � �4} t �j j�| ��} |r�d d � | D �} W d � n1 s�0 Y � jD �]�} t| tj��r^|�� j| j�} | jj�r t�| jj�} |� | ��r,q�| jj} |�rB| �� } | | v�rXd|f S d}q�t| tj�r�|�� j| j�} | jj�r�t�| jj�} |� | ��r�q�| jj}| j� | j�}|j �r�|j j jj}|�r�|�� }|| v�r�d|f S d}q�j � � � } �j j� | |�}W d � n1 �s&0 Y |D ]:}|j!}|j}|�rZ|�� }|�� }||k�r4d} q�q4d|f S q�|�p�||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 p,tj� 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�proxyZmanagedr Z allow_migrater �aliasrW r[ )r( �modelr- r r �should_skip_detecting_model: s ���zJMigrationExecutor.detect_soft_applied.<locals>.should_skip_detecting_modelNc 3 s | ]\}}|� j kV qd S r )rW )r ZapprX )r( r r r K r8 z8MigrationExecutor.detect_soft_applied.<locals>.<genexpr>FT)Zat_endc S s h | ] }|� � �qS r )�casefold)r rX r r r r. ` s z8MigrationExecutor.detect_soft_applied.<locals>.<setcomp>)"�initial�any� dependenciesr � project_staterW rX r1 r r �featuresZignores_table_name_case�cursor�setZ introspectionZ table_names� operations� isinstancer Z CreateModelZ get_modelr\ Zswapped� global_apps�db_tablera ZAddFieldr[ � get_fieldZ many_to_manyZ remote_fieldZthroughZget_table_description�column)r re r( r` Z after_stater Z found_create_model_migrationZfound_add_field_migrationZfold_identifier_caserg Zexisting_table_namesZ operationr_ rl �table�fieldZthrough_db_table�columnsrn Z field_columnZ column_namer )r( r r rQ 3 s� � ��$ �$ z%MigrationExecutor.detect_soft_applied)N)F)F)NNFF)FF)F)�__name__� __module__� __qualname__�__doc__r r! r6 rA r> r? rH rT rL r@ rQ r r r r r s 3 5> r N)Zdjango.apps.registryr rk Z django.dbr r � exceptionsr r r r r r3 r r r r r r �<module> s