OwlCyberSecurity - MANAGER
Edit File: loader.cpython-39.pyc
a ���e�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�NodeNotFoundErrorZ migrationsc @ s� e Zd ZdZd 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 s. || _ d | _d | _|| _|| _|r*| �� d S �N)� connection�disk_migrations�applied_migrations�ignore_no_migrations�replace_migrations� build_graph)�selfr �loadr r � r �\/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-packages/django/db/migrations/loader.py�__init__- s zMigrationLoader.__init__c C s8 |t jv rt j| dfS t�|�j}d|tf dfS dS )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.%sFN)r ZMIGRATION_MODULESr Zget_app_config�name�MIGRATIONS_MODULE_NAME)�cls� app_labelZapp_package_namer r r �migrations_module< s z!MigrationLoader.migrations_modulec C s� i | _ t� | _t� | _t�� D �]�}| �|j�\}}|du rL| j�|j� q |t j v }z t |�}W n^ t y� } zF|r|| j s�|s�t|j�d�v r�| j�|j� W Y d}~q � W Y d}~nbd}~0 0 t|d�s�| j�|j� q t|dd�du �r t|jt��s | j�|j� q |�rt|� | j�|j� dd� t�|j�D �}|D ]�}d||f } z t | �} W nF t�y� } z,dt|�v �r�td | �|�n� W Y d}~n d}~0 0 t| d ��s�td ||jf ��| � ||j�| j |j|f< �qBq dS ) 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 Zis_pkgr r r � <setcomp>o 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 Zget_app_configsr �label�add�sys�modulesr �ModuleNotFoundErrorr r r �split�hasattr�getattr� isinstancer! �listr �pkgutil� iter_modules� ImportError�strr r& ) r Z app_config� module_name�explicitZ was_loaded�module�eZmigration_names�migration_nameZmigration_pathZmigration_moduler r r � load_diskI sz �� � � ��� �����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 s� g }| j D ](\}}||kr |�|�r |�||f� q t|�dkrRtd||f ��n*|sntd|� d|� d���n| j |d S dS )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'zThere is no migration for 'z' with the prefix '�'r N)r � startswith�append�lenr �KeyError)r r rA �resultsZmigration_app_labelr< 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 |kr6d S |d | jv rHd S |d | jv r�z<|d dkrz| j �|d �d W S | j �|d �d W S W n. ty� | jr�Y d S td|d ��Y n0 td|d ��d S )Nr � __first__Z __latest__r z(Dependency on app with no migrations: %sz Dependency on unknown app: %s)r? r( r) Z root_nodes� leaf_nodes� IndexErrorr � ValueError)r �keyZ current_appr r r � check_key� s$ " � zMigrationLoader.check_keyc C s@ |j D ]4}|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 rJ T�Zskip_validationN)� dependenciesr? �add_dependency)r rN � migration�parentr r r �add_internal_dependencies� s z)MigrationLoader.add_internal_dependenciesc C s� |j D ]B}|d |d kr q| �||d �}|d ur| jj|||dd� q|jD ]0}| �||d �}|d urP| jj|||dd� qPd S )Nr TrP )rQ rO r? rR Z run_before)r rN rS rT �childr r r �add_external_dependencies� s z)MigrationLoader.add_external_dependenciesc s, � � � � jdu ri � _nt� j�}|�� � _t� � _i � _� j�� D ]&\}}� j� ||� |j rF|� j|< qF� j�� D ]\}}� � ||� qx� j�� D ]\}}� � ||� q�� j �r6� j�� D ]t\}}� fdd�|j D �}t|�r�|� j|< n� j�|d� t|��st|��s$� j�||j � q�� j�||j � q�z� j�� W n� t�y } z�i }� j�� D ],\}}|j D ]}|�|t� ��|� �qr�qd|j|v �r|�|jt� �}t� fdd�|D ��} | �sd�dd� |D ��} td�|j |jd |jd | �|j�|�� W Y d}~n d}~0 0 � 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 r> )r# � candidaterY r r � <genexpr>" s z.MigrationLoader.build_graph.<locals>.<genexpr>z, c s s | ]}d | V qdS )r Nr )r# �cr r r r\ & � 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 ) r= r r r r r? � replacementsr �items�add_node�replacesrU rW r �all�pop�anyZremove_replaced_nodesZremove_replacement_nodeZvalidate_consistencyr � setdefaultr' r+ �node�get�join�format�originZensure_not_cyclic) r �recorderrN rS Zapplied_statuses�excZreverse_replacementsZreplaced� candidatesZ is_replaced�triesr rY r r � sd � ���� zMigrationLoader.build_graphc s� t |�}|�� � � D ]�}|| jjvr&q| jj| jD ]^}|� vr4|| jv rjt� fdd�| j| jD ��rjq4t d� |d |d |d |d |j ���q4qdS )zs Raise InconsistentMigrationHistory if any applied migrations have unapplied dependencies. c 3 s | ]}|� v V qd S r r )r# �m�Zappliedr r r\ C s z;MigrationLoader.check_consistent_history.<locals>.<genexpr>zHMigration {}.{} is applied before its dependency {}.{} on database '{}'.r r N) r r r? r@ Znode_map�parentsr_ rc rb r rj �alias)r r rl rS rT r rq r �check_consistent_history3 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 �Z seen_appsr r � <dictcomp>^ s z4MigrationLoader.detect_conflicts.<locals>.<dictcomp>)r' r? rK r+ rf )r Zconflicting_appsr r< r rv r �detect_conflictsR s �z MigrationLoader.detect_conflictsNc C s | j j||| jd�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". )r@ �at_endZ real_apps)r? Z make_stater( )r r@ ry r r r � project_stateb s �z MigrationLoader.project_statec C s� g }d}|D ]�\}}| j jd|jd��T}|du rF| j|j|jfdd�}|s\|j||dd�}n|j||dd�}W d � n1 s�0 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)ry )r{ ) r � schema_editorr| rz r r �applyZunapply�extendZ collected_sql)r ZplanZ statements�staterS Z backwardsr} r r r r{ m s � �.zMigrationLoader.collect_sql)TFT)NT)�__name__� __module__� __qualname__�__doc__r � classmethodr r= rB rI rO rU rW r rt rx rz r{ r r r r r s$ � D U r )r4 r, � importlibr r Z django.appsr Z django.confr Zdjango.db.migrations.graphr Z django.db.migrations.recorderr � exceptionsr r r r r r r r r r �<module> s