File: //usr/share/lve/modlscapi/user/__pycache__/stat_utils.cpython-36.pyc
3
��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� 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,