File: //lib/python3/dist-packages/django/db/models/__pycache__/deletion.cpython-310.pyc
o
3�a&M � @ s� d dl Z d dlmZmZ d dlmZmZ d dlmZ d dl m Z d dl
m Z m Z m
Z
d dlmZmZmZ G dd � d e �ZG d
d � d e �Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� ZG d d � d �Z dS ) � N)�Counter� defaultdict)�partial�reduce)�chain)�
attrgetter)�IntegrityError� connections� transaction)� query_utils�signals�sqlc � e Zd Z� fdd�Z� ZS )�ProtectedErrorc � || _ t� �||� d S �N)�protected_objects�super�__init__)�self�msgr �� __class__� �;/usr/lib/python3/dist-packages/django/db/models/deletion.pyr � zProtectedError.__init__��__name__�
__module__� __qualname__r �
__classcell__r r r r r � r c r )�RestrictedErrorc r r )�restricted_objectsr r )r r r# r r r r r zRestrictedError.__init__r r r r r r" r! r" c C sH | j ||jj|j|jdd� |jr t| jjs"| �|d |� d S d S d S )NF)�source� source_attr�nullable�fail_on_restricted) �collect� remote_field�model�name�nullr �features�can_defer_constraint_checks�add_field_update�� collector�field�sub_objs�usingr r r �CASCADE s ��r5 c C s$ t d|jjj|d jj|jf |��)NzoCannot delete some instances of model '%s' because they are referenced through a protected foreign key: '%s.%s'r )r r) r* r r r+ r0 r r r �PROTECT s ���r6 c C s" | � ||� | �|jj|j� d S r )�add_restricted_objects�add_dependencyr) r* r0 r r r �RESTRICT* s r9 c s4 t � �r � fdd�}n� fdd�}� fdd�|_|S )Nc s | � |� � |� d S r �r/ r0 ��valuer r �
set_on_delete1 s zSET.<locals>.set_on_deletec s | � |� |� d S r r: r0 r; r r r= 4 � c s d� fi fS )Nzdjango.db.models.SETr r r; r r �<lambda>6 s zSET.<locals>.<lambda>)�callable� deconstruct)r< r= r r; r �SET/ s
rB c C s | � |d |� d S r r: r0 r r r �SET_NULL: r> rC c C s | � ||�� |� d S r )r/ � get_defaultr0 r r r � SET_DEFAULT> s rE c C s d S r r r0 r r r �
DO_NOTHINGB s rF c C s dd� | j dd�D �S )Nc s s, � | ]}|j r|js|js|jr|V qd S r )� auto_created�concrete�
one_to_one� one_to_many)�.0�fr r r � <genexpr>I s � ����
�z4get_candidate_relations_to_delete.<locals>.<genexpr>T)�include_hidden)�
get_fields)�optsr r r �!get_candidate_relations_to_deleteF s
�rQ c @ s� e Zd Zdd� Zd"dd�Zd#dd�Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
d$dd�Z dd� Z d%dd�Z
dd� Zd d � Zd d � Zd d!� ZdS )&� Collectorc C sD || _ tt�| _tttt��| _tttt��| _g | _tt�| _d S r ) r4 r �set�datar �
field_updatesr# � fast_deletes� dependencies)r r4 r r r r P s
zCollector.__init__NFc C sf |sg S g }|d j }| j| }|D ] }||vr |�|� q|�|� |dur1|s1| j|||d� |S )a7
Add 'objs' to the collection of objects to be deleted. If the call is
the result of a cascade, 'source' should be the model that caused it,
and 'nullable' should be set to True if the relation can be null.
Return a list of all objects that were not already collected.
r N��reverse_dependency)r rT �append�updater8 ) r �objsr$ r& rY �new_objsr* � instances�objr r r �addc s
�
z
Collector.addc C s> |r||}}| j |jj �|jj� | j�|| j�� � d S r )rW �_meta�concrete_modelr` rT �
setdefault�default_factory)r r* �
dependencyrY r r r r8 { s
zCollector.add_dependencyc C s. |sdS |d j }| j| ||f �|� dS )z�
Schedule a field update. 'objs' must be a homogeneous iterable
collection of model instances (e.g. a QuerySet).
Nr )r rU r[ )r r2 r< r\ r* r r r r/ � s
zCollector.add_field_updatec C s* |r|d j }| j| | �|� d S d S )Nr )r r# r[ )r r2 r\ r* r r r r7 � s
�z Collector.add_restricted_objectsc s4 || j v r� fdd�| j | �� D �| j |< d S d S )Nc s i | ] \}}||� �qS r r )rK r2 �items�r\ r r �
<dictcomp>� s ��z?Collector.clear_restricted_objects_from_set.<locals>.<dictcomp>)r# rf )r r* r\ r rg r �!clear_restricted_objects_from_set� s
��z+Collector.clear_restricted_objects_from_setc C sB || j v r t|jdd� | j | �� D �d��}| �||� d S d S )Nc S s g | ]
}|D ]}|j �qqS r ��pk)rK r\ r_ r r r �
<listcomp>� s ���zDCollector.clear_restricted_objects_from_queryset.<locals>.<listcomp>)�pk__in)r# rS �filter�valuesri )r r* �qsr\ r r r �&clear_restricted_objects_from_queryset� s
��z0Collector.clear_restricted_objects_from_querysetc C s t j�|�p t j�|�S r )r �
pre_delete�
has_listeners� post_delete)r r* r r r �_has_signal_listeners� s
�z Collector._has_signal_listenersc s� � r
� j jtur
dS t|d�r|jj}nt|d�r"t|d�r"|j}ndS | �|�r+dS |j}t� fdd�|jjj �
� D ��oStdd� t |�D ��oSt dd� |j
D �� S ) a�
Determine if the objects in the given queryset-like or single object
can be fast-deleted. This can be done if there are no cascades, no
parents and no signal listeners for the object class.
The 'from_field' tells where we are coming from - we need this to
determine if the objects are in fact to be deleted. Allow also
skipping parent -> child -> parent chain preventing fast delete of
the child.
Fra r* � _raw_deletec 3 s � | ]}|� kV qd S r r )rK �link��
from_fieldr r rM � s � z,Collector.can_fast_delete.<locals>.<genexpr>c s s � | ]
}|j jjtu V qd S r )r2 r) � on_deleterF )rK �relatedr r r rM � s
� �
�c s s � | ]}t |d �V qdS )�bulk_related_objectsN)�hasattr�rK r2 r r r rM � s � )r) rz r5 r} ra r* ru �allrb �parentsro rQ �any�private_fields)r r\ ry r* rP r rx r �can_fast_delete� s"
���zCollector.can_fast_deletec sZ dd� |D �}t t| j j�|��d�� t��� kr*� �fdd�tdt��� �D �S �gS )zT
Return the objs in suitably sized batches for the used connection.
c S � g | ]}|j �qS r )r+ r~ r r r rl � � z-Collector.get_del_batches.<locals>.<listcomp>� c s g | ]
}�||� � �qS r r )rK �i��conn_batch_sizer\ r r rl � s �r )�maxr r4 �ops�bulk_batch_size�len�range)r r\ �fields� field_namesr r� r �get_del_batches� s � �zCollector.get_del_batchesTc sl | � |�r
| j�|� dS | j||||d�} | sdS | d j}
|sF|
jj} | jj�� D ]� � rE� fdd�| D �} | j | |
� j
j dddd� q+|sJdS |rSt |
j�
� �}
tt�}tt�}t|
j�D ]�}|rj|j|
v rjq`|j}|j
jtkrtq`|j}| j ||d �r�|| �|� q`| �| |g�}|D ]\}| �||g|�}|jjs�| �|�s�t t� d
d � t|j�D ���}|j t |�� }|r�z |j
�| ||| j � W q� t y� } zd |jj!|j"f }|| |j#7 < W Y d}~q�d}~ww q�q`|�rt d
|
j!d�$|�f t t� |�� ����|�%� D ] \}}| �| |�}|D ]}| �|||�}| j�|� �q�q|
jj&D ]}t'|d��rC|�(| | j �}| j ||
ddd� �q+|�r�| j)�%� D ] \}}| �*||� �qM| jD ]
}| �+|j|� �q\| j,�� �r�tt�} | j,�%� D ]"\}} | �%� D ]\}}|�r�d |j!|j"f }| | |7 < �q~�qv| �r�t-d|
j!d�$| �f t t� | �� ����dS dS dS )a�
Add 'objs' to the collection of objects to be deleted as well as all
parent instances. 'objs' must be a homogeneous iterable collection of
model instances (e.g. a QuerySet). If 'collect_related' is True,
related objects will be handled by their respective on_delete handler.
If the call is the result of a cascade, 'source' should be the model
that caused it and 'nullable' should be set to True, if the relation
can be null.
If 'reverse_dependency' is True, 'source' will be deleted before the
current model, rather than after. (Needed for cascading to parent
models, the one case in which the cascade follows the forwards
direction of an FK rather than the reverse direction.)
If 'keep_parents' is True, data of parent model's will be not deleted.
If 'fail_on_restricted' is False, error won't be raised even if it's
prohibited to delete such objects due to RESTRICT, that defers
restricted object checking in recursive calls where the top-level call
may need to collect more objects to determine whether restricted ones
can be deleted.
NrX r c s g | ]}t |� j��qS r )�getattrr+ �rK r_ ��ptrr r rl � z%Collector.collect.<locals>.<listcomp>FT)r$ r% �collect_relatedrY r' rx c s s" � | ] }d d� |j jD �V qdS )c s s � | ]}|j V qd S r )�attname)rK �rfr r r rM # s � z.Collector.collect.<locals>.<genexpr>.<genexpr>N)r2 �foreign_related_fields)rK �relr r r rM " s
� �
�z$Collector.collect.<locals>.<genexpr>z'%s.%s'zhCannot delete some instances of model %r because they are referenced through protected foreign keys: %s.z, r| )r$ r& r' ziCannot delete some instances of model %r because they are referenced through restricted foreign keys: %s.).r� rV rZ r` r ra rb r� ro r( r) � related_namerS �get_parent_listr �listrQ r* r2 rz rF �
related_modelr� �related_objects�query�select_relatedru r �
from_iterable�only�tupler4 r r r+ r �joinrf r� r} r| rT ri rq r# r" ) r r\ r$ r& r� r% rY � keep_parentsr' r] r* rb � parent_objsr� �model_fast_deletesr r{ r2 r� �batches�batchr3 �referenced_fields�error�key�related_fieldsr^ rp r# r� r r� r r( � s�
�
��
� ������� � �
������zCollector.collectc s. t tj� fdd�|D ��}|j�| j��|�S )zQ
Get a QuerySet of the related model to objs via related fields.
c 3 s* � | ]}t jdi d |j � i��V qdS )z%s__inNr )r �Qr+ )rK �
related_fieldrg r r rM ^ s
� �
�z,Collector.related_objects.<locals>.<genexpr>)r �operator�or_�
_base_managerr4 rn )r r� r� r\ � predicater rg r r� Z s �zCollector.related_objectsc c s. � | j �� D ]\}}|D ]}||fV q qd S r )rT rf )r r* r^ r_ r r r �instances_with_modeld s � ��z Collector.instances_with_modelc s� g }t � }t� j�}t|�t|�k rGd}|D ]$}||v r q� j�|jj�}|r,|�|�s:|� |� |�
|jj� d}q|s?d S t|�t|�k s� fdd�|D �� _d S )NFTc s i | ]}|� j | �qS r )rT )rK r* �r r r rh y r� z"Collector.sort.<locals>.<dictcomp>) rS r� rT r� rW �getra rb �
differencerZ r` )r �
sorted_models�concrete_models�models�foundr* rW r r� r �sorti s$
�� zCollector.sortc
C s� | j �� D ]\}}t|td�d�| j |< q| �� t� }t| j �dkrgt|�dkrgt|�d }| �|�rgt �
| j �� t �
|��|jg| j �}W d � n1 sQw Y t||jjjd � ||jj|ifS t j| j dd��� | �� D ]\}}|jjs�tjj||| j d� qt| jD ]}|j| j d�}|r�||j jj |7 < q�| j �� D ]$\}}|�� D ]\\} }
}t � |�} | � d d
� |D �| j |
i| j � q�q�| j �!� D ]}|�"� q�| j �� D ]8\}}t �
|�} d d
� |D �} | �| | j �}|r�||jj |7 < |jj�s|D ]
}tj#j||| j d� �qq�W d � n 1 �s w Y | j �!� D ] }|�� D ]\\} }
}|D ]
}t|| j|
� �q8�q0�q*| j �� D ]\}}|D ] }t||jjjd � �qR�qLt$|�!� �t%|�fS ) Nrk )r� r� r F)r4 � savepoint)�sender�instancer4 )r4 c S r� r rj r� r r r rl � r� z$Collector.delete.<locals>.<listcomp>c S r� r rj r� r r r rl � r� )&rT rf �sortedr r� r r� r� r� r
�mark_for_rollback_on_errorr4 r
� DeleteQuery� delete_batchrk �setattrra r� �label�atomicr� rG r rr �sendrV rv r* rU � UpdateQuery� update_batchr+ ro �reversert �sum�dict)
r r* r^ �deleted_counterr� �countr_ rp �instances_for_fieldvaluesr2 r< r� �pk_listr r r �delete{ st
���
�
��
��� �(���zCollector.delete)NFF)Fr )NFTNFFT)r r r r r` r8 r/ r7 ri rq ru r� r� r( r� r� r� r� r r r r rR O s(
$
�
rR ) r� � collectionsr r � functoolsr r � itertoolsr r � django.dbr r r
�django.db.modelsr r r
r r" r5 r6 r9 rB rC rE rF rQ rR r r r r �<module> s$