XylotrechusZ
�
iAhl � �� � d dl Z d dlZd dlmZmZ d dlmZ d dlmZm Z m
Z
m Z m Z m
Z
mZmZmZ d� Z ed� � d� � � Zd� Zd � Z ed
� � d � � � Zd � ZdS )
� N)�run�task)�install_package) � STATE_FAILED� STATE_MASKED�
STATE_RUNNING�
STATE_STOPPED�STATE_UNMASKED�
STATE_ENABLED�STATE_DISABLED�get_service_state�set_service_statec �D � t dd�� � t � � d S )Nz lve-statsT)� reinstall)r �enable_lve_services� � �py/clconfigure/limits.py�initialize_lvestatsr s' � ��K�4�0�0�0�0������r z2Changing default limits state to '{desired_state}'c �R � | dk rt g d�� � dS t d� � �)z�
Brings default limits to given state (unlimited | default).
May be executed more that once, does't crash on future calls
� unlimited)�lvectl�set�defaultz --unlimitedz$Another states temporary unavailableN)r � RuntimeError)�
desired_states r �set_default_limits_stater s9 � � � �#�#� � 7� 7� 7�8�8�8�8�8��A�B�B�Br c �^ � t t t fD ]} dD ]}t | |� � ��dS )z,
Turn off all lve-related services.
��lve�lve_namespaces�lvestatsr N)r r r r ��state�services r �disable_lve_servicesr& * sK � � � � �>� .� .��F� .� .�G� �e�W� -� -� -� -� .�.� .r c �^ � t t t fD ]} dD ]}t | |� � ��dS )z+
Turn on all lve-related services.
r N)r
r r r r# s r r r 3 sK � � !�-� �?� .� .��F� .� .�G� �e�W� -� -� -� -� .�.� .r zUnloading lve modulec � �� t j d� � � � � sdS t ddg� � } | j �| j � � � ng ��fd�dD � � }|rt ddg|�� � t g d�� � } t j d � � � � � rt
j d
� � dS dS ) zO
Restart services that still use lve device and unload lve module then
z/dev/lveN�lsofc �L ��� g | ] �t �fd ��D � � � � � ��� S )c 3 �B �K � | ]}|� �� � V � �d S )N)�
startswith)�.0�liner% s �r � <genexpr>z/unload_lve_module.<locals>.<listcomp>.<genexpr>K s/ �� � � �E�E�D�t���w�'�'�E�E�E�E�E�Er )�any)r- r% �res_stdout_liness @�r �
<listcomp>z%unload_lve_module.<locals>.<listcomp>I sN ��� � � � �� �E�E�E�E�4D�E�E�E� E� E���� � r )�mysqld�mysql�mariadb�httpd�apache2� systemctl�restart)�modprobez-rf�kmodlvez/sys/module/kmodlvez:Failed to unload the lve module. Please reboot the server.)�pathlib�Path�existsr �stdout�split�logging�warning)�res�servicesr1 s @r �unload_lve_modulerE < s� �� �
�<�
� #� #� *� *� ,� ,� ��� �v�z�"�
#�
#�C�-0�Z�-C�s�z�'�'�)�)�)���� � � � Q�� � �H�
� 1� �[�)� /�h� /�0�0�0�
�,�,�,�
-�
-�C��|�)�*�*�1�1�3�3� V���T�U�U�U�U�U�V� Vr c �� � t d� � t k rdS t t d� � t t d� � t t
d� � t t d� � dS )z�
Apply workaround for the case
When `systemctl stop lve` makes it transition to "failed" state instead of "inactive"
r N)r
r r r
r r r r r r �apply_workaround_lve_failedrG V sb � �
�� � �<�/�/����n�e�,�,�,��m�U�+�+�+��m�U�+�+�+��l�E�*�*�*�*�*r )rA r< � clconfigurer r �clconfigure.packagesr �clconfigure.servicesr r r r r
r r r
r r r r& r rE rG r r r �<module>rK sK �� ���� ���� !� !� !� !� !� !� !� !� 0� 0� 0� 0� 0� 0�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�� � � ��:�;�;�C� C� <�;�C�.� .� .�.� .� .� �� � � �V� V� � �V�2
+�
+�
+�
+�
+r