Edit File: 15fed017ca0b_add_uid_column.cpython-311.pyc
� L( e# � �X � d Z ddlmZ ddlZddlZddlZddlmZ ddlmZ ddl m Z m Z m Z m Z mZmZmZ ddlmZ dZd ZdZdZd Z e� � Z ej d ej � e � � z � � Z e d e d d d�� � e d e d� � d d�� � e de � � e de � � e de � � e de � � e de � � e de � � e de � � e d e d� � d d�� � f Z e d e d d�� � ed � e� � d � � e d e d d�� � e de � � e de � � e de � � e de � � e de � � e de � � e de � � e d e d� � d d�� � f Z d � Z!d � Z"dS ) zzReplace columns username to uid Revision ID: 15fed017ca0b Revises: 48efb1674139 Create Date: 2016-07-22 17:19:55.478609 � )�absolute_importN)�op)�context)�Column�Integer�String�Float�Index�MetaData�Table)� uidconverter� 15fed017ca0b� 48efb1674139�lve_stats2_history_govzalembic.script.%s�tsTF)�index� primary_key� autoincrement�username�@ )r r �sum_cpu� sum_write�sum_read�limit_cpu_on_period_end�limit_read_on_period_end�limit_write_on_period_end�cause_of_restrict� server_id� z ix_{}_ts_2�uidc �� � t j � � } t j j � d� � d }t j t � � }t dz t z }t j t |� � t j t gt �R � }t |t gt �R � }| � � � }| � |� � � � � }|� � � D ]�}t |� � } | d } | � d� � } | || v r|| | } nKt- j | || | � � pd} | dk r!t0 � d| �d| �d �� � | || | < | dk r-| | d < | � |� | � � � � ��|� � � t j |� � d S ) N� lve-statsr �_r ���r zCan't obtain uid for user "�"; server_id "�"r ) r �get_bindr �config� attributes�get� collections� defaultdict�dict�HISTORY_GOV_TABLE_NAME� down_revision� rename_table� create_table�history_gov_new_itemsr �metadata�history_gov_old_items�begin�execute�select�fetchall�popr �username_to_uid�logger�warning�insert�commit� drop_table) � connection�local_server_id�username_to_uid_cache�tablename_renamed� history_gov�history_gov_old�trans�data_history_gov_old�history_gov_old_row�history_gov_new_rowr r r s ��/builddir/build/BUILDROOT/lve-stats-4.2.1-1.el8.x86_64/opt/cloudlinux/venv/lib/python3.11/site-packages/lvestats/utils/dbmigrator/versions/15fed017ca0b_add_uid_column.py�upgraderK ? s � ��� � �J� �n�/�3�3�K�@�@��M�O� '� 3�D� 9� 9��.��4�}�D�� �O�*�,=�>�>�>��/� �8�!6�8� 8� 8�K� � -�x�P�:O�P�P�P�O� � � � � �E�%�-�-�o�.D�.D�.F�.F�G�G�� 3� <� <� >� >� H� H�� "�#6� 7� 7��'� �4� �&�*�*�:�6�6�� �,�Y�7� 7� 7�'� �2�8�<�C�C� �.��/�9�j�B� B� H�EG� ��Q�w�w����� '�x�x���� 4� 5� 5� 5�9<� !�)� ,�X� 6� ��7�7�),� �� &� � � �{� 1� 1�2E� F� F� G� G� G�� �L�L�N�N�N��M�#�$�$�$�$�$� c �� � t j � � } t j j � d� � d }t j t � � }t dz t z }t j t |� � t j t gt �R � }t |t gt �R � }| � � � }| � |� � � � � }|� � � D ]�}t |� � } | d } | � d� � } | || v r|| | } nEt- j | || | � � } | �!t0 � d| �d| �d�� � | || | < | r-| | d< | � |� | � � � � ��|� � � t j |� � d S ) Nr" r r# r z Can't obtain user name for uid "r% r& r ) r r' r r( r) r* r+ r, r- r. �revisionr0 r1 r4 r r3 r2 r5 r6 r7 r8 r9 r �uid_to_usernamer; r< r= r> r? ) r@ rA �uid_to_username_cacherC rE rD rF �data_history_gov_newrI �history_gov_old_dictr r r s rJ � downgraderS c s� � ��� � �J� �n�/�3�3�K�@�@��M�O� '� 3�D� 9� 9��.��4�x�?�� �O�*�,=�>�>�>��o� �8�!6�8� 8� 8�O� �)�8�L�6K�L�L�L�K� � � � � �E�%�-�-�k�.@�.@�.B�.B�C�C�� 3� <� <� >� >� M� M�� #�$7� 8� 8� �(��5� �"�&�&�u�-�-�� �'� �2� 2� 2�,�Y�7��<�H�H�#�3��_�i��=� =�H�� ������s�s�I�I�I�'�(� (� (� 5=� !�)� ,�S� 1� � M�/7� �� ,� � � �� 5� 5�6J� K� K� L� L� L�� �L�L�N�N�N� �M�#�$�$�$�$�$rL )#�__doc__� __future__r �logging�osr+ �alembicr r � sqlalchemyr r r r r r r � lvestats.libr rN r/ � branch_labels� depends_onr. r3 � getLogger�path�basename�__name__r; r4 �formatr2 rK rS � rL rJ �<module>rc s� ��� � '� &� &� &� &� &� ���� � � � � � � � � � � � � � � � � � � � � M� M� M� M� M� M� M� M� M� M� M� M� M� M� M� M� M� M� %� %� %� %� %� %� �� � �� � � �1� � �8�:�:�� �� �.���1A�1A�(�1K�1K�K� L� L�� �F�4���$�e�L�L�L� �F�:�v�v�b�z�z�t�5�I�I�I� �F�9�e� � � �F�;�� � � �F�:�u� � � �F� $�g�.�.� �F� %�w�/�/� �F� &��0�0� �F� ��(�(� �F�;���s� � ��U�K�K�K� � � �F�4��d�%�@�@�@� �E�,� � � 4� 5� 5�t�<�<� �F�5�'�t�5�A�A�A� �F�9�e� � � �F�;�� � � �F�:�u� � � �F� $�g�.�.� �F� %�w�/�/� �F� &��0�0� �F� ��(�(� �F�;���s� � ��U�K�K�K� � � !%� !%� !%�H#%� #%� #%� #%� #%rL