Current File : //opt/alt/python37/lib/python3.7/site-packages/wmt/common/__pycache__/utils.cpython-37.pyc
B
��wa6 � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl
m Z d dl m
Z
mZ dd� Zdd � Ze jeed
�d d ��Zd
d� Zdd� Zdd� ZdS )� N)�cpapi)�get_rhn_systemid_value)�parse)�CLICKHOUSE_WMT_ENDPOINT�UNKNOWN_RHN_IDc C s� t � } tjtjkr*dd� tjdd�D �}nt�� }|s:g S tjdkrPt� tjj_ x2|D ]*}x$t�
|�D ]\}}| � t |�� qfW qVW | S )Nc S s g | ] }|d �qS )r � )�.0Z_cpinfor r �A/opt/alt/python37/lib/python3.7/site-packages/wmt/common/utils.py�
<listcomp> s z get_domains.<locals>.<listcomp>)Zcplogin)ZkeylsZcPanel)
�setr ZCP_NAMEZ
PLESK_NAMEZcpinfoZcpusers�dictZpluginsZcpanelZ_user_to_domains_map_cpanelZ userdomains�addr )ZdomainsZusers�userZdomain�_r r r � get_domains s
r c C sT t �| �}|�t j� yt �d�}W n tk
r8 Y nX t �d�|_|�|� |S )Nz/var/log/cl_wmt.logz*[%(levelname)s | %(asctime)s]: %(message)s) �loggingZ getLoggerZsetLevel�DEBUGZ FileHandler�IOErrorZ FormatterZ formatterZ
addHandler)Z logger_nameZ
app_loggerZfhr r r � setup_logger. s
r )�file�pidc
c s� t j�| �sdnd}t| |�}y t�|�� tjtjB � W n t k
rZ t d| ��Y nX z&|�
|� |� � t � |� d V W d t�|�� tj
� |�� t �| � X d S )Nzw+zr+zProcess %s already running!)�os�path�exists�open�fcntlZflock�filenoZLOCK_EXZLOCK_NB�OSError�write�flush�fsyncZLOCK_UN�close�remove)r r �mode�fr r r �save_pid_and_lock; s
r% c C s2 i }x(| D ] }||kr
| | || f||< q
W |S )Nr )Zd1Zd2�result�keyr r r � intersectN s
r( c C sz t �d�}xjtd�D ]Z}y t j| |dd�}|�� W n6 t jk
rj } z|}t�d| � W d d }~X Y qX P qW |�d S )Nz(Error while sending report to ClickHouse� �< )ZjsonZtimeout� )�requestsZRequestException�rangeZpostZraise_for_status�timeZsleep)Zurl�report�error�iZresponse�errr r r � send_reportV s
r3 c C s8 t d�p
t}|�dd�| d< | �d�}tt| |�� d S )NZ system_idzID-� Z server_idZsummary_report)r r �replace�popr3 r )r/ Z
systemd_idZsummaryr r r �send_report_to_clickhousee s
r7 )�
contextlibr r r r. r, Zclcommonr Zclcommon.utilsr Zwmt.common.url_parserr Zwmt.common.constr r r r �contextmanager�strr% r( r3 r7 r r r r �<module>
s
Mini Shell