Current File : //usr/share/lve/modlscapi/user/stat_utils.pyo
�
� cc @ s d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d e
f d � � YZ d � Z d � Z
d � Z d � Z d
d � Z d d
� Z d d � Z d d � Z d � Z d � Z d � Z d d � Z e d � Z d S( i����N( t StringIO( t quotet ModLsapiStatExceptionc B s e Z RS( ( t __name__t
__module__( ( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pyR s c K s� d j g | j � D]$ \ } } t d j | | � � ^ q � } t j d j d | d | � � } y t j d j | � � } Wn5 t k
r� t d j | | d j | � � � � n Xy | d SWn& t k
r� t d j | � � � n Xd
S( s�
Perform cPanel WHM API console request and return data from result
:param cmd: whm api command
:return: data dict from result
t s {0}={1}s* /usr/sbin/whmapi1 {cmd} {kw} --output jsont cmdt kwt s> Failed to get JSON from this API request: {0} {1}; output: {2}t datas, Failed to get data from this API result: {0}N(
t joint iteritemsR t formatt exec_commandt jsont loadst
ValueErrorR t KeyError( R t kwargst kt vt
joined_kwargst resultt dict_result( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pyt
cpanel_whmapi s @!
c K s� d j g | j � D]$ \ } } d j | t | � � ^ q � } t j d j d | d | � � } y t j d j | � � SWn5 t k
r� t d j | | d j | � � � � n Xd S( s�
Perform Plesk php_handler utility console request and return result
:param cmd: php_handler command
:return: dict result
R s -{0} {1}s6 /usr/local/psa/bin/php_handler --{cmd} {kw} -json trueR R R sJ Failed 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 ( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pyt plesk_bin_php_handler( s @
c C s� d } d } t j | � } t j | � } yh xa | D]Y } t | j d | � d � �6 } t g | j � D] } | j � ^ qh � | | <Wd QXq4 WWn; t t
f k
r� t d j t
j � j d � � � � n X| S( se
Get domains per user
:return: dict(
user: list of domains
)
s! /usr/local/directadmin/data/userss5 /usr/local/directadmin/data/users/{user}/domains.listt usert rbNR s
( t ost listdirt dictt fromkeyst openR t sett readlinest stript OSErrort IOErrorR R
t tracebackt
format_exct split( t
da_users_patht da_domains_patht da_userst domainsR t domains_list_filet l( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pyt get_da_domains; s
=(c C s� d } y^ t | � \ } } | j | d � } | j | d � } | j | d � } | j | d � } WnA t j k
r� t d j d j t j � j d � � � � � n Xi i | d 6| d
k r� d n | d 6d
6i | d 6| d
k r� d n | d 6d 6S( s�
Get php settings from options.conf
:return: dict(
first php setting: {version, mode},
second php setting: {version, mode},
)
s/ /usr/local/directadmin/custombuild/options.conft php1_releaset php2_releaset php1_modet php2_modes No option found: {0}R s
t versiont lsphpt lsapit handleri i (
t read_da_configt gett ConfigParsert
NoOptionErrorR R R
R& R' R( ( t options_patht
config_parsert global_sectiont php1_vert php2_vert php1_handlert php2_handler( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pyt get_da_php_optionsP s 1 t
dummy_sectionc C s� yT t | � �) } t d j d | � | j � � } Wd QXt j � } | j | � Wn; t t f k
r� t d j
t j � j
d � � � � n X| | f S( s�
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
s [{s}]
t sNR s
( R R R t readR: t RawConfigParsert readfpR$ R% R R
R& R' R( ( t conf_filet append_section_namet ft file_contentR= ( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pyR8 o s ( (t phpc C sC d j | � } y | t | j d � � SWn t k
r> | SXd S( s�
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}
s {0}%s%st .N( R t tupleR( t Exception( t php_vert pret template( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pyt pretty_version_keys� s
s /usr/sbin/httpdc C s] t j d j d | � � } y0 t g | D] } | j � j d � ^ q( � SWn t � SXd S( s+
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
)
s {ctl} -t -D DUMP_RUN_LSAPIt ctlR N( R
R R t lowerR( ( t ctl_patht apache_conf_dataR. ( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pyt
dump_lsapi� s
0c C s] t j d j d | � � } y0 t g | D] } | j � j d � ^ q( � SWn t � SXd S( s
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
)
s {ctl} -MRU R N( R
R R RV R( ( RW t apache_modulesR. ( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pyt dump_loaded_modules� s
0c C s+ t j d k } d j d | r$ d n d � S( sR
Retrieve path to liblsapi, depends on arch
:return: path to liblsapi
i i s /usr/lib{a}/liblscapi.sot at 64R I ( t syst maxsizeR ( t is_64bits( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pyt
liblsapi_path� s c C so yV d j t j d | d � � j d � \ } } d j d | d | j d � d � SWn t k
rj d
SXd
S( sr
Get version-release from rpm -q `pkg`
:param pkg: package name to query
:return: version-release
R s /bin/rpm -q s --qf %{v}-%{r}t -s {ver}-{rel}t vert relRN i N( R
R
R( R R t None( t pkgRc Rd ( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pyt rpm_query� s
/'
c C sf yD g t | � D] } | | k r | ^ q d j | � d j � SWn t t t f k
ra d SXd S( s�
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
i i N( t stringsR( R# t
IndexErrorR% R$ Re ( t fnameRS R. ( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pyt
query_strings� s Di c c s� t | d � �} } d } xS | j � D]E } | t j k rJ | | 7} q% n t | � | k rd | Vn d } q% Wt | � | k r� | Vn Wd QXd S( s�
Strings utility analog.
Finds printable strings in executable
:param fname: path to file
:param n: minimum string length
:return: generator, yeilds string
R R N( R RF t stringt printablet len( Rj t nRK R t c( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pyRh � s
c C s� t � } x: | j � D], \ } } t d � | j � D� � | | <q Wy | rW | d S| SWn t k
r| t j | d � SXd S( s�
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 | � f Vq d S( N( Rn ( t .0R R ( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pys <genexpr>� s R6 i N( R R R R ( t handler_structt default_keyst
only_lsapit result_statt hR ( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pyt
count_domains� s $
( Rl R^ R R& R R: R
R t pipesR RP R R R R/ RC R8 RT RY R[ Ra Rg Rk Rh t TrueRw ( ( ( s+ /usr/share/lve/modlscapi/user/stat_utils.pyt <module> s,
Mini Shell