o
    ˷e                     @   s  d dl mZ d dlmZ d dlmZ d dlmZ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
 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$G d-d. d.e Z%d/S )0    )models)	Operation)
ModelState)field_referencesresolve_relation)normalize_together)cached_property   )AddField
AlterFieldFieldOperationRemoveFieldRenameFieldc                 C   s6   t  }|D ]}||v rtd|| f || qd S )Nz4Found duplicate value %s in CreateModel %s argument.)set
ValueErroradd)arg_nameobjs	used_valsval r   ]/var/www/ideatree/venv/lib/python3.10/site-packages/django/db/migrations/operations/models.py_check_for_duplicates   s   
r   c                       s@   e Zd Zdd Zedd Zdd Z fddZd	d
 Z  Z	S )ModelOperationc                 C   s
   || _ d S Nname)selfr   r   r   r   __init__      
zModelOperation.__init__c                 C   
   | j  S r   )r   lowerr   r   r   r   
name_lower      
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| ||S r   )superreducecan_reduce_throughr   	operationr'   	__class__r   r   r*       s   zModelOperation.reducec                 C   s   | | j| S r   )r(   r   r,   r   r   r   r+   %      z!ModelOperation.can_reduce_through)
__name__
__module____qualname__r   r   r#   r(   r*   r+   __classcell__r   r   r.   r   r      s    
r   c                       sr   e Zd ZdZg d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 )Nr6   c                 s       | ]\}}|V  qd S r   r   .0r   _r   r   r   	<genexpr>6       z'CreateModel.__init__.<locals>.<genexpr>basesc                 s   s8    | ]}t |d r|jjn
t|tr| n|V  qdS )_metaN)hasattrr@   label_lower
isinstancestrr!   )r;   baser   r   r   r=   9   s    


r8   c                 s   r9   r   r   r:   r   r   r   r=   B   r>   )	r6   r7   r   Modelr?   r8   r)   r   r   )r   r   r6   r7   r?   r8   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r0| jdt fgkr0| j|d< | jj	g |fS )N)r   r6   r7   r?   objectsr8   )
r   r6   r7   r?   r   rF   r8   Managerr/   r3   r   kwargsr   r   r   deconstructD   s   


zCreateModel.deconstructc              
   C   s6   | t|| jt| jt| jt| jt| j	 d S r   )
	add_modelr   r   listr6   dictr7   tupler?   r8   r   r'   stater   r   r   state_forwardsQ   s   zCreateModel.state_forwardsc                 C   2   |j || j}| |jj|r|| d S d S r   apps	get_modelr   allow_migrate_model
connectionaliascreate_modelr   r'   schema_editor
from_stateto_statemodelr   r   r   database_forwards]      zCreateModel.database_forwardsc                 C   2   |j || j}| |jj|r|| d S d S r   rU   rV   r   rW   rX   rY   delete_modelr[   r   r   r   database_backwardsb   ra   zCreateModel.database_backwardsc                 C   s    d| j ddr
dnd| jf S )NzCreate %smodel %sproxyFzproxy  )r7   getr   r"   r   r   r   describeg   s   zCreateModel.describec                 C   s   | j S r   r#   r"   r   r   r   migration_name_fragmentm   s   z#CreateModel.migration_name_fragmentc                 C   s   |  }|| jkrdS ||f}| jD ]}|tjur,t|tjjtfr,t	|||kr, dS q| j
D ]\}}t|| jf||r@ dS q0dS )NTF)r!   r#   r?   r   rF   rC   rE   	ModelBaserD   r   r6   r   )r   r   r'   r#   reference_model_tuplerE   _namefieldr   r   r   r(   q   s$   


