U
    Ha[                     @   s  d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	 ddl
mZmZmZmZmZ ddlmZmZmZ d	d
 ZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd  d eZG d!d" d"eZ G d#d$ d$eZ!G d%d& d&e!Z"G d'd( d(e!Z#G d)d* d*e!Z$G d+d, d,e!Z%d-S ).    )models)	Operation)
ModelState)normalize_together)cached_property   )AddField
AlterFieldFieldOperationRemoveFieldRenameField)field_referencesget_referencesresolve_relationc                 C   s6   t  }|D ]&}||kr&td|| f || q
d S )Nz4Found duplicate value %s in CreateModel %s argument.)set
ValueErroradd)Zarg_nameobjsZ	used_valsval r   J/tmp/pip-unpacked-wheel-3jxiddxt/django/db/migrations/operations/models.py_check_for_duplicates   s    
r   c                       s8   e Zd Zdd Zedd Zdd Z fddZ  ZS )	ModelOperationc                 C   s
   || _ d S Nname)selfr   r   r   r   __init__   s    zModelOperation.__init__c                 C   s
   | j  S r   )r   lowerr   r   r   r   
name_lower   s    zModelOperation.name_lowerc                 C   s   |  | jkS r   r   r    r   r   	app_labelr   r   r   references_model   s    zModelOperation.references_modelc                    s   t  ||p|| j| S r   )superreducer$   r   r   	operationr#   	__class__r   r   r&   "   s    zModelOperation.reduce)	__name__
__module____qualname__r   r   r    r$   r&   __classcell__r   r   r)   r   r      s
   
r   c                       st   e Zd ZdZdddgZd fdd	Zdd	 Zd
d Zdd Zdd Z	dd Z
edd Zdd Z fddZ  ZS )CreateModelzCreate a model's table.fieldsoptionsmanagersNc                    sz   || _ |pi | _|ptjf| _|p$g | _t | tddd | j D  tddd | jD  tddd | jD  d S )Nr0   c                 s   s   | ]\}}|V  qd S r   r   .0r   _r   r   r   	<genexpr>6   s     z'CreateModel.__init__.<locals>.<genexpr>basesc                 s   s6   | ].}t |d r|jjnt|tr*| n|V  qdS )_metaN)hasattrr8   Zlabel_lower
isinstancestrr   )r4   baser   r   r   r6   7   s   r2   c                 s   s   | ]\}}|V  qd S r   r   r3   r   r   r   r6   <   s     )	r0   r1   r   Modelr7   r2   r%   r   r   )r   r   r0   r1   r7   r2   r)   r   r   r   .   s    


zCreateModel.__init__c                 C   sn   | j | jd}| jr| j|d< | jr<| jtjfkr<| j|d< | jr`| jdt fgkr`| j|d< | jj	g |fS )N)r   r0   r1   r7   Zobjectsr2   )
r   r0   r1   r7   r   r=   r2   Managerr*   r-   r   kwargsr   r   r   deconstruct>   s    


zCreateModel.deconstructc              
   C   s6   | t|| jt| jt| jt| jt| j	 d S r   )
Z	add_modelr   r   listr0   dictr1   tupler7   r2   r   r#   stater   r   r   state_forwardsO   s    zCreateModel.state_forwardsc                 C   s.   |j || j}| |jj|r*|| d S r   apps	get_modelr   allow_migrate_model
connectionaliasZcreate_modelr   r#   schema_editor
from_stateto_statemodelr   r   r   database_forwardsY   s    zCreateModel.database_forwardsc                 C   s.   |j || j}| |jj|r*|| d S r   rI   rJ   r   rK   rL   rM   Zdelete_modelrN   r   r   r   database_backwards^   s    zCreateModel.database_backwardsc                 C   s    d| j ddrdnd| jf S )NzCreate %smodel %sproxyFzproxy  )r1   getr   r   r   r   r   describec   s    zCreateModel.describec                 C   s   | j S r   r    r   r   r   r   migration_name_fragmentf   s    z#CreateModel.migration_name_fragmentc                 C   s   |  }|| jkrdS ||f}| jD ]4}|tjk	r$t|tjjtfr$t	|||kr$ dS q$| j
D ] \}}t|| jf||r` dS q`dS )NTF)r   r    r7   r   r=   r:   r<   Z	ModelBaser;   r   r0   r   )r   r   r#   r    Zreference_model_tupler<   _namefieldr   r   r   r$   j   s    

