관리-도구
편집 파일: stat_utils.cpython-39.pyc
a �q?h�$ � @ 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� d� dd� |�� D ��}t�dj| |d��}zt�d� |��}W n* tyj td�| |d� |����Y n0 z |d W S ty� td �|���Y n0 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 �||���qS )z{0}={1})r �format��.0�k�vr r r � <listcomp>$ � 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 d� dd� |�� D ��}t�dj| |d��}zt�d� |��W S tyh td�| |d� |����Y n0 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 � |t|���qS )z-{0} {1})r r r r r r r = r z)plesk_bin_php_handler.<locals>.<listcomp>z6/usr/local/psa/bin/php_handler --{cmd} {kw} -json truer 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 �| �}t�|�}zZ|D ]P}t|j|d�d��*}tdd� |�� D ��||< W d � q"1 sh0 Y q"W n. tt fy� t d� t � � �d ����Y n0 |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 | ] }|� � �qS r )�strip�r �lr r r r X r 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 > r6 c C s� d} z@t | �\}}|�|d�}|�|d�}|�|d�}|�|d�}W n2 tjyv td�d�t�� � d�����Y n0 ||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 r2 r3 r4 )Z options_path� config_parser�global_section�php1_ver�php2_ver� php1_handler� php2_handlerr r r �get_da_php_options^ s $���rG � dummy_sectionc C s� zZt | ��(}tdj|d�|�� �}W d � n1 s80 Y tjdd�}|�|� W n. ttfy� t d� t � � � d����Y n0 ||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_filer0 r1 r r r2 r3 r4 )Z conf_fileZappend_section_name�fZ file_contentrA r r r r= } s 6 r= �phpc C s: d� |�}z|t| �d�� W S ty4 | Y S 0 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 �tupler4 � Exception)Zphp_ver�pre�templater r r �pretty_version_keys� s rU �/usr/sbin/httpdc C s> t � dj| d��}ztdd� |D ��W S t� Y S 0 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�Zctlc S s g | ]}|� � �d ��qS �r ��lowerr4 r& r r r r � r z dump_lsapi.<locals>.<listcomp>N�r r r+ )�ctl_pathZapache_conf_datar r r � dump_lsapi� s r] c C s> t � dj| d��}ztdd� |D ��W S t� Y S 0 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} -MrW c S s g | ]}|� � �d ��qS rX rY r& r r r r � r z'dump_loaded_modules.<locals>.<listcomp>Nr[ )r\ �apache_modulesr r r �dump_loaded_modules� s r_ c C s t jdk} dj| rdndd�S )zR Retrieve path to liblsapi, depends on arch :return: path to liblsapi l z/usr/lib{a}/liblscapi.soZ64r )�a)�sys�maxsizer )Z is_64bitsr r r � liblsapi_path� s rc c C sV z<d� t�d| d ���d�\}}dj||�d�d d�W S tyP Y d S 0 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}rP r )�ver�relN)r r r4 r r )Zpkgre rf r r r � rpm_query� s " rg c sJ z*� fdd�t | �D �d �� �d �� W S tttfyD Y dS 0 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 | ]}� |v r|�qS r r r& �rT r r r � r z!query_strings.<locals>.<listcomp>r r; N)�stringsr4 r% � IndexErrorr1 r0 )�fnamerT r rh r � query_strings� s *rl � c c s~ t | dd��^}d}|�� D ].}|tjv r2||7 }qt|�|krD|V d}qt|�|kr\|V W d � n1 sp0 Y 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- rK �stringZ printable�len)rk �nrN r �cr r r ri � s ri Tc C sr t � }| �� D ]"\}}t dd� |�� D ��||< qz|r@|d n|W S tyl t �dd� |D �d� Y S 0 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)rq r r r r � <genexpr> r z count_domains.<locals>.<genexpr>r8 c S s g | ]}|d kr|�qS )�nor )r �xr r r r r 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 ry )rH )rO )rV )rV )rm )T)rp ra r) r2 r r? �ior Zpipesr r rR r r! r" r6 rG r= rU r] r_ rc rg rl ri ry r r r r �<module> s,