OwlCyberSecurity - MANAGER
Edit File: server_id.cpython-311.pyc
� Ѝig� � � � d dl mZ d dlmZmZ d dlmZmZ d dlm Z m Z d dl m Z d dl mZ d dlmZ d dlmZ G d � d e� � Zd S ) � )�uuid4)�change_settings�load)�config� lveinfolib)�history� history_x60)� history_gov)�incident)�servers)�userc �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.11-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: N)r r �get� Exception�print)r �cfg�es r �_get_current_server_id_cfgz1UpdateOrCreateServerID._get_current_server_id_cfg" sl � � ��D� ,�-�-�-�C��7�7�;�'�'� '��� � � � �-�A�-�-� � � ��4�4�4�4�4����� ���s �), � A�A�Ac � � 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# s r �_set_server_id_cfgz)UpdateOrCreateServerID._set_server_id_cfg- se � � � �[�)� 4�d�6F� G� G� G� � ��� � � � �-�A�-�-� � � ��4�4�4�4�4����� ���s � � 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_table9 sn � � �_�+�+�-�-�3�3� �O� 0� 0� � � � � � �Z�&�&� �-� (� � � � � � � r c �6 � t t t t t g}t j | j |� � }| j � � � }|� � � } |D ]}| � ||||� � �|� t j � � � � ||d�� � � � |� t j � � � � t j |k � � � � |� � � |� � � |S # t* $ rE}|� � � |� � � 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�beginr2 r+ r r( �insertr, �deleter* r �commit�closer �rollbackr! ) r r/ r0 � sql_tablesr4 r- �txr. r# s r �_update_server_id_dbz+UpdateOrCreateServerID._update_server_id_dbH sy � � �{�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�:F�FTc �( � t t � � � � d| j � }|r�t d|� d�� � }t |� � |r|� � � }|| j v rt d|� d�� � dS ||k rt d|� 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] NzEnter new server ID ['z']: �Server ID cannot be '�'z Server ID is already set to '� z(Server ID length should be less than 255r ) �strr �server_id_length�inputr! �strip�restricted_server_id_list�lenr$ r? r& )r �promptr/ r �res� cur_server_ids r �_resolve_localhostz)UpdateOrCreateServerID._resolve_localhostf sN � � ����L�L�!7�$�"7�!7�8� � � ��@��@�@�@�A�A�C� �#�J�J�J�� (� �I�I�K�K� ��D� :�:�:��:�i�:�:�:�;�;�;��t�� -�-�-��B�i�B�B�B�C�C�C� � ��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 rt d|� d�� � | � |�� � S )r rA rB )r/ )r$ � isinstancerD rG rH r! rM )r �cur_cfg_server_ids r rJ z UpdateOrCreateServerID.prompt� sw � � !� ;� ;� =� =�� �'�� -� -� @�!�'�'�)�)�T�-K�K�K� �>�*;�>�>�>� ?� ?� ?��&�&�9J�&�K�K�Kr c �. � | � d�� � S )r F)rJ )rM )r s r �autozUpdateOrCreateServerID.auto� s � � �&�&�e�&�4�4�4r )N)TN)�__name__� __module__� __qualname__�__doc__rH rE r r$ r& r2 r? rM rJ rR � r r r r s� � � � � � �� � !/� ���H� H� H� H� � � � � � � � � � !� !� !�<� � � �: L� L� L�5� 5� 5� 5� 5r r N)�uuidr �clcommon.clconfparsr r � lvestats.libr r �lvestats.orm.historyr r �lvestats.orm.history_govr �lvestats.orm.incidentr �lvestats.orm.serversr �lvestats.orm.userr �objectr rW r r �<module>ra 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