zCreateModel.references_modelc                    s  t  tr(| j jkr(| jdds(g S t  tr\| j jkr\t j| j	| j| j
| jdgS t  tr| j jkr| j j} jD ]}| jkr||d  qt| j| j	|| j
| jdgS t  tr | j jkr t| j| j	| j j ji| j
| jdgS t  trB| j jkrBt| j| j	| jd ji| j
| jdgS t  tr| j jkrt  trt| j| j	 j jfg | j| j
| jdgS t  trt| j fdd| j	D | j| j
| jdgS t  trp| j }dD ]B}||d }|rttt fdd	|D }|r|||< q|d}| jkrF|d= t| j fd
d| j	D || j
| jdgS t  tr| j }dD ],}||}|r fdd|D ||< q|d}| j kr؈ j|d< t| j fdd| j	D || j
| jdgS t! " |S )NrV   F)r0   r1   r7   r2   order_with_respect_toc                    s(   g | ] \}}|| j kr jn|fqS r   )r   r]   r4   nvr(   r   r   
<listcomp>   s   z&CreateModel.reduce.<locals>.<listcomp>)unique_togetherindex_togetherc                 3   s$   | ]}t  fd d|D V  qdS )c                 3   s   | ]}| j kr|V  qd S r   rZ   r4   frb   r   r   r6      s     
 z/CreateModel.reduce.<locals>.<genexpr>.<genexpr>NrD   r4   r0   rb   r   r   r6      s    z%CreateModel.reduce.<locals>.<genexpr>c                    s&   g | ]\}}|   jkr||fqS r   r!   r_   rb   r   r   rc      s   c                    s"   h | ]}t  fd d|D qS )c                 3   s"   | ]}| j kr jn|V  qd S r   old_namenew_namerf   rb   r   r   r6      s     z/CreateModel.reduce.<locals>.<setcomp>.<genexpr>rh   ri   rb   r   r   	<setcomp>   s   z%CreateModel.reduce.<locals>.<setcomp>c                    s(   g | ] \}}| j kr jn||fqS r   rj   r_   rb   r   r   rc      s   )#r:   DeleteModelr    r1   rX   RenameModelold_name_lowerr/   rl   r0   r7   r2   AlterModelOptionsALTER_OPTION_KEYSpopr   AlterTogetherOptionOperationoption_nameoption_valueAlterOrderWithRespectTor^   r
   model_name_lowerr   r]   r	   r   copyr   filterboolr   rk   r%   r&   )r   r(   r#   r1   keyru   optionr^   r)   rb   r   r&   |   s    

	

				










zCreateModel.reduce)NNN)r+   r,   r-   __doc__serialization_expand_argsr   rA   rG   rS   rU   rY   propertyr[   r$   r&   r.   r   r   r)   r   r/   )   s   


r/   c                   @   sL   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	e
dd ZdS )rn   zDrop a model's table.c                 C   s   d| j i}| jjg |fS Nr   )r   r*   r-   r?   r   r   r   rA      s     zDeleteModel.deconstructc                 C   s   | || j d S r   )remove_modelr    rE   r   r   r   rG     s    zDeleteModel.state_forwardsc                 C   s.   |j || j}| |jj|r*|| d S r   rT   rN   r   r   r   rS   
  s    zDeleteModel.database_forwardsc                 C   s.   |j || j}| |jj|r*|| d S r   rH   rN   r   r   r   rU     s    zDeleteModel.database_backwardsc                 C   s   dS )NTr   r"   r   r   r   r$     s    zDeleteModel.references_modelc                 C   s
   d| j  S )NzDelete model %sr   r   r   r   r   rY     s    zDeleteModel.describec                 C   s
   d| j  S )Nz	delete_%srZ   r   r   r   r   r[     s    z#DeleteModel.migration_name_fragmentN)r+   r,   r-   r~   rA   rG   rS   rU   r$   rY   r   r[   r   r   r   r   rn      s   
rn   c                       s   e Zd ZdZ fddZedd Zedd Zdd	 Zd
d Z	dd Z
dd Zdd Zdd Zedd Z fddZ  ZS )ro   zRename a model.c                    s   || _ || _t | d S r   )rk   rl   r%   r   )r   rk   rl   r)   r   r   r   $  s    zRenameModel.__init__c                 C   s
   | j  S r   )rk   r   r   r   r   r   rp   )  s    zRenameModel.old_name_lowerc                 C   s
   | j  S r   )rl   r   r   r   r   r   new_name_lower-  s    zRenameModel.new_name_lowerc                 C   s   | j | jd}| jjg |fS )Nrj   )rk   rl   r*   r-   r?   r   r   r   rA   1  s    zRenameModel.deconstructc                 C   s   |j || jf  }| j|_||j || jf< || jf}d|| jf }t }t||D ]d\}}}	}
