Edit File: cl_summary_utils.cpython-37.pyc
B 5/9e�+ � @ s` d dl mZmZ d dlmZmZmZmZmZ d dl m Z d dl m Z m Z d dlmZmZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d d l m Z d d l m Z d d l!m"Z"m#Z# d d � Z$y d dl%m&Z& d dl'm(Z( W n e)k �r e$Z&e$Z(Y nX dZ*dZ+dZ,dZ-ej.�/e-d�Z0dZ1da2dZ3dZ4G dd� de5�Z6dd� Z7d d � Z8dLd d �Z9d!d"� Z:d#d$� Z;dMe<e=e=d&�d'd(�Z>e<e=dd)�d*d+�Z?e=d,�d-d.�Z@e=d,�d/d0�ZAe=dd1�d2d3�ZBe=dd1�d4d5�ZCd6d7� ZDd8d9� ZEd:d;� ZFd<d=� ZGd>d?� ZHd@dA� ZIdBdC� ZJdDdE� ZKdNeeeeee=f f d,�dFdG�ZLdHdI� ZMee d,�dJdK�ZNdS )O� )�absolute_import�division)�Optional�Dict�AnyStr�Union�List)�DEFAULT_JWT_ES_TOKEN_PATH)�process_is_running�get_cl_version)�decode� exceptionsN)�mkdir)�CL_CONFIG_FILE)�write_file_via_tempfile)�check_cagefs_initialized)� WhmApiRequest� WhmApiErrorc O s d S )N� )�a�kwr r � py/clsummary/cl_summary_utils.py�dummy_none_function s r )�"interpreter_versions_short_summary)�is_clpassenger_activeZ license_checkZcl_statistics_enabledZcl_statistics_rpm_enabledz/var/lvezsummary_status.jsonz%/var/run/cloudlinux_summary.send.lockz./usr/share/cloudlinux/cl_plus/clplus_sender.pyz#/opt/cloudlinux/venv/bin/virtualenvc @ s e Zd ZdZdZdZdZdS )� SummaryStatusz� Status of both, collecting and sending statistics If process still collects statistics -> IN_PROGRESS If statistics collected and sent correctly -> SUCCESS If any error during collecting or sending -> FAILED ZsuccessZ in_progressZfailedN)�__name__� __module__� __qualname__�__doc__�SUCCESS� IN_PROGRESS�FAILEDr r r r r 7 s r c C s t j�t�S )zS Checks is virtualenv installed :return: True/False - installed or not )�os�path�exists�ALT_PYTHON_VIRTUALENV_BINr r r r �is_virtualenv_installedC s r' c C s` t �dtjddddd�}y$t�| tj|�}t �d|�d }W n ttfk rT dS X |tjkS )a$ Check if file is locked by another process without acquiring lock. IMPORTANT! This function should NOT be used to check lock acquired by the same process that executes the is_locked() function. For example, when process executes fcntl.lockf(LOCK_FILE), and then the same process executes is_locked(LOCK_FILE), the is_locked(LOCK_FILE) call returns False. Use is_locked() function to check lock acquired by another process only. :param lock_file: file to check lock on :type lock_file: file object or descriptor Zhhllhhr F) �structZpack�fcntlZF_WRLCKZF_GETLKZunpack�OSError�IOErrorZF_UNLCK)Z lock_fileZ lock_dataZ lock_queryZ lock_statusr r r � is_lockedK s r, Fc C sL t td�a| stt�S yt�ttjtjB � W n tt fk rF dS X dS )z� Check if processes collecting stats are running already (with --send option in command line) :param acquire_lock: True = acquire lock when possible :type acquire_lock: bool :return bool: True = Processes are running �wTF) �open�_CL_STATISTICS_LOCK_PATH�_CL_STATISTICS_LOCK_FILEr, r) �lockfZLOCK_EXZLOCK_NBr* r+ )� acquire_lockr r r �is_sending_process_running` s r3 c C s" t jddd�} t| _| �t� | S )zW Retrieves ConfigParser object for /etc/sysconfig/cloudlinux file :return: NT)Z interpolation�strict)� configparserZ ConfigParser�strZ optionxform�readr )�configr r r � _get_configr s r9 c C sL y.t �� }| �|� |�� }t|tddd� W n ttfk rF dS X dS )z� write config to /etc/sysconfig/cloudlinux file :param config: configParser object to write return boolean: True|False i� � cloudlinux_)�prefixFT)�io�StringIO�write�getvaluer r r* r+ )r8 Z string_fp�contentr r r � _write_config~ s rA T)� parameter�default�returnc C s> t � }|}y|�t| �}W n tjtjtfk r8 Y nX |S )zi Retrieves parameter's value from /etc/sysconfig/cloudlinux file, _CL_STATISTICS_SECTION section )r9 Z getboolean�_CL_STATISTICS_SECTIONr5 ZNoSectionErrorZ NoOptionError� ValueError)rB rC r8 Zresr r r �_get_config_value� s rG )rB �valuerD c C s( t � }|�t| tt|��� t|� dS )zb Sets parameter's value to /etc/sysconfig/cloudlinux file, _CL_STATISTICS_SECTION section N)r9 �setrE r6 �intrA )rB rH r8 r r r �_set_config_value� s rK )rD c C s t t�S )z~ Retrieves statistic collection status from /etc/sysconfig/cloudlinux file :return: True/False - enabled/disabled )rG �#_CL_STATISTICS_COLLECT_STATE_OPTIONr r r r �is_statistic_enabled� s rM c C s t t�S )z� Retrieves rpm statistic collection status from /etc/sysconfig/cloudlinux file :return: True/False - enabled/disabled )rG �'_CL_STATISTICS_COLLECT_RPM_STATE_OPTIONr r r r �is_statistic_rpm_enabled� s rO )� is_enabledrD c C s t t| � dS )z� Set statistic collection status to /etc/sysconfig/cloudlinux file :param is_enabled: True/False - enabled/disabled N)rK rL )rP r r r � set_statistic_collection_enabled� s rQ c C s t t| � dS )z� Set rpm statistic collection status to /etc/sysconfig/cloudlinux file :param is_enabled: True/False - enabled/disabled N)rK rN )rP r r r �$set_statistic_rpm_collection_enabled� s rR c C sP y2t j�t�stt� t�| �}t|tddd� W n t t fk rJ Y nX dS )z� Writes statistics send status to file /var/lve/summary_status.json :param status_dict: status dictionary for write to file :return: i� r: )r; N) r# r$ r% �_CL_STATISTICS_DIR�mkdir_p�json�dumpsr � _CL_STATISTICS_SEND_STATUS_FILEr* r+ )� status_dictr@ r r r �$write_statistics_send_status_to_file� s rY c C sx d} yPt t��}|�� }W dQ R X t�|�} | d tjkrRt� sRtj| d< d| d< W n t t t t t fk rr Y nX | S )z� Retrieves statistics send status from file /var/lve/summary_status.json :return: Dictionary with last send status. None if any error N�resultzaCollecting statistics was failed. Error report has been sent to developers and will be fixed soon�reason)r. rW r7 rU �loadsr r! r3 r" r* r+ rF �AttributeError� TypeError)rX �fZ s_contentr r r �$get_statistics_send_status_from_file� s r` c C s dd� t | �D �S )z� Returns list of installed interpreters :param interpreter: str - name of interpreter :rtype: List of InterpreterSummary c S s g | ]}|j r|�qS r )Z installed)�.0�ir r r � <listcomp>� s z/installed_interpreters_list.<locals>.<listcomp>)r )Z interpreterr r r �installed_interpreters_list� s rd c C s t � o t� S )z� Checks that python selector is installed Installed if: - ea-apache24-mod-alt-passenger or alt-mod-passenger is installed - alt-python-virtualenv is installed :rtype: bool )r r' r r r r � is_python_selector_installed� s re c C s t � o t� S )z� Checks that ruby selector is installed Installed if: - ea-apache24-mod-alt-passenger or alt-mod-passenger is installed - alt-python-virtualenv is installed :rtype: bool )r r' r r r r �is_ruby_selector_installed� s rf c C s t � ottd��S )z� Checks that nodejs selector is installed Installed if: - ea-apache24-mod-alt-passenger or alt-mod-passenger is installed - At least one version is installed :rtype: bool Znodejs)r �boolrd r r r r � is_nodejs_selector_installed s rh c C s t t� �S )zn Checks that php selector is installed Installed if: - CageFS is initialized :rtype: bool )rg r r r r r �is_php_selector_installed s ri c C s< yt d��� } W n tk r$ g S X dd� | d D �}|S )z9 Gets packages with set lve limits via extension Zlistpkgsc S s( g | ] }d |kr|d dkr|d �qS )Z_PACKAGE_EXTENSIONSZlve�namer )ra �itemr r r rc & s z4get_packages_with_lve_extensions.<locals>.<listcomp>Zpkg)r Zcallr )rZ Zlve_extensions_packagesr r r � get_packages_with_lve_extensions s rl c C s~ y(t tdd��}|�� �� }W dQ R X W n ttfk r@ dS X y t|dgd| ddd�dd �}|S tjk rx dS X dS ) z� Gets (if any) fields cl_plus and client_id from jwt token :return: decoded jwt_token value, None if error jwt_token: result of the successful decoding �rb)�modeNZHS256TF)Z require_expZ verify_expZ verify_issZverify_signatureZ CloudLinux)Z algorithmsZoptionsZissuer) r. r r7 �stripr* r+ r r Z PyJWTError)Zcheck_expiration�fileZ file_contentZ jwt_tokenr r r � get_client_data_from_jwt_token+ s rq c C s, | sdS | � d�sdS tt�� �| � d�k S )z. Checks whether license ts is expired NZexp)�getrJ �time)Z token_datar r r � is_active_cloudlinux_licenseD s rt c C s4 yt td�} W n tk r& d} Y nX | r0dS dS )z� Retrieves data from status of cl_plus_sender service :return: status of service, Optional[AnyStr] - 'active' - 'inactive' FZactiveZinactive)r �CL_PLUS_SENDER_FILE_PATH�FileNotFoundError)rZ r r r �get_cl_plus_sender_statusO s rw )F)T)T)OZ __future__r r �typingr r r r r Zclcommon.lib.constsr Zclcommon.utilsr r Zjwtr r r5 r< r# rU r) r( rs Zclcommon.clcaptainr rT Z cldetectlibr Zsecureior Z!clconfig.cagefs_statistics_configr Zclcommon.lib.whmapi_libr r r Zclselect.clselectctlr Z clselector.clpassenger_detectlibr � ImportErrorrE rL rN rS r$ �joinrW r/ r0 ru r& �objectr r' r, r3 r9 rA r6 rg rG rK rM rO rQ rR rY r` rd re rf rh ri rl rq rt rw r r r r �<module> sn $