Edit File: server_id.cpython-311.pyc
� L( e� � �� � d dl mZ d dl mZ d dlmZ d dlmZ d dlmZm Z d dl m Z d dl m Z d dl mZmZ d d lmZ d d lmZ d d lmZ d d lmZ d d lmZ G d� de� � ZdS )� )�print_function)�absolute_import)� basestring)�uuid4)�load�change_settings)�config)� lveinfolib)�history� history_x60)� history_gov)�incident)�servers)�user)� raw_inputc �N � e Zd ZdZdZdZdd�Zd� Zd� Zd� Z d � Z dd �Z d � Z d � Z dS )�UpdateOrCreateServerIDzeChange old-style server ID 'localhost' to auto-generated uuid User can set his own server ID )� localhost� Nc �: � || _ |p t j | _ dS )zi :type engine: sqlalchemy.engine.base.Engine :type config_path: Union[str, None] N)�enginer �GLOBAL_CONFIG_LOCATION� config_path)�selfr r s ��/builddir/build/BUILDROOT/lve-stats-4.2.1-1.el8.x86_64/opt/cloudlinux/venv/lib/python3.11/site-packages/lvestats/lib/server_id.py�__init__z UpdateOrCreateServerID.__init__ s! � � �� �&�G�&�*G����� c �� � t | j �� � }|� d� � S # t $ r-}t d� |� � � � Y d}~dS d}~ww xY w)�* :rtype: Union[None, str] )�path� server_id� Error: {0}N)r r �get� Exception�print�format)r �cfg�es r �_get_current_server_id_cfgz1UpdateOrCreateServerID._get_current_server_id_cfg( st � � ��D� ,�-�-�-�C��7�7�;�'�'� '��� � � � �,�%�%�a�(�(� )� )� )��4�4�4�4�4����� ���s �), � A#�"A � A#c � � t d|i| j � � |S # t $ r-}t d� |� � � � Y d}~dS d}~ww xY w)zmSets option 'server_id' in config file :type server_id: str :rtype: Union[None, str] r! r" N)r r r$ r% r& )r r! r( s r �_set_server_id_cfgz)UpdateOrCreateServerID._set_server_id_cfg3 sm � � � �[�)� 4�d�6F� G� G� G� � ��� � � � �,�%�%�a�(�(� )� )� )��4�4�4�4�4����� ���s � � A�"A � Ac �� � |j � � � � |j |k � � }|� |� d|i� � � � dS )a updates server_id in one of the tables :type conn: sqlalchemy.engine.base.Connection :type table: Union[history, history_gov, incident, user, history_x60] :type current_server_id: str :type new_server_id: str r! N)� __table__�update�wherer! �execute�values)r �conn�table�current_server_id� new_server_id� sql_updates r �_update_server_id_tablez.UpdateOrCreateServerID._update_server_id_table? sn � � �_�+�+�-�-�3�3� �O� 0� 0� � � � � � �Z�&�&� �-� (� � � � � � � r c �V � t t t t t g}t j | j |� � }| j � � � }|� � � } |D ]}| � ||||� � �|� t j � � � � ||d�� � � � |� t j � � � � t j |k � � � � |� � � |� � � |S # t* $ rU}|� � � |� � � t/ d� |� � � � Y d}~dS d}~ww xY w)z� updates server_id for all tables in the database :type current_server_id: str :type new_server_id: str :rtype: Union[None, str] )r! � lve_versionr" N)r r r r r r �get_lve_versionr �connect�beginr7 r0 r r- �insertr1 �deleter/ r! �commit�closer$ �rollbackr% r& ) r r4 r5 � sql_tablesr9 r2 �txr3 r( s r �_update_server_id_dbz+UpdateOrCreateServerID._update_server_id_dbN s� � � �{�H�d�K�H� � �0���>O�P�P� ��{�"�"�$�$�� �Z�Z�\�\�� !�#� \� \���,�,�T�5�:K�]�[�[�[�[� �L�L��*�1�1�3�3�:�:� +�K�H�H�� � � � � �L�L��*�1�1�3�3�9�9�'�:K�O`�:`�a�a� b� b� b� �I�I�K�K�K� �J�J�L�L�L� � �� � � � � �K�K�M�M�M� �J�J�L�L�L� �,�%�%�a�(�(� )� )� )��4�4�4�4�4����� ���s �)B6E � F(�A F#�#F(Tc � � t t � � � � d| j � }|r�t d� |� � � � }t |� � |r|� � � }|| j v r$t d� |� � � � dS ||k r$t d� |� � � � |S t |� � dk rt d� � dS | � � � }|� |s || j vr|S | � |p|pd|� � r| � |� � S |S )zw :type current_server_id: Union[None, str] :type prompt: bool :rtype: Union[None, str] Nz Enter new server ID ['{0}']: �Server ID cannot be '{0}'z!Server ID is already set to '{0}'� z(Server ID length should be less than 255r ) �strr �server_id_lengthr r& r% �strip�restricted_server_id_list�lenr) rD r+ )r �promptr4 r! �res� cur_server_ids r �_resolve_localhostz)UpdateOrCreateServerID._resolve_localhostl sW � � ����L�L�!7�$�"7�!7�8� � � �� =� D� D�Y� O� O�P�P�C� �#�J�J�J�� (� �I�I�K�K� ��D� :�:�:��3�:�:�9�E�E�F�F�F��t�� -�-�-��9�@�@��K�K�L�L�L� � ��9�~�~��#�#��@�A�A�A��t� �7�7�9�9� � � $�V� $� �T�%C�C�C� � � � $� $�%6�%V�-�%V�;�Xa� b� b� 6��*�*�9�5�5� 5��r c � � | � � � }t |t � � r=|� � � | j v r"t d� |� � � � | � |�� � S )r rF )r4 )r) � isinstancer rJ rK r% r& rP )r �cur_cfg_server_ids r rM z UpdateOrCreateServerID.prompt� sz � � !� ;� ;� =� =�� �'�� 4� 4� K�!�'�'�)�)�T�-K�K�K� �/�6�6�7H�I�I� J� J� J��&�&�9J�&�K�K�Kr c �. � | � d�� � S )r F)rM )rP )r s r �autozUpdateOrCreateServerID.auto� s � � �&�&�e�&�4�4�4r )N)TN)�__name__� __module__� __qualname__�__doc__rK rI r r) r+ r7 rD rP rM rU � r r r r s� � � � � � �� � !/� ���H� H� H� H� � � � � � � � � � !� !� !�<� � � �: L� L� L�5� 5� 5� 5� 5r r N)� __future__r r � past.builtinsr �uuidr �clcommon.clconfparsr r � lvestats.libr r �lvestats.orm.historyr r �lvestats.orm.history_govr �lvestats.orm.incidentr �lvestats.orm.serversr �lvestats.orm.userr r �objectr rZ r r �<module>rf s; �� &� %� %� %� %� %� &� &� &� &� &� &� $� $� $� $� $� $� � � � � � � 5� 5� 5� 5� 5� 5� 5� 5� � � � � � � #� #� #� #� #� #� 5� 5� 5� 5� 5� 5� 5� 5� 0� 0� 0� 0� 0� 0� *� *� *� *� *� *� (� (� (� (� (� (� "� "� "� "� "� "� #� #� #� #� #� #�@5� @5� @5� @5� @5�V� @5� @5� @5� @5� @5r