EVOLUTION-NINJA
Edit File: stat_utils.cpython-36.pyc
3 ���i�$ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl Z G dd� de �Z dd� Z dd � Zd d � Zd d � Zd%dd�Zd&dd�Zd'dd�Zd(dd�Zdd� Zdd � Zd d � Zd)d d!�Zd*d#d$�ZdS )+� N)�StringIO)�quotec @ s e Zd ZdS )�StatUtilsExceptionN)�__name__� __module__� __qualname__� r r �+/usr/share/lve/modlscapi/user/stat_utils.pyr s r c K s� dj dd� |j� D ��}tjdj| |d��}ytjdj |��}W n, tk rl tdj| |dj |����Y nX y|d S tk r� td j|���Y nX d S ) z� Perform cPanel WHM API console request and return data from result :param cmd: whm api command :return: data dict from result � c S s g | ]\}}t d j||���qS )z{0}={1})r �format)�.0�k�vr r r � <listcomp>$ s z!cpanel_whmapi.<locals>.<listcomp>z*/usr/sbin/whmapi1 {cmd} {kw} --output json)�cmd�kw� z>Failed to get JSON from this API request: {0} {1}; output: {2}�dataz,Failed to get data from this API result: {0}N) �join�items� exec_commandr �json�loads� ValueErrorr �KeyError)r �kwargs� joined_kwargs�resultZ dict_resultr r r � cpanel_whmapi s r c K sn dj dd� |j� D ��}tjdj| |d��}ytjdj |��S tk rh tdj| |dj |����Y nX dS ) z� Perform Plesk php_handler utility console request and return result :param cmd: php_handler command :return: dict result r c S s g | ]\}}d j |t|���qS )z-{0} {1})r r )r r r r r r r = s z)plesk_bin_php_handler.<locals>.<listcomp>z6/usr/local/psa/bin/php_handler --{cmd} {kw} -json true)r r r zJFailed to get JSON from this API request: php_handler {0} {1}; output: {2}N)r r r r r r r r )r r r r r r r �plesk_bin_php_handler6 s r c C s� d} d}t j| �}tj|�}yJxD|D ]<}t|j|d�d�� }tdd� |j� D ��||< W dQ R X q$W W n0 tt fk r� t dj t j � jd ����Y nX |S ) ze Get domains per user :return: dict( user: list of domains ) z!/usr/local/directadmin/data/usersz5/usr/local/directadmin/data/users/{user}/domains.list)�user�rc S s g | ] }|j � �qS r )�strip)r �lr r r r X s z"get_da_domains.<locals>.<listcomp>Nr � )�os�listdir�dict�fromkeys�openr �set� readlines�OSError�IOErrorr r � traceback� format_exc�split)Z da_users_pathZda_domains_pathZda_users�domainsr Zdomains_list_filer r r �get_da_domainsI s , r2 c C s� d} y@t | �\}}|j|d�}|j|d�}|j|d�}|j|d�}W n4 tjk rx tdjdjtj� j d�����Y nX ||d kr�d n|d �||d kr�d n|d �d �S ) z� Get php settings from options.conf :return: dict( first php setting: {version, mode}, second php setting: {version, mode}, ) z//usr/local/directadmin/custombuild/options.confZ php1_releaseZ php2_releaseZ php1_modeZ php2_modezNo option found: {0}r r$ �lsphp�lsapi)�version�handler)� � ) �read_da_config�get� configparser� NoOptionErrorr r r r. r/ r0 )Z options_path� config_parser�global_section�php1_ver�php2_ver� php1_handler� php2_handlerr r r �get_da_php_options^ s $rC � dummy_sectionc C s� yFt | �� }tdj|d�|j� �}W dQ R X tjdd�}|j|� W n0 ttfk rv t dj t j � j d����Y nX ||fS )z� Read DA config file with ConfigParser. Need to add dummy section for success :param conf_file: config file name :param append_section_name: name of section to place in the beginning of file :return: RawConfigParser instance z[{s}] )�sNF)�strictr r$ )r) r r �readr; �RawConfigParser� read_filer, r- r r r. r/ r0 )Z conf_fileZappend_section_name�fZ file_contentr= r r r r9 } s " r9 �phpc C s6 dj |�}y|t| jd�� S tk r0 | S X dS )z� Convert simple php versions to pretty format :param php_ver: {major}.{minor} version :param pre: desired key start :return: alt-php{major}{minor} or desired `pre`{major}{minor} z{0}%s%s�.N)r �tupler0 � Exception)Zphp_ver�pre�templater r r �pretty_version_keys� s rQ �/usr/sbin/httpdc C s6 t j dj| d��}ytdd� |D ��S t� S dS )a+ Get `httpd -t -D DUMP_RUN_LSAPI` info For httpd24 this default path is `/opt/rh/httpd24/root/usr/sbin/httpd`, generated in make_from_templates.sh script :param ctl_path: path to httpd (also apachectl may be used) :return: dict( lsapi_option: value ) z{ctl} -t -D DUMP_RUN_LSAPI)�ctlc S s g | ]}|j � jd ��qS )r )�lowerr0 )r r# r r r r � s z dump_lsapi.<locals>.<listcomp>N)r r r' )�ctl_pathZapache_conf_datar r r � dump_lsapi� s rV c C s6 t j dj| d��}ytdd� |D ��S t� S dS )a Get `httpd -M` For httpd24 this default path is `/opt/rh/httpd24/root/usr/sbin/httpd`, generated in make_from_templates.sh script :param ctl_path: path to httpd (also apachectl may be used) :return: dict( apache_module: value ) z{ctl} -M)rS c S s g | ]}|j � jd ��qS )r )rT r0 )r r# r r r r � s z'dump_loaded_modules.<locals>.<listcomp>N)r r r' )rU �apache_modulesr r r �dump_loaded_modules� s rX c C s t jdk} dj| rdndd�S )zR Retrieve path to liblsapi, depends on arch :return: path to liblsapi r8 � z/usr/lib{a}/liblscapi.soZ64r )�al )�sys�maxsizer )Z is_64bitsr r r � liblsapi_path� s r] c C sT y:dj tjd| d ��jd�\}}dj||jd�d d�S tk rN d S X d S ) zr Get version-release from rpm -q `pkg` :param pkg: package name to query :return: version-release r z /bin/rpm -q z --qf %{v}-%{r}�-z {ver}-{rel}rL r )�ver�relN)r r r0 r r )Zpkgr_ r` r r r � rpm_query� s "ra c sH y(� fdd�t | �D �d j� �d j� S tttfk rB dS X dS )z� Filter strings by given template Also split string upon given template :param fname: path to file :param template: template to find in string :return: first template occurrence splitted by template c s g | ]}� |kr|�qS r r )r r# )rP r r r � s z!query_strings.<locals>.<listcomp>r r7 N)�stringsr0 r" � IndexErrorr- r, )�fnamerP r )rP r � query_strings� s (re � c c sn t | dd��X}d}x:|j� D ].}|tjkr4||7 }q t|�|krF|V d}q W t|�|kr`|V W dQ R X dS )z� Strings utility analog. Finds printable strings in executable :param fname: path to file :param n: minimum string length :return: generator, yeilds string �ignore)�errorsr N)r) rG �stringZ printable�len)rd �nrJ r �cr r r rb � s rb Tc C sr t � }x.| j� D ]"\}}t dd� |j� D ��||< qW y|rD|d S |S tk rl t jdd� |D �d�S X dS )a� Count domains :param handler_struct: handler: version: set_of_domains structure :param default_keys: sequence of keys to add as default if no `lsapi` found :param only_lsapi: return only lsapi statistics :return: statistics - number of lsapi domains per version if only_lsapi=True number of lsapi domains per version per handler otherwise c s s | ]\}}|t |�fV qd S )N)rj )r r r r r r � <genexpr> s z count_domains.<locals>.<genexpr>r4 c S s g | ]}|d kr|�qS )�nor )r �xr r r r s z!count_domains.<locals>.<listcomp>r N)r' r r r( )Zhandler_structZ default_keysZ only_lsapi� result_stat�hr r r r � count_domains� s rr )rD )rK )rR )rR )rf )T)ri r[ r% r. r r; �ior Zpipesr r rN r r r r2 rC r9 rQ rV rX r] ra re rb rr r r r r �<module> s,