Edit File: lve_stats2_reseller_lib.cpython-37.pyc
B 5/9e, � @ s� d dl mZ d dlZd dlZddlmZ ddlmZmZmZm Z ddl m Z m Z m Z mZmZmZmZ d dlmZ ed Zd d d d �Zd dd�Zd dd�Zddd�Zdd� Zd dd�Zdd� ZdS ) � )�absolute_importN� )�lve_stats2_lib)�StatsNotifierDir�LveStats2Exception� faults_dict�StatsNotifierConfig)�str_to_boolean� repack_dict�time_convertor_to_dict�is_positive_int�time_unit_to_letter�boolean_to_yes_no�min_num_notify_converter)� mod_makedirsz /resellers.d/�notifyReseller�notifyResellerOnCustomers�notifyCustomers)� NOTIFY_RESELLER_ON_TOTAL_FAULTS�#NOTIFY_RESELLER_ON_CUSTOMERS_FAULTS�NOTIFY_CUSTOMERS_ON_FAULTS�reseller�customer)�NOTIFY_MIN_FAULTS_RESELLER�NOTIFY_MIN_FAULTS_CUSTOMER)�NOTIFY_INTERVAL_RESELLER�NOTIFY_INTERVAL_CUSTOMER�rc C s t | |d�S )N)�mode)�open)� file_namer � r! �&py/clconfig/lve_stats2_reseller_lib.py�_open1 s r# c s� t �� �d�� � fdd�dD �} d� d � d d � d d d �� d � d d d � d d d d �� d d d � d d d d �d �| � d d�iS )z; Creates default StatsNotifier config for reseller �faultsNotificationc s i | ]}� d | |�qS )�faultsToIncluder! )�.0�key)� admin_configr! r" � <dictcomp>: s z3_create_default_reseller_config.<locals>.<dictcomp>)ZmemZiops�ioZnprocZconcurrentConnectionsZcpuZnotifyResellers� minimumNumberOfFaultsToNotifyZadmin�user)r r �notify� unitOfTime�period)r. r/ ZnotifyResellerCustomers)r r+ r r- r% r )r �get_notification�get)Z admin_faultsr! )r( r" � _create_default_reseller_config5 s r2 Fc C sB | dkrt d��tj�td| �}tj�|�s<|r8t� S i S y t|�}|�� }|� � W n@ t t fk r� } z t dt |� d|id���W dd}~X Y nX i }x>|D ]6}|�d�s�|�� dkr�q�|�d �\}} | �� ||< q�W t} t} t} t} t| |t�}t| |t�}t| |td d �}t| |t�}||d < ||d < ||d< d|iS )ak Retrieves lve-stats2 notifications parameters for resellers :param bool defaults: True if we want get defaults when config is not set, else False :param str reseller: reseller name, if we want notifications parameters for a reseller :return: dict. For example: {'faultsNotification': {'notifyResellerCustomers': True, 'notifyResellers': True, 'minimumNumberOfFaultsToNotify': {'admin': 0, 'user': 0}, 'notifyAdmin': True, 'notify': {'admin': {'unitOfTime': 'hours', 'period': 12}, 'user': {'unitOfTime': 'hours', 'period': 12} }, 'faultsToInclude': {'mem': True, 'iops': False, 'io': False, 'nproc': False, 'concurrentConnections': False, 'cpu': True}, 'notifyCustomers': False } } Nzreseller can`t be Nonez%s_StatsNotifier.cfgz %(lvestats_cfg)s read error: � lvestats_cfg)�message�context�#� �=r )�defaultr% r+ r- r$ )� ValueError�os�path�join�ResellerStatsNotifierDir�existsr2 r# � readlines�close�IOError�OSErrorr �str� startswith�strip�split�reseller_notify_dictr �reseller_number_notify_dict�reseller_notify_time_dictr r r r )r �defaults� config_path�fZconfig�eZ config_dict�liner' �valueZnotify_dZfaults_dZnumber_notify_dZ notify_time_dZfaults_notification_dictZfaults_include_dictZmin_num_notifyZ period_notifyr! r! r" r0 M s< r0 c s� t j�t�rt��dkr dS i �|dkr0td��t j�td| �}t j�|�s\��t � � n��t |d�� t j�t j� |��s�t t j� |�d� t j�|�s�t |d��� y&t |�}dd � |�� D �}|�� W nB ttfk �r } z td t|� d tid ���W dd}~X Y nX d d� dfdd� dfdd� dfdd� dfdd� dfdd� dfd�}|�dd� d d fd d� d d fd!d� d"fd#d� d$fd%d� d&fd'd(d)�� i }��fd*d+�� xv|�� D ]j\}} yDt| t��rވ | �||< n&| d � fd,d-�| d.d� D �� ||< W n tk �r �w�Y nX �q�W d/d � |D �} y�x�t|�D ]�\} } | �d0��sB| �� d1k�rh�qBy| �d2�\}} W n4 tk �r� td3�| d. | �d |id ���Y nX ||k�rBd4||| f | | < ||= �qBW W nB tt fk �r } z td5t|� d |id ���W dd}~X Y nX x(|�� D ] \}} | � d4|| f � �q*W y&t |d�}|� d6�| �� |�� W nB ttfk �r� } z td7t|� d |id ���W dd}~X Y nX t j!d8d9d:d;� dS )<a� Sets lve-stats2 notifications parameters :param parameters_dict: Parametres to set. For example: {u'notifyResellers': True, u'faultsToInclude': {u'mem': True, u'iops': False, u'io': False, u'nproc': False, u'concurrentConnections': False, u'cpu': True}, u'minimumNumberOfFaultsToNotify': {u'admin': 0, u'user': 0}, u'notifyAdmin': True, u'notify': {u'admin': {u'unitOfTime': u'hours', u'period': 12}, u'user': {u'unitOfTime': u'hours', u'period': 12} }, u'notifyResellerCustomers': True, u'notifyCustomers': False } :param str reseller: either reseller name or None r Nzreseller can`t be Nonez%s_StatsNotifier.cfg)r i� �wc S s g | ] }|� � �qS r! )rF )r& �lr! r! r" � <listcomp>� s z$set_notification.<locals>.<listcomp>z %(lvestats_cfg)s read error: r3 )r4 r5 c S s t | �S )N)r )�xr! r! r" �<lambda>� � z"set_notification.<locals>.<lambda>zfaultsToInclude/cpuc S s t | �S )N)r )rT r! r! r" rU � rV zfaultsToInclude/nprocc S s t | �S )N)r )rT r! r! r" rU � rV zfaultsToInclude/ioc S s t | �S )N)r )rT r! r! r" rU � rV zfaultsToInclude/memc S s t | �S )N)r )rT r! r! r" rU � rV z%faultsToInclude/concurrentConnectionsc S s t | �S )N)r )rT r! r! r" rU � rV zfaultsToInclude/iops)Z NOTIFY_CPUZ NOTIFY_NPROCZ NOTIFY_IOZ NOTIFY_MEMORYZ NOTIFY_EPZ NOTIFY_IOPSc S s dt | �t|�f S )Nz%d%s)r r )rT �yr! r! r" rU � rV znotify/reseller/periodznotify/reseller/unitOfTimec S s dt | �t|�f S )Nz%d%s)r r )rT rW r! r! r" rU � rV znotify/customer/periodznotify/customer/unitOfTimec S s t | �S )N)r )rT r! r! r" rU � rV r c S s t | �S )N)r )rT r! r! r" rU � rV r c S s t | �S )N)r )rT r! r! r" rU � rV r z&minimumNumberOfFaultsToNotify/resellerz&minimumNumberOfFaultsToNotify/customer)r r r r r r r c s` | � d�}y �}x|D ] }|| }qW W n2 tk rZ � d }x|D ] }|| }qFW Y nX |S )N�/r$ )rG �KeyError)r< � path_partsZpoint�part)�notifier_params�parameters_dictr! r" �get_val_by_path� s z)set_notification.<locals>.get_val_by_pathc 3 s | ]}� |�V qd S )Nr! )r& rT )r^ r! r" � <genexpr>� s z#set_notification.<locals>.<genexpr>r c S s g | ] }|� � �qS r! )rF )r& rR r! r! r" rS � s r6 r7 r8 z(%(lvestats_cfg)s format error : {0}: {1}z%s=%sz %(lvestats_cfg)s format error: � z %(lvestats_cfg)s write error: z+/sbin/service lvestats reload &>/dev/null &Tz /bin/bash)�shell� executable)"r; r< r? r �lenr: r= r> �updater2 r0 �dirnamer r# rA r@ rB rC r rD �items� isinstancerY � enumeraterE rF rG �format� IndexError�appendr � writelines� subprocess�run)r] r rL rM Z config_linesrN Zfaults_config_mapZ new_configr' �valZconfig_to_write�idxrO rP r! )r^ r\ r] r" �set_notification� s� * rq )r )F) Z __future__r r; rm r7 r r r r r Zclconfig_utilsr r r r r r r Zclcommon.utilsr r> rH rI rJ r# r2 r0 rq r! r! r! r" �<module> s$ $ :