File: //opt/alt/python37/lib/python3.7/site-packages/xray/adviser/__pycache__/clwpos_get.cpython-37.pyc
B
�V�di � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z ddl
m Z ddl m
Z
ddlmZ dd lmZ e�d
�ZG d d � d �ZdS )
zG
This module contains a wrapper around `clwpos-user get` local utility
� N)�Optional)�Feature)�is_panel_feature_supported)�CloudlinuxSummary� )�
get_client)�NginxUserCacheZ clwpos_utilc @ s� e Zd ZdZeedd�dd�Zeeed�dd��Ze e
e d �d
d ��Z e ed �d d
��Z eee
e
d�dd�Zeee
d�dd�Zeed �dd��Zee
dd�dd��ZdS )� ClWposGetterz/usr/bin/clwpos-userN)�username�domain�returnc C sV | � |�rt�d� dS | �||�}t�dt|�� |rD| �|� nt�d||� dS )z8Construct and POST metadata to Smart Advice microservicez)ea-nginx detected, skipping metadata sendNz Got WPOS: %sz4Metadata for user %s with domain %s will not be sent)�nginx_cache_for_user�logger�info�construct_metadata�debug�str�send�warning)�selfr
r � json_data� r �H/opt/alt/python37/lib/python3.7/site-packages/xray/adviser/clwpos_get.py�
post_metadata s
zClWposGetter.post_metadata)r
r c C s
t | �jS )z9
Check nginx cache status for given user
)r �
is_enabled)r
r r r r
+ s z!ClWposGetter.nginx_cache_for_user)r c C s t tj�sdS dS )z0Special hack for executing user commands on Soloz%sudo -u %(username)s -s /bin/bash -cN)r r �CAGEFS)r r r r �wrapper2 s
zClWposGetter.wrapperc C s( t tj�s| j� d�S d| j� d�S dS )zResolve command to executez scan --website %(website)sz%/sbin/cagefs_enter_user %(username)s N)r r r �util)r r r r �cmd8 s
zClWposGetter.cmd)r
�
domainnamer c C s. t j�| j�sdS | jdk rH| j||d� �� }|�| j||d� � n| j||d� �� }ytj |ddd�}W n: t
t tj fk
r� } zt
�d| j|� dS d}~X Y nX |jdkr�t
�d| jtt|dd��tt|d d��� dS yt�|j�� �S tjk
�r( t
�d
| jtt|dd��� dS X dS ) z<
External call of `clwpos-user get` utility
N)r
�websiteT)�capture_output�textzError running %s: %sr z9Metadata collection via %s failed. stdout: %s, stderr: %s�stdout�stderrz8Invalid JSON from %s for metadata collection. stdout: %s)�os�path�isfiler r �split�appendr �
subprocess�run�OSError�
ValueError�SubprocessErrorr �error�
returncoder �getattr�json�loadsr# �strip�JSONDecodeError)r r
r �_exec�result�er r r �utility@ s0
zClWposGetter.utilityc sD d}| � ||�� � dk r@||� fdd�� d �� D �| �� d�}|S )zX
Ensure format accepted by Smart Advice POST requests/metadata endpoint
Nc s. g | ]&\}}t d |� �� d �|g �d��qS )�/�issues)r& r; )�dict�get)�.0�siter; )�datar r �
<listcomp>q s z3ClWposGetter.construct_metadata.<locals>.<listcomp>r; )r
r �websites�server_load_rate)r9 �itemsrC )r r
r Z dummy_resultr )r@ r r f s
z ClWposGetter.construct_metadatac
C sT y t �� } W n2 tk
r> } zt�dt|�� dS d}~X Y nX t�� pJd}| | S )� z%Unable to get domains_total stats: %sg �Ng �?)r Z_get_total_domains_amount� Exceptionr r/ r �multiprocessing� cpu_count)Z
domains_totalr8 rH r r r rC x s z ClWposGetter.server_load_rate)�statr c C s" | d r t d�}|� j| d� dS )zo
Send gathered metadata to adviser miscroservice.
Ignore sending if websites are empty
rB �adviser)r@ N)r Z send_stat)rI �clientr r r r � s zClWposGetter.send)�__name__�
__module__� __qualname__r r r � staticmethod�boolr
�propertyr r r r< r9 r �floatrC r r r r r r s &
r )�__doc__r2 �loggingr% r* rG �typingr �clcommon.constr �clcommon.cpapir Zclsummary.summaryr � apiclientr �internal.nginx_utilsr � getLoggerr r r r r r �<module>
s