OwlCyberSecurity - MANAGER
Edit File: related_descriptors.cpython-39.pyc
a ���e�� � @ s d Z ddlmZ ddlmZ ddlmZmZmZm Z m Z ddl m Z m Z mZ ddlmZ ddlmZmZ ddlmZ dd lmZ dd lmZmZ dd lm Z G d d � d e�Z dd� Z G dd� d�Z G dd� de �Z G dd� d�Z!G dd� d�Z"dd� Z#G dd� de"�Z$d d � Z%d S ) a� Accessors for related objects. When a field defines a relation between two models, each model class provides an attribute to access related instances of the other model class (unless the reverse accessor has been disabled with related_name='+'). Accessors are implemented as descriptors in order to customize access and assignment. This module defines the descriptor classes. Forward accessors follow foreign keys. Reverse accessors trace them back. For example, with the following models:: class Parent(Model): pass class Child(Model): parent = ForeignKey(Parent, related_name='children') ``child.parent`` is a forward many-to-one relation. ``parent.children`` is a reverse many-to-one relation. There are three types of relations (many-to-one, one-to-one, and many-to-many) and two directions (forward and reverse) for a total of six combinations. 1. Related instance on the forward side of a many-to-one relation: ``ForwardManyToOneDescriptor``. Uniqueness of foreign key values is irrelevant to accessing the related instance, making the many-to-one and one-to-one cases identical as far as the descriptor is concerned. The constraint is checked upstream (unicity validation in forms) or downstream (unique indexes in the database). 2. Related instance on the forward side of a one-to-one relation: ``ForwardOneToOneDescriptor``. It avoids querying the database when accessing the parent link field in a multi-table inheritance scenario. 3. Related instance on the reverse side of a one-to-one relation: ``ReverseOneToOneDescriptor``. One-to-one relations are asymmetrical, despite the apparent symmetry of the name, because they're implemented in the database with a foreign key from one table to another. As a consequence ``ReverseOneToOneDescriptor`` is slightly different from ``ForwardManyToOneDescriptor``. 4. Related objects manager for related instances on the reverse side of a many-to-one relation: ``ReverseManyToOneDescriptor``. Unlike the previous two classes, this one provides access to a collection of objects. It returns a manager rather than an instance. 5. Related objects manager for related instances on the forward or reverse sides of a many-to-many relation: ``ManyToManyDescriptor``. Many-to-many relations are symmetrical. The syntax of Django models requires declaring them on one side but that's an implementation detail. They could be declared on the other side without any change in behavior. Therefore the forward and reverse descriptors can be the same. If you're looking for ``ForwardManyToManyDescriptor`` or ``ReverseManyToManyDescriptor``, use ``ManyToManyDescriptor`` instead. � )� sync_to_async)� FieldError)�DEFAULT_DB_ALIAS�NotSupportedError� connections�router� transaction)�Q�Window�signals)� RowNumber)� GreaterThan�LessThanOrEqual)�QuerySet)�DeferredAttribute)� AltersData�resolve_callables)�cached_propertyc @ s e Zd Zdd� ZdS )�ForeignKeyDeferredAttributec C s> |j �| jj�|kr,| j�|�r,| j�|� ||j | jj< d S �N)�__dict__�get�field�attname� is_cached�delete_cached_value��self�instance�value� r �l/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-packages/django/db/models/fields/related_descriptors.py�__set__V s � z#ForeignKeyDeferredAttribute.__set__N)�__name__� __module__� __qualname__r" r r r r! r U s r c C s� t f i |� d�|i��}| jp t}| jjr�t| jjs>td��| jj | jj }}dd� | jj |d�� � D �}t t� ||d�}|t||�M }|d ur�|t||�M }| j�� | �|�S )N�__inz`Prefetching from a limited queryset is only supported on backends that support window functions.c S s g | ] \}}|�qS r r )�.0�expr�_r r r! � <listcomp>h s z-_filter_prefetch_queryset.<locals>.<listcomp>)�using)Z partition_by�order_by)r �_dbr �queryZ is_slicedr �featuresZsupports_over_clauser �low_mark� high_markZ get_compilerZ get_order_byr r r r Z clear_limits�filter) �queryset� field_name� instances� predicate�dbr0 r1 r, Zwindowr r r! �_filter_prefetch_queryset^ s" �� r8 c @ s` e Zd ZdZdd� Zedd� �Zdd� Zdd � Zdd d �Z d d� Z ddd�Z dd� Z dd� Z d S )�ForwardManyToOneDescriptoraJ Accessor to the related object on the forward side of a many-to-one or one-to-one (via ForwardOneToOneDescriptor subclass) relation. In the example:: class Child(Model): parent = ForeignKey(Parent, related_name='children') ``Child.parent`` is a ``ForwardManyToOneDescriptor`` instance. c C s || _ d S r )r )r Zfield_with_relr r r! �__init__� s z#ForwardManyToOneDescriptor.__init__c C s6 t d| jjjjtf| jjjd| jjj| jjf d��S �N�RelatedObjectDoesNotExistz %s.%s.RelatedObjectDoesNotExist)r$ r% ) �typer � remote_field�model� DoesNotExist�AttributeErrorr$ r% �name�r r r r! r< � s ����z4ForwardManyToOneDescriptor.RelatedObjectDoesNotExistc C s | j �|�S r )r r �r r r r r! r � s z$ForwardManyToOneDescriptor.is_cachedc K s | j jjjj|d��� S �N)�hints)r r> r? � _base_manager� db_manager�all�r rF r r r! � get_queryset� s z'ForwardManyToOneDescriptor.get_querysetNc s� |d u r| � � }|j|d d� | jj}| jj� � fdd�|D �}| jjd }| jj}|�� snt| jj�dkr�d|j � fdd�|D �i}nd| j� � |i}|j f i |��}|j s�|D ] }|||� } |� || � q�||� d | j�� d fS ) Nr �r c s i | ]}� |�|�qS r r �r' �inst�Z instance_attrr r! � <dictcomp>� � zDForwardManyToOneDescriptor.get_prefetch_queryset.<locals>.<dictcomp>� �%s__inc s h | ]}� |�d �qS )r r rM rO r r! � <setcomp>� rQ zCForwardManyToOneDescriptor.get_prefetch_queryset.<locals>.<setcomp>TF)rK � _add_hintsr �get_foreign_related_value�get_local_related_value�foreign_related_fieldsr> Z is_hidden�lenrB �related_query_namer2 �multiple�set_cached_value�get_cache_name) r r5 r3 � rel_obj_attr�instances_dictZ related_fieldr> r. �rel_objr r rO r! �get_prefetch_queryset� s6 �� �z0ForwardManyToOneDescriptor.get_prefetch_querysetc C s | j |d�}|�| j�|��S �NrL )rK r r Zget_reverse_related_filter)r r �qsr r r! � get_object� s z%ForwardManyToOneDescriptor.get_objectc C s� |du r | S z| j �|�}W n� ty� d| j �|�v}|rN|j�| j j�nd}|r||�|�r||�|�}| j j|dd�}nd}|du r�|r�| �|�}| j j }|j s�|� ||� | j � ||� Y n0 |du r�| j j s�| � d| j jj| j jf ��n|S dS )a= Get the related instance through the forward relation. With the example above, when getting ``child.parent``: - ``self`` is the descriptor managing the ``parent`` attribute - ``instance`` is the ``child`` instance - ``cls`` is the ``Child`` class (we don't need it) N��default� %s has no %s.)r �get_cached_value�KeyErrorrW �_metaZget_ancestor_linkr? r rd r> r[ r\ �nullr< r# rB )r r �clsr` Z has_valueZ ancestor_linkZancestorr> r r r! �__get__� s2 �� �z"ForwardManyToOneDescriptor.__get__c C sT |durDt || jjjjj�sDtd||jj| jj| jjjjjf ��n`|dur�|j j du rlt j |j |d�|j _ |j j du r�t j |j |d�|j _ t �||�s�td| ��| jj}|du r�| jj|dd�}|dur�|�|d� | jjD ]\}}t||jd� q�n*| jjD ] \}}t||jt||j�� �q| j�||� |du�rP|j�sP|�||� dS )aX Set the related instance through the forward relation. With the example above, when setting ``child.parent = parent``: - ``self`` is the descriptor managing the ``parent`` attribute - ``instance`` is the ``child`` instance - ``value`` is the ``parent`` instance on the right of the equal sign N�4Cannot assign "%r": "%s.%s" must be a "%s" instance.rL �GCannot assign "%r": the current database router prevents this relation.re )� isinstancer r> r? rj Zconcrete_model� ValueError� object_namerB �_stater7 r � db_for_write� __class__�allow_relationrh r\ �related_fields�setattrr �getattrr[ )r r r r> �related�lh_field�rh_fieldr r r! r" � sN � ��� � � �� z"ForwardManyToOneDescriptor.__set__c C s t | jj| jjffS )z� Pickling should return the instance attached by self.field on the model, not a new copy of that descriptor. Use getattr() to retrieve the instance directly from the model. )ry r r? rB rC r r r! � __reduce__H s z%ForwardManyToOneDescriptor.__reduce__)N)N)r# r$ r% �__doc__r: r r< r rK ra rd rm r" r} r r r r! r9 s s ) 3Lr9 c s, e Zd ZdZ� fdd�Z� fdd�Z� ZS )�ForwardOneToOneDescriptora Accessor to the related object on the forward side of a one-to-one relation. In the example:: class Restaurant(Model): place = OneToOneField(Place, related_name='restaurant') ``Restaurant.place`` is a ``ForwardOneToOneDescriptor`` instance. c s� | j jjr���� }| j jj}dd� |jjD �� t� fdd�|D ��s��fdd�� D �}|f i |��}�jj |j_ �jj |j_ |S t � � ��S )Nc S s g | ] }|j �qS r )r �r' r r r r! r* c rQ z8ForwardOneToOneDescriptor.get_object.<locals>.<listcomp>c 3 s | ]}|� v V qd S r r r� )�fieldsr r! � <genexpr>h rQ z7ForwardOneToOneDescriptor.get_object.<locals>.<genexpr>c s i | ]}|t � |��qS r )ry r� rL r r! rP i rQ z8ForwardOneToOneDescriptor.get_object.<locals>.<dictcomp>) r r> � parent_linkZget_deferred_fieldsr? rj �concrete_fields�anyrs �addingr7 �superrd )r r �deferredZ rel_model�kwargs�obj�ru )r� r r! rd ] s z$ForwardOneToOneDescriptor.get_objectc st t � �||� | jjrp| jjjrp|j}dd� |jD �}|D ]4}|jjjj j }|d ur^t ||�nd }t |||� q:d S )Nc S s g | ]}|j r|jr|�qS r )� primary_keyr> r� r r r! r* w s �z5ForwardOneToOneDescriptor.__set__.<locals>.<listcomp>) r� r" r r� r> r� rj r� r? �pkr ry rx )r r r �optsZinherited_pk_fieldsr Zrel_model_pk_nameZ raw_valuer� r r! r" p s ��z!ForwardOneToOneDescriptor.__set__)r# r$ r% r~ rd r" � __classcell__r r r� r! r Q s r c @ sX e Zd ZdZdd� Zedd� �Zdd� Zdd � Zdd d �Z dd d�Z dd� Z dd� Z d S )�ReverseOneToOneDescriptora Accessor to the related object on the reverse side of a one-to-one relation. In the example:: class Restaurant(Model): place = OneToOneField(Place, related_name='restaurant') ``Place.restaurant`` is a ``ReverseOneToOneDescriptor`` instance. c C s || _ d S r )rz )r rz r r r! r: � s z"ReverseOneToOneDescriptor.__init__c C s4 t d| jjjtf| jjjd| jjj| jjf d��S r; ) r= rz � related_modelr@ rA r? r$ r% rB rC r r r! r< � s ����z3ReverseOneToOneDescriptor.RelatedObjectDoesNotExistc C s | j �|�S r )rz r rD r r r! r � s z#ReverseOneToOneDescriptor.is_cachedc K s | j jjj|d��� S rE )rz r� rG rH rI rJ r r r! rK � s z&ReverseOneToOneDescriptor.get_querysetNc s� |d u r| � � }|j|d d� | jjj}| jjj� � fdd�|D �}d| jjj |i}|jf i |��}|D ] }|||� }| jj�||� ql||� d| j� � dfS )Nr rL c s i | ]}� |�|�qS r r rM rO r r! rP � rQ zCReverseOneToOneDescriptor.get_prefetch_queryset.<locals>.<dictcomp>rS TF) rK rU rz r rW rV rB r2 r\ r] )r r5 r3 r^ r_ r. r` r r rO r! ra � s$ �z/ReverseOneToOneDescriptor.get_prefetch_querysetc C s� |du r | S z| j �|�}W n� ty� |j}|du r>d}nX| j j�|�}z | j|d�jf i |��}W n | j jj y� d}Y n0 | j j� ||� | j � ||� Y n0 |du r�| � d|j j | j �� f ��n|S dS )a� Get the related instance through the reverse relation. With the example above, when getting ``place.restaurant``: - ``self`` is the descriptor managing the ``restaurant`` attribute - ``instance`` is the ``place`` instance - ``cls`` is the ``Place`` class (unused) Keep in mind that ``Restaurant`` holds the foreign key to ``Place``. NrL rg )rz rh ri r� r Zget_forward_related_filterrK r r� r@ r\ r< ru r# �get_accessor_name)r r rl r` � related_pkZ filter_argsr r r! rm � s, ��z!ReverseOneToOneDescriptor.__get__c s6 |du r@| j j� dd�}|dur>| j �� � t|| j jjd� n�t|| j j�svtd|� j j | j � � | j jj j f ��n�� j j du r�tj� j|d�� j _ |j j du r�tj|j� d�|j _ t�|� �s�td| ��t� fdd�| j jjD ��}t| j jj�D ]\}}t||j|| � q�| j �� |� | j j�|� � dS )a� Set the related instance through the reverse relation. With the example above, when setting ``place.restaurant = restaurant``: - ``self`` is the descriptor managing the ``restaurant`` attribute - ``instance`` is the ``place`` instance - ``value`` is the ``restaurant`` instance on the right of the equal sign Keep in mind that ``Restaurant`` holds the foreign key to ``Place``. Nre rn rL ro c 3 s | ]}t � |j�V qd S r �ry r r� rL r r! r� ) s �z4ReverseOneToOneDescriptor.__set__.<locals>.<genexpr>)rz rh r rx r rB rp r� rq rj rr r� rs r7 r rt ru rv �tuplerX � enumerate�local_related_fieldsr r\ )r r r r` r� �indexr r rL r! r" � sH ��� � � �� �z!ReverseOneToOneDescriptor.__set__c C s t | jj| jjffS r )ry rz r? rB rC r r r! r} : s z$ReverseOneToOneDescriptor.__reduce__)N)N) r# r$ r% r~ r: r r< r rK ra rm r" r} r r r r! r� � s -Gr� c @ s> e Zd ZdZdd� Zedd� �Zd dd�Zd d � Zd d � Z dS )�ReverseManyToOneDescriptora� Accessor to the related objects manager on the reverse side of a many-to-one relation. In the example:: class Child(Model): parent = ForeignKey(Parent, related_name='children') ``Parent.children`` is a ``ReverseManyToOneDescriptor`` instance. Most of the implementation is delegated to a dynamically defined manager class built by ``create_forward_many_to_many_manager()`` defined below. c C s || _ |j| _d S r )�relr )r r� r r r! r: O s z#ReverseManyToOneDescriptor.__init__c C s | j j}t|jj| j �S r )r� r� �"create_reverse_many_to_one_manager�_default_managerru �r r� r r r! �related_manager_clsS s �z.ReverseManyToOneDescriptor.related_manager_clsNc C s |du r | S | � |�S )a9 Get the related objects through the reverse relation. With the example above, when getting ``parent.children``: - ``self`` is the descriptor managing the ``children`` attribute - ``instance`` is the ``parent`` instance - ``cls`` is the ``Parent`` class (unused) N)r� )r r rl r r r! rm \ s z"ReverseManyToOneDescriptor.__get__c C s d| j �� fS )Nz reverse side of a related set)r� r� rC r r r! � _get_set_deprecation_msg_paramsk s �z:ReverseManyToOneDescriptor._get_set_deprecation_msg_paramsc C s t d| �� ��d S )Nz@Direct assignment to the %s is prohibited. Use %s.set() instead.)� TypeErrorr� r r r r! r" q s ��z"ReverseManyToOneDescriptor.__set__)N) r# r$ r% r~ r: r r� rm r� r" r r r r! r� ? s r� c s G � �fdd�d| t �� � S )z� Create a manager for the reverse side of a many-to-one relation. This manager subclasses another manager, generally the default manager of the related model, and adds behaviors specific to many-to-one relations. c s� e Zd Z� �fdd�Z�fdd�ZdZdd� Zdd � Zd d � Z� fd d �Z d2� fdd� Z dd�dd�Z de _ dd�dd�Z de _ �� fdd�Zde_ dd� Zde_ �� fdd�Zde_ d d � Zde_ �� fd d �Zde_ d d!� Zde_ �jj�rNdd�d"d#�Zde_ dd�d$d%�Zde_ dd�d&d'�Zde_ dd�d(d)�Zde_ d*d+� Zde_ dd,d-�d.d/�Zde_ dd,d-�d0d1�Z de _ � Z S )3z:create_reverse_many_to_one_manager.<locals>.RelatedManagerc s2 t � �� || _�j| _�j| _| jj|i| _d S r )r� r: r r� r? r rB � core_filtersrD )ru r� r r! r: � s zCcreate_reverse_many_to_one_manager.<locals>.RelatedManager.__init__c s" t | j|�}t|j� �}|| j�S r )ry r? r� ru r �r �managerZ manager_class)r� r r! �__call__� s zCcreate_reverse_many_to_one_manager.<locals>.RelatedManager.__call__Tc S s> | j jD ]0}t| j|j�d u rtd| j�d|j� d���qd S )N�"z#" needs to have a value for field "z'" before this relationship can be used.)r rX ry r r rq )r r r r r! � _check_fk_val� s ��zHcreate_reverse_many_to_one_manager.<locals>.RelatedManager._check_fk_valc s � j ptj� j� jd�}t| jj}|j� jd� � j rD|� � j �}d|_ |j f i � j ��}� j jD ]2}t� j|j�}|du s�|dkrd|rd|�� S qd� j j�rz � j j}W n2 ty� t� fdd�� j jd jD ��}Y n0 t� j|j�}� j |� jii|_|S )�X Filter the queryset for the instance this manager is bound to. rL TN� c s g | ]}t � j|j��qS r )ry r r )r' � target_fieldrC r r! r* � s �zacreate_reverse_many_to_one_manager.<locals>.RelatedManager._apply_rel_filters.<locals>.<listcomp>���)r- r Z db_for_readr? r r r/ Z!interprets_empty_strings_as_nullsrU r+ �_defer_next_filterr2 r� r rX ry r �noneZ many_to_oner� r r� Z path_infosZ target_fieldsZ_known_related_objects)r r3 r7 Zempty_strings_as_nullr �valr� Z rel_obj_idr rC r! �_apply_rel_filters� s4 � �� �zMcreate_reverse_many_to_one_manager.<locals>.RelatedManager._apply_rel_filtersc S s6 z| j j�| jj�� � W n ttfy0 Y n0 d S r )r �_prefetched_objects_cache�popr r> r] rA ri rC r r r! �_remove_prefetched_objects� s �zUcreate_reverse_many_to_one_manager.<locals>.RelatedManager._remove_prefetched_objectsc sf | j jd u r t| j jj�d���z| j j| jj�� W S t t fy` t � � � }| � |� Y S 0 d S )NzQ instance needs to have a primary key value before this relationship can be used.)r r� rq ru r# r� r r> r] rA ri r� rK r� �r r3 r� r r! rK � s � � zGcreate_reverse_many_to_one_manager.<locals>.RelatedManager.get_querysetNc s� |d u rt � �� }|j|d d� |�|jp0| j�}| jj}| jj� � fdd�|D �}t|| jj |�}|D ],}| j� |�sj|||� }t || jj |� qj| jj � � }||� d|dfS )Nr rL c s i | ]}� |�|�qS r r rM rO r r! rP � rQ zdcreate_reverse_many_to_one_manager.<locals>.RelatedManager.get_prefetch_queryset.<locals>.<dictcomp>F)r� rK rU r+ r- r rW rV r8 rB r rx r> r] )r r5 r3 r^ r_ r` r Z cache_namer� rO r! ra � s zPcreate_reverse_many_to_one_manager.<locals>.RelatedManager.get_prefetch_queryset��bulkc s� � � � � �� tj� j� jd�}� fdd�}|r�g }|D ]8}||� |jjsZ|jj|krft d| ��|� |j � q:� jj � |�j|d�jf i � jj� ji�� nHtj|dd��* |D ]}||� |�� q�W d � n1 s�0 Y d S )NrL c s8 t | � j�s"td� jjj| f ��t| � jj� j� d S )N� '%s' instance expected, got %r) rp r? r� rj rr rx r rB r )r� rC r r! �check_and_update_obj� s ���z\create_reverse_many_to_one_manager.<locals>.RelatedManager.add.<locals>.check_and_update_objzA%r instance isn't saved. Use bulk=False or save the object first.�Zpk__inF�r+ Z savepoint)r� r� r rt r? r rs r� r7 rq �appendr� rG r+ r2 �updater rB r �atomic�save)r r� �objsr7 r� Zpksr� r rC r! �add� s. �� ��z>create_reverse_many_to_one_manager.<locals>.RelatedManager.addc � s t | j�|d|i�I d H S �Nr� �r r� �r r� r� r r r! �aadd s z?create_reverse_many_to_one_manager.<locals>.RelatedManager.aaddc sD | � � | j|| jj< tj| j| jd�}t� | �|��j f i |��S rb ) r� r r rB r rt r? r� rH �create�r r� r7 ��RelatedManagerru r r! r� s zAcreate_reverse_many_to_one_manager.<locals>.RelatedManager.createc � s t | j�f i |��I d H S r �r r� �r r� r r r! �acreate# s zBcreate_reverse_many_to_one_manager.<locals>.RelatedManager.acreatec sD | � � | j|| jj< tj| j| jd�}t� | �|��j f i |��S rb ) r� r r rB r rt r? r� rH � get_or_creater� r� r r! r� ( s zHcreate_reverse_many_to_one_manager.<locals>.RelatedManager.get_or_createc � s t | j�f i |��I d H S r �r r� r� r r r! �aget_or_create0 s zIcreate_reverse_many_to_one_manager.<locals>.RelatedManager.aget_or_createc sD | � � | j|| jj< tj| j| jd�}t� | �|��j f i |��S rb ) r� r r rB r rt r? r� rH �update_or_creater� r� r r! r� 5 s zKcreate_reverse_many_to_one_manager.<locals>.RelatedManager.update_or_createc � s t | j�f i |��I d H S r �r r� r� r r r! �aupdate_or_create= s zLcreate_reverse_many_to_one_manager.<locals>.RelatedManager.aupdate_or_createc W s� |sd S | � � | j�| j�}t� }|D ]^}t|| j�sNtd| jjj |f ��| j� |�|krl|� |j � q(| jj j�d|| jf ��q(| �| j|d�|� d S )Nr� z%r is not related to %r.r� )r� r rV r �setrp r? r� rj rr rW r� r� r> r@ �_clearr2 )r r� r� r� �old_idsr� r r r! �removeF s( ��� �zAcreate_reverse_many_to_one_manager.<locals>.RelatedManager.removec � s t | j�|d|i�I d H S r� �r r� r� r r r! �aremove` s zBcreate_reverse_many_to_one_manager.<locals>.RelatedManager.aremovec S s | � � | �| |� d S r )r� r� �r r� r r r! �cleare s z@create_reverse_many_to_one_manager.<locals>.RelatedManager.clearc � s t | j�|d�I d H S )Nr� �r r� r� r r r! �acleark s zAcreate_reverse_many_to_one_manager.<locals>.RelatedManager.aclearc S s� | � � tj| j| jd�}|�|�}|rB|jf i | jjd i�� nZt j |dd��< |D ]&}t || jjd � |j | jjgd� qVW d � n1 s�0 Y d S )NrL Fr� )Z update_fields) r� r rt r? r r+ r� r rB r r� rx r� )r r3 r� r7 r� r r r! r� p s zAcreate_reverse_many_to_one_manager.<locals>.RelatedManager._clearF)r� r� c S s� | � � t|�}| jjr�tj| j| jd�}tj |dd��� |r\| j |d� | j |d|i� n^t | � |��� �}g }|D ]"}||v r�|�|� qv|�|� qv| j|d|i� | j |d|i� W d � q�1 s�0 Y n| j |d|i� d S )NrL Fr� r� r� )r� r� r rk r rt r? r r r� r� r� r� r+ rI r� r� )r r� r� r� r7 Zold_objs�new_objsr� r r r! r� s" 0z>create_reverse_many_to_one_manager.<locals>.RelatedManager.setc � s t | j�|||d�I d H S )N)r� r� r� �r r� )r r� r� r� r r r! �aset� s z?create_reverse_many_to_one_manager.<locals>.RelatedManager.aset)N) r# r$ r% r: r� �do_not_call_in_templatesr� r� r� rK ra r� � alters_datar� r� r� r� r� r� r� r rk r� r� r� r� r� r� r� r� r �r� r� r� r! r� � sN ( % r� �r )� superclassr� r r� r! r� x s "r� c sB e Zd ZdZd � fdd� Zedd� �Zedd� �Zd d � Z � Z S ) �ManyToManyDescriptora� Accessor to the related objects manager on the forward and reverse sides of a many-to-many relation. In the example:: class Pizza(Model): toppings = ManyToManyField(Topping, related_name='pizzas') ``Pizza.toppings`` and ``Topping.pizzas`` are ``ManyToManyDescriptor`` instances. Most of the implementation is delegated to a dynamically defined manager class built by ``create_forward_many_to_many_manager()`` defined below. Fc s t � �|� || _d S r )r� r: �reverse)r r� r� r� r r! r: � s z ManyToManyDescriptor.__init__c C s | j jS r )r� �throughrC r r r! r� � s z ManyToManyDescriptor.throughc C s, | j r| jjn| jj}t|jj| j| j d�S )N)r� )r� r� r� r? �#create_forward_many_to_many_managerr� ru r� r r r! r� � s �z(ManyToManyDescriptor.related_manager_clsc C s* d| j r dnd | j r | j�� n| jjfS )Nz %s side of a many-to-many setr� �forward)r� r� r� r rB rC r r r! r� � s ��z4ManyToManyDescriptor._get_set_deprecation_msg_params)F) r# r$ r% r~ r: �propertyr� r r� r� r� r r r� r! r� � s r� c s G � ��fdd�d| t �� � S )z� Create a manager for the either side of a many-to-many relation. This manager subclasses another manager, generally the default manager of the related model, and adds behaviors specific to many-to-many relations. c s� e Zd Zd:� ��fdd� Z��fdd�ZdZdd� Zd d � Zd d � Z� fd d�Z d;� fdd� Z dd�dd�Z de _ dd�dd�Z de _ dd� Zde_ dd� Zde_ � fdd�Zde_ d d � Zde_ d dd �d d!�Zde_ d dd �d"d#�Zde_ dd��� fd$d%� Zde_ dd�d&d'�Zde_ dd��� fd(d)� Zde_ dd�d*d+�Zde_ dd��� fd,d-� Zde_ dd�d.d/�Zde_ d0d1� Zd2d3� Zd4d5� Z dd�d6d7�Z � fd8d9�Z � Z S )<z?create_forward_many_to_many_manager.<locals>.ManyRelatedManagerNc sb t � �� || _�sT�j| _�j�� | _�jj| _�j� � | _ �j� � | _ �j | _ n<�j| _�jj| _�j�� | _�j� � | _ �j� � | _ d| _ �j| _�| _| jj�| j �| _| jj�| j �| _i | _i | _| jjD ]8\}}d| j|jf }t||j�| j|< |j| j|j< q�| j�|�| _d | jv �rBt d|| j| j f ��|j d u �r^t d|j j ��d S )NFz%s__%sz\"%r" needs to have a value for field "%s" before this many-to-many relationship can be used.z]%r instance needs to have a primary key value before a many-to-many relationship can be used.) r� r: r r? r rZ �query_field_namerB �prefetch_cache_nameZm2m_field_name�source_field_nameZm2m_reverse_field_name�target_field_name� symmetricalr� r� r� rj � get_fieldZ source_fieldr� r� Zpk_field_namesrw ry r rV � related_valrq r� ru r# )r r r{ r| Zcore_filter_key)ru r� r� r r! r: � sL �� ��zHcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.__init__c s& t | j|�}t|j� ��}|| jd�S rb )ry r? r� ru r r� )r� r� r r! r� s �zHcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.__call__Tc S s� t �| j| jfg�}t|t� p&|�� }|rH|t �| j� d�|fg�M }| jr�t �| j| jfg�}|r�|t �| j� d�|fg�M }||O }|S )Nr& ) r r� r� r� rp r � _has_filtersr� r� )r Z removed_vals�filtersZremoved_vals_filtersZsymmetrical_filtersr r r! �_build_remove_filters s � ��zUcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager._build_remove_filtersc S s< |j | jd� | jr |�| j�}d|_|�� jf i | j��S )r� rL T)rU r r- r+ r� �_next_is_stickyr2 r� r� r r r! r� * s zRcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager._apply_rel_filtersc S s0 z| j j�| j� W n ttfy* Y n0 d S r )r r� r� r� rA ri rC r r r! r� 4 s zZcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager._remove_prefetched_objectsc s@ z| j j| j W S ttfy: t� �� }| �|� Y S 0 d S r )r r� r� rA ri r� rK r� r� r� r r! rK : s zLcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.get_querysetc s� |d u rt � �� }|j|d d� |�|jp0| j�}t|�� | j|�}| jj � | j ���j j j �t|j � � jj�|j��fdd��jD �d�}|�fdd�� �fdd�d | jd fS ) Nr rL c s* i | ]"}d |j d�� ��|j�f �qS )�_prefetch_related_val_%sz%s.%s)r �column�r' �f)� join_table�qnr r! rP W s ���zicreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.get_prefetch_queryset.<locals>.<dictcomp>)�selectc s t � fdd��jD ��S )Nc 3 s | ]}t � d |j �V qdS )r� Nr� r� ��resultr r! r� ` s ��zcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.get_prefetch_queryset.<locals>.<lambda>.<locals>.<genexpr>)r� r� r� )�fkr� r! �<lambda>` s �zgcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.get_prefetch_queryset.<locals>.<lambda>c s t �� fdd��jD ��S )Nc 3 s" | ]}|� t�|j�� �V qd S r )Zget_db_prep_valuery r r� )� connectionrN r r! r� d s �r� )r� rX �rN )r� r� r r! r� d s �F)r� rK rU r+ r- r8 r� r� r� rj r� r� r? Zdb_tabler r7 �opsZ quote_name�extrar� r� )r r5 r3 r� )r� r� r� r� r! ra A s. � �� �zUcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.get_prefetch_queryset��through_defaultsc W s� | � � tj| j| jd�}tj|dd��R | j| j| j g|�R d|i� | j rl| j| j | jg|�R d|i� W d � n1 s�0 Y d S )NrL Fr� r ) r� r rt r� r r r� � _add_itemsr� r� r� )r r r� r7 r r r! r� m s( ������zCcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.addc � s t | j�|d|i�I d H S �Nr r� )r r r� r r r! r� � s ��zDcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.aaddc W s$ | � � | j| j| jg|�R � d S r )r� � _remove_itemsr� r� �r r� r r r! r� � s zFcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.removec � s t | j�|� I d H S r r� r r r r! r� � s zGcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.aremovec s� t j| j| jd�}tj|dd��� tjj| jd| j| j | j d |d� | � � | � t � �� �|��}| jj�|��|��� tjj| jd| j| j | j d |d� W d � n1 s�0 Y d S )NrL Fr� Z pre_clear�Zsender�actionr r� r? Zpk_setr+ Z post_clear)r rt r� r r r� r � m2m_changed�sendr� r? r� r� r� rK r+ r� r2 �delete)r r7 r� r� r r! r� � s. � �zEcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.clearc � s t | j�� I d H S r r� rC r r r! r� � s zFcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.aclearF)r� r c S s� t |�}tj| j| jd�}tj|dd��� |rH| �� | j|d|i� n�t | � |�j | j j j dd��}g }|D ]J}t|| j�r�| j �|�d n | j �|�}||v r�|�|� qn|�|� qn| j|� | j|d|i� W d � n1 s�0 Y d S )NrL Fr� r T�Zflatr )r� r rt r� r r r� r� r� r� r+ � values_listr� r rp r? rV �get_prep_valuer� r� ) r r� r� r r7 r� r� r� �fk_valr r r! r� � s, �� � � zCcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.setc � s t | j�|||d�I d H S )N)r� r� r r� )r r� r� r r r r! r� � s �zDcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.asetc sB t j| jj| jd�}t� | �|��jf i |��}| j||d� |S �NrL r )r rt r ru r� rH r� r� )r r r� r7 Znew_obj��ManyRelatedManagerru r r! r� � s zFcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.createc � s t | j�f d|i|��I d H S r r� �r r r� r r r! r� � s ��zGcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.acreatec sN t j| jj| jd�}t� | �|��jf i |��\}}|rF| j||d� ||fS r )r rt r ru r� rH r� r� �r r r� r7 r� �createdr r r! r� � s � zMcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.get_or_createc � s t | j�f d|i|��I d H S r r� r r r r! r� � s ��zNcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.aget_or_createc sN t j| jj| jd�}t� | �|��jf i |��\}}|rF| j||d� ||fS r )r rt r ru r� rH r� r� r r r r! r� s �� zPcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.update_or_createc � s t | j�f d|i|��I d H S r r� r r r r! r� s ��zQcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.aupdate_or_createc S s� ddl m} t� }| jj�|�}|D ]�}t|| j�r�t� || j �s^t d|| j j j |j j f ��|�|�d }|du r�t d||f ��|�|� q$t||�r�td| jjj|f ��q$|�|�|�� q$|S )z[ Return the set of ids of `objs` that the target field references. r )�ModelzHCannot add "%r": instance is on database "%s", value is on database "%s"Nz1Cannot add "%r": the value for field "%s" is Noner� )�django.db.modelsr r� r� rj r� rp r? r rv r rq rs r7 rV r� r� rr r )r r� r� r � target_idsr� r� � target_idr r r! �_get_target_ids s6 ���� ��zOcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager._get_target_idsc S s@ | j j�|�j|dd�jf i || jd d| |i��}|�|�S )z{ Return the subset of ids of `objs` that aren't already assigned to this relationship. Tr r rS )r� r� r+ r r2 r� � difference)r r� r� r7 r �valsr r r! �_get_missing_target_ids6 s � ��� zWcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager._get_missing_target_idsc S sH | j jjduot| jj}| js*|| jko6tj � | j �}|||oD| fS )aw Return a boolean triple of the way the add should be performed. The first element is whether or not bulk_create(ignore_conflicts) can be used, the second whether or not signals must be sent, and the third element is whether or not the immediate bulk insertion with conflicts ignored can be performed. F) r� rj Z auto_createdr r/ Zsupports_ignore_conflictsr� r� r r Z has_listeners)r r7 r� �can_ignore_conflicts�must_send_signalsr r r! � _get_add_planI s �� ��zMcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager._get_add_planc s8 |sd S t t�pi ���� ��|�}tj� j� jd�}� �|��\}}} | rz� jj� |�j � ��fdd�|D �dd� d S � � ��||�} t j |dd��� |r�tjj� jd� j� j� j| |d � � jj� |�j � ���fd d�| D �|d� |�rtjj� jd � j� j� j| |d � W d � n1 �s*0 Y d S ) NrL c s2 g | ]*}� j f i d � � jd d � |i���qS �z%s_idr �r� r� �r' r )r r� r� r r! r* | s ���z^create_forward_many_to_many_manager.<locals>.ManyRelatedManager._add_items.<locals>.<listcomp>T)Zignore_conflictsFr� Zpre_addr c s6 g | ].}� j f i ��d � � jd d � |i���qS r# r$ r% �r r� r� r r r! r* � s ����Zpost_add)�dictr r r rt r� r r" r� r+ Z bulk_creater r r� r r r r� r? ) r r� r� r r� r r7 r r! Z can_fast_addZmissing_target_idsr r& r! r j s\ � � � �� � ��zJcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager._add_itemsc s2 |sd S t � }|D ]6}t|| j�r>| j�|�d }|�|� q|�|� qtj| j| j d�}t j |dd��� t j j| jd| j | j| j||d� t� �� }|�� r�|�|�jf i d| jjj |i��} n|} | �| �} | jj�|��| ��� t j j| jd| j | j| j||d� W d � n1 �s$0 Y d S ) Nr rL Fr� Z pre_remover rS Z post_remove)r� rp r? r� rV r� r rt r� r r r� r r r r� r� rK r� r+ r2 r r� r� r ) r r� r� r� r� r� r r7 Ztarget_model_qsZold_valsr� r� r r! r � sH � � �zMcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager._remove_items)N)N) r# r$ r% r: r� r� r� r� r� rK ra r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r" r r r� r �r r� r� r� r! r � sT 2 , "� Gr r� )r� r� r� r r( r! r� � s r� N)&r~ Z asgiref.syncr Zdjango.core.exceptionsr Z django.dbr r r r r r r r r Zdjango.db.models.functionsr Zdjango.db.models.lookupsr r Zdjango.db.models.queryr Z django.db.models.query_utilsr Zdjango.db.models.utilsr r Zdjango.utils.functionalr r r8 r9 r r� r� r� r� r� r r r r! �<module> s. A _3 <9 -/