Edit File: dbmigrate_lib.cpython-311.pyc
� L( e� � �< � d dl mZ d dl mZ d dlZ d dlmZ d dlmZmZm Z n3# e $ r+Z e dk r e de � �� � ed� � n� Y dZ [ ndZ [ ww xY wej � ej � e� � d � � Zd � Zd � Zd � Zd � Zd� Zdd�ZdS )� )�print_function)�absolute_importN)�MigrationContext)�config�command�script�__main__z0Alembic Python library is not installed. Error: � z alembic.inic � � t � � }t j � | � � }d� |� � � D � � }d}|r,|� |� � |� |� � }|�,|S )Nc �, � i | ]}|j pd |j ��S )�base)� down_revision�revision)�.0�ss ��/builddir/build/BUILDROOT/lve-stats-4.2.1-1.el8.x86_64/opt/cloudlinux/venv/lib/python3.11/site-packages/lvestats/utils/dbmigrator/dbmigrate_lib.py� <dictcomp>z,generate_revisions_chain.<locals>.<dictcomp> s/ � � 0� 0� 0��a�o�/���� 0� 0� 0� r )�listr �ScriptDirectory� from_config�walk_revisions�append�get)� alembic_cfg�chain_�script_�revision_pairs�curent_revisions r �generate_revisions_chainr s� � � �V�V�F��$�0�0��=�=�G�0� 0� �,�,�.�.�0� 0� 0�N� �O� � >�� � �o�&�&�&�(�,�,�_�=�=�� � >� �Mr c � � | � � � }t j |� � }|� � � }|pdS )Nr )�connectr � configure�get_current_revision)�engine� connection�context� current_revs r �get_database_versionr) ' s= � �� � �!�!�J� �(��4�4�G��.�.�0�0�K� � �&� r c �x � t | � � }|� |� � |� |� � k S )zq Check need migrate or downgrade to specific revisions :return bool: True - upgrade; False downgrade )r �index)r � revision_from� revision_to� rev_chains r � migration_wayr/ . s4 � � )��5�5�I� �?�?�=� )� )�Y�_�_�[�-I�-I� I�Ir c � � | � dd� � }| � dd� � }t j � t j � ||� � � � }| � dd|� � | S )zr correct script_location in alembic config for relative path support :param alembic_cfg: :return: �alembic�script_location�here)�get_section_option�os�path�abspath�join�set_section_option)r �locationr3 �location_correcteds r �correct_script_locationr< 7 sz � � �-�-�i�9J�K�K�H� � )� )�)�V� <� <�D��� � ��� � �d�H�)E�)E�F�F���"�"��$�&8�:� :� :� �r c �H � t t ddg�� � dd� � } | S )Nz lvestats.orm�alembic_version)�fromlistr )�getattr� __import__)r> s r �get_orm_versionrB E s2 � � ��>�->�,?�@�@�@��6�#� #�O� �r Fc � � |�t � � }t t j t � � � � }||j d< | � � � 5 }||j d< |rt j ||� � nKt | � � }t |||� � rt j ||� � nt j ||� � d d d � � d S # 1 swxY w Y d S )Nz lve-statsr& ) rB r< r �Config� ALEMBIC_CONF� attributes�beginr �stampr) r/ �upgrade� downgrade)r% r> rH � lve_stats_cfgr r& �database_versions r �alembic_migraterM L s0 � �� �)�+�+��)�&�-� �*E�*E�F�F�K�*7�K��;�'� � � � � � @�:�/9� � �|�,� � @� �M�+�� 7� 7� 7� 7� 3�F� ;� ;� � �[�*:�O�L�L� @��� �_�=�=�=�=��!�+��?�?�?� @� @� @� @� @� @� @� @� @� @� @� @���� @� @� @� @� @� @s �A.C�C�C)NFN)� __future__r r r5 �alembic.migrationr r1 r r r � ImportError�e�__name__�print�exitr6 r8 �dirname�__file__rE r r) r/ r< rB rM � r r �<module>rX se �� &� %� %� %� %� %� &� &� &� &� &� &� � � � � �2�2�2�2�2�2�/�/�/�/�/�/�/�/�/�/�/��� � � ��:� � � ��D��D�D�E�E�E� ��Q����� � ��������� ���� �w�|�|�B�G�O�O�H� 5� 5�}�E�E� � � � �!� !� !�J� J� J� � � � � � �@� @� @� @� @� @s �# �A�!A�A