zCreateModel.references_modelc                    sF  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`i | j j} jD ]}| jvrQ||d  qDt| j| j	|| j
| jdgS t  trz| j jkrzt| j| j	| j| j
 jdgS t  tr| j jkrt| j| j	i | j j ji| j
| jdgS t  tr| j jkrt| j| j	i | jd ji| j
| jdgS t  tr| j jkrt  trt| j| j	 j jfg | j| j
| jdgS t  trt| j fdd| j	D | j| j
| jdgS t  trS| j }dD ]!}||d }|r.ttt fdd	|D }|r.|||< q|d}| jkr>|d= t| j fd
d| j	D || j
| jdgS t  t r| j }dD ]}||}|ru 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 )Nrf   F)r6   r7   r?   r8   order_with_respect_toc                    s(   g | ]\}}|| j kr jn|fqS r   )r   ro   r;   nvr-   r   r   
<listcomp>       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   rj   r;   frt   r   r   r=      s    z/CreateModel.reduce.<locals>.<genexpr>.<genexpr>NrO   r;   r6   rt   r   r   r=      s    
z%CreateModel.reduce.<locals>.<genexpr>c                    s&   g | ]\}}|   jkr||fqS r   r%   rq   rt   r   r   ru     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_namery   rt   r   r   r=     s
    
z/CreateModel.reduce.<locals>.<setcomp>.<genexpr>r{   r|   rt   r   r   	<setcomp>  s    z%CreateModel.reduce.<locals>.<setcomp>c                    s(   g | ]\}}| j kr jn||fqS r   r}   rq   rt   r   r   ru   )  rv   )$rC   DeleteModelr#   r7   rh   RenameModelold_name_lowerr5   r   r6   r?   r8   AlterModelOptionsALTER_OPTION_KEYSpopr   AlterModelManagersAlterTogetherOptionOperationoption_nameoption_valueAlterOrderWithRespectTorp   r   model_name_lowerr
   ro   r   r   copyr   filterboolr   r~   r)   r*   )r   r-   r'   r7   keyr   optionrp   r.   rt   r   r*      s*  






	











zCreateModel.reduce)NNN)r1   r2   r3   __doc__serialization_expand_argsr   rK   rR   r`   re   ri   propertyrk   r(   r*   r4   r   r   r.   r   r5   )   s    
r5   c                   @   L   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 )r   zDrop a model's table.c                 C   s   d| j i}| jjg |fS Nr   )r   r/   r3   rI   r   r   r   rK   8  s   zDeleteModel.deconstructc                 C   s   | || j d S r   )remove_modelr#   rP   r   r   r   rR   >     zDeleteModel.state_forwardsc                 C   rb   r   rc   r[   r   r   r   r`   A  ra   zDeleteModel.database_forwardsc                 C   rS   r   rT   r[   r   r   r   re   F  ra   zDeleteModel.database_backwardsc                 C   s   dS )NTr   r&   r   r   r   r(   K  s   zDeleteModel.references_modelc                 C   
   d| j  S )NzDelete model %sr   r"   r   r   r   ri   P  r   zDeleteModel.describec                 C   r   )Nz	delete_%srj   r"   r   r   r   rk   S  r$   z#DeleteModel.migration_name_fragmentN)r1   r2   r3   r   rK   rR   r`   re   r(   ri   r   rk   r   r   r   r   r   5  s    r   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 )r   zRename a model.c                    s   || _ || _t | d S r   )r~   r   r)   r   )r   r~   r   r.   r   r   r   [  s   zRenameModel.__init__c                 C   r    r   r~   r!   r"   r   r   r   r   `  r$   zRenameModel.old_name_lowerc                 C   r    r   r   r!   r"   r   r   r   new_name_lowerd  r$   zRenameModel.new_name_lowerc                 C      | j | jd}| jjg |fS )Nr}   )r~   r   r/   r3   rI   r   r   r   rK   h     zRenameModel.deconstructc                 C      | || j| j d S r   )rename_modelr~   r   rP   r   r   r   rR   o     zRenameModel.state_forwardsc                 C   sR  |j || j}| |jj|r|j || j}|jj}|jj}||kr&d S |	||| |jj