d }|
jrz|	 }||j	_
|
jr|d kr|	 }||j	_|rT||j|< ||j|jf qT|j|dd ||| j |j|| jdd d S )Nz%s.%sTdelay)r   rp   clonerl   r   r   r   r   toremote_fieldrR   throughr0   r   r#   r    Zreload_modelsr   reload_model)r   r#   rF   Zrenamed_modelZold_model_tupleZnew_remote_modelZ	to_reloadmodel_stater   r]   	referenceZchanged_fieldr   r   r   rG   <  s*    

zRenameModel.state_forwardsc                 C   s<  |j || j}| |jj|r8|j || j}|||jj	|jj	 |jj
D ]`}|j|krn|}|| jf}	n|j}|jjj|jjjf}	|j j|	 j|jj}
|||j|
 qPt|jj|jj}|D ]n\}}|j|jks|jjjjsq|jj}|jj}|||jj	|jj	 |||j|jj|j|jj qd S r   )rI   rJ   rl   rK   rL   rM   rk   alter_db_tabler8   db_tableZrelated_objectsZrelated_modelr   r#   
model_name	get_fieldr]   r   Zalter_fieldziplocal_many_to_manyrR   r   r   auto_created)r   r#   rO   rP   rQ   	new_model	old_modelZrelated_objectrR   Zrelated_keyZto_fieldr0   	old_field	new_fieldZold_m2m_modelZnew_m2m_modelr   r   r   rS   W  sT    
zRenameModel.database_forwardsc                 C   s\   | j | j | _| _ | j| j | _| _| |||| | j | j | _| _ | j| j | _| _d S r   )rp   r   rk   rl   rS   r   r#   rO   rP   rQ   r   r   r   rU     s
    zRenameModel.database_backwardsc                 C   s   |  | jkp|  | jkS r   )r   rp   r   r"   r   r   r   r$     s    zRenameModel.references_modelc                 C   s   d| j | jf S )NzRename model %s to %srj   r   r   r   r   rY     s    zRenameModel.describec                 C   s   d| j | jf S )Nzrename_%s_%s)rp   r   r   r   r   r   r[     s    z#RenameModel.migration_name_fragmentc                    sH   t |tr&| j|jkr&t| j|jgS tt| ||pF|	| j| S r   )
r:   ro   r   rp   rk   rl   r%   r   r&   r$   r'   r)   r   r   r&     s    

	zRenameModel.reduce)r+   r,   r-   r~   r   r   rp   r   rA   rG   rS   rU   r$   rY   r   r[   r&   r.   r   r   r)   r   ro   !  s   

3	
ro   c                       s   e Zd Z fddZ  ZS )ModelOptionOperationc                    s0   t || jtfr"| j|jkr"|gS t ||S r   )r:   r*   rn   r    r%   r&   r'   r)   r   r   r&     s    zModelOptionOperation.reduce)r+   r,   r-   r&   r.   r   r   r)   r   r     s   r   c                       sT   e Zd ZdZ fddZdd Zdd Zdd	 Zd
d Zdd Z	e
dd Z  ZS )AlterModelTablezRename a model's table.c                    s   || _ t | d S r   )tabler%   r   )r   r   r   r)   r   r   r     s    zAlterModelTable.__init__c                 C   s   | j | jd}| jjg |fS )Nr   r   )r   r   r*   r-   r?   r   r   r   rA     s    zAlterModelTable.deconstructc                 C   s.   | j |j|| jf jd< |j|| jdd d S )Nr   Tr   )r   r   r    r1   r   rE   r   r   r   rG     s    zAlterModelTable.state_forwardsc           	      C   s   |j || j}| |jj|r|j || j}|||jj|jj t	|jj
|jj
D ]6\}}|jjjjrX||jj|jjjj|jjjj qXd S r   )rI   rJ   r   rK   rL   rM   r   r8   r   r   r   r   r   r   )	r   r#   rO   rP   rQ   r   r   r   r   r   r   r   rS     s    

z!AlterModelTable.database_forwardsc                 C   s   |  ||||S r   rS   r   r   r   r   rU     s    z"AlterModelTable.database_backwardsc                 C   s   d| j | jd k	r| jndf S )NzRename table for %s to %sz	(default)r   r   r   r   r   rY     s    zAlterModelTable.describec                 C   s
   d| j  S )Nzalter_%s_tablerZ   r   r   r   r   r[     s    z'AlterModelTable.migration_name_fragment)r+   r,   r-   r~   r   rA   rG   rS   rU   rY   r   r[   r.   r   r   r)   r   r     s   r   c                       sh   e Zd ZdZ fddZedd Zdd Zdd	 Zd
d Z	dd Z
dd Zdd Zedd Z  ZS )rt   Nc                    s.   |rt t|}t| | j| t | d S r   )r   r   setattrru   r%   r   )r   r   rv   r)   r   r   r     s    z%AlterTogetherOptionOperation.__init__c                 C   s   t | | jS r   )getattrru   r   r   r   r   rv     s    z)AlterTogetherOptionOperation.option_valuec                 C   s    d| j | j| ji}| jjg |fS r   )r   ru   rv   r*   r-   r?   r   r   r   rA     s      z(AlterTogetherOptionOperation.deconstructc                 C   s4   |j || jf }| j|j| j< |j|| jdd d S NTr   )r   r    rv   r1   ru   r   r   r#   rF   r   r   r   r   rG     s    z+AlterTogetherOptionOperation.state_forwardsc                 C   sl   |j || j}| |jj|rh|j || j}t|d| j }||t|j| jt	 t|j| jt	  d S )Nzalter_%s)
rI   rJ   r   rK   rL   rM   r   ru   r8   r   )r   r#   rO   rP   rQ   r   r   Zalter_togetherr   r   r   rS     s    z.AlterTogetherOptionOperation.database_forwardsc                 C   s   |  ||||S r   r   r   r   r   r   rU     s    z/AlterTogetherOptionOperation.database_backwardsc                    s,   |  ||o*| j p*t fdd| jD S )Nc                 3   s   | ]} |kV  qd S r   r   ri   r   r   r   r6     s     z@AlterTogetherOptionOperation.references_field.<locals>.<genexpr>)r$   rv   anyr   r   r   r#   r   r   r   references_field  s    z-AlterTogetherOptionOperation.references_fieldc                 C   s   d| j | jt| jpdf S )Nz"Alter %s for %s (%s constraint(s))rW   )ru   r   lenrv   r   r   r   r   rY   "  s    z%AlterTogetherOptionOperation.describec                 C   s   d| j | jf S )Nzalter_%s_%s)r    ru   r   r   r   r   r[   %  s    z4AlterTogetherOptionOperation.migration_name_fragment)r+   r,   r-   ru   r   r   rv   rA   rG   rS   rU   r   rY   r   r[   r.   r   r   r)   r   rt     s   
	rt   c                       s$   e Zd ZdZdZ fddZ  ZS )AlterUniqueTogetherz|
    Change the value of unique_together to the target one.
    Input value of unique_together must be a set of tuples.
    rd   c                    s   t  || d S r   r%   r   )r   r   rd   r)   r   r   r   1  s    zAlterUniqueTogether.__init__r+   r,   r-   r~   ru   r   r.   r   r   r)   r   r   *  s   r   c                       s$   e Zd ZdZdZ fddZ  ZS )AlterIndexTogetherzz
    Change the value of index_together to the target one.
    Input value of index_together must be a set of tuples.
    re   c                    s   t  || d S r   r   )r   r   re   r)   r   r   r   <  s    zAlterIndexTogether.__init__r   r   r   r)   r   r   5  s   r   c                       s`   e Zd ZdZdZ fddZdd Zdd Zd	d
 Zdd Z	dd Z
