B
    ©0H[4Á  ã               @   sž   d Z ddlmZmZmZ ddlmZmZ ddlm	Z	 ddl
mZ G dd„ dƒZG dd	„ d	eƒZG d
d„ dƒZG dd„ dƒZdd„ ZG dd„ deƒ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.
é    )ÚconnectionsÚrouterÚtransaction)ÚQÚsignals)ÚQuerySet)Úcached_propertyc               @   sX   e Zd ZdZdd„ Zedd„ ƒZdd„ Zdd	„ Zddd„Z	dd„ Z
ddd„Zdd„ 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 )N)Úfield)ÚselfZfield_with_rel© r   úoC:\Users\HIRONO~1\AppData\Local\Temp\pip-install-6bm3nxem\django\django\db\models\fields\related_descriptors.pyÚ__init__U   s    z#ForwardManyToOneDescriptor.__init__c             C   s   t d| jjjjtfi ƒS )NÚRelatedObjectDoesNotExist)Útyper
   Úremote_fieldÚmodelÚDoesNotExistÚAttributeError)r   r   r   r   r   X   s    z4ForwardManyToOneDescriptor.RelatedObjectDoesNotExistc             C   s   | j  |¡S )N)r
   Ú	is_cached)r   Úinstancer   r   r   r   c   s    z$ForwardManyToOneDescriptor.is_cachedc             K   s   | j jjjj|d� ¡ S )N)Úhints)r
   r   r   Ú_base_managerÚ
db_managerÚall)r   r   r   r   r   Úget_querysetf   s    z'ForwardManyToOneDescriptor.get_querysetNc       
         sì   |d kr|   ¡ }|j|d d� | jj}| jj‰ ‡ fdd„|D ƒ}| jjd }| jj}| ¡ snt| jjƒdkrŒd|j	 ‡ fdd„|D ƒi}nd| j 
¡  |i}|jf |Ž}|jsÖx$|D ]}|||ƒ }	| ||	¡ q¶W ||ˆ d	| j ¡ d
fS )Nr   )r   c                s   i | ]}|ˆ |ƒ“qS r   r   )Ú.0Úinst)Úinstance_attrr   r   ú
<dictcomp>p   s    zDForwardManyToOneDescriptor.get_prefetch_queryset.<locals>.<dictcomp>é   z%s__inc                s   h | ]}ˆ |ƒd  ’qS )r   r   )r   r   )r   r   r   ú	<setcomp>{   s    zCForwardManyToOneDescriptor.get_prefetch_queryset.<locals>.<setcomp>TF)r   Ú
_add_hintsr
   Úget_foreign_related_valueÚget_local_related_valueÚforeign_related_fieldsr   Z	is_hiddenÚlenÚnameÚrelated_query_nameÚfilterÚmultipleÚset_cached_valueÚget_cache_name)
r   Ú	instancesÚquerysetÚrel_obj_attrÚinstances_dictZrelated_fieldr   ÚqueryÚrel_objr   r   )r   r   Úget_prefetch_queryseti   s"    
z0ForwardManyToOneDescriptor.get_prefetch_querysetc             C   s   | j |d�}| | j |¡¡S )N)r   )r   Úgetr
   Zget_reverse_related_filter)r   r   Úqsr   r   r   Ú
