관리-도구
편집 파일: dom_collector.cpython-311.pyc
� �EBg[ � �t � d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m Z m Z m Z G d� d� � Z dS )� N)�ClPwd)�Feature)�is_panel_feature_supported� list_users� userdomainsc �l � e Zd ZdZd� Zed� � � Zd� Zd� Zd� Z ed� � � Z ed� � � Z d S ) �DomainCollectorzR Class for print panel users domain docroots. See LU-1751 for details c �� � t t j � d� � � � | _ | � � � | _ t � � | _ d S )N�DOM_COLL_RUNNING_IN_LVE) �bool�os�environ�get� is_not_in_lve�is_not_limited�is_not_limited_processr �_clpwd)�selfs �$py/cl_dom_collector/dom_collector.py�__init__zDomainCollector.__init__ sG � �!%�b�j�n�n�5N�&O�&O�!P�!P� P���&*�&9�&9�&;�&;��#��g�g�� � � � c � � t t j � d� � � � } t t j � d� � � � }| o| S )Nr �DOM_COLL_RUNNING_IN_NICE)r r r r )�is_limited_by_lve�is_limited_by_nices r r z DomainCollector.is_not_limited sN � � ��� � �0I�!J�!J� K� K�� !�"�*�.�.�1K�"L�"L� M� M��$�$�?�-?�)?�?r c � � t t � � � � � � � }g }|D ]M} | j � |� � d }|� |� � �9# t j $ r Y �Jw xY w|S )zL Get panel user names list :return: user names list r )�listr �keysr � get_names�appendr �NoSuchUserException)r �panel_uids_list�panel_users_list�uid�names r �_get_panel_user_names_listz*DomainCollector._get_panel_user_names_list# s� � � �z�|�|� 0� 0� 2� 2�3�3�� ��"� � �C� ��{�,�,�S�1�1�!�4�� �'�'��-�-�-�-���,� � � ��� ���� � s �5A+�+A=�<A=c � � | � � � }|D ]v} t |� � D ]S\ }}t j |� d� � � � � d� � }t |� d|� �� � �T�g# t $ r Y �sw xY wdS )z� Method prints pair `domain,doc_root` for an each user on a server domain is encoded in punycode if it's needed doc_root is always encoded in base64 zutf-8�,N)r& r �base64� b64encode�encode�decode�print� Exception)r r# �username�domain�doc_root�encoded_doc_roots r �"_print_users_doc_roots_and_domainsz2DomainCollector._print_users_doc_roots_and_domains2 s� � � �:�:�<�<��(� � �H� �(3�H�(=�(=� :� :�$�F�H�'-�'7� ����0�0�(� (� �f�W�o�o� %� �V�8�8�&6�8�8�9�9�9�9� :�� � � � � �� ���� � s �A#A>�> B � B c � � | j r�t t j � � rp | � � � }t j |� � d S # t j $ r3}d|� �}t |� � t j d� � Y d }~d S d }~ww xY w| � � � }t j |� � d S | � � � d S )Nz"failed to run task in lve, error: ���) r r r �LVE�_run_self_in_lve�sys�exit� lvectllib� PyLveErrorr- � _run_by_nicer3 )r �rc�e� error_msgs r �runzDomainCollector.runF s� � � � &� 6�)�'�+�6�6� �!� �.�.�0�0�B��H�R�L�L�L�L�L�� �+� !� !� !� H�Q� H� H�I��)�$�$�$��H�R�L�L�L�L�L�L�L�L�L�����!���� �&�&�(�(����� � � � � � � 3� 3� 5� 5� 5� 5� 5s �(A � B�(B � Bc �0 � t j t j ddd�� � � � } t j | � � 5 }t j ddt |� � dgt t j d� � � � � � cd d d � � S # 1 swxY w Y d S ) z� Run same command in lve and set environ RUNNING_IN_LVE=True in order to check it in child process. :return: � r i @)�ls_cpu�ls_cpus� ls_memory_phyz/bin/lve_suwrapper�-n�%/usr/bin/cloudlinux_domains_collector�1)r ��envN) r: �make_liblve_settings�create_liblve_settings� temporary_lve� subprocess�call�str�dictr r )�settings�lve_ids r r7 z DomainCollector._run_self_in_lveY s� � � �1� � ,��� .� � � � � �� � $�X� .� .� C�&� �?�%�t�S��[�[�:a�b����S�A�A�A�C� C� C� C� C� C� C� C� C� C� C� C� C� C� C���� C� C� C� C� C� Cs �A B � B�Bc �d � t j g d�t t j d�� � �� � S )N)z /usr/bin/nicerF �19rG rH )r rI )rN rO rQ r r � r r r< z DomainCollector._run_by_nicel sD � ��� 6� 6� 6��� �),�� � � � � � r N) �__name__� __module__� __qualname__�__doc__r � staticmethodr r& r3 r@ r7 r< rV r r r r s� � � � � � �� � � � � �@� @� �\�@� � � � � � �(6� 6� 6�& �C� C� �\�C�$ � � � �\� � � r r )r) r rN r8 r: �clcommon.clpwdr �clcommon.constr �clcommon.cpapir r r r rV r r �<module>r_ s� �� � � � � � � � � � � � � � � � � � � � � � � � � � � "� "� "� "� "� "� N� N� N� N� N� N� N� N� N� N�b � b � b � b � b � b � b � b � b � b r