dd Zedd Z  ZS )rw   z9Represent a change with the order_with_respect_to option.r^   c                    s   || _ t | d S r   )r^   r%   r   )r   r   r^   r)   r   r   r   E  s    z AlterOrderWithRespectTo.__init__c                 C   s   | j | jd}| jjg |fS )Nr   r^   )r   r^   r*   r-   r?   r   r   r   rA   I  s    z#AlterOrderWithRespectTo.deconstructc                 C   s2   |j || jf }| j|jd< |j|| jdd d S )Nr^   Tr   )r   r    r^   r1   r   r   r   r   r   rG   T  s    z&AlterOrderWithRespectTo.state_forwardsc                 C   s   |j || j}| |jj|r|j || j}|jjrV|jjsV|||j	d n6|jjr|jjs|j	d}|
 sd|_||| d S )N_orderr   )rI   rJ   r   rK   rL   rM   r8   r^   Zremove_fieldr   Zhas_defaultdefaultZ	add_field)r   r#   rO   rP   rQ   Zto_modelZ
from_modelr]   r   r   r   rS   Y  s    z)AlterOrderWithRespectTo.database_forwardsc                 C   s   |  |||| d S r   r   r   r   r   r   rU   k  s    z*AlterOrderWithRespectTo.database_backwardsc                 C   s    |  ||o| jd kp|| jkS r   )r$   r^   r   r   r   r   r   n  s    
z(AlterOrderWithRespectTo.references_fieldc                 C   s   d| j | jf S )Nz%Set order_with_respect_to on %s to %sr   r   r   r   r   rY   w  s    z AlterOrderWithRespectTo.describec                 C   s
   d| j  S )Nzalter_%s_order_with_respect_torZ   r   r   r   r   r[   z  s    z/AlterOrderWithRespectTo.migration_name_fragment)r+   r,   r-   r~   ru   r   rA   rG   rS   rU   r   rY   r   r[   r.   r   r   r)   r   rw   @  s   	rw   c                       sn   e Zd ZdZdddddddd	d
