a
    ;]ri¿	  ã                   @   sd   d Z ddlmZ ddlZddlmZ ddlZddlZdZ	dZ
dZdZdZdd	„ Zd
d„ Zdd„ ZdS )zlremove satbot jwt

Revision ID: fed7415a80da
Revises: 9e43d8bea8c8
Create Date: 2025-11-07 14:46:58.776815

é    )ÚopN)ÚCONFIGZfed7415a80daÚ9e43d8bea8c8é
   c                 C   s`   t j| jtd d ddgd}dD ]2}| |¡r"dd„ ||  d	d¡ d
¡D ƒ||< q"t |¡S )NÚsatbotsÚ
jwt_secretZHS256Ú )ÚkeyÚ
algorithms)Zupdate_displacementsz--datesc                 S   s   g | ]}t |ƒ‘qS © )Úint©Ú.0Úxr   r   úX/var/www/staging/api/app_home/migrations/users/alembic/versions/038_remove_satbot_jwt.pyÚ
<listcomp>   ó    z_parameters.<locals>.<listcomp>ú'ú,)	ÚjwtÚdecodeÚ
parametersr   ÚgetÚreplaceÚsplitÚjsonÚdumps)Úsatbotr   r	   r   r   r   Ú_parameters   s
    
&r   c               
   C   sô   t  ¡ } t  dt dt ¡ ¡¡ |  t d¡¡ ¡ }dd„ |D ƒ}t	dt
|ƒtƒD ]€}|||t … }g }|  t dd d	d„ |D ƒ¡› d
¡¡ ¡ }|D ] }| dt|ƒ› d|j› ¡ qš|  t d |¡¡¡ qRt  dd¡ t jdddd d S )Nr   ÚparamsúSELECT id FROM satbotsc                 S   s   g | ]}|d  ‘qS ©r   r   r   r   r   r   r   "   r   zupgrade.<locals>.<listcomp>r   ú0SELECT id, parameters FROM satbots WHERE id in (r   c                 S   s   g | ]}t |ƒ‘qS r   ©Ústrr   r   r   r   r   '   r   ú)úUPDATE satbots SET params = 'ú' WHERE id = ú;r   ©Únew_column_name)r   Úget_bindÚ
add_columnÚsaÚColumnÚJSONÚexecuteÚtextÚfetchallÚrangeÚlenÚ
BATCH_SIZEÚjoinÚappendr   ÚidÚdrop_columnÚalter_column©Úusers_connectionZ
satbot_idsÚbatchZsatbot_batch_idsZqueriesr   r   r   r   r   Úupgrade   s    ,r>   c               
   C   s  t  ¡ } t  dt dt ¡ ¡¡ |  t d¡¡ ¡ }dd„ |D ƒ}t	dt
|ƒtƒD ]}|||t … }g }|  t dd d	d„ |D ƒ¡› d
¡¡ ¡ }|D ]0}| dtj|jtd d d› d|j› ¡ qš|  t d |¡¡¡ qRt  dd¡ t jdddd d S )Nr   r   r    c                 S   s   g | ]}|d  ‘qS r!   r   r   r   r   r   r   8   r   zdowngrade.<locals>.<listcomp>r   r"   r   c                 S   s   g | ]}t |ƒ‘qS r   r#   r   r   r   r   r   =   r   r%   r&   r   )r	   r'   r(   r   r)   )r   r+   r,   r-   r.   ÚStringr0   r1   r2   r3   r4   r5   r6   r7   r   Úencoder   r   r8   r9   r:   r;   r   r   r   Ú	downgrade2   s    ,.rA   )Ú__doc__Úalembicr   Ú
sqlalchemyr-   Úsetupr   r   r   ÚrevisionÚdown_revisionÚbranch_labelsÚ
depends_onr5   r   r>   rA   r   r   r   r   Ú<module>   s   