
    Rj                         S r SSKJr  SSKrSrSrSrSrS\	\
   4S jrS\
S	\
S
\
SS4S jrS\
S	\
S
\
SS4S jrSS jrSS jrg)zKRevision: renombrar tablas agent_v2_* a agents / agents_document_templates.    )opN0012_rename_agent_v2_to_agents0011_agents_token_usagereturnc                 \    [        [        R                  " U 5      R                  5       5      $ )N)setsainspectget_table_names)binds    aC:\Users\jesus\Desktop\proyecto_pie360\backend\alembic\versions\0012_rename_agent_v2_to_agents.py_tablesr      s    rzz$//122    tableold_namenew_namec                 &   [         R                  " 5       n[        R                  " U5      R	                  U 5       Vs1 s H  oDS   iM	     nnX;   a;  X%;  a5  [         R
                  " [        R                  " SU  SU SU S35      5        g g g s  snf )NnamezALTER TABLE `z` RENAME INDEX `z` TO ``)r   get_bindr	   r
   get_indexesexecutetext)r   r   r   r   idxindexess         r   _rename_index_if_existsr      s    ;;=D&(jj&6&B&B5&IJ&Is6{&IGJx6


277]5'1A(6RZQ[[\]^_  7 Ks   Bc                    [         R                  " 5       n[        R                  " U5      R	                  U 5       Vs1 s H  oDS   iM	     nnX;   a5  X%;  a/  [         R
                  " XSS9  [         R                  " X SS/5        g g g s  snf )Nr   unique)
table_nametype_agent_iddocument_id)r   r   r	   r
   get_unique_constraintsdrop_constraintcreate_unique_constraint)r   r   r   r   ucuniquess         r   _rename_unique_if_existsr(      sw    ;;=D$&JJt$4$K$KE$RS$Rb&z$RGSx6
8XF
##Hj-5PQ  7 Ts   Bc                     [         R                  " 5       n [        U 5      nSU;   ae  SU;   aH  SU;   a,   [         R                  " SSS9  [         R
                  " S5        [         R
                  " S5        [         R                  " SS5        [        U 5      nSU;   a2  [         R                  " SS5        [        SSS	5        [        SS
S5        g g ! [         a     Nf = f)Nagent_v2_agentsagentsagent_filesix_agent_files_agent_id)r   agent_v2_document_templatesagents_document_templates'ix_agent_v2_document_templates_agent_id%ix_agents_document_templates_agent_iduq_agent_v2_document_templateuq_agents_document_template)	r   r   r   
drop_index	Exception
drop_tablerename_tabler   r(   r   tabless     r   upgrader:       s    ;;=DT]FF"v&MM";V m,MM(#
)84T]F$.
57RS'53	

 	!'+)	
 / ! s   C 
CCc                     [         R                  " 5       n [        U 5      nSU;   a1  [        SSS5        [	        SSS5        [         R
                  " SS5        [        U 5      nSU;   a  SU;  a  [         R
                  " SS5        g g g )	Nr/   r3   r2   r1   r0   r.   r+   r*   )r   r   r   r(   r   r7   r8   s     r   	downgrader<   ?   s    ;;=DT]F"f, ')+	

 	 '35	

 	35RST]F6/v=
"34 >r   )r   N)__doc__alembicr   
sqlalchemyr	   revisiondown_revisionbranch_labels
depends_onr   strr   r   r(   r:   r<    r   r   <module>rF      s    Q  +)
3SX 3`3 `# ` ` `RC R3 R# R$ R
>5r   