ddgZ fddZdd Zdd Zdd Zdd Z	dd Z
edd Z  ZS )rq   z
    Set new model options that don't directly affect the database schema
    (like verbose_name, permissions, ordering). Python code in migrations
    may still need them.
    Zbase_manager_nameZdefault_manager_nameZdefault_related_nameZget_latest_byZmanagedZorderingpermissionsZdefault_permissionsZselect_on_saveZverbose_nameZverbose_name_pluralc                    s   || _ t | d S r   )r1   r%   r   )r   r   r1   r)   r   r   r     s    zAlterModelOptions.__init__c                 C   s   | j | jd}| jjg |fS )N)r   r1   )r   r1   r*   r-   r?   r   r   r   rA     s    zAlterModelOptions.deconstructc                 C   sX   |j || jf }|j| j|_| jD ]}|| jkr$|j|d q$|j|| jdd d S )NFTr   )r   r    r1   rr   rs   r   )r   r#   rF   r   r|   r   r   r   rG     s    

z AlterModelOptions.state_forwardsc                 C   s   d S r   r   r   r   r   r   rS     s    z#AlterModelOptions.database_forwardsc                 C   s   d S r   r   r   r   r   r   rU     s    z$AlterModelOptions.database_backwardsc                 C   s
   d| j  S )NzChange Meta options on %sr   r   r   r   r   rY     s    zAlterModelOptions.describec                 C   s
   d| j  S )Nzalter_%s_optionsrZ   r   r   r   r   r[     s    z)AlterModelOptions.migration_name_fragment)r+   r,   r-   r~   rr   r   rA   rG   rS   rU   rY   r   r[   r.   r   r   r)   r   rq     s*   rq   c                       sZ   e Zd ZdZdgZ fddZdd Zdd Zd	d
 Zdd Z	dd Z
edd Z  ZS )AlterModelManagerszAlter the model's managers.r2   c                    s   || _ t | d S r   )r2   r%   r   )r   r   r2   r)   r   r   r     s    zAlterModelManagers.__init__c                 C   s   | j j| j| jgi fS r   )r*   r-   r   r2   r   r   r   r   rA     s    
zAlterModelManagers.deconstructc                 C   s2   |j || jf }t| j|_|j|| jdd d S r   )r   r    rB   r2   r   r   r   r   r   rG     s    z!AlterModelManagers.state_forwardsc                 C   s   d S r   r   r   r   r   r   rS     s    z$AlterModelManagers.database_forwardsc                 C   s   d S r   r   r   r   r   r   rU     s    z%AlterModelManagers.database_backwardsc                 C   s
   d| j  S )NzChange managers on %sr   r   r   r   r   rY     s    zAlterModelManagers.describec                 C   s
   d| j  S )Nzalter_%s_managersrZ   r   r   r   r   r[     s    z*AlterModelManagers.migration_name_fragment)r+   r,   r-   r~   r   r   rA   rG   rS   rU   rY   r   r[   r.   r   r   r)   r   r     s   r   c                   @   s   e Zd ZdZedd ZdS )IndexOperationindexesc                 C   s
   | j  S r   )r   r   r   r   r   r   rx     s    zIndexOperation.model_name_lowerN)r+   r,   r-   ru   r   rx   r   r   r   r   r     s   r   c                   @   sL   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	e
dd ZdS )AddIndexzAdd an index on a model.c                 C   s"   || _ |jstd| || _d S )NzSIndexes passed to AddIndex operations require a name argument. %r doesn't have one.)r   r   r   index)r   r   r   r   r   r   r     s    zAddIndex.__init__c                 C   sF   |j || jf }|j| j | j f|j| j< |j|| jdd d S r   )r   rx   r1   ru   r   r   r   r   r   r   r   rG     s     zAddIndex.state_forwardsc                 C   s2   |j || j}| |jj|r.||| j d S r   )rI   rJ   r   rK   rL   rM   	add_indexr   rN   r   r   r   rS     s    zAddIndex.database_forwardsc                 C   s2   |j || j}| |jj|r.||| j d S r   )rI   rJ   r   rK   rL   rM   remove_indexr   rN   r   r   r   rU     s    zAddIndex.database_backwardsc                 C   s   | j | jd}| jjg |fS )N)r   r   )r   r   r*   r-   r?   r   r   r   rA      s    zAddIndex.deconstructc                 C   sN   | j jr0d| j jddd | j jD | jf S d| j jd| j j| jf S )Nz!Create index %s on %s on model %sz, c                 S   s   g | ]}t |qS r   )r;   )r4   Z
expressionr   r   r   rc     s     z%AddIndex.describe.<locals>.<listcomp>z*Create index %s on field(s) %s of model %s)r   Zexpressionsr   joinr   r0   r   r   r   r   rY     s    zAddIndex.describec                 C   s   d| j | jj f S Nz%s_%s)rx   r   r   r   r   r   r   r   r[     s    z AddIndex.migration_name_fragmentNr+   r,   r-   r~   r   rG   rS   rU   rA   rY   r   r[   r   r   r   r   r     s   	r   c                   @   sL   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	e
dd ZdS )RemoveIndexzRemove an index from a model.c                 C   s   || _ || _d S r   r   r   r   r   r   r   r   r   r      s    zRemoveIndex.__init__c                    sL   |j | jf }|j j } fdd|D |j j< |j| jdd d S )Nc                    s   g | ]}|j  j kr|qS r   r   )r4   idxr   r   r   rc   '  s      z.RemoveIndex.state_forwards.<locals>.<listcomp>Tr   r   rx   r1   ru   r   )r   r#   rF   r   r   r   r   r   rG   $  s    zRemoveIndex.state_forwardsc                 C   sL   |j || j}| |jj|rH|j|| jf }|| j	}|
|| d S r   )rI   rJ   r   rK   rL   rM   r   rx   get_index_by_namer   r   )r   r#   rO   rP   rQ   rR   from_model_stater   r   r   r   rS   *  s
    zRemoveIndex.database_forwardsc                 C   sL   |j || j}| |jj|rH|j|| jf }|| j	}|
|| d S r   )rI   rJ   r   rK   rL   rM   r   rx   r   r   r   )r   r#   rO   rP   rQ   rR   to_model_stater   r   r   r   rU   1  s
    zRemoveIndex.database_backwardsc                 C   s   | j | jd}| jjg |fS Nr   )r   r   r*   r-   r?   r   r   r   rA   8  s    zRemoveIndex.deconstructc                 C   s   d| j | jf S )NzRemove index %s from %sr   r   r   r   r   r   rY   C  s    zRemoveIndex.describec                 C   s   d| j | j f S Nzremove_%s_%srx   r   r   r   r   r   r   r[   F  s    z#RemoveIndex.migration_name_fragmentNr   r   r   r   r   r     s   r   c                   @   sL   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	e
dd ZdS )AddConstraintconstraintsc                 C   s   || _ || _d S r   r   
constraint)r   r   r   r   r   r   r   N  s    zAddConstraint.__init__c                 C   sB   |j || jf }|j| j | jf|j| j< |j|| jdd d S r   )r   rx   r1   ru   r   r   r   r   r   r   rG   R  s    zAddConstraint.state_forwardsc                 C   s2   |j || j}| |jj|r.||| j d S r   )rI   rJ   r   rK   rL   rM   add_constraintr   rN   r   r   r   rS   W  s    zAddConstraint.database_forwardsc                 C   s2   |j || j}| |jj|r.||| j d S r   )rI   rJ   r   rK   rL   rM   remove_constraintr   rN   r   r   r   rU   \  s    z AddConstraint.database_backwardsc                 C   s   | j jg | j| jdfS )Nr   )r*   r+   r   r   r   r   r   r   rA   a  s    zAddConstraint.deconstructc                 C   s   d| j j| jf S )Nz Create constraint %s on model %s)r   r   r   r   r   r   r   rY   g  s    zAddConstraint.describec                 C   s   d| j | jj f S r   )rx   r   r   r   r   r   r   r   r[   j  s    z%AddConstraint.migration_name_fragmentNr+   r,   r-   ru   r   rG   rS   rU   rA   rY   r   r[   r   r   r   r   r   K  s   r   c                   @   sL   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	e
dd ZdS )RemoveConstraintr   c                 C   s   || _ || _d S r   r   r   r   r   r   r   r  s    zRemoveConstraint.__init__c                    sL   |j | jf }|j j } fdd|D |j j< |j| jdd d S )Nc                    s   g | ]}|j  j kr|qS r   r   )r4   cr   r   r   rc   y  s      z3RemoveConstraint.state_forwards.<locals>.<listcomp>Tr   r   )r   r#   rF   r   r   r   r   r   rG   v  s    zRemoveConstraint.state_forwardsc                 C   sL   |j || j}| |jj|rH|j|| jf }|| j	}|
|| d S r   )rI   rJ   r   rK   rL   rM   r   rx   get_constraint_by_namer   r   )r   r#   rO   rP   rQ   rR   r   r   r   r   r   rS   |  s
    z"RemoveConstraint.database_forwardsc                 C   sL   |j || j}| |jj|rH|j|| jf }|| j	}|
|| d S r   )rI   rJ   r   rK   rL   rM   r   rx   r   r   r   )r   r#   rO   rP   rQ   rR   r   r   r   r   r   rU     s
    z#RemoveConstraint.database_backwardsc                 C   s   | j jg | j| jdfS r   )r*   r+   r   r   r   r   r   r   rA     s    zRemoveConstraint.deconstructc                 C   s   d| j | jf S )Nz"Remove constraint %s from model %sr   r   r   r   r   rY     s    zRemoveConstraint.describec                 C   s   d| j | j f S r   r   r   r   r   r   r[     s    z(RemoveConstraint.migration_name_fragmentNr   r   r   r   r   r   o  s   r   N)&Z	django.dbr   Z$django.db.migrations.operations.baser   Zdjango.db.migrations.stater   Zdjango.db.models.optionsr   Zdjango.utils.functionalr   r0   r   r	   r
   r   r   utilsr   r   r   r   r   r/   rn   ro   r   r   rt   r   r   rw   rq   r   r   r   r   r   r   r   r   r   r   <module>   s4   
 R' 6<?;#8.$