get_objectˆ   s    z%ForwardManyToOneDescriptor.get_objectc             C   s´   |dkr| S y| j  |¡}W n` tk
r|   | j  |¡}d|krFd}n$|  |¡}| j j}|jsj| ||¡ | j  ||¡ Y nX |dkr¬| j js¬|  	d| j j
j| j jf ¡‚n|S dS )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)
        Nz%s has no %s.)r
   Úget_cached_valueÚKeyErrorr$   r6   r   r*   r+   Únullr   r   Ú__name__r'   )r   r   Úclsr2   Úvalr   r   r   r   Ú__get__�   s"    

z"ForwardManyToOneDescriptor.__get__c             C   s|  |dk	rDt || jjjjjƒsDtd||jj| jj| jjjjjf ƒ‚n||dk	rÀ|j	j
dkrntj|j|d�|j	_
nR|j	j
dkr�tj|j|d�|j	_
n0|j	j
dk	rÀ|j	j
dk	rÀt ||¡sÀtd| ƒ‚| jj}|dk�r| jj|dd�}|dk	rö| |d¡ xT| jjD ]\}}t||jdƒ �q W n.x,| jjD ] \}}t||jt||jƒƒ �q(W | j ||¡ |dk	�rx|j�sx| ||¡ dS )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
        Nz4Cannot assign "%r": "%s.%s" must be a "%s" instance.)r   zGCannot assign "%r": the current database router prevents this relation.)Údefault)Ú
isinstancer
   r   r   Ú_metaZconcrete_modelÚ
ValueErrorÚobject_namer'   Ú_stateÚdbr   Údb_for_writeÚ	__class__Úallow_relationr7   r+   Úrelated_fieldsÚsetattrÚattnameÚgetattrr*   )r   r   Úvaluer   ÚrelatedÚlh_fieldÚrh_fieldr   r   r   Ú__set__´   s6    
z"ForwardManyToOneDescriptor.__set__)N)N)r:   Ú
__module__Ú__qualname__Ú__doc__r   r   r   r   r   r3   r6   r=   rP   r   r   r   r   r	   H   s   

'r	   c                   s,   e Zd ZdZ‡ fdd„Z‡ fdd„Z‡  ZS )Ú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 jjr|ˆ ¡ }| j jj}dd„ |jjD ƒ‰ t‡ fdd„|D ƒƒs|‡fdd„ˆ D ƒ}|f |Ž}ˆjj	|j_	ˆjj
|j_
|S tƒ  ˆ¡S )Nc             S   s   g | ]
}|j ‘qS r   )rJ   )r   r
   r   r   r   ú
<listcomp>
  s    z8ForwardOneToOneDescriptor.get_object.<locals>.<listcomp>c             3   s   | ]}|ˆ kV  qd S )Nr   )r   r
   )Úfieldsr   r   ú	<genexpr>  s    z7ForwardOneToOneDescriptor.get_object.<locals>.<genexpr>c                s   i | ]}t ˆ |ƒ|“qS r   )rK   )r   r
   )r   r   r   r     s    z8ForwardOneToOneDescriptor.get_object.<locals>.<dictcomp>)r
   r   Úparent_linkZget_deferred_fieldsr   r@   Úconcrete_fieldsÚanyrC   ÚaddingrD   Úsuperr6   )r   r   ÚdeferredZ	rel_modelÚkwargsÚobj)rF   )rV   r   r   r6     s    


z$ForwardOneToOneDescriptor.get_objectc                sx   t ƒ  ||¡ | jjrt| jjjrt|j}dd„ |jD ƒ}x<|D ]4}|jjjj	j
}|d k	r`t||ƒnd }t|||ƒ q<W d S )Nc             S   s   g | ]}|j r|jr|‘qS r   )Úprimary_keyr   )r   r
   r   r   r   rU     s    z5ForwardOneToOneDescriptor.__set__.<locals>.<listcomp>)r\   rP   r
   r`   r   rX   r@   rY   r   ÚpkrJ   rK   rI   )r   r   rL   ÚoptsZinherited_pk_fieldsr
   Zrel_model_pk_nameZ	raw_value)rF   r   r   rP     s    
z!ForwardOneToOneDescriptor.__set__)r:   rQ   rR   rS   r6   rP   Ú__classcell__r   r   )rF   r   rT   ø   s   
rT   c               @   sP   e Zd ZdZdd„ Zedd„ ƒZdd„ Zdd	„ Zddd„Z	ddd„Z
dd„ 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 )N)rM   )r   rM   r   r   r   r   5  s    z"ReverseOneToOneDescriptor.__init__c             C   s   t d| jjjtfi ƒS )Nr   )r   rM   Úrelated_modelr   r   )r   r   r   r   r   :  s    z3ReverseOneToOneDescriptor.RelatedObjectDoesNotExistc             C   s   | j  |¡S )N)rM   r   )r   r   r   r   r   r   D  s    z#ReverseOneToOneDescriptor.is_cachedc             K   s   | j jjj|d� ¡ S )N)r   )rM   re   r   r   r   )r   r   r   r   r   r   G  s    z&ReverseOneToOneDescriptor.get_querysetNc                s¤   |d kr|   ¡ }|j|d d� | jjj}| jjj‰ ‡ fdd„|D ƒ}d| jjj |i}|jf |Ž}x(|D ] }|||ƒ }| jj ||¡ qjW ||ˆ d| j 	¡ dfS )Nr   )r   c                s   i | ]}|ˆ |ƒ“qS r   r   )r   r   )r   r   r   r   Q  s    zCReverseOneToOneDescriptor.get_prefetch_queryset.<locals>.<dictcomp>z%s__inTF)
r   r"   rM   r
   r$   r#   r'   r)   r+   r,   )r   r-   r.   r/   r0   r1   r2   r   r   )r   r   r3   J  s    


z/ReverseOneToOneDescriptor.get_prefetch_querysetc             C   sØ   |dkr| S y| j  |¡}W nŒ tk
r¨   |j}|dkr@d}nV| j j |¡}y| j|d�jf |Ž}W n | j jj	k
r„   d}Y nX | j j 
||¡ | j  
||¡ Y nX |dkrÐ|  d|jj| j  ¡ f ¡‚n|S dS )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``.
        N)r   z%s has no %s.)rM   r7   r8   ra   r
   Zget_forward_related_filterr   r4   re   r   r+   r   rF   r:   Úget_accessor_name)r   r   r;   r2   Ú
related_pkZfilter_argsr   r   r   r=   \  s*    
z!ReverseOneToOneDescriptor.__get__c                sZ  |dkrB| j jˆ dd�}|dk	r>| j  ˆ ¡ t|| j jjdƒ �nt|| j jƒsxtd|ˆ j	j
| j  ¡ | j jj	j
f ƒ‚nÞˆ jjdkrštjˆ j|d�ˆ j_nR|jjdkr¼tj|jˆ d�|j_n0|jjdk	rìˆ jjdk	rìt |ˆ ¡sìtd| ƒ‚t‡ fdd„| j jjD ƒƒ}x.t| j jjƒD ]\}}t||j|| ƒ �qW | j  ˆ |¡ | j j |ˆ ¡ dS )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``.
        N)r>   z4Cannot assign "%r": "%s.%s" must be a "%s" instance.)r   zGCannot assign "%r": the current database router prevents this relation.c             3   s   | ]}t ˆ |jƒV  qd S )N)rK   rJ   )r   r
   )r   r   r   rW   º  s    z4ReverseOneToOneDescriptor.__set__.<locals>.<genexpr>)rM   r7   Zdelete_cached_valuerI   r
   r'   r?   re   rA   r@   rB   rf   rC   rD   r   rE   rF   rG   Útupler%   Ú	enumerateÚlocal_related_fieldsrJ   r+   )r   r   rL   r2   rg   Úindexr
   r   )r   r   rP   ‹  s0    z!ReverseOneToOneDescriptor.__set__)N)N)r:   rQ   rR   rS   r   r   r   r   r   r3   r=   rP   r   r   r   r   rd   (  s   


/rd   c               @   s>   e Zd ZdZdd„ Zedd„ ƒZddd„Zd	d
„ Zdd„ Z	dS )Ú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 )N)Úrelr
   )r   rm   r   r   r   r   Ø  s    z#ReverseManyToOneDescriptor.__init__c             C   s   | j j}t|jj| j ƒS )N)rm   re   Ú"create_reverse_many_to_one_managerÚ_default_managerrF   )r   re   r   r   r   Úrelated_manager_clsÜ  s    z.ReverseManyToOneDescriptor.related_manager_clsNc             C   s   |dk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)rp   )r   r   r;   r   r   r   r=   å  s    
z"ReverseManyToOneDescriptor.__get__c             C   s   d| j  ¡ fS )Nzreverse side of a related set)rm   rf   )r   r   r   r   Ú_get_set_deprecation_msg_paramsô  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.)Ú	TypeErrorrq   )r   r   rL   r   r   r   rP   ú  s    z"ReverseManyToOneDescriptor.__set__)N)
r:   rQ   rR   rS   r   r   rp   r=   rq   rP   r   r   r   r   rl   È  s   	
rl   c                s   G ‡ ‡fdd„d| ƒ‰ ˆ 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 Zd Z‡ ‡fdd„Z‡fdd„ZdZdd„ Zdd	„ Z‡ fd
d„Zd"‡ fdd„	Z	ddœdd„Z
de
_‡‡ fdd„Zde_‡‡ fdd„Zde_‡‡ fdd„Zde_”jjrÞddœdd„Zde_ddœdd„Zde_dd„ Zde_dddœd d!„Zde_‡  ZS )#z:create_reverse_many_to_one_manager.<locals>.RelatedManagerc                s2   t ƒ  ¡  || _ˆj| _ˆj| _| jj|i| _d S )N)r\   r   r   re   r   r
   r'   Úcore_filters)r   r   )rF   rm   r   r   r   
  s
    
zCcreate_reverse_many_to_one_manager.<locals>.RelatedManager.__init__c               s"   t | j|ƒ}t|jˆ ƒ}|| jƒS )N)rK   r   rn   rF   r   )r   ÚmanagerÚmanager_class)rm   r   r   Ú__call__  s    zCcreate_reverse_many_to_one_manager.<locals>.RelatedManager.__call__Tc             S   s¨   | j ptj| j| jd�}t| jj}|j| jd� | j rD| 	| j ¡}|j
f | jŽ}x:| jjD ].}t| j|jƒ}|dks‚|dkr\|r\| ¡ S q\W | j| jj| jii|_|S )zX
            Filter the queryset for the instance this manager is bound to.
            )r   NÚ )Ú_dbr   Zdb_for_readr   r   r   ÚfeaturesZ!interprets_empty_strings_as_nullsr"   Úusingr)   rs   r
   r%   rK   rJ   Únonera   Z_known_related_objects)r   r.   rD   Zempty_strings_as_nullr
   r<   r   r   r   Ú_apply_rel_filters  s    zMcreate_reverse_many_to_one_manager.<locals>.RelatedManager._apply_rel_filtersc          	   S   s6   y| j j | j ¡ ¡ W n ttfk
r0   Y nX d S )N)r   Ú_prefetched_objects_cacheÚpopr
   r(   r   r8   )r   r   r   r   Ú_remove_prefetched_objects*  s    zUcreate_reverse_many_to_one_manager.<locals>.RelatedManager._remove_prefetched_objectsc          	      s@   y| j j| j ¡  S  ttfk
r:   tƒ  ¡ }|  |¡S X d S )N)	r   r}   r
   r(   r   r8   r\   r   r|   )r   r.   )rF   r   r   r   0  s
    
zGcreate_reverse_many_to_one_manager.<locals>.RelatedManager.get_querysetNc       	         s¶   |d krt ƒ  ¡ }|j|d d� | |jp0| j¡}| jj}| jj‰ ‡ fdd„|D ƒ}d| jj |i}|j	f |Ž}x(|D ] }|||ƒ }t
|| jj|ƒ qxW | j ¡ }||ˆ d|dfS )Nr   )r   c                s   i | ]}|ˆ |ƒ“qS r   r   )r   r   )r   r   r   r   @  s    zdcreate_reverse_many_to_one_manager.<locals>.RelatedManager.get_prefetch_queryset.<locals>.<dictcomp>z%s__inF)r\   r   r"   rz   rx   r
   r$   r#   r'   r)   rI   r(   )	r   r-   r.   r/   r0   r1   r2   r   Z
cache_name)rF   )r   r   r3   7  s    


zPcreate_reverse_many_to_one_manager.<locals>.RelatedManager.get_prefetch_queryset)Úbulkc         	      sÞ   ˆ   ¡  t|ƒ}tjˆ jˆ jd�}‡ fdd„}|r¢g }x@|D ]8}||ƒ |jjs\|jj|krht	d| ƒ‚| 
|j¡ q<W ˆ jj |¡j|d�jf ˆ jjˆ jiŽ n8tj|dd��$ x|D ]}||ƒ | ¡  q¸W W d Q R X d S )N)r   c                s8   t | ˆ jƒs"tdˆ jjj| f ƒ‚t| ˆ jjˆ jƒ d S )Nz'%s' instance expected, got %r)	r?   r   rr   r@   rB   rI   r
   r'   r   )r_   )r   r   r   Úcheck_and_update_objQ  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.)Úpk__inF)rz   Ú	savepoint)r   Úlistr   rE   r   r   rC   r[   rD   rA   Úappendra   r   rz   r)   Úupdater
   r'   r   ÚatomicÚsave)r   r€   ÚobjsrD   r�   Zpksr_   r   )r   r   ÚaddL  s&    

z>create_reverse_many_to_one_manager.<locals>.RelatedManager.addc                s8   | j || jj< tj| j| j d�}tˆ |  |¡ƒjf |ŽS )N)r   )	r   r
   r'   r   rE   r   r\   r   Úcreate)r   r^   rD   )ÚRelatedManagerrF   r   r   r‹   l  s    zAcreate_reverse_many_to_one_manager.<locals>.RelatedManager.createc                s8   | j || jj< tj| j| j d�}tˆ |  |¡ƒjf |ŽS )N)r   )	r   r
   r'   r   rE   r   r\   r   Úget_or_create)r   r^   rD   )rŒ   rF   r   r   r�   r  s    zHcreate_reverse_many_to_one_manager.<locals>.RelatedManager.get_or_createc                s8   | j || jj< tj| j| j d�}tˆ |  |¡ƒjf |ŽS )N)r   )	r   r
   r'   r   rE   r   r\   r   Úupdate_or_create)r   r^   rD   )rŒ   rF   r   r   rŽ   x  s    zKcreate_reverse_many_to_one_manager.<locals>.RelatedManager.update_or_createc            W   sz   |sd S | j  | j¡}tƒ }xD|D ]<}| j  |¡|krD| |j¡ q"| j jj 	d|| jf ¡‚q"W |  
| j|d�|¡ d S )Nz%r is not related to %r.)r‚   )r
   r#   r   Úsetr$   rŠ   ra   r   r   r   Ú_clearr)   )r   r€   r‰   r<   Úold_idsr_   r   r   r   Úremove€  s    

zAcreate_reverse_many_to_one_manager.<locals>.RelatedManager.removec            S   s   |   | |¡ d S )N)r�   )r   r€   r   r   r   Úclear�  s    z@create_reverse_many_to_one_manager.<locals>.RelatedManager.clearc          	   S   sŒ   |   ¡  tj| j| jd�}| |¡}|r>|jf | jjd iŽ nJt	j
|dd��6 x.|D ]&}t|| jjd ƒ |j| jjgd� qTW W d Q R X d S )N)r   F)rz   rƒ   )Zupdate_fields)r   r   rE   r   r   rz   r†   r
   r'   r   r‡   rI   rˆ   )r   r.   r€   rD   r_   r   r   r   r�   ”  s    

zAcreate_reverse_many_to_one_manager.<locals>.RelatedManager._clearF)r€   r“   c         	   S   sÒ   t |ƒ}| jjr¾tj| j| jd�}tj|dd��† |rP|  	¡  | j
|d|iŽ nbt|  |¡ ¡ ƒ}g }x*|D ]"}||kr„| |¡ ql| |¡ qlW | j|d|iŽ | j
|d|iŽ W d Q R X n| j
|d|iŽ d S )N)r   F)rz   rƒ   r€   )rh   r
   r9   r   rE   r   r   r   r‡   r“   rŠ   r�   rz   r   r’   r…   )r   r‰   r€   r“   rD   Zold_objsÚnew_objsr_   r   r   r   r�   ¢  s     
z>create_reverse_many_to_one_manager.<locals>.RelatedManager.set)N)r:   rQ   rR   r   rv   Údo_not_call_in_templatesr|   r   r   r3   rŠ   Úalters_datar‹   r�   rŽ   r
   r9   r’   r“   r�   r�   rc   r   )rŒ   rm   )rF   r   rŒ   	  s0   	rŒ   r   )Ú
superclassrm   r   )rŒ   rm   r   rn     s     4rn   c                   sB   e Zd ZdZd‡ fdd„	Zedd„ ƒZedd„ ƒZd	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 )N)r\   r   Úreverse)r   rm   r™   )rF   r   r   r   Ð  s    zManyToManyDescriptor.__init__c             C   s   | j jS )N)rm   Úthrough)r   r   r   r   rš   Õ  s    zManyToManyDescriptor.throughc             C   s,   | j r| jjn| jj}t|jj| j| j d�S )N)r™   )r™   rm   re   r   Ú#create_forward_many_to_many_managerro   rF   )r   re   r   r   r   rp   Ü  s
    z(ManyToManyDescriptor.related_manager_clsc             C   s*   d| j rdnd | j r | j ¡ n| jjfS )Nz%s side of a many-to-many setr™   Zforward)r™   rm   rf   r
   r'   )r   r   r   r   rq   æ  s    z4ManyToManyDescriptor._get_set_deprecation_msg_params)F)r:   rQ   rR   rS   r   Úpropertyrš   r   rp   rq   rc   r   r   )rF   r   r˜   ¿  s
   
r˜   c                s   G ‡ ‡‡fdd„d| ƒ‰ ˆ 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 Zd Zd%‡ ‡‡fdd„	Z‡‡fdd„ZdZdd„ Zd	d
„ Zdd„ Z‡ fdd„Z	d&‡ fdd„	Z
‡fdd„Zde_‡fdd„Zde_‡ fdd„Zde_ddœ‡fdd„
Zde_‡‡ fdd„Zde_‡‡ fdd„Zde_‡‡ fdd „Zde_d!d"„ Z‡ fd#d$„Z‡  ZS )'z?create_forward_many_to_many_manager.<locals>.ManyRelatedManagerNc                sf  t ƒ  ¡  || _ˆsTˆj| _ˆj ¡ | _ˆjj| _ˆj 	¡ | _
ˆj ¡ | _ˆj| _n<ˆj| _ˆjj| _ˆj ¡ | _ˆj ¡ | _
ˆj 	¡ | _d| _ˆj| _ˆ| _| jj | j
¡| _| jj | j¡| _i | _i | _xD| jjD ]8\}}d| j|jf }t||jƒ| j|< |j| j|j< qØW | j |¡| _d | jk�rFtd|| j| j
 f ƒ‚|jd k�rbt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
   r(   Úquery_field_namer'   Úprefetch_cache_nameZm2m_field_nameÚsource_field_nameZm2m_reverse_field_nameÚtarget_field_nameÚsymmetricalre   rš   r™   r@   Ú	get_fieldZsource_fieldÚtarget_fieldrs   Zpk_field_namesrH   rK   rJ   r#   Úrelated_valrA   ra   rF   r:   )r   r   rN   rO   Zcore_filter_key)rF   rm   r™   r   r   r   ö  s@    



zHcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.__init__c               s&   t | j|ƒ}t|jˆ ˆƒ}|| jd�S )N)r   )rK   r   r›   rF   r   )r   rt   ru   )rm   r™   r   r   rv   $  s    zHcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.__call__Tc             S   s‚   t f | j| jiŽ}t|tƒ p$| ¡ }|rB|t f d| j |iŽM }| jr~t f | j| jiŽ}|rv|t f d| j |iŽM }||O }|S )Nz%s__in)r   rŸ   r¤   r?   r   Ú_has_filtersr    r¡   )r   Zremoved_valsÚfiltersZremoved_vals_filtersZsymmetrical_filtersr   r   r   Ú_build_remove_filters*  s    zUcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager._build_remove_filtersc             S   s2   |j | jd� | jr | | j¡}| ¡ jf | jŽS )zX
            Filter the queryset for the instance this manager is bound to.
            )r   )r"   r   rx   rz   Ú_next_is_stickyr)   rs   )r   r.   r   r   r   r|   :  s    zRcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager._apply_rel_filtersc          	   S   s2   y| j j | j¡ W n ttfk
r,   Y nX d S )N)r   r}   r~   rž   r   r8   )r   r   r   r   r   C  s    zZcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager._remove_prefetched_objectsc          	      s<   y| j j| j S  ttfk
r6   tƒ  ¡ }|  |¡S X d S )N)r   r}   rž   r   r8   r\   r   r|   )r   r.   )rF   r   r   r   I  s
    
zLcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.get_querysetc                sÀ   |d krt ƒ  ¡ }|j|d d� | |jp0| j¡}d| j |i}| ¡ jf |Ž}| jj	 
| j¡‰ˆjj	j‰t|j ‰ ˆ jj‰|j‡‡fdd„ˆjD ƒd�}|‡fdd„‡ ‡fd	d„d
| jd
fS )Nr   )r   z%s__inc                s*   i | ]"}d ˆˆ ƒˆ|j ƒf d|j “qS )z%s.%sz_prefetch_related_val_%s)ÚcolumnrJ   )r   Úf)Ú
join_tableÚqnr   r   r   e  s   zicreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.get_prefetch_queryset.<locals>.<dictcomp>)Úselectc                s   t ‡ fdd„ˆjD ƒƒS )Nc             3   s   | ]}t ˆ d |j ƒV  qdS )z_prefetch_related_val_%sN)rK   rJ   )r   rª   )Úresultr   r   rW   k  s   zzcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.get_prefetch_queryset.<locals>.<lambda>.<locals>.<genexpr>)rh   rj   )r®   )Úfk)r®   r   Ú<lambda>j  s   
zgcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.get_prefetch_queryset.<locals>.<lambda>c                s   t ‡‡ fdd„ˆjD ƒƒS )Nc             3   s"   | ]}|  tˆ|jƒˆ ¡V  qd S )N)Zget_db_prep_valuerK   rJ   )r   rª   )Ú
connectionr   r   r   rW   o  s   zzcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.get_prefetch_queryset.<locals>.<lambda>.<locals>.<genexpr>)rh   r%   )r   )r±   r¯   )r   r   r°   n  s   F)r\   r   r"   rz   rx   r�   r¨   r)   rš   r@   r¢   rŸ   r   Zdb_tabler   rD   ÚopsZ
quote_nameÚextrarj   rž   )r   r-   r.   r1   )rF   )r±   r¯   r«   r¬   r   r3   P  s$    
	


zUcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.get_prefetch_querysetc          	      s�   ˆ j jjs&| j j}td|j|jf ƒ‚|  ¡  tj| j | j	d�}t
j|dd��8 | j| j| jf|žŽ  | jr‚| j| j| jf|žŽ  W d Q R X d S )NziCannot use add() on a ManyToManyField which specifies an intermediary model. Use %s.%s's Manager instead.)r   F)rz   rƒ   )rš   r@   Úauto_createdr   Ú	app_labelrB   r   r   rE   r   r   r‡   Ú
_add_itemsrŸ   r    r¡   )r   r‰   rb   rD   )rm   r   r   rŠ   w  s    
zCcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.addc                sH   ˆ j jjs&| j j}td|j|jf ƒ‚|  ¡  | j| j| j	f|žŽ  d S )NzlCannot use remove() on a ManyToManyField which specifies an intermediary model. Use %s.%s's Manager instead.)
rš   r@   r´   r   rµ   rB   r   Ú_remove_itemsrŸ   r    )r   r‰   rb   )rm   r   r   r’   ‰  s    
zFcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.removec          
      sª   t j| j| jd�}tj|dd��€ tjj| jd| j| j	| j
d |d� |  ¡  |  tƒ  ¡  |¡¡}| jj |¡ |¡ ¡  tjj| jd| j| j	| j
d |d� W d Q R X d S )N)r   F)rz   rƒ   Z	pre_clear)ÚsenderÚactionr   r™   r   Úpk_setrz   Z
post_clear)r   rE   rš   r   r   r‡   r   Úm2m_changedÚsendr™   r   r   r§   r\   r   rz   ro   r)   Údelete)r   rD   r¦   )rF   r   r   r“   •  s    zEcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.clearF)r“   c      	   	      sø   ˆ j jjs&| j j}td|j|jf ƒ‚t|ƒ}tj| j | j	d�}t
j|dd��  |rh|  ¡  | j|Ž  n‚t|  |¡j| jjjdd�ƒ}g }xJ|D ]B}t|| jƒr°| j |¡d n|}||krÈ| |¡ q�| |¡ q�W | j|Ž  | j|Ž  W d Q R X d S )NzjCannot set values on a ManyToManyField which specifies an intermediary model. Use %s.%s's Manager instead.)r   F)rz   rƒ   T)Úflatr   )rš   r@   r´   r   rµ   rB   rh   r   rE   r   r   r‡   r“   rŠ   r�   rz   Úvalues_listr£   rJ   r?   r   r#   r’   r…   )	r   r‰   r“   rb   rD   r‘   r”   r_   Úfk_val)rm   r   r   r�   ¨  s(    

 
zCcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.setc                s`   | j jjs&| j j}td|j|jf ƒ‚tj| jj	| jd�}t
ˆ |  |¡ƒjf |Ž}|  |¡ |S )NzlCannot use create() on a ManyToManyField which specifies an intermediary model. Use %s.%s's Manager instead.)r   )rš   r@   r´   r   rµ   rB   r   rE   r   rF   r\   r   r‹   rŠ   )r   r^   rb   rD   Znew_obj)ÚManyRelatedManagerrF   r   r   r‹   Ì  s    

zFcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.createc                sF   t j| jj| jd�}tˆ |  |¡ƒjf |Ž\}}|r>|  |¡ ||fS )N)r   )r   rE   r   rF   r\   r   r�   rŠ   )r   r^   rD   r_   Úcreated)rÁ   rF   r   r   r�   Ü  s
    
zMcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.get_or_createc                sF   t j| jj| jd�}tˆ |  |¡ƒjf |Ž\}}|r>|  |¡ ||fS )N)r   )r   rE   r   rF   r\   r   rŽ   rŠ   )r   r^   rD   r_   rÂ   )rÁ   rF   r   r   rŽ   æ  s
    
zPcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager.update_or_createc       
   
      sÂ  ddl m} |�r¾tƒ }xª|D ]¢}t|ˆ jƒr”t |ˆ j¡sXtd|ˆ jj	j
|j	j
f ƒ‚ˆ jj ˆ¡ |¡d }|d krˆtd|ˆf ƒ‚| |¡ qt||ƒr¶tdˆ jjj|f ƒ‚q| |¡ qW tjˆ jˆ jd�}ˆ jj |¡jˆdd�jf ˆˆ jd d	ˆ |iŽ}	| |	¡ tj|d
d��˜ ˆ j�s6ˆˆ jk�rXtjjˆ jdˆ jˆ jˆ j||d� ˆ jj |¡  ‡ ‡‡fdd„|D ƒ¡ ˆ j�s’ˆˆ jk�r´tjjˆ jdˆ jˆ jˆ j||d� W d Q R X d S )Nr   )ÚModelzHCannot add "%r": instance is on database "%s", value is on database "%s"z1Cannot add "%r": the value for field "%s" is Nonez'%s' instance expected, got %r)r   T)r¾   z%s__inF)rz   rƒ   Zpre_add)r¸   r¹   r   r™   r   rº   rz   c                s.   g | ]&}ˆ j f d ˆ ˆ jd d ˆ |iŽ‘qS )z%s_idr   )rš   r¤   )r   Zobj_id)r   rŸ   r    r   r   rU   %  s   z^create_forward_many_to_many_manager.<locals>.ManyRelatedManager._add_items.<locals>.<listcomp>Zpost_add)!Údjango.db.modelsrÃ   r�   r?   r   r   rG   r   rA   rC   rD   rš   r@   r¢   r#   rŠ   rr   rB   rE   ro   rz   r¿   r)   r¤   Údifference_updater   r‡   r™   rŸ   r   r»   r¼   Zbulk_create)
r   rŸ   r    r‰   rÃ   Znew_idsr_   rÀ   rD   Úvalsr   )r   rŸ   r    r   r¶   ð  sR    




zJcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager._add_itemsc          
      s  |sd S t ƒ }x>|D ]6}t|| jƒr@| j |¡d }| |¡ q| |¡ qW tj| j| j	d�}t
j|dd��¤ tjj| jd| j	| j| j||d� tƒ  ¡ }| ¡ rÆ| |¡jf d| jjj |iŽ}	n|}	|  |	¡}
| jj |¡ |
¡ ¡  tjj| jd| j	| j| j||d� W d Q R X d S )	Nr   )r   F)rz   rƒ   Z
pre_remove)r¸   r¹   r   r™   r   rº   rz   z%s__inZpost_remove)r�   r?   r   r£   r#   rŠ   r   rE   rš   r   r   r‡   r   r»   r¼   r™   r\   r   r¥   rz   r)   rJ   r§   ro   r½   )r   rŸ   r    r‰   r‘   r_   rÀ   rD   Ztarget_model_qsZold_valsr¦   )rF   r   r   r·   5  s2    


zMcreate_forward_many_to_many_manager.<locals>.ManyRelatedManager._remove_items)N)N)r:   rQ   rR   r   rv   r•   r§   r|   r   r   r3   rŠ   r–   r’   r“   r�   r‹   r�   rŽ   r¶   r·   rc   r   )rÁ   rm   r™   )rF   r   rÁ   õ  s0   .	'
"ErÁ   r   )r—   rm   r™   r   )rÁ   rm   r™   r   r›   í  s      ir›   N)rS   Z	django.dbr   r   r   rÄ   r   r   Zdjango.db.models.queryr   Zdjango.utils.functionalr   r	   rT   rd   rl   rn   r˜   r›   r   r   r   r   Ú<module>@   s    10 !9 ?.