D ]0}	|	j|kr@|}
|| jf}n|	j}
|	jjj|	jjjf}|j j| j|	jj}||
|	j| q1t|jj|jj}|D ]9\}}|j|jks}|jjjjs~qm|jj}|jj}|	||jj|jj |||j|jj|j|jj qmd S d S r   )rU   rV   r   rW   rX   rY   r~   r@   db_tablealter_db_tablerelated_objectsrelated_modelr   r'   
model_name	get_fieldro   r   alter_fieldziplocal_many_to_manyr_   remote_fieldthroughauto_created)r   r'   r\   r]   r^   	new_model	old_modelold_db_tablenew_db_tablerelated_objectr_   related_keyto_fieldr6   	old_field	new_fieldold_m2m_modelnew_m2m_modelr   r   r   r`   r  s\   

 zRenameModel.database_forwardsc                 C   s\   | j | j| _| _ | j| j| _| _| |||| | j | j| _| _ | j| j| _| _d S r   )r   r   r~   r   r`   r   r'   r\   r]   r^   r   r   r   re     s   

zRenameModel.database_backwardsc                 C   s   |  | jkp|  | jkS r   )r!   r   r   r&   r   r   r   r(     s   zRenameModel.references_modelc                 C      d| j | jf S )NzRename model %s to %sr}   r"   r   r   r   ri     r0   zRenameModel.describec                 C   r   )Nrename_%s_%s)r   r   r"   r   r   r   rk        z#RenameModel.migration_name_fragmentc                    sH   t |tr| j|jkrt| j|jgS tt| ||p#|	| j| S r   )
rC   r   r   r   r~   r   r)   r   r*   r(   r,   r.   r   r   r*     s   
zRenameModel.reduce)r1   r2   r3   r   r   r   r   r   rK   rR   r`   re   r(   ri   r   rk   r*   r4   r   r   r.   r   r   X  s     

9
r   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   )rC   r/   r   r#   r)   r*   r,   r.   r   r   r*     s
   zModelOptionOperation.reduce)r1   r2   r3   r*   r4   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                       || _ t | d S r   )tabler)   r   )r   r   r   r.   r   r   r        zAlterModelTable.__init__c                 C   r   )Nr   r   )r   r   r/   r3   rI   r   r   r   rK     r   zAlterModelTable.deconstructc                 C   s   | || jd| ji d S )Nr   )alter_model_optionsr#   r   rP   r   r   r   rR     s   zAlterModelTable.state_forwardsc           	      C   s   |j || j}| |jj|rH|j || j}|||jj|jj t	|jj
|jj
D ]\}}|jjjjrG||jj|jjjj|jjjj q,d S d S r   )rU   rV   r   rW   rX   rY   r   r@   r   r   r   r   r   r   )	r   r'   r\   r]   r^   r   r   r   r   r   r   r   r`     s(   

z!AlterModelTable.database_forwardsc                 C      |  ||||S r   r`   r   r   r   r   re     r0   z"AlterModelTable.database_backwardsc                 C   s"   d| j | jd ur| jf S df S )NzRename table for %s to %sz	(default)r   r"   r   r   r   ri   
  s   zAlterModelTable.describec                 C   r   )Nzalter_%s_tablerj   r"   r   r   r   rk     r$   z'AlterModelTable.migration_name_fragment)r1   r2   r3   r   r   rK   rR   r`   re   ri   r   rk   r4   r   r   r.   r   r     s    r   c                       st   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 fddZ  ZS )r   Nc                    s.   |rt t|}t| | j| t | d S r   )r   r   setattrr   r)   r   )r   r   r   r.   r   r   r     s   z%AlterTogetherOptionOperation.__init__c                 C   s   t | | jS r   )getattrr   r"   r   r   r   r     s   z)AlterTogetherOptionOperation.option_valuec                 C   s    d| j | j| ji}| jjg |fS r   )r   r   r   r/   r3   rI   r   r   r   rK   "  s   z(AlterTogetherOptionOperation.deconstructc                 C      | || j| j| ji d S r   )r   r#   r   r   rP   r   r   r   rR   )  
   
z+AlterTogetherOptionOperation.state_forwardsc                 C   sp   |j || j}| |jj|r6|j || j}t|d| j }||t|j| jt	 t|j| jt	  d S d S )Nzalter_%s)
rU   rV   r   rW   rX   rY   r   r   r@   r   )r   r'   r\   r]   r^   r   r   alter_togetherr   r   r   r`   0  s   z.AlterTogetherOptionOperation.database_forwardsc                 C   r   r   r   r   r   r   r   re   ;  r0   z/AlterTogetherOptionOperation.database_backwardsc                    s,   |  ||o| j pt fdd| jD S )Nc                 3   s    | ]} |v V  qd S r   r   r|   r   r   r   r=   A  r>   z@AlterTogetherOptionOperation.references_field.<locals>.<genexpr>)r(   r   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))rg   )r   r   lenr   r"   r   r   r   ri   D  s
   z%AlterTogetherOptionOperation.describec                 C   r   )Nzalter_%s_%s)r#   r   r"   r   r   r   rk   K  r   z4AlterTogetherOptionOperation.migration_name_fragmentc                    s(   t  ||pt|tot|t| uS r   )r)   r+   rC   r   typer,   r.   r   r   r+   O  s   
