File: //usr/lib/python3/dist-packages/django/db/migrations/__pycache__/state.cpython-310.pyc
o
3�akb � @ s d dl Z d dlmZ d dlmZ d dlmZmZ d dl m
Z
d dl m Z d dl
mZ d dlmZmZ d d lmZ d d
lmZ d d lmZ d d lmZ d
dlmZ d dd�Z dd� Z dd� Z dd� Z G dd� d�Z G dd� de�Z!G d d � d e�Z"G d d � d �Z#dS )!� N)�contextmanager)� AppConfig)�Apps�apps)�settings)�models)� RECURSIVE_RELATIONSHIP_CONSTANT)�
DEFAULT_NAMES�normalize_together)�make_model_tuple)�cached_property)�
import_string)�get_docs_version� )�InvalidBasesError� c C sF t | t�r| �dd�}t|�dkrt|�S ||d fS | jj| jjfS )N�.r � r )�
isinstance�str�split�len�tuple�_meta� app_label�
model_name)�modelr r � r �</usr/lib/python3/dist-packages/django/db/migrations/state.py� _get_app_label_and_model_name s
r c C s� dd� | � � D �}t� }| jjddd�D ] }|jr0|jdur0t|jt�s0|�|j � |�
|j� q| j}|j rA| |v rA|�
|j � |S )zEReturn all models that have a direct relationship to the given model.c S s g | ]
}t |tj�r|�qS r )�
issubclassr �Model)�.0�subclassr r r �
<listcomp> s
��z'_get_related_models.<locals>.<listcomp>T)�include_parents�include_hiddenN)
�__subclasses__�setr �
get_fields� is_relation�
related_modelr r �addr �append�proxy�concrete_model)�m�related_models�related_fields_models�f�optsr r r �_get_related_models s � � r5 c C s dd� t | �D �S )zq
Return a list of typical (app_label, model_name) tuples for all related
models for the given model.
c S s h | ]
}|j j|j jf�qS r )r r r )r" �rel_modr r r � <setcomp>2 s ��z,get_related_models_tuples.<locals>.<setcomp>)r5 �r r r r �get_related_models_tuples- s �r9 c C sj t � }t| �}|D ] }|jj|jj}}||f|v rq |�||f� |�t|�� q || jj| jjfh S )a�
Return all models that have a direct or indirect relationship
to the given model.
Relationships are either defined by explicit relational fields, like
ForeignKey, ManyToManyField or OneToOneField, or by inheriting from another
model (a superclass is related to its subclasses, but not vice versa). Note,
however, that a model inheriting from a concrete model is also related to
its superclass through the implicit *_ptr OneToOneField on the subclass.
)r( r5 r r r r, �extend)r �seen�queuer6 �
rel_app_label�rel_model_namer r r � get_related_models_recursive8 s r? c @ s� e Zd ZdZd dd�Zdd� Zdd� Zd d
d �Zd d d
�Zd!dd�Z dd� Z
dd� Z dd� Z e
dd� �Zedd� �Zedd � �Zd d � ZdS )"� ProjectStatez�
Represent the entire project's overall state. This is the item that is
passed around - do it here rather than at the app level so that cross-app
FKs/etc. resolve properly.
Nc C s |pi | _ |pg | _d| _d S �NF)r � real_apps�
is_delayed)�selfr rB r r r �__init__U s
zProjectState.__init__c C s: |j |j}}|| j||f< d| jv r| �||� d S d S �Nr )r �
name_lowerr �__dict__� reload_model)rD � model_stater r r r r � add_model[ s
�zProjectState.add_modelc C s6 | j ||f= d| jv r| j�||� | j�� d S d S rF )r rH r �unregister_model� clear_cache�rD r r r r r � remove_modela s
�zProjectState.remove_modelFc C s |rd| _ t� }z | j�||�}W n ty Y n w |r"t|�}nt|�}| j||f }t� }|j� � D ] }|j
rR|j j t
krAq5t|j|�\} }
|�| |
�� f� q5|�|� |D ](\} }
z | j�| |
�} W n typ Y qZw |r{|�t| �� qZ|�t| �� qZ|�||f� |S )NT)rC r( r � get_model� LookupErrorr9 r? r �fields�valuesr* � remote_fieldr r r r+ r, �lower�update) rD r r �delayr1 � old_modelrJ �direct_related_models�fieldr= r> � rel_modelr r r �_find_reload_modeli s@ �
�
�z ProjectState._find_reload_modelc C s* d| j v r| �|||�}| �|� d S d S rF )rH r\ �_reload)rD r r rW r1 r r r rI � s
�zProjectState.reload_modelTc C sD d| j v r t� }|D ]\}}|�| �|||�� q
| �|� d S d S rF )rH r( rV r\ r] )rD r rW r1 r r r r r �
reload_models� s
�zProjectState.reload_modelsc C s� | j �� � |D ] \}}| j �||� qW d � n1 s w Y g }| j jD ]}|j|jf|v r8|�|� q)|D ] \}}z | j||f }W n tyQ Y q;w |�|� q;| j � |� d S �N)
r � bulk_updaterL � real_modelsr rG r- r �KeyError�render_multiple)rD r1 r= r> �states_to_be_renderedrJ r r r r] � s$ � �
� � zProjectState._reloadc C s@ t dd� | j�� D �| jd�}d| jv r| j�� |_| j|_|S )z*Return an exact copy of this ProjectState.c S � i | ] \}}||� � �qS r ��clone)r" �k�vr r r �
<dictcomp>� � z&ProjectState.clone.<locals>.<dictcomp>)r rB r )r@ r �itemsrB rH r rg rC )rD � new_stater r r rg � s �
zProjectState.clonec C s$ | j rd| jv r| jd= d S d S d S rF )rC rH �rD r r r �clear_delayed_apps_cache� s �z%ProjectState.clear_delayed_apps_cachec C s t | j| j�S r_ )� StateAppsrB r rn r r r r � s zProjectState.appsc C s t | j| jdd�| _| jS )NT)�ignore_swappable)rp rB r r rn r r r �
concrete_apps� s zProjectState.concrete_appsc C s8 i }|j dd�D ]}t�|�}|||j|jf< q| |�S )z3Take an Apps and return a ProjectState matching it.T)�include_swapped)�
get_models�
ModelState�
from_modelr rG )�clsr �
app_modelsr rJ r r r � from_apps� s
zProjectState.from_appsc C s | j |j kot| j�t|j�kS r_ )r r( rB �rD �otherr r r �__eq__� s zProjectState.__eq__)NN�F)T)�__name__�
__module__� __qualname__�__doc__rE rK rO r\ rI r^ r] rg ro r r �propertyrr � classmethodry r| r r r r r@ N s$
0
r@ c @ s e Zd ZdZdd� Zdd� ZdS )�
AppConfigStubzAStub of an AppConfig. Only provides a label and a dict of models.c C s d | _ i | _|| _|| _d S r_ )r r �label�name)rD r� r r r rE � s
zAppConfigStub.__init__c C s | j j| j | _d S r_ )r �
all_modelsr� r rn r r r �
import_models� s zAppConfigStub.import_modelsN)r~ r r� r� rE r� r r r r r� � s r� c sN e Zd ZdZd� fdd� Zedd� �Zdd� Zd d
� Zd d � Z d
d� Z
� Z S )rp zo
Subclass of the global Apps registry class to better handle dynamic model
additions and removals.
Fc s� g | _ |D ]}t�|�}|�� D ]
}| j �tj|dd�� qqdd� |�� D �}dd� tg |�|��D �}t � �
|� d | _ d | _ | �
g |�� �| j �� ddlm} |r[ttj�hnt� }
| | |
d �} | rrtd
�d d � | D ����d S )
NT)� exclude_relsc S � h | ]}|j �qS r )r )r" rJ r r r r7 � z%StateApps.__init__.<locals>.<setcomp>c S s g | ]}t |��qS r )r� )r" r� r r r r$ � z&StateApps.__init__.<locals>.<listcomp>r )�_check_lazy_references)�ignore�
c s s � | ]}|j V qd S r_ )�msg)r" �errorr r r � <genexpr> s � z%StateApps.__init__.<locals>.<genexpr>)ra � global_apps�get_app_configrt r- ru rv rS �sorted�superrE �_lock� ready_eventrc � django.core.checks.model_checksr� r r �AUTH_USER_MODELr( �
ValueError�join) rD rB r rq r �appr �
app_labels� app_configsr� r� �errors�� __class__r r rE � s$
� �zStateApps.__init__c c s: � | j }d| _ z
d V W || _ | �� d S || _ | �� w rA )�readyrM )rD r� r r r r` s � �
zStateApps.bulk_updatec
C s� |sd S | � � �F |}|rFg }|D ]}z|�| � W q ty( |�|� Y qw t|�t|�kr:td|t� f ��|}|s
W d � d S W d � d S 1 sQw Y d S )Nz�Cannot resolve bases for %r
This can happen if you are inheriting models from an app with migrations (e.g. contrib.auth)
in an app with no migrations; see https://docs.djangoproject.com/en/%s/topics/migrations/#dependencies for more)r` �renderr r- r r )rD � model_states�unrendered_models�new_unrendered_modelsr r r r rc # s.
�����"�zStateApps.render_multiplec C sH t g i �}t�| j�|_t�| j�|_|j�� D ]}||_q| j|_|S )z Return a clone of this registry.)rp �copy�deepcopyr� r� rS r ra )rD rg �
app_configr r r rg = s
zStateApps.clonec C s` || j | |jj< || jvrt|�| j|< | | j| _|| j| j|jj< | �|� | �� d S r_ ) r� r r r� r� r r �do_pending_operationsrM )rD r r r r r �register_modelI s
zStateApps.register_modelc C s6 z| j | |= | j| j|= W d S ty Y d S w r_ )r� r� r rb rN r r r rL R s �zStateApps.unregister_modelr} ) r~ r r� r� rE r r` rc rg r� rL �
__classcell__r r r� r rp � s
rp c @ sl e Zd ZdZddd�Zedd� �Zeddd ��Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )ru a�
Represent a Django Model. Don't use the actual Model class as it's not
designed to have its options changed - instead, mutate this one and then
render it into a Model as required.
Note that while you are allowed to mutate .fields, you are not allowed
to mutate the Field instances inside there themselves - you must instead
assign new ones, as these are not detached during a clone.
Nc C s� || _ || _t|�| _|pi | _| j�dg � | j�dg � |p#tjf| _|p(g | _ | j�
� D ].\}}t |d�r>t d| ��|j
rMt |jd�rMt d| ��|jr]t |jjd�r]t d| ��q/| jd D ] }|jsnt d| ��qcd S ) N�indexes� constraintsr z7ModelState.fields cannot be bound to a model - "%s" is.r z_ModelState.fields cannot refer to a model class - "%s.to" does. Use a string reference instead.zdModelState.fields cannot refer to a model class - "%s.through" does. Use a string reference instead.zKIndexes passed to ModelState require a name attribute. %r doesn't have one.)r r� �dictrR �options�
setdefaultr r! �bases�managersrl �hasattrr� r* r+ � many_to_manyrT �through) rD r r� rR r� r� r� rZ �indexr r r rE e sF
����������zModelState.__init__c C s
| j �� S r_ )r� rU rn r r r rG � s
zModelState.name_lowerFc s& g }�j jD ]7}t|dd�r|rqt|tj�rq|j}z |�||�� f� W q t y= } z t d|�j j
|f ��d}~ww |sl�j j D ]'}|j}z |�||�� f� W qD t yk } z t d|�j j |f ��d}~ww i }t
D ]j}|dv rwqp|�j jv r�|dkr��j jd }tt|��||< qp|dkr��j jd } tt| ��||< qp|dkr�d d
� �j jD �}
|
D ]
} | js�| ��� q�|
|d< qp|d kr�d d
� �j jD �|d < qp�j j| ||< qp|r�d
D ] } | |v r�|| = q�n|�d�dd� �j jD �v r�|d= � fdd�� tt� ����fdd�d�}
tdd� |
D ��}tdd� |D ���s&tjf}g }t� }d}�j jD ]J}|j|v �r;�q1|j�rIt � |�}|� � n#|�j u �sU|�j u �rjt� � }|j!|_!|j|_|�j u �ri|}n�q1|�"|j� |�|j|f� �q1|d|fgk�r�g }| �j j#�j j ||||�S )z3Given a model, return a ModelState representing it.rT Nz'Couldn't reconstruct field %s on %s: %sz+Couldn't reconstruct m2m field %s on %s: %s)r r �unique_together�index_togetherr� c S � g | ]}|� � �qS r rf )r" �idxr r r r$ � r� z)ModelState.from_model.<locals>.<listcomp>r� c S r� r rf )r" �conr r r r$ � r� )r� r� �order_with_respect_tor� c S r� r )r� )r" rZ r r r r7 � r� z(ModelState.from_model.<locals>.<setcomp>c s@ g }| j D ]}t|d�r|jjr|�� |�� q|�|� q|S )Nr )� __bases__r� r �abstractr: r- )r r� �base)�
flatten_basesr r r� � s
z,ModelState.from_model.<locals>.flatten_basesc s � j �| �S r_ )�__mro__r� )�xr8 r r �<lambda>� s z'ModelState.from_model.<locals>.<lambda>��keyc s s& � | ]}t |d �r
|jjn|V qdS )r N)r� r � label_lower�r" r� r r r r� � s � �
�
�z(ModelState.from_model.<locals>.<genexpr>c s s& � | ]}t |t�pt|tj�V qd S r_ )r r r r r! r� r r r r� � s �$ �objects)$r � local_fields�getattrr r �OrderWrtr� r- rg � TypeErrorr� �local_many_to_many� object_namer �original_attrsr( r
r� �set_name_with_modelr� �get�private_fieldsr� r �anyr! r� �use_in_migrationsr� �_set_creation_counter�
_base_manager�_default_manager�Managerr r, r )rw r r� rR rZ r� �er� �ut�itr� r� r� �flattened_basesr� r� �
manager_names�default_manager_shim�manager� new_managerr )r� r r rv � s� ��� ���
�
��� �
� �zModelState.from_modelc c sp � t | jdd� d�}|D ])\}}|�� \}}}}}|r't|�} || �� fV q t|�}
||
|i |��fV q dS )z-Deep-clone the managers using deconstruction.c S s
| d j S )Nr )�creation_counter)ri r r r r� s
z/ModelState.construct_managers.<locals>.<lambda>r� N)r� r� � deconstructr
�
as_manager) rD �sorted_managers�mgr_namer� r� � manager_path�qs_path�args�kwargs�qs_class�
manager_classr r r �construct_managers s � �z ModelState.construct_managersc C s. | j | j| jt| j�t| j�| jt| j�d�S )z(Return an exact copy of this ModelState.)r r� rR r� r� r� ) r� r r� r� rR r� r� �listr� rn r r r rg s �zModelState.clonec s� | j � d�| j�}tdd|�}zt� fdd�| jD ��}W n ty, td| jf ��w dd� | j�� D �}||d< d |d
< |� | �
� � t| j ||�S ) zACreate a Model object from our current state into the given apps.)r r �Metar c 3 s( � | ]}t |t�r� �|�n|V qd S r_ )r r rP r� �r r r r� . s
� �
�z$ModelState.render.<locals>.<genexpr>z(Cannot resolve one or more bases from %rc S re r rf )r" r� rZ r r r rj 5 rk z%ModelState.render.<locals>.<dictcomp>�__fake__r ) r r� �typer r� rQ r rR rl rV r� r� )rD r �
meta_contents�metar� �bodyr r� r r� ' s � �zModelState.renderc C �4 | j d D ] }|j|kr| S qtd|| jf ��)Nr� z No index named %s on model %s�r� r� r� )rD r� r� r r r �get_index_by_name> �
�z ModelState.get_index_by_namec C r� )Nr� z"No constraint named %s on model %sr� )rD r� �
constraintr r r �get_constraint_by_nameD r� z!ModelState.get_constraint_by_namec C s d| j j| j| jf S )Nz
<%s: '%s.%s'>)r� r~ r r� rn r r r �__repr__J s zModelState.__repr__c C s| | j |j ko=| j|jko=t| j�t|j�ko=tdd� tt| j�� �t|j�� ��D ��o=| j|jko=| j |j ko=| j
|j
kS )Nc s sD � | ] \\}}\}}||ko |� � d d� |� � d d� kV qdS )r N)r� )r" �k1�f1�k2�f2r r r r� R s
� &�
�z$ModelState.__eq__.<locals>.<genexpr>) r r� r rR �all�zipr� rl r� r� r� rz r r r r| M s$
�� ���
�
�
�zModelState.__eq__)NNNr} )r~ r r� r� rE r rG r� rv r� rg r� r� r� r� r| r r r r ru Z s
"
ru )r )$r� �
contextlibr � django.appsr �django.apps.registryr r r� � django.confr � django.dbr � django.db.models.fields.relatedr �django.db.models.optionsr r
�django.db.models.utilsr �django.utils.functionalr �django.utils.module_loadingr
�django.utils.versionr �
exceptionsr r r5 r9 r? r@ r� rp ru r r r r �<module> s,
g