Edit File: dom_collector.cpython-37.pyc
B 5/9e? � @ sf d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZ d dl m Z d dlm Z G dd� d�Z dS )� N)�ClPwd)� userdomains� list_users)�Feature)�is_panel_feature_supportedc @ sT e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd d � Z ed d � �Z edd� �Z dS )�DomainCollectorzR Class for print panel users domain docroots. See LU-1751 for details c C s* t tj�d�� | _| �� | _t� | _d S )N�DOM_COLL_RUNNING_IN_LVE) �bool�os�environ�getZ is_not_in_lve�is_not_limited�is_not_limited_processr �_clpwd)�self� r �$py/cl_dom_collector/dom_collector.py�__init__ s zDomainCollector.__init__c C s, t tj�d��} t tj�d��}| o*| S )Nr �DOM_COLL_RUNNING_IN_NICE)r r r r )Zis_limited_by_lveZis_limited_by_nicer r r r s z DomainCollector.is_not_limitedc C sZ t t� �� �}g }xB|D ]:}y | j�|�d }|�|� W q tjk rP Y qX qW |S )zL Get panel user names list :return: user names list r )�listr �keysr Z get_names�appendr ZNoSuchUserException)r Zpanel_uids_list�panel_users_listZuid�namer r r �_get_panel_user_names_list& s z*DomainCollector._get_panel_user_names_listc C sr | � � }xd|D ]\}yBx<t|�D ]0\}}t�|�d���d�}t|� d|� �� q W W q tk rh Y qX qW dS )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 �base64Z b64encode�encode�decode�print� Exception)r r ZusernameZdomainZdoc_rootZencoded_doc_rootr r r �"_print_users_doc_roots_and_domains5 s z2DomainCollector._print_users_doc_roots_and_domainsc C s� | j rvttj�rdy| �� }t|� W qt tjk r` } z d| }t|� td� W d d }~X Y qtX q~| � � }t|� n| � � d S )Nz$failed to run task in lve, error: %s���) r r r ZLVE�_run_self_in_lve�exit� lvectllibZ PyLveErrorr � _run_by_nicer! )r Zrc�eZ error_msgr r r �runI s zDomainCollector.runc C sJ t jdddd�} t �| ��&}tjddt|�dgttjdd �d �S Q R X 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 @)Zls_cpuZls_cpusZ ls_memory_phyz/bin/lve_suwrapperz-nz%/usr/bin/cloudlinux_domains_collector�1)r )�envN) r% Zmake_liblve_settingsZ temporary_lve� subprocess�call�str�dictr r )ZsettingsZlve_idr r r r# \ s z DomainCollector._run_self_in_lvec C s t jddddgttjdd�d�S )Nz /usr/bin/nicez-nZ19z%/usr/bin/cloudlinux_domains_collectorr* )r )r+ )r, r- r/ r r r r r r r& m s z DomainCollector._run_by_niceN) �__name__� __module__� __qualname__�__doc__r � staticmethodr r r! r( r# r& r r r r r s r ) r r r% r, Zclcommon.clpwdr Zclcommon.cpapir r Zclcommon.constr r r r r r r �<module> s