z/AlterTogetherOptionOperation.can_reduce_through)r1   r2   r3   r   r   r   r   rK   rR   r`   re   r   ri   r   rk   r+   r4   r   r   r.   r   r     s    

r   c                       $   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.
    rw   c                       t  || d S r   r)   r   )r   r   rw   r.   r   r   r   ^  r   zAlterUniqueTogether.__init__r1   r2   r3   r   r   r   r4   r   r   r.   r   r   V      r   c                       r   )AlterIndexTogetherzz
    Change the value of index_together to the target one.
    Input value of index_together must be a set of tuples.
    rx   c                    r   r   r   )r   r   rx   r.   r   r   r   j  r   zAlterIndexTogether.__init__r   r   r   r.   r   r   b  r   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 )r   z9Represent a change with the order_with_respect_to option.rp   c                    r   r   )rp   r)   r   )r   r   rp   r.   r   r   r   s  r   z AlterOrderWithRespectTo.__init__c                 C   r   )Nr   rp   )r   rp   r/   r3   rI   r   r   r   rK   w  r   z#AlterOrderWithRespectTo.deconstructc                 C   r   r   )r   r#   r   rp   rP   r   r   r   rR   ~  r   z&AlterOrderWithRespectTo.state_forwardsc                 C   s   |j || j}| |jj|rI|j || j}|jjr,|jjs,|||j	d d S |jjrK|jjsM|j	d}|
 sAd|_||| d S d S d S d S )N_orderr   )rU   rV   r   rW   rX   rY   r@   rp   remove_fieldr   has_defaultdefault	add_field)r   r'   r\   r]   r^   to_model
from_modelro   r   r   r   r`     s.   z)AlterOrderWithRespectTo.database_forwardsc                 C   s   |  |||| d S r   r   r   r   r   r   re     s   z*AlterOrderWithRespectTo.database_backwardsc                 C   s    |  ||o| jd u p|| jkS r   )r(   rp   r   r   r   r   r     s   z(AlterOrderWithRespectTo.references_fieldc                 C   r   )Nz%Set order_with_respect_to on %s to %sr   r"   r   r   r   ri     s   z AlterOrderWithRespectTo.describec                 C   r   )Nzalter_%s_order_with_respect_torj   r"   r   r   r   rk     r$   z/AlterOrderWithRespectTo.migration_name_fragment)r1   r2   r3   r   r   r   rK   rR   r`   re   r   ri   r   rk   r4   r   r   r.   r   r   n  s    r   c                       s\   e Zd ZdZg 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 )r   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.
    )base_manager_namedefault_manager_namedefault_related_nameget_latest_bymanagedorderingpermissionsdefault_permissionsselect_on_saveverbose_nameverbose_name_pluralc                    r   r   )r7   r)   r   )r   r   r7   r.   r   r   r     r   zAlterModelOptions.__init__c                 C   r   )N)r   r7   )r   r7   r/   r3   rI   r   r   r   rK     r   zAlterModelOptions.deconstructc                 C   s   | || j| j| j d S r   )r   r#   r7   r   rP   r   r   r   rR     s   z AlterModelOptions.state_forwardsc                 C      d S r   r   r   r   r   r   r`        z#AlterModelOptions.database_forwardsc                 C   r   r   r   r   r   r   r   re     r   z$AlterModelOptions.database_backwardsc                 C   r   )NzChange Meta options on %sr   r"   r   r   r   ri     r   zAlterModelOptions.describec                 C   r   )Nzalter_%s_optionsrj   r"   r   r   r   rk     r$   z)AlterModelOptions.migration_name_fragment)r1   r2   r3   r   r   r   rK   rR   r`   re   ri   r   rk   r4   r   r   r.   r   r     s    r   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 )r   zAlter the model's managers.r8   c                    r   r   )r8   r)   r   )r   r   r8   r.   r   r   r     r   zAlterModelManagers.__init__c                 C   s   | j j| j| jgi fS r   )r/   r3   r   r8   r"   r   r   r   rK     r   zAlterModelManagers.deconstructc                 C   r   r   )alter_model_managersr#   r8   rP   r   r   r   rR     r   z!AlterModelManagers.state_forwardsc                 C   r   r   r   r   r   r   r   r`     r   z$AlterModelManagers.database_forwardsc                 C   r   r   r   r   r   r   r   re     r   z%AlterModelManagers.database_backwardsc                 C   r   )NzChange managers on %sr   r"   r   r   r   ri     r   zAlterModelManagers.describec                 C   r   )Nzalter_%s_managersrj   r"   r   r   r   rk     r$   z*AlterModelManagers.migration_name_fragment)r1   r2   r3   r   r   r   rK   rR   r`   re   ri   r   rk   r4   r   r   r.   r   r     s    r   c                   @   s   e Zd ZdZedd ZdS )IndexOperationindexesc                 C   r    r   )r   r!   r"   r   r   r   r   	  r$   zIndexOperation.model_name_lowerN)r1   r2   r3   r   r   r   r   r   r   r   r     s    r   c                   @   r   )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   r   r   )	add_indexr   r   rP   r   r   r   rR     r   zAddIndex.state_forwardsc                 C   6   |j || j}| |jj|r||| j d S d S r   )rU   rV   r   rW   rX   rY   r   r   r[   r   r   r   r`        zAddIndex.database_forwardsc                 C   s6   |j || j}| |jj|r||| j d S d S r   )rU   rV   r   rW   rX   rY   remove_indexr   r[   r   r   r   re   "  r   zAddIndex.database_backwardsc                 C   r   )N)r   r   )r   r   r/   r3   rI   r   r   r   rK   '     zAddIndex.deconstructc                 C   sN   | j jrd| 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 %s, c                 S   s   g | ]}t |qS r   )rD   )r;   
