|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/django/db/migrations/__pycache__/ |
Upload File : |
o
3�a�? � @ s~ d dl Z d dlZd dlmZmZ d dlmZ d dlmZ d dl m
Z
d dl m Z ddl
mZmZmZmZ d ZG d
d � d �ZdS ) � N)�
import_module�reload)�apps)�settings)�MigrationGraph)�MigrationRecorder� )�AmbiguityError�BadMigrationError� InconsistentMigrationHistory�NodeNotFoundError�
migrationsc @ s� e Zd ZdZ d dd�Zedd� �Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd dd �Zd d � ZdS )!�MigrationLoaderal
Load migration files from disk and their status from the database.
Migration files are expected to live in the "migrations" directory of
an app. Their names are entirely unimportant from a code perspective,
but will probably follow the 1234_name.py convention.
On initialization, this class will scan those directories, and open and
read the Python files, looking for a class called Migration, which should
inherit from django.db.migrations.Migration. See
django.db.migrations.migration for what that looks like.
Some migrations will be marked as "replacing" another set of migrations.
These are loaded into a separate set of migrations away from the main ones.
If all the migrations they replace are either unapplied or missing from
disk, then they are injected into the main set, replacing the named migrations.
Any dependency pointers to the replaced migrations are re-pointed to the
new migration.
This does mean that this class MUST also talk to the database as well as
to disk, but this is probably fine. We're already not just operating
in memory.
TFc C s2 || _ d | _d | _|| _|| _|r| �� d S d S �N)�
connection�disk_migrations�applied_migrations�ignore_no_migrations�replace_migrations� build_graph)�selfr �loadr r � r �=/usr/lib/python3/dist-packages/django/db/migrations/loader.py�__init__+ s �zMigrationLoader.__init__c C s4 |t jv r t j| dfS t�|�j}d|tf dfS )z�
Return the path to the migrations module for the specified app_label
and a boolean indicating if the module is specified in
settings.MIGRATION_MODULE.
T�%s.%sF)r �MIGRATION_MODULESr �get_app_config�name�MIGRATIONS_MODULE_NAME)�cls� app_label�app_package_namer r r �migrations_module7 s
z!MigrationLoader.migrations_modulec C s� i | _ t� | _t� | _t�� D ]�}| �|j�\}}|du r%| j�|j� q|t j
v }zt |�}W n) t yY } z |r=| j
sG|sTt|j�d�v rT| j�|j� W Y d}~q� d}~ww t|d�sg| j�|j� qt|dd�du r}t|jt�s}| j�|j� q|r�t|� | j�|j� dd� t�|j�D �}|D ]D}d||f } zt | �}
W n ty� } zdt|�v r�td | �|�� d}~ww t|
d
�s�td ||jf ��|
� ||j�| j |j|f< q�qdS ) z6Load the migrations from all INSTALLED_APPS from disk.N�.�__path__�__file__c S s& h | ]\}}}|s|d dvr|�qS )r z_~r )�.0�_r �is_pkgr r r � <setcomp>l s � �z,MigrationLoader.load_disk.<locals>.<setcomp>r zbad magic numberz9Couldn't import %r as it appears to be a stale .pyc file.� Migrationz-Migration %s in app %s has no Migration class) r �set�unmigrated_apps�
migrated_appsr �get_app_configsr# �label�add�sys�modulesr �ModuleNotFoundErrorr r r �split�hasattr�getattr�
isinstancer% �listr �pkgutil� iter_modules� ImportError�strr
r+ ) r �
app_config� module_name�explicit�
was_loaded�module�e�migration_names�migration_name�migration_path�migration_moduler r r � load_diskD s�
��� ��
�
� �����
����zMigrationLoader.load_diskc C s | j j||f S )z6Return the named migration or raise NodeNotFoundError.��graph�nodes)r r! � name_prefixr r r �
get_migration� s z MigrationLoader.get_migrationc C sr g }| j D ]\}}||kr|�|�r|�||f� qt|�dkr(td||f ��|s2td||f ��| j |d S )zZ
Return the migration(s) which match the given app label and name_prefix.
r z>There is more than one migration for '%s' with the prefix '%s'z1There no migrations for '%s' with the prefix '%s'r )r �
startswith�append�lenr �KeyError)r r! rL �results�migration_app_labelrE r r r �get_migration_by_prefix� s �
�z'MigrationLoader.get_migration_by_prefixc C s� |d dkr |d dks|| j v r|S |d |krd S |d | jv r$d S |d | jv r]z |d dkr=| j �|d �d W S | j �|d �d W S ty\ | jrTY d S td|d ��w td|d ��)Nr � __first__�
__latest__r z(Dependency on app with no migrations: %sz Dependency on unknown app: %s)rJ r- r. �
root_nodes�
leaf_nodes�
IndexErrorr �
ValueError)r �key� current_appr r r � check_key� s" " �zMigrationLoader.check_keyc C s@ |j D ]}|d |d kr |d dkr | jj|||dd� qdS )z�
Internal dependencies need to be added first to ensure `__first__`
dependencies find the correct root node.
r r rU T��skip_validationN)� dependenciesrJ �add_dependency)r r[ � migration�parentr r r �add_internal_dependencies� s
��z)MigrationLoader.add_internal_dependenciesc C s� |j D ]!}|d |d krq| �||d �}|d ur$| jj|||dd� q|jD ]}| �||d �}|d ur@| jj|||dd� q(d S )Nr Tr^ )r` r] rJ ra �
run_before)r r[ rb rc �childr r r �add_external_dependencies� s
�
��z)MigrationLoader.add_external_dependenciesc
s � � � � jdu r
i � _n
t� j�}|�� � _t� � _i � _� j�� D ]\}}� j� ||� |j
r6|� j|< q#� j�� D ]
\}}� � ||� q<� j�� D ]
\}}� � ||� qL� j
r�� j�� D ]8\}}� fdd�|j
D �}t|�rw|� j|< n� j�|d� t|�s�t|�s�� j�||j
� q_� j�||j
� q_z� j�� W nb t�y } zUi }� j�� D ]\}}|j
D ] }|�|t� ��|� q�q�|j|v r�|�|jt� �}t� fdd�|D ��} | s�d�dd� |D ��}
td�|j |jd |jd
|
�|j�|�� d}~ww � j� � dS ) z�
Build a migration dependency graph using both the disk and database.
You'll need to rebuild the graph if you apply migrations. This isn't
usually a problem as generally migration stuff runs in a one-shot process.
Nc s g | ]}|� j v �qS r )r )r' �target�r r r �
<listcomp>� s z/MigrationLoader.build_graph.<locals>.<listcomp>c 3 s � | ] }|� j jv V qd S r rI )r' � candidateri r r � <genexpr> s � z.MigrationLoader.build_graph.<locals>.<genexpr>z, c s s � | ]}d | V qdS )r Nr )r' �cr r r rl � � z�Migration {0} depends on nonexistent node ('{1}', '{2}'). Django tried to replace migration {1}.{2} with any of [{3}] but wasn't able to because some of the replaced migrations are already applied.r r ) rH r r r r rJ � replacementsr �items�add_node�replacesrd rg r �all�pop�any�remove_replaced_nodes�remove_replacement_node�validate_consistencyr �
setdefaultr, r1 �node�get�join�format�origin�ensure_not_cyclic) r �recorderr[ rb �applied_statuses�exc�reverse_replacements�replaced�
candidates� is_replaced�triesr ri r r � sf
�
�
��� ��zMigrationLoader.build_graphc
s� t |�}|�� � � D ]@}|| jjvrq
| jj| jD ]/}|� vrI|| jv r5t� fdd�| j| jD ��r5qt d�
|d |d |d |d |j ���qq
dS )zs
Raise InconsistentMigrationHistory if any applied migrations have
unapplied dependencies.
c 3 s � | ]}|� v V qd S r r )r' �m��appliedr r rl 0 rn z;MigrationLoader.check_consistent_history.<locals>.<genexpr>zHMigration {}.{} is applied before its dependency {}.{} on database '{}'.r r N) r r rJ rK �node_map�parentsro rs rr r r} �alias)r r r� rb rc r r� r �check_consistent_history s(
�����z(MigrationLoader.check_consistent_historyc sV i � t � }| j�� D ]\}}|� v r|�|� � �|t � ��|� q
� fdd�|D �S )z�
Look through the loaded graph and detect any conflicts - apps
with more than one leaf migration. Return a dict of the app labels
that conflict with the migration names that conflict.
c s i | ] }|t � | ��qS r )�sorted)r' r! �� seen_appsr r �
<dictcomp>F s z4MigrationLoader.detect_conflicts.<locals>.<dictcomp>)r, rJ rX r1 ry )r �conflicting_appsr! rE r r� r �detect_conflicts: s
z MigrationLoader.detect_conflictsNc C s | j j||t| j�d�S )z�
Return a ProjectState object representing the most recent state
that the loaded migrations represent.
See graph.make_state() for the meaning of "nodes" and "at_end".
)rK �at_end� real_apps)rJ �
make_stater9 r- )r rK r� r r r �
project_stateH s z MigrationLoader.project_statec C s� g }d}|D ]E\}}| j jd|jd��*}|du r#| j|j|jfdd�}|s.|j||dd�}n|j||dd�}W d � n1 s@w Y |�|j � q|S )z�
Take a migration plan and return a list of collected SQL statements
that represent the best-efforts version of that plan.
NT)� collect_sql�atomicF)r� )r� )
r �
schema_editorr� r� r! r �apply�unapply�extend�
collected_sql)r �plan�
statements�staterb � backwardsr� r r r r� Q s � �zMigrationLoader.collect_sql)TFT)NT)�__name__�
__module__� __qualname__�__doc__r � classmethodr# rH rM rT r] rd rg r r� r� r� r� r r r r r s$
�
B
Q
r )r: r2 � importlibr r � django.appsr � django.confr �django.db.migrations.graphr � django.db.migrations.recorderr �
exceptionsr r
r r r r r r r r �<module> s