Edit File: dom_collector.cpython-311.pyc
� c|9e? � �t � d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZ d dl m Z d dlm Z G d� d� � Z dS )� N)�ClPwd)� userdomains� list_users)�Feature)�is_panel_feature_supportedc �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 �U/opt/cloudlinux/venv/lib64/python3.11/site-packages/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_domains5 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 � � rf | � � � }t |� � d S # t j $ r.}d|z }t |� � t d� � Y d }~d S d }~ww xY w| � � � }t |� � d S | � � � d S )Nz$failed to run task in lve, error: %s���) r r r �LVE�_run_self_in_lve�exit� lvectllib� PyLveErrorr- � _run_by_nicer3 )r �rc�e� error_msgs r �runzDomainCollector.runI s� � � � &� 6�)�'�+�6�6� � � �.�.�0�0�B���H�H�H�H�H�� �+� � � � F�� J�I��)�$�$�$���H�H�H�H�H�H�H�H�H����� ���� �&�&�(�(���R������ � 3� 3� 5� 5� 5� 5� 5s �#A �B�#A?�?Bc � � 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) r9 �make_liblve_settings� temporary_lve� subprocess�call�str�dictr r )�settings�lve_ids r r7 z DomainCollector._run_self_in_lve\ 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 A9�9A=� A=c �d � t j g d�t t j d�� � �� � S )N)z /usr/bin/nicerE �19rF rG )r rH )rL rM rO r r � r r r; z DomainCollector._run_by_nicem sD � ��� 6� 6� 6��� �),�� � � � � � r N) �__name__� __module__� __qualname__�__doc__r � staticmethodr r& r3 r? r7 r; rT r r r r s� � � � � � �� � � � � �@� @� �\�@� � � � � � �(6� 6� 6�& �C� C� �\�C� � � � �\� � � r r ) r r) r9 rL �clcommon.clpwdr �clcommon.cpapir r �clcommon.constr r r rT r r �<module>r] s� �� � � � � � � � � � � � � � � � � � � � � � � 2� 2� 2� 2� 2� 2� 2� 2� "� "� "� "� "� "� 5� 5� 5� 5� 5� 5�` � ` � ` � ` � ` � ` � ` � ` � ` � ` r