expressionr   r   r   ru   6  s    z%AddIndex.describe.<locals>.<listcomp>z*Create index %s on field(s) %s of model %s)r   expressionsr   joinr   r6   r"   r   r   r   ri   2  s   zAddIndex.describec                 C      d| j | jj f S Nz%s_%s)r   r   r   r!   r"   r   r   r   rk   ?     z AddIndex.migration_name_fragmentNr1   r2   r3   r   r   rR   r`   re   rK   ri   r   rk   r   r   r   r   r     s    	r   c                   @   r   )RemoveIndexzRemove an index from a model.c                 C      || _ || _d S r   r   r   r   r   r   r   r   r   r   G     
zRemoveIndex.__init__c                 C   r   r   )r   r   r   rP   r   r   r   rR   K  r   zRemoveIndex.state_forwardsc                 C   sP   |j || j}| |jj|r&|j|| jf }|| j	}|
|| d S d S r   )rU   rV   r   rW   rX   rY   r   r   get_index_by_namer   r   )r   r'   r\   r]   r^   r_   from_model_stater   r   r   r   r`   N     zRemoveIndex.database_forwardsc                 C   P   |j || j}| |jj|r&|j|| jf }|| j	}|
|| d S d S r   )rU   rV   r   rW   rX   rY   r   r   r  r   r   )r   r'   r\   r]   r^   r_   to_model_stater   r   r   r   re   U  r  zRemoveIndex.database_backwardsc                 C   r   Nr   )r   r   r/   r3   rI   r   r   r   rK   \  r   zRemoveIndex.deconstructc                 C   r   )NzRemove index %s from %sr   r   r"   r   r   r   ri   g  r0   zRemoveIndex.describec                 C      d| j | j f S Nzremove_%s_%sr   r   r!   r"   r   r   r   rk   j     z#RemoveIndex.migration_name_fragmentNr   r   r   r   r   r   D  s    r   c                       sv   e Zd ZdZd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edd Z fddZ  ZS )RenameIndexzRename an index.Nc                 C   s<   |s|st d|r|rt d|| _|| _|| _|| _d S )NzHRenameIndex requires one of old_name and old_fields arguments to be set.z;RenameIndex.old_name and old_fields are mutually exclusive.)r   r   r   r~   
old_fields)r   r   r   r~   r  r   r   r   r   r  s   
zRenameIndex.__init__c                 C   r    r   r   r"   r   r   r   r     r$   zRenameIndex.old_name_lowerc                 C   r    r   r   r"   r   r   r   r     r$   zRenameIndex.new_name_lowerc                 C   s<   | j | jd}| jr| j|d< | jr| j|d< | jjg |fS )N)r   r   r~   r  )r   r   r~   r  r/   r3   rI   r   r   r   rK     s   

