OwlCyberSecurity - MANAGER
Edit File: deletion.cpython-39.pyc
a ���ekR � @ s� d dl mZmZ d dlmZmZ d dlmZ d dlm Z m Z d dl m Z m Z mZmZ d dlmZmZmZ G dd� de �ZG d d � d e �Zd d � Zd d� Zdd� Zdd� Zdd� Zde_dd� Z de _dd� Z dd� Z G d d � d �Z d S ) � )�Counter� defaultdict)�partial�reduce)�chain)� attrgetter�or_)�IntegrityError� connections�models� transaction)� query_utils�signals�sqlc s e Zd Z� fdd�Z� ZS )�ProtectedErrorc s || _ t� �||� d S �N)�protected_objects�super�__init__)�self�msgr �� __class__� �Z/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-packages/django/db/models/deletion.pyr s zProtectedError.__init__��__name__� __module__� __qualname__r � __classcell__r r r r r s r c s e Zd Z� fdd�Z� ZS )�RestrictedErrorc s || _ t� �||� d S r )�restricted_objectsr r )r r r! r r r r s zRestrictedError.__init__r r r r r r s r c C s@ | j ||jj|j|jdd� |jr<t| jjs<| �|d |� d S )NF)�source� source_attr�nullable�fail_on_restricted) �collect� remote_field�model�name�nullr �featuresZcan_defer_constraint_checks�add_field_update�� collector�field�sub_objs�usingr r r �CASCADE s �r2 c C s( t d|jjj|d jj|jf |��d S )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) r- r r r �PROTECT" s ���r3 c C s" | � ||� | �|jj|j� d S r )�add_restricted_objects�add_dependencyr' r( r- r r r �RESTRICT/ s r6 c s: t � �r� fdd�}n � fdd�}� fdd�|_d|_|S )Nc s | � |� � |� d S r �r, r- ��valuer r � set_on_delete7 s zSET.<locals>.set_on_deletec s | � |� |� d S r r7 r- r8 r r r: < s c s d� fi fS )Nzdjango.db.models.SETr r r8 r r �<lambda>? � zSET.<locals>.<lambda>T)�callableZ deconstruct� lazy_sub_objs)r9 r: r r8 r �SET4 s r? c C s | � |d |� d S r r7 r- r r r �SET_NULLD s r@ Tc C s | � ||�� |� d S r )r, � get_defaultr- r r r � SET_DEFAULTK s rB c C s d S r r r- r r r � DO_NOTHINGR s rC c C s dd� | j dd�D �S )Nc s s* | ]"}|j r|js|js |jr|V qd S r )� auto_createdZconcreteZ one_to_oneZ one_to_many)�.0�fr r r � <genexpr>Y s �z4get_candidate_relations_to_delete.<locals>.<genexpr>T)Zinclude_hidden)Z get_fields)�optsr r r �!get_candidate_relations_to_deleteV s �rI c @ s� e Zd Zd"dd�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 )'� CollectorNc C sD || _ || _tt�| _tt�| _tttt��| _g | _ tt�| _ d S r ) r1 �originr �set�data�list� field_updatesr r! � fast_deletes� dependencies)r r1 rK r r r r a s zCollector.__init__Fc C sf |sg S g }|d j }| j| }|D ]}||vr$|�|� q$|�|� |durb|sb| 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 rM �append�updater5 ) r �objsr" r$ rS �new_objsr( � instances�objr r r �addv s z Collector.addc C s> |r|| }}| j |jj �|jj� | j�|| j�� � d S r )rQ �_meta�concrete_modelrZ rM � setdefault�default_factory)r r( � dependencyrS r r r r5 � s �zCollector.add_dependencyc C s | j ||f �|� dS )z� Schedule a field update. 'objs' must be a homogeneous iterable collection of model instances (e.g. a QuerySet). N)rO rT )r r/ r9 rV r r r r, � s zCollector.add_field_updatec C s& |r"|d j }| j| | �|� d S )Nr )r r! rU )r r/ rV r( r r r r4 � s z Collector.add_restricted_objectsc s0 || j v r,� fdd�| j | �� D �| j |< d S )Nc s i | ]\}}||� �qS r r )rE r/ �items�rV r r � <dictcomp>� s �z?Collector.clear_restricted_objects_from_set.<locals>.<dictcomp>)r! r` )r r( rV r ra r �!clear_restricted_objects_from_set� s �z+Collector.clear_restricted_objects_from_setc C s> || j v r:t|jdd� | j | �� D �d��}| �||� d S )Nc S s g | ]}|D ] }|j �q qS r ��pk)rE rV rY r r r � <listcomp>� s �zDCollector.clear_restricted_objects_from_queryset.<locals>.<listcomp>)Zpk__in)r! rL �filter�valuesrc )r r( �qsrV r r r �&clear_restricted_objects_from_queryset� s ��� z0Collector.clear_restricted_objects_from_querysetc C s t j�|�pt j�|�S r )r � pre_deleteZ has_listeners� post_delete)r r( r r r �_has_signal_listeners� s � �z Collector._has_signal_listenersc s� � r� j jturdS t|d�r(|jj}n t|d�rDt|d�rD|j}ndS | �|�rVdS |j}t� fdd�|jjj � � D ��o�tdd� t |�D ��o�t 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. Fr[ r( � _raw_deletec 3 s | ]}|� kV qd S r r )rE �link�� from_fieldr r rG � s �z,Collector.can_fast_delete.<locals>.<genexpr>c s s | ]}|j jjtu V qd S r )r/ r' � on_deleterC )rE �relatedr r r rG � s �c s s | ]}t |d �V qdS )�bulk_related_objectsN)�hasattr�rE r/ r r r rG � s �)r' rr r2 ru r[ r( rm �allr\ �parentsrh rI �any�private_fields)r rV rq r( rH r rp r �can_fast_delete� s* ��� ��zCollector.can_fast_deletec s^ dd� |D �}t t| j j�|��d�� t��� krT� �fdd�tdt��� �D �S �gS dS )zT Return the objs in suitably sized batches for the used connection. c S s g | ] }|j �qS r �r) rv r r r rf � r< z-Collector.get_del_batches.<locals>.<listcomp>� c s g | ]}�||� � �qS r r )rE �i�Zconn_batch_sizerV r r rf � s �r N)�maxr r1 �opsZbulk_batch_size�len�range)r rV �fields� field_namesr r r �get_del_batches� s � �zCollector.get_del_batchesTc s� | � |�r| j�|� dS | j||||d�} | s4dS | d j} |s�| jj} | jj�� D ]4� � rV� fdd�| D �} | j | | � j j dddd� qV|s�dS |r�t | j� � �} tt�}tt�}t| j�D �]2}|r�|j| v r�q�|j}|j j}|tkr�q�|j}| j ||d ��r|| �|� q�| �| |g�}|D ]�}| �||g|�}|jj�s|| �|��s|t t� d d � t|j�D ���}|j t |�� }t |d d��s�|�r&z|| ||| j � W nJ t!�y� } z0d |jj"|j#f }|| |j$7 < W Y d}~n d}~0 0 �q&q�|�r$t!d| j"d�%|�f t t� |�� ����|�&� D ]<\}}| �| |�}|D ] }| �|||�}| j�|� �qD�q,| jj'D ]2}t(|d��rr|�)| | j �}| j || ddd� �qr|�r|| j*�&� D ]\}}| �+||� �q�| jD ]} | �,| j| � �q�| j-�� �r|tt�} | j-�&� D ]D\}} | �&� D ]0\}}|�rd |j"|j#f }| | |7 < �q�q| �r|t.d| j"d�%| �f t t� | �� ����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. NrR r c s g | ]}t |� j��qS r )�getattrr) �rE rY �Zptrr r rf * r< z%Collector.collect.<locals>.<listcomp>FT)r" r# �collect_relatedrS r% rp c s s | ]}d d� |j jD �V qdS )c s s | ] }|j V qd S r )�attname)rE �rfr r r rG U r< z.Collector.collect.<locals>.<genexpr>.<genexpr>N)r/ Zforeign_related_fields)rE �relr r r rG T s �z$Collector.collect.<locals>.<genexpr>r> z'%s.%s'zhCannot delete some instances of model %r because they are referenced through protected foreign keys: %s.z, rt )r" r$ r% ziCannot delete some instances of model %r because they are referenced through restricted foreign keys: %s.)/r{ rP rT rZ r r[ r\ rx rh r& r' Z related_namerL Zget_parent_listr rN rI r( r/ rr rC � related_modelr� �related_objects�queryZselect_relatedrm r � from_iterable�only�tupler� r1 r r r) r �joinr` rz ru rt rM rc rj r! r ) r rV r" r$ r� r# rS Z keep_parentsr% rW r( r\ Z parent_objsrx Zmodel_fast_deletesr rs r/ rr r� Zbatches�batchr0 Zreferenced_fields�error�key�related_fieldsrX ri r! r� r r� r r& � s� " � ��� ���.��� � ���zCollector.collectc s6 t jj� fdd�|D �t jjd�}|j�| j��|�S )zQ Get a QuerySet of the related model to objs via related fields. c s g | ]}|j � d �� f�qS )Z__inr| )rE Z related_fieldra r r rf � r< z-Collector.related_objects.<locals>.<listcomp>)Z connector)r �Q�create�ORZ _base_managerr1 rg )r r� r� rV � predicater ra r r� � s �zCollector.related_objectsc c s, | j �� D ] \}}|D ]}||fV qq d S r )rM r` )r r( rX rY r r r �instances_with_model� s z Collector.instances_with_modelc s� g }t � }t� j�}t|�t|�k r�d}|D ]H}||v r:q,� j�|jj�}|rX|�|�s,|� |� |� |jj� d}q,|sd S q� fdd�|D �� _d S )NFTc s i | ]}|� j | �qS r )rM )rE r( �r r r rb � r< z"Collector.sort.<locals>.<dictcomp>) rL rN rM r� rQ �getr[ r\ � differencerT rZ )r Z sorted_modelsZconcrete_modelsr �foundr( rQ r r� r �sort� s zCollector.sortc C s | j �� D ] \}}t|td�d�| j |< q | �� t� }t| j �dkr�t|�dkr�t|�d }| �|�r�t � | j ��( t � |��|jg| j �}W d � n1 s�0 Y t||jjjd � ||jj|ifS t j| j dd���� | �� D ](\}}|jjs�tjj||| j | jd� q�| jD ]0}|j | j d�}|�r||j jj |7 < �q| j �� D ]�\\}} } g } g } | D ]6}t |t j!��r�|j"d u �r�| �#|� n | �$|� �qn| �r�t%t&| �} | j'f i |j(| i�� | �rV| d j)}t �*|�}|�+td d � | D ��|j(| i| j � �qV| j �,� D ]}|�-� �q| j �� D ]x\}}t � |�}d d � |D �}|�|| j �}|�rz||jj |7 < |jj�s2|D ] }tj.j||| j | jd� �q��q2W d � n1 �s�0 Y | j �� D ](\}}|D ]}t||jjjd � �q�q�t/|�,� �t0|�fS ) Nre )r� r} r F)r1 Z savepoint)Zsender�instancer1 rK )r1 c S s h | ] }|j �qS r rd r� r r r � <setcomp>� r< z#Collector.delete.<locals>.<setcomp>c S s g | ] }|j �qS r rd r� r r r rf � r< z$Collector.delete.<locals>.<listcomp>)1rM r` �sortedr r� r r� rN r{ r Zmark_for_rollback_on_errorr1 r Z DeleteQueryZ delete_batchre �setattrr[ r� �labelZatomicr� rD r rk �sendrK rP rn r( rO � isinstancer ZQuerySetZ _result_cacherT �extendr r rU r) r Z UpdateQueryZ update_batchrh �reverserl �sum�dict)r r( rX Zdeleted_counterr� �countrY ri r/ r9 Zinstances_listZupdatesrV Zcombined_updatesr� Zpk_listr r r �delete� s� �"� �� � �.zCollector.delete)N)NFF)F)N)NFTNFFT)r r r r rZ r5 r, r4 rc rj rm r{ r� r& r� r� r� r� r r r r rJ ` s. , � rJ N) � collectionsr r � functoolsr r � itertoolsr �operatorr r Z django.dbr r r r Zdjango.db.modelsr r r r r r2 r3 r6 r? r@ r> rB rC rI rJ r r r r �<module> s$