a
ABMf�$ � @ 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,