OwlCyberSecurity - MANAGER
Edit File: autodetector.cpython-39.pyc
a ���e�4 � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z d dl m Z d dl m Z d dlmZ d d lmZ d d lmZ d d lmZmZmZ d d lmZ G d d� d�ZdS )� N)� defaultdict)�chain)�settings)�models)� operations)� Migration)�AlterModelOptions)�MigrationOptimizer)�MigrationQuestioner)�COMPILED_REGEX_TYPE� RegexObject�resolve_relation)�stable_topological_sortc @ s� e Zd ZdZdfdd�Zdgdd�Zdd� Zd d � Zdhd d �Zd d� Z dd� Z e dd� �Z didd�Z dd� Zdd� Zdd� Zdjd d �Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#dBdC� Z$dDdE� Z%e dFdG� �Z&dHdI� Z'dJdK� Z(dLdM� Z)dNdO� Z*dPdQ� Z+dRdS� Z,dTdU� Z-dVdW� Z.dXdY� Z/dZd[� Z0d\d]� Z1d^d_� Z2dkd`da�Z3dbdc� Z4e5ddde� �Z6dS )l�MigrationAutodetectora Take a pair of ProjectStates and compare them to see what the first would need doing to make it match the second (the second usually being the project's current state). Note that this naturally operates on entire projects at a time, as it's likely that changes interact (for example, you can't add a ForeignKey without having a migration to add the table it depends on first). A user interface may offer single-app usage if it wishes, with the caveat that it may not always be possible. Nc C s. || _ || _|pt� | _dd� |jD �| _d S )Nc S s h | ] \}}|�qS � r )�.0Zapp�modelr r �b/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-packages/django/db/migrations/autodetector.py� <setcomp>& � z1MigrationAutodetector.__init__.<locals>.<setcomp>)� from_state�to_stater � questionerr � existing_apps)�selfr r r r r r �__init__" s z MigrationAutodetector.__init__c C s. | � ||�}| �|||�}|r*| �||�}|S )z� Main entry point to produce a list of applicable changes. Take a graph to base names on and an optional set of apps to try and restrict to (restriction is not guaranteed) )�_detect_changes�arrange_for_graph� _trim_to_apps)r �graphZ trim_to_apps� convert_apps�migration_name�changesr r r r" ( s z MigrationAutodetector.changesc s t |t�r � fdd�|D �S t |t�r<t� fdd�|D ��S t |t�r\� fdd�|�� D �S t |tj�r�|j� �|j �� �|j �fS t |t �r�t |�S t |t �r�|S t|d��r|�� }t |tj�r�|dd � }|\}}}|� fd d�|D �� fd d�|�� D �fS |S d S ) z� Recursive deconstruction for a field and its arguments. Used for full comparison for rename/alter; sometimes a single-level deconstruction will not compare correctly. c s g | ]}� � |��qS r ��deep_deconstruct�r �value�r r r � <listcomp>; r z:MigrationAutodetector.deep_deconstruct.<locals>.<listcomp>c 3 s | ]}� � |�V qd S �Nr# r% r' r r � <genexpr>= r z9MigrationAutodetector.deep_deconstruct.<locals>.<genexpr>c s i | ]\}}|� � |��qS r r# �r �keyr& r' r r � <dictcomp>? r z:MigrationAutodetector.deep_deconstruct.<locals>.<dictcomp>� deconstruct� Nc s g | ]}� � |��qS r r# r% r' r r r( T r c s i | ]\}}|� � |��qS r r# r+ r' r r r- U r )� isinstance�list�tuple�dict�items� functools�partial�funcr$ �args�keywordsr r �type�hasattrr. r �Field)r �objZ deconstructed�pathr8 �kwargsr r' r r$ 4 s2 � �z&MigrationAutodetector.deep_deconstructc C sP g }t |�� �D ]:\}}| �|�}|jr@|jjr@|d �dd� |�|� q|S )z� Return a definition of the fields that ignores field names and what related fields actually relate to. Used for detecting renames (as the related fields change during renames). � �toN)�sortedr4 r$ � remote_fieldr �pop�append)r �fieldsZ fields_def�name�fieldZdeconstructionr r r � only_relation_agnostic_fieldsZ s z3MigrationAutodetector.only_relation_agnostic_fieldsc C sT i | _ i | _i | _i | _t� | _t� | _t� | _t� | _t� | _ t� | _ | j j � � D ]f\\}}}|j�dd�s�| j�||f� qT|| j jvrT|j�d�r�| j�||f� qT| j�||f� qT| jj � � D ]v\\}}}|j�dd�s�| j �||f� q�|| j jv�s|r�||v r�|j�d��r.| j �||f� q�| j�||f� q�| j �� | j�� | �� | �� | �� | �� | �� | �� | �� | �� | � � | � � | � � | � � | � � | �!� | �"� | �#� | �$� | �%� | �&� | �'� | �(� | �)� | �*� | �+� | �,� | �-� | �.� | �/� | �0� | �1|� | �2� | j3S )aX Return a dict of migration plans which will achieve the change from from_state to to_state. The dict has app labels as keys and a list of migrations as values. The resulting migrations aren't specially named, but the names do matter for dependencies inside the set. convert_apps is the list of apps to convert to use migrations (i.e. to make initial migrations for, in the usual case) graph is an optional argument that, if provided, can help improve dependency generation and avoid potential circular dependencies. �managedT�proxy)4�generated_operations�altered_indexes�altered_constraints�renamed_fields�set�old_model_keys�old_proxy_keys�old_unmanaged_keys�new_model_keys�new_proxy_keys�new_unmanaged_keysr r r4 �options�get�addZ real_appsr Z resolve_fields_and_relations�generate_renamed_models�_prepare_field_lists�_generate_through_model_map�generate_deleted_models�generate_created_models�generate_deleted_proxies�generate_created_proxies�generate_altered_options�generate_altered_managers�!generate_altered_db_table_comment�create_renamed_fields�create_altered_indexes�create_altered_constraints� generate_removed_constraints�generate_removed_indexes�generate_renamed_fields�generate_renamed_indexes�(generate_removed_altered_unique_together�'generate_removed_altered_index_together�generate_removed_fields�generate_added_fields�generate_altered_fields�&generate_altered_order_with_respect_to� generate_altered_unique_together� generate_altered_index_together�generate_added_indexes�generate_added_constraints�generate_altered_db_table�_sort_migrations�_build_migration_list�_optimize_migrations� migrations)r r r � app_label� model_name� model_stater r r r h s| �� z%MigrationAutodetector._detect_changesc s` � j � j@ � _� j� j@ � _� j� j@ � _i � _ � fdd�� jD �� _ � fdd�� jD �� _ dS )z� Prepare field lists and a list of the fields that used through models in the old state so dependencies can be made from the through model deletion to the field that uses it. c s@ h | ]8\}}� j j|� j�||f|�f jD ]}|||f�q,qS r )r r �renamed_modelsrX rF �r rz r{ � field_namer' r r r � s � �z=MigrationAutodetector._prepare_field_lists.<locals>.<setcomp>c s2 h | ]*\}}� j j||f jD ]}|||f�q qS r )r r rF r~ r' r r r � s �N) rQ rT �kept_model_keysrR rU �kept_proxy_keysrS rV �kept_unmanaged_keys� through_users�old_field_keys�new_field_keysr' r r' r r[ � s � �z*MigrationAutodetector._prepare_field_listsc C s� t | j�D ]v\}}| j�||f|�}| jj||f }|j�� D ]@\}}t|d�r>t |j dd�r>t |j j ||�}|||f| j |< q>q dS )z Through model map generation.rC �throughN)rB rQ r} rX r r rF r4 r; �getattrrC r r� r� )r rz r{ �old_model_name�old_model_stater rH Z through_keyr r r r\ � s � � ��z1MigrationAutodetector._generate_through_model_mapc C sH | d dkr| dfS t t| d ��d�\}}||�� f| dd� dfS ) zp Return the resolved dependency and a boolean denoting whether or not it was swappable. r � __setting__Fr/ �.r@ NT)r� r �split�lower)� dependencyZresolved_app_labelZresolved_object_namer r r �_resolve_dependency� s ��z)MigrationAutodetector._resolve_dependencyc C sr i | _ tdd� | j�� D ��}d}|�rnt| j�D �]�}g }t� }t| j| �D �]<}d}t� } |jD ]�} | } | �| �\} } | d |krf| j� | d g �D ]} | � | | �r�d} q�q�|s� �q`qf| r�| � | d | d f� qf| d | j v �r| � | d | j | d d j f� qf|�rZ|�rF|� | d ��rF| � |� | d �d � n| � | d df� qfd}qf|�r�|�|� |�| � | j| d= qP �q�qP|�s�|r2| j| �r�|�rtd tfg g d ��}|d t| j � |g ��d |�}t|�|_||_|| jv|_| j �|g ��|� d}q2|| j| | j|< q2td d� | j�� D ��}||k�rh|�sZd}ntd | j ��|}q"dS )ag Chop the lists of operations up into migrations with dependencies on each other. Do this by going through an app's list of operations until one is found that has an outgoing dependency that isn't in another app's migration yet (hasn't been chopped off its list). Then chop off the operations before it into a migration and move onto the next app. If the loops completes without doing anything, there's a circular dependency (which _should_ be impossible as the operations are all split at this point so they can't depend and be depended on). c s s | ]}t |�V qd S r) ��len�r �xr r r r* r z>MigrationAutodetector._build_migration_list.<locals>.<genexpr>FTr r/ ���� __first__r )r � dependencieszauto_%ic s s | ]}t |�V qd S r) r� r� r r r r* r r z)Cannot resolve operation dependencies: %rN)ry �sumrL �valuesrB rP r1 � _auto_depsr� rX �check_dependencyrY rG � leaf_nodesrE �updater: r r� r� r r �initial� setdefault� ValueError)r r Znum_opsZ chop_moderz Zchoppedr� � operationZdeps_satisfiedZoperation_dependencies�depZ original_depZis_swappable_depZother_operation�subclass�instanceZ new_num_opsr r r rw s� � ��� � �� � ��z+MigrationAutodetector._build_migration_listc C s� t | j�� �D ]z\}}dd� |D �}|D ]N}|jD ]B}| �|�d }|d |kr2|D ] }| �||�rT|| �|� qTq2q(t||�| j|< qdS )z{ Reorder to make things possible. Reordering may be needed so FKs work nicely inside the same app. c S s i | ] }|t � �qS r )rP )r �opr r r r- � r z:MigrationAutodetector._sort_migrations.<locals>.<dictcomp>r N)rB rL r4 r� r� r� rY r )r rz �opsZdependency_graphr� r� Zop2r r r rv } s �z&MigrationAutodetector._sort_migrationsc C s� | j �� D ]6\}}t||dd � �D ]\}}|j�||jf� q$q | j �� D ] }|D ]}tt|j��|_qTqL| j �� D ]$\}}|D ]}t � � |j |�|_ q�qvd S )Nr/ ) ry r4 �zipr� rE rG r� r1 rP r �optimizer )r rz ry �m1�m2� migrationr r r rx � s �z*MigrationAutodetector._optimize_migrationsc s> � d du r6� d du r6t |tj�o4|j� d �� kS � d dur�� d du r�t |tj�r�|j� d �� kr�t� fdd�|jD ��p�t |tj�o�|j� d �� ko�|j� d �� kS � d du�r� d du �rt |tj ��o|j� d �� k�o|j� d �� kS � d du �r@� d du �r@t |tj ��o>|j� d �� kS � d du�r�� d d k�r�t |tj ��o�|j� d �� k�o�|j� d �� kS � d du�r� d d k�r�t |tj ��o�|j� d �� k�o�|j �p�d �� � d �� kS � d du�r,� d d k�r,t |tjtjf��o*|j� d �� kS td � f ��dS )zn Return True if the given operation depends on the given dependency, False otherwise. r@ N� Tr/ c 3 s | ]\}}� d |kV qdS )r@ Nr )r r� �y�r� r r r* � r z9MigrationAutodetector.check_dependency.<locals>.<genexpr>F�alter�order_wrt_unset� �foo_together_changezCan't handle dependency %r)r0 r � CreateModel� name_lowerr� �anyrF �AddFieldZmodel_name_lower� RemoveField� DeleteModel� AlterField�AlterOrderWithRespectTo�order_with_respect_to�AlterUniqueTogether�AlterIndexTogetherr� )r r� r� r r� r r� � s^ � �� �� �� � �� � �� �� z&MigrationAutodetector.check_dependencyFc C s> |pg |_ |r&| j�|g ��d|� n| j�|g ��|� d S )Nr )r� rL r� �insertrE )r rz r� r� � beginningr r r � add_operation� s z#MigrationAutodetector.add_operationc C s� z|| j j| }dd� |jD �}d|d |d f }|j�d�s`d|v s`d|v s`tj�� |�� krzd |d d |d fW S W n ty� Y n0 |S ) z| Place potential swappable models first in lists of created models (only real way to solve #22783). c S s h | ]}t |t�r|n|j�qS r )r0 �str�__name__)r �baser r r r � s �z<MigrationAutodetector.swappable_first_key.<locals>.<setcomp>�%s.%sr r/ Z swappableZ AbstractUserZAbstractBaseUserZ___) r r �basesrW rX r ZAUTH_USER_MODELr� � LookupError)r �itemr| Z base_namesZstring_versionr r r �swappable_first_key� s$ � ���� z)MigrationAutodetector.swappable_first_keyc C sp i | _ i | _| j| j }t|�D �]H\}}| jj||f }| �|j�}| j| j }|D �]\}}||krV| j j||f } | �| j�} || krV| j � | |�rVg } t |j� � �dd� | jj||f � � D � } | D ]"} | jr�| �| �||| | j�� q�| j|tj| j|jd�| d� || j ||f< d| j| jf }d|j|jf | j|< | j�||f� | j�||f� q qVq dS )z� Find any renamed models, generate the operations for them, and remove the old entry from the model lists. Must be run before other model-level generation. c S s g | ]}|� � D ] }|j�qqS r )r� rC )r � relationsrH r r r r( s �zAMigrationAutodetector.generate_renamed_models.<locals>.<listcomp>)�old_name�new_name�r� r� N)r} �renamed_models_relrT rQ rB r r rI rF r r Zask_rename_modelr1 r� r� Z is_relation�extend�!_get_dependencies_for_foreign_keyr� r Z RenameModelrG rz r� �removerY )r � added_modelsrz r{ r| Zmodel_fields_defZremoved_models� rem_app_label�rem_model_nameZrem_model_stateZrem_model_fields_defr� rF rH Zrenamed_models_rel_keyr r r rZ st �������������z-MigrationAutodetector.generate_renamed_modelsc s� | j | jB }| j| }| j| }tt|| jdd�t|| jdd��}|D �]�\� �| jj� �f }i �d}|j � � D ]L\}}|j rr|j j r�|j r�|j j }n|j js�|�|< t|j dd�rr|�|< qr|j�d�} |j�d�} |j�dd�} |j�dd�} |j�d d�} � �dd fg}|jD ]�}t|t��rd |v �r|�d d �\}}|�||ddf� | jj�||f�}| jj�||f�}|�r|�rt|j ��|j ��|j �}|D ]}|�|||d f� �q��q|�r�|�t |� ��d � | j � t j |j �fdd�|j � � D �|j|j|j!d�|dd� |j�dd��s*qFt�� � �D ]R\}}| �"� �|| j�}|�� �ddf� | j � t j#�||d�t$t|��d� �q6| �r�| j � t j%�| d�� �| df� �ddfgd� � �fdd�t��D �}|�� �ddf� | D ] }| j � t j&�|d�|d� �q�| D ] }| j � t j'�|d�|d� �q| �rT| j � t j(�| d�|d� | �rt| j � t j)�| d�|d� | jj*}� �f| j+v rF|� �f � � D ]T\}�| jj| } �� � D ]4\} } | j | j,t j-| j | | d�� �ddfgd� �q��q�qFdS )a� Find all new models (both managed and unmanaged) and make create operations for them as well as separate operations to create any foreign key or M2M relationships (these are optimized later, if possible). Defer any model options that refer to collections of fields that might be deferred (e.g. unique_together, index_together). T)r, �reverseNr� �indexes� constraints�unique_together�index_togetherr� Fr� r/ )NTc s g | ]}|d � vr|�qS )r r �r �d)�related_fieldsr r r( � s �zAMigrationAutodetector.generate_created_models.<locals>.<listcomp>�rG rF rW r� �managers)r� r� rJ �r{ rG rH r� �rG r� c s g | ]}� �|d f�qS )Tr )r rG )rz r{ r r r( � s �r{ �index�r{ � constraint�rG r� �rG r� ).rQ rS rT rV r rB r� r r rF r4 rC r Z primary_keyZ parent_linkr� rW rD r� r0 r� r� rE r rX rP � difference� intersectionr r� r r� rG r� r� r� r1 r� �AddIndex� AddConstraintr� r� r� rR rz r� ) r Zold_keysr� Zadded_unmanaged_modelsZall_added_modelsr| Zprimary_key_relr rH r� r� r� r� r� r� r� �base_app_label� base_nameZold_base_model_stateZnew_base_model_stateZremoved_base_fieldsZremoved_base_fieldrG Zrelated_dependenciesr� r� r� Zrelated_model_keyZrelated_model_stateZrelated_field_name� related_fieldr )rz r{ r� r r^ H s@ � � � �� ��� � ��� �� ��� � � �� ��� �� �� �� � � �z-MigrationAutodetector.generate_created_modelsc C s� | j | j }t|�D ]�\}}| jj||f }|j�d�s<J �||ddfg}|jD ]8}t|t �rPd|v rP|� dd�\}}|� ||ddf� qP| j |t j|jg |j|j|jd�|d� qdS ) z� Make CreateModel statements for proxy models. Use the same statements as that way there's less code duplication, but for proxy models it's safe to skip all the pointless field stuff and chuck out an operation. rK NFr� r/ Tr� r� )rU rR rB r r rW rX r� r0 r� r� rE r� r r� rG r� ) r �addedrz r{ r| r� r� r� r� r r r r` s* � ��z.MigrationAutodetector.generate_created_proxiesc C s� | j | jB }| j| }| j| }tt|�t|��}|D �]�\}}| jj||f }i }|j� � D ]4\} } | j r^| j j r|| || < t | j dd�r^| || < q^|j �dd�} |j �dd�} | r�| �|tj|dd�� | r�| �|tj|dd�� t|�D ]} | �|tj|| d�� q�g }| jj}|||f � � D ]N\\}}}|� � D ]6\} } |�||| df� | j�s8|�||| d f� �q8�q$t|�D ]} |�||| df� �q|| j�||jf�}|�r�|�|d |d |d df� | j|tj|jd �t t |��d� q6dS )a� Find all deleted models (managed and unmanaged) and make delete operations for them as well as separate operations to delete any foreign key or M2M relationships (these are optimized later, if possible). Also bring forward removal of any model options that refer to collections of fields - the inverse of generate_created_models(). r� Nr� r� r� r� �r{ rG Fr� r r/ r@ �rG r� ) rT rV rQ rS r rB r r rF r4 rC r r� rW rD r� r r� r� r� r� rE � many_to_manyr� rX r� r� rG r1 rP )r Znew_keysZdeleted_modelsZdeleted_unmanaged_modelsZall_deleted_modelsrz r{ r| r� r rH r� r� rG r� r� Zrelated_object_app_labelZ object_nameZrelation_related_fieldsZ through_userr r r r] , s� � ���� �� � ��� �� �z-MigrationAutodetector.generate_deleted_modelsc C sX | j | j }t|�D ]>\}}| jj||f }|j�d�s<J �| �|tj |j d�� qdS )z*Make DeleteModel options for proxy models.rK r� N) rR rU rB r r rW rX r� r r� rG )r Zdeletedrz r{ r| r r r r_ � s ��z.MigrationAutodetector.generate_deleted_proxiesc C s� g | _ | j�� }t| j| �D �]�\}}}| j�||f|�}| jj||f }| j j||f }|� |�}| � |�} t|| j �D �]$\} } } | |kr~| |kr~|� | �} | � | �}|j r�|j j r�d|d v r�|d d }|| jv r�| j| |d d< | �| � | �� d }|| k�s@|dd� | dd� kr~t|d |d�| d kr~| j�|| ||�r~| j �| | | j| ||||f� |�| | | f� |�|||f� | | j|||f< q q~q dS )zWork out renamed fields.rA r@ r/ r )� db_columnN)�renamed_operationsr� �copyrB r� r} rX r r r � get_fieldr$ rC r r� Zset_attributes_from_nameZget_attname_columnr3 r Z ask_renamerE r� r� rY rO )r r� rz r{ r r� r� �new_model_staterH Z field_decr� r� �rem_field_name� old_field� old_field_decZ old_rel_toZ old_db_columnr r r rd � sv �� � �� � ������ ���z+MigrationAutodetector.create_renamed_fieldsc C s� | j D ]�\}}}}}}}}||jkrJ|�� } || _| �|tj||| d�� | �|tj|||d�� | j� |||f� | j� |||f� qdS )z Generate RenameField operations.r� )r{ r� r� N) r� r� �clonerG r� r r� Z RenameFieldr� r� rY ) r r� r� Z rem_db_columnr� rz r{ rH r Z altered_fieldr r r ri � s> � ����z-MigrationAutodetector.generate_renamed_fieldsc C s. t | j| j �D ]\}}}| �|||� qdS )zMake AddField operations.N)rB r� r� �_generate_added_field�r rz r{ r r r r rn � s �z+MigrationAutodetector.generate_added_fieldsc C s | j j||f �|�}|||dfg}|jrJ|jjrJ|�| �|||| j �� tjtjtj f}|j p�|� � p�|j p�|j rz|jp�t||�o�|j}|s�|�� }t||�r�|jr�| j�||�|_n| j�||�|_|jr�|jtjur�t|j�r�| j�||� | j|t j ||||d�|d� d S )NF�r{ rG rH �preserve_defaultr� ) r r r� rC r r� r� Z DateFieldZ DateTimeFieldZ TimeField�null� has_defaultr� �blankZempty_strings_allowedr0 Zauto_nowr� Z auto_now_addr Zask_auto_now_add_addition�defaultZask_not_null_addition�unique� NOT_PROVIDED�callableZ$ask_unique_callable_default_additionr� r r� )r rz r{ r rH r� Z time_fieldsr� r r r r� s^ �� �� ����� ����z+MigrationAutodetector._generate_added_fieldc C s. t | j| j �D ]\}}}| �|||� qdS )z Make RemoveField operations.N)rB r� r� �_generate_removed_fieldr� r r r rm 7 s �z-MigrationAutodetector.generate_removed_fieldsc C s2 | j |tj||d�|||df|||dfgd� d S )Nr� r� r� r� )r� r r� r� r r r r > s � ��z-MigrationAutodetector._generate_removed_fieldc s� t �j�j@ �D �]p\}}}�j�||f|�}�j�|||f|�}�jj||f �|�}�j j||f �|�}g }t |d��rZt |j dd��rZt |j j||����jv r�|j j|j _t |j dd�} | r�| f } | �jv r�|j j|j _|j j|j _t |dd�} | �rB||f� t� �fdd�| D ��|_t��fdd�|jD ��|_|���|||�j �� t |d��r�t |j d d��r�t |j j||����jv �r�|j j|j _��|�} ��|�} | | kr||kr|j�o�|j}|j �o�|j }|�s�|�rfd }|j�rB|j�sB|�� �sB|j�sB|�� }�j� ||�}|tj u�rF||_ d }n|}�j |t j!||||d �|d � q��"|||� ��#|||� qdS )zp Make AlterField operations, or possibly RemovedField/AddField if alter isn't possible. rC r Nr � from_fieldsc s g | ]}�j �� |f |��qS r �rO rX )r Z from_field)�from_rename_keyr r r r( s � �zAMigrationAutodetector.generate_altered_fields.<locals>.<listcomp>c s g | ]}�j �� |f |��qS r r )r Zto_field)� rename_keyr r r r( � s �r� TFr� r� )$rB r� r� r} rX rO r r r� r r; r� rC r r r r2 r Z to_fieldsr� r� r� r$ r� r� r� r� r Zask_not_null_alterationr r r� r r� r r� )r rz r{ r r� Zold_field_namer� Z new_fieldr� Zremote_field_nameZto_field_rename_keyr r� Z new_field_decZboth_m2mZ neither_m2mr� rH Z new_defaultr )r r r r ro N s� �� ���� � � �� ����� � ����� �� z-MigrationAutodetector.generate_altered_fieldsc s: t jj}tt�| _t| j�D �]\}}| j� ||f|�}| j j ||f }| j j ||f }|j | �|j | � �fdd�� D �}� fdd��D �}g } g �g �|D ]n} | �� } | d �d�} |D ]N} | �� }|d �d�}| |kr�| |kr�| �|| d f� ��| � ��| � q�q�| �t jj�D ]�\}}}}}||k�s ||k�rD�q |�|�}|D ]�}g }|D ]F} | �� \}}}|�s^| jt|�k�r^t|�ddhk�r^|�| � �q^t|�dk�rR|d }��|� | �d |j|f� | j||f �|� �qR�q �fd d�|D �}�fd d�|D �}| j�||f||| d �i� q d S ) Nc s g | ]}|� vr|�qS r r �r �idx)� old_indexesr r r( � r z@MigrationAutodetector.create_altered_indexes.<locals>.<listcomp>c s g | ]}|� vr|�qS r r r )� new_indexesr r r( � r r@ rG rF r/ r c s g | ]}|� vr|�qS r r r )�remove_from_addedr r r( s c s g | ]}|� vr|�qS r r r )�remove_from_removedr r r( s )� added_indexes�removed_indexes�renamed_indexes)r r� � option_namer r1 � renamed_index_together_valuesrB r� r} rX r r r rW r. rD rE �$_get_altered_foo_together_operationsr� r� rF rP r� rG rM r� ) r r rz r{ r� r� r� r r r Z new_indexZ new_index_dec�new_index_nameZ old_indexZ old_index_dec�old_index_name� old_value� new_valueZindex_together_app_labelZindex_together_model_namer� Zremoved_valuesZremoved_index_togetherZ renamed_index_together_indexes�_r8 r? Z renamed_indexr )r r r r r re � s� � �� ���� � �� ��� � ����z,MigrationAutodetector.create_altered_indexesc C s@ | j �� D ]0\\}}}|d D ]}| �|tj||d�� q q d S )Nr r� )rM r4 r� r r� �r rz r{ � alt_indexesr� r r r rs s ��z,MigrationAutodetector.generate_added_indexesc C sB | j �� D ]2\\}}}|d D ] }| �|tj||jd�� q q d S )Nr r� )rM r4 r� r Z RemoveIndexrG r r r r rh ) s ��z.MigrationAutodetector.generate_removed_indexesc C sJ | j �� D ]:\\}}}|d D ]$\}}}| �|tj||||d�� q q d S )Nr )r{ r� r� � old_fields)rM r4 r� r Z RenameIndex)r rz r{ r r r r r r r rj 4 s ���z.MigrationAutodetector.generate_renamed_indexesc s� t jj}t| j�D ]�\}}| j�||f|�}| jj||f }| j j||f }|j | �|j | � �fdd�� D �}� fdd��D �}| j � ||f||d�i� qd S )Nc s g | ]}|� vr|�qS r r �r �c)�old_constraintsr r r( N r zDMigrationAutodetector.create_altered_constraints.<locals>.<listcomp>c s g | ]}|� vr|�qS r r r )�new_constraintsr r r( O r )�added_constraints�removed_constraints) r r� r rB r� r} rX r r r rW rN r� ) r r rz r{ r� r� r� Zadd_constraintsZrem_constraintsr )r r r rf C s$ � ���z0MigrationAutodetector.create_altered_constraintsc C s@ | j �� D ]0\\}}}|d D ]}| �|tj||d�� q q d S )Nr! r� )rN r4 r� r r� �r rz r{ Zalt_constraintsr� r r r rt Z s � ��z0MigrationAutodetector.generate_added_constraintsc C sB | j �� D ]2\\}}}|d D ] }| �|tj||jd�� q q d S )Nr" r� )rN r4 r� r ZRemoveConstraintrG r# r r r rg h s � ��z2MigrationAutodetector.generate_removed_constraintsc s� d }t � jd�r� jj}nP|j| |f }|�� D ]8\\}}}t� fdd�|�� D ��r0|� d|� �} qjq0t� dd �} | d ur�d} | } nt|| |�\} } | | d dfg} t� jdd �r�t� jj | |�\} }| � | |d df� | S ) Nr c 3 s | ]}� |j kV qd S r) )rC )r r� �rH r r r* ~ s �zJMigrationAutodetector._get_dependencies_for_foreign_key.<locals>.<genexpr>r� �swappable_settingr� Tr� ) r; rC r r� r4 r� r� r� r r� rE )rz r{ rH Z project_stateZremote_field_modelr� Zremote_app_labelZremote_model_namerF r% � dep_app_labelZdep_object_namer� Zthrough_app_labelZthrough_object_namer r$ r r� v s8 � ��z7MigrationAutodetector._get_dependencies_for_foreign_keyc # s� t �j�D ]�\� ��j�� �f��}�jj� |f }�jj� �f }|j�|�}|rj� ��fdd�|D �nt� }|j�|�}|r�t|�nt� }||kr g }|D ]>}|D ]4} |� | �} | j r�| j j r�|� �� � �| �j�� q�q�||� �|fV q d S )Nc s& h | ] }t � ��fd d�|D ���qS )c 3 s" | ]}�j �� �|f|�V qd S r) r )r �n�rz r{ r r r r* � s �zWMigrationAutodetector._get_altered_foo_together_operations.<locals>.<setcomp>.<genexpr>)r2 )r r r( r r r � s ��zMMigrationAutodetector._get_altered_foo_together_operations.<locals>.<setcomp>)rB r� r} rX r r r rW rP r� rC r r� r� ) r r r� r� r� r r r� Z foo_togethersr rH r r( r r � sF � ��� �� �z:MigrationAutodetector._get_altered_foo_together_operationsc sv �� |j�D ]d\}}� �}|tjkr:� ��fdd�|D �}|�|�}|sL|r �j� |f d�i|j|i��|d� q d S )Nc s" h | ]}|�j � �f vr|�qS r )r r% r( r r r � s ��zOMigrationAutodetector._generate_removed_altered_foo_together.<locals>.<setcomp>rG r� )r r r r� r� r� )r r� r r r� � removal_valuer r( r �&_generate_removed_altered_foo_together� s, � � ���z<MigrationAutodetector._generate_removed_altered_foo_togetherc C s | � tj� d S r) )r* r r� r' r r r rk � s z>MigrationAutodetector.generate_removed_altered_unique_togetherc C s | � tj� d S r) )r* r r� r' r r r rl � s z=MigrationAutodetector.generate_removed_altered_index_togetherc C sV | � |j�D ]D\}}}}}|�|�}||kr | j||f d|i|j|i��|d� q d S )NrG r� )r r r� r� )r r� r r rz r{ r� r) r r r � _generate_altered_foo_together� s � �z4MigrationAutodetector._generate_altered_foo_togetherc C s | � tj� d S r) )r+ r r� r' r r r rq � s z6MigrationAutodetector.generate_altered_unique_togetherc C s | � tj� d S r) )r+ r r� r' r r r rr � s z5MigrationAutodetector.generate_altered_index_togetherc C s� | j �| j| j�}t|�D ]p\}}| j�||f|�}| jj||f }| j j||f }|j �d�}|j �d�}||kr| � |t j ||d�� qd S )NZdb_table)rG �table)r� �unionr� r� rB r} rX r r r rW r� r ZAlterModelTable) r �models_to_checkrz r{ r� r� r� Zold_db_table_nameZnew_db_table_namer r r ru � s& �� ��z/MigrationAutodetector.generate_altered_db_tablec C s� | j �| j| j�}t|�D ]p\}}| j�||f|�}| jj||f }| j j||f }|j �d�}|j �d�}||kr| � |t j ||d�� qd S )NZdb_table_comment)rG Z table_comment)r� r- r� r� rB r} rX r r r rW r� r ZAlterModelTableComment) r r. rz r{ r� r� r� Zold_db_table_commentZnew_db_table_commentr r r rc s& �� ��z7MigrationAutodetector.generate_altered_db_table_commentc C s� | j �| j| j| j| j@ | j| j@ �}t|�D ]�\}}| j � ||f|�}| j j ||f }| j j ||f }dd� |j�� D �}dd� |j�� D �}||kr.| �|tj||d�� q.dS )z� Work out if any non-schema-affecting options have changed and make an operation to represent them in state changes (in case Python code in migrations needs them). c S s i | ]\}}|t jv r||�qS r �r ZALTER_OPTION_KEYSr+ r r r r- ? s �zBMigrationAutodetector.generate_altered_options.<locals>.<dictcomp>c S s i | ]\}}|t jv r||�qS r r/ r+ r r r r- D s �)rG rW N)r� r- r� r� rS rT rQ rV rB r} rX r r r rW r4 r� r r ) r r. rz r{ r� r� r� Z old_options� new_optionsr r r ra * s4 � �����z.MigrationAutodetector.generate_altered_optionsc C s� t | j�D ]�\}}| j�||f|�}| jj||f }| jj||f }|j�d�|j�d�kr g }|j�d�r�|�|||jd df� | j |t j ||j�d�d�|d� q d S )Nr� Tr� r� ) rB r� r} rX r r r rW rE r� r r� )r rz r{ r� r� r� r� r r r rp R s< �� � �� ���z<MigrationAutodetector.generate_altered_order_with_respect_toc C sn t | j�D ]^\}}| j�||f|�}| jj||f }| jj||f }|j|jkr | �|t j ||jd�� q d S )N)rG r� ) rB r� r} rX r r r r� r� r ZAlterModelManagers)r rz r{ r� r� r� r r r rb t s � ��z/MigrationAutodetector.generate_altered_managersc s~ |� � }i � t|�� �D �],\}}|s(qd}|D ]}|d |kr0|} qJq0|du r�| j�|�s�|D ]} |df� || jf< qb||= q|du r�d} n| �|d �p�dd } t|�D ]�\} } | dkr�|r�| j� |� d| g} |r�| � |� n2| dk�r|�s| � d� n| � | � � dd� � d� | �} || f� || jf< | d7 } | | _q�q|� � D ](}|D ] } � fd d �| jD �| _�qX�qP|S ) z� Take a result from changes() and a MigrationGraph, and fix the names and dependencies of the changes so they extend the graph from the leaf nodes for each app. Nr r� r/ z%04ir� �d r c s g | ]}� � ||��qS r �rX r� �Zname_mapr r r( � s z;MigrationAutodetector.arrange_for_graph.<locals>.<listcomp>) r� r1 r4 r Z ask_initialrG � parse_number� enumerater� rE Z suggest_name�joinr� )r r" r r! �leavesrz ry Zapp_leaf�leafr� Z next_number�iZnew_name_partsr� r r3 r r � sJ �z'MigrationAutodetector.arrange_for_graphc s� i � |� � D ]6\}}|D ](}|jD ] \}}� �|t� ��|� q"qq t|�}d} | |krzt|�} |j� fdd�|D �� qPt|�D ]}||vr�||= q�|S )a1 Take changes from arrange_for_graph() and set of app labels, and return a modified set of changes which trims out as many migrations that are not in app_labels as possible. Note that some other migrations may still be present as they may be required dependencies. Nc s g | ]}� � |d ��qS )r r2 )r rz �Zapp_dependenciesr r r( � r z7MigrationAutodetector._trim_to_apps.<locals>.<listcomp>)r4 r� r� rP rY r� r1 ) r r" Z app_labelsrz ry r� r& rG Z required_appsZold_required_appsr r: r r � s � z#MigrationAutodetector._trim_to_appsc C s< t �d|� }r t|d �S t �d|�}|r8t|d �S dS )u� Given a migration name, try to extract a number from the beginning of it. For a squashed migration such as '0001_squashed_0004…', return the second number. If no number is found, return None. z.*_squashed_(\d+)r/ z^\d+r N)�re�search�int�match)�clsrG Zsquashed_matchr> r r r r4 � s z"MigrationAutodetector.parse_number)N)NNN)NN)N)NF)N)7r� � __module__� __qualname__�__doc__r r" r$ rI r r[ r\ � staticmethodr� rw rv rx r� r� r� rZ r^ r` r] r_ rd ri rn r� rm r ro re rs rh rj rf rt rg r� r r* rk rl r+ rq rr ru rc ra rp rb r r � classmethodr4 r r r r r sl & k r? C D!a >%3wY "/(" 4r )r5 r; � collectionsr � itertoolsr Z django.confr Z django.dbr Zdjango.db.migrationsr Z django.db.migrations.migrationr Z&django.db.migrations.operations.modelsr Z django.db.migrations.optimizerr Z django.db.migrations.questionerr Zdjango.db.migrations.utilsr r r Z django.utils.topological_sortr r r r r r �<module> s