zRenameIndex.deconstructc                 C   sX   | j r||| jtj| j | jd ||| jtj| j  d S |	|| j| j
| j d S )Nr6   r   )r  r   r   r   Indexr   remove_model_optionsr   r   rename_indexr~   rP   r   r   r   rR     s   zRenameIndex.state_forwardsc                    s   |j || j}| |jj|sd S | jrP|j || j  fdd| jD }|j |dd}t|dkrEt	dt| j
jd|f tj| j|d d	}n|j|| jf }	|	| j}|j| jkrfd S |j|| jf }
|
| j}|||| d S )
Nc                    s   g | ]	} j |jqS r   )r@   r   column)r;   ro   r   r   r   ru     s    z1RenameIndex.database_forwards.<locals>.<listcomp>T)column_namesr   r	   z.Found wrong number (%s) of indexes for %s(%s).r   r   r  )rU   rV   r   rW   rX   rY   r  _constraint_namesr   r   r@   r   r   r   r  r   r  r~   r   r   r  )r   r'   r\   r]   r^   r_   columnsmatching_index_name	old_indexr  r  	new_indexr   r  r   r`     s>   
zRenameIndex.database_forwardsc                 C   sf   | j rd S | j| j| _| _| j| j| _| _| |||| | j| j| _| _| j| j| _| _d S r   )r  r   r   r~   r   r`   r   r   r   r   re     s   

zRenameIndex.database_backwardsc                 C   s>   | j rd| j  d| j d| j S d| j d| j d| j S )NzRename index z on z to zRename unnamed index for )r~   r   r   r  r"   r   r   r   ri     s   zRenameIndex.describec                 C   s0   | j rd| j| jf S d| jd| j| jf S )Nr   zrename_%s_%s_%sr<   )r~   r   r   r   r   r  r"   r   r   r   rk     s   
z#RenameIndex.migration_name_fragmentc                    sP   t |tr!| j|jkr!|jr!| j|jkr!t| j|j| j| jdgS t	 
||S )N)r   r~   r  )rC   r  r   r~   r   r   r   r   r  r)   r*   r,   r.   r   r   r*     s   zRenameIndex.reduce)NN)r1   r2   r3   r   r   r   r   r   rK   rR   r`   re   ri   r   rk   r*   r4   r   r   r.   r   r  o  s    


%

	r  c                   @   r   )AddConstraintconstraintsc                 C   r   r   r   
constraint)r   r   r  r   r   r   r     r  zAddConstraint.__init__c                 C   r   r   )add_constraintr   r  rP   r   r   r   rR     r   zAddConstraint.state_forwardsc                 C   r   r   )rU   rV   r   rW   rX   rY   r   r  r[   r   r   r   r`     r   zAddConstraint.database_forwardsc                 C   r   r   )rU   rV   r   rW   rX   rY   remove_constraintr  r[   r   r   r   re     r   z AddConstraint.database_backwardsc                 C      | j jg | j| jdfS )Nr  )r/   r1   r   r  r"   r   r   r   rK        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   ri   "  s   zAddConstraint.describec                 C   r   r   )r   r  r   r!   r"   r   r   r   rk   (  r   z%AddConstraint.migration_name_fragmentNr1   r2   r3   r   r   rR   r`   re   rK   ri   r   rk   r   r   r   r   r    s    
r  c                   @   r   )RemoveConstraintr  c                 C   r   r   r   r  r   r   r   r   0  r  zRemoveConstraint.__init__c                 C   r   r   )r!  r   r   rP   r   r   r   rR   4  r   zRemoveConstraint.state_forwardsc                 C   sP   |j || j}| |jj|r&|j|| jf }|| j	}|
|| d S d S r   )rU   rV   r   rW   rX   rY   r   r   get_constraint_by_namer   r!  )r   r'   r\   r]   r^   r_   r  r  r   r   r   r`   7  r  z"RemoveConstraint.database_forwardsc                 C   r  r   )rU   rV   r   rW   rX   rY   r   r   r&  r   r   )r   r'   r\   r]   r^   r_   r  r  r   r   r   re   >  r  z#RemoveConstraint.database_backwardsc                 C   r"  r  )r/   r1   r   r   r"   r   r   r   rK   E  r#  zRemoveConstraint.deconstructc                 C   r   )Nz"Remove constraint %s from model %sr	  r"   r   r   r   ri   O  r0   zRemoveConstraint.describec                 C   r
  r  r  r"   r   r   r   rk   R  r  z(RemoveConstraint.migration_name_fragmentNr$  r   r   r   r   r%  -  s    
r%  N)&	django.dbr   $django.db.migrations.operations.baser   django.db.migrations.stater   django.db.migrations.utilsr   r   django.db.models.optionsr   django.utils.functionalr   r6   r
   r   r   r   r   r   r   r5   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r%  r   r   r   r   <module>   s<    
  # 
3AD76+ )