Edit File: lve_stats2_lib.cpython-37.opt-1.pyc
B 5/9eZ* � @ s� d dl mZ d dlZd dlZddlmZmZmZmZm Z m Z m Z d dl m Z d dlmZ G dd� de �Zd Zed Zd d d dd�Zddddddd�Zddddd�Zd d d �Zd d d �Zd'd!d"�Zd(d#d$�Zd)d%d&�ZdS )*� )�absolute_importN� )�str_to_boolean� repack_dict�time_convertor_to_dict�is_positive_int�time_unit_to_letter�boolean_to_yes_no�min_num_notify_converter)�FormattedException)� mod_makedirsc @ s e Zd ZdS )�LveStats2ExceptionN)�__name__� __module__� __qualname__� r r � py/clconfig/lve_stats2_lib.pyr s r z /etc/sysconfig/lvestats.configz/StatsNotifier.cfg� notifyAdmin�notifyResellers�notifyCustomers�notifyResellerCustomers)� NOTIFY_ADMIN�NOTIFY_RESELLER�NOTIFY_CUSTOMER� NOTIFY_INCLUDE_RESELLER_CUSTOMERZmemZiopsZnproc�ioZconcurrentConnectionsZcpu)� NOTIFY_MEMORY� NOTIFY_IOPS� NOTIFY_NPROC� NOTIFY_IO� NOTIFY_EP� NOTIFY_CPUZnotifyFromEmailZreportAdminMailZ notifySubjectZ notifyCharset)�NOTIFY_FROM_EMAIL�REPORT_ADMIN_EMAIL�NOTIFY_FROM_SUBJECT�NOTIFY_CHARSET_EMAIL�userZadmin)�NOTIFY_MIN_FAULTS_USER�NOTIFY_MIN_FAULTS_ADMIN)�NOTIFY_INTERVAL_ADMIN�NOTIFY_INTERVAL_USER�rc C s t | |d�S )N)�mode)�open)� file_namer, r r r �_open9 s r/ c C s2 t }tj�|�si S y t|�}|�� }|�� W n@ ttfk rp } 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| |tdd �}t| |t�}tt|t dd �}|| d < || d < || d < || d < d| iS )a Retrieves lve-stats2 notifications parameters :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 } } z %(lvestats_cfg)s read error: � lvestats_cfg)�message�contextN�#� �=r )�defaultZfaultsToIncludeZ minimumNumberOfFaultsToNotifyZnotifyZemail�faultsNotification)�StatsNotifierConfig�os�path�existsr/ � readlines�close�IOError�OSErrorr �str� startswith�strip�split� notify_dict� faults_dict�number_notify_dict�notify_time_dictr r r r �email_notify_dict)�reseller� config_path�fZconfig�eZ config_dict�line�key�valueZnotify_dZfaults_dZnumber_notify_dZ notify_time_dZfaults_notification_dictZfaults_include_dictZmin_num_notifyZ period_notifyZemail_settingsr r r �get_notification= s< rP c s� t j�t�rt��dkr dS i �t}t� �t j�t j�|��sPtt j�|�d� t j�|�sjt|d�� � y&t|�}dd� |� � D �}|� � W n@ t t fk r� } z t dt|� dtid ���W dd}~X Y nX d d � d fd d � dfdd � dfdd � dfdd � dfdd � dfd�}|�dd � ddfdd � d d fd d � d fd d � d!fd"d � d#fd$d � d%fd&d'd(�� |�d)d*d+d,d-�� i }��fd.d/�� xv|�� D ]j\}} yDt| t��r�� | �||< n&| d � fd0d1�| d2d� D �� ||< W n tk �r� �w�Y nX �q�W d3d� |D �} y�x�t|�D ]�\} } | �d4��s | �� d5k�rB�q y| �d6�\}} W n4 tk �r� t d7�| d2 | �d|id ���Y nX ||k�r d8||| f | | < ||= �q W W nB ttfk �r� } z t d9t|� d|id ���W dd}~X Y nX x(|�� D ] \}} | �d8|| f � �qW d:d;� }tt || ��} y&t |d�}|� d<� | �� |� � W nB t t fk �r� } z t d=t|� d|id ���W dd}~X Y nX t j!d>d?d@dA� dS )Ba� 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 Ni� �wc S s g | ] }|� � �qS r )rB )�.0�lr r r � <listcomp>� s z$set_notification.<locals>.<listcomp>z %(lvestats_cfg)s read error: r0 )r1 r2 c S s t | �S )N)r )�xr r r �<lambda>� � z"set_notification.<locals>.<lambda>zfaultsToInclude/cpuc S s t | �S )N)r )rU r r r rV � rW zfaultsToInclude/nprocc S s t | �S )N)r )rU r r r rV � rW zfaultsToInclude/ioc S s t | �S )N)r )rU r r r rV � rW zfaultsToInclude/memc S s t | �S )N)r )rU r r r rV � rW z%faultsToInclude/concurrentConnectionsc S s t | �S )N)r )rU r r r rV � rW zfaultsToInclude/iops)r! r r r r r c S s dt | �t|�f S )Nz%d%s)r r )rU �yr r r rV � rW znotify/admin/periodznotify/admin/unitOfTimec S s dt | �t|�f S )Nz%d%s)r r )rU rX r r r rV � rW znotify/user/periodznotify/user/unitOfTimec S s t | �S )N)r )rU r r r rV � rW r c S s t | �S )N)r )rU r r r rV � rW r c S s t | �S )N)r )rU r r r rV � rW r c S s t | �S )N)r )rU r r r rV � rW r z#minimumNumberOfFaultsToNotify/adminz"minimumNumberOfFaultsToNotify/user)r) r* r r r r r( r' zemail/notifyFromEmailzemail/notifySubjectzemail/reportAdminMailzemail/notifyCharset)r" r$ r# r% c s` | � d�}y �}x|D ] }|| }qW W n2 tk rZ � d }x|D ] }|| }qFW Y nX |S )N�/r7 )rC �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 )rR rU )r_ r r � <genexpr>� s z#set_notification.<locals>.<genexpr>r c S s g | ] }|� � �qS r )rB )rR rS r r r rT � s r3 r4 r5 z(%(lvestats_cfg)s format error : {0}: {1}z%s=%sz %(lvestats_cfg)s format error: c S s0 | � d�s| �� dkrdS | �d�\}}|dkS )Nr3 r4 Tr5 �None)rA rB rC )�_line�_Z_valuer r r �filter_config_lines� s z-set_notification.<locals>.filter_config_lines� z %(lvestats_cfg)s write error: z+/sbin/service lvestats reload &>/dev/null &Tz /bin/bash)�shell� executable)"r9 r: r; r8 �lenrP �dirnamer r/ r= r< r> r? r r@ �update�items� isinstancerZ � enumeraterA rB rC � ValueError�format� IndexError�append�list�filterr- � writelines�join� subprocess�run)r^ rI rJ rK Z config_linesrL Zfaults_config_mapZ new_configrN �valZconfig_to_write�idxrM rO rd r )r_ r] r^ r �set_notificationv s� * rz )r+ )N)N)Z __future__r r9 rv Zclconfig_utilsr r r r r r r Zclcommon.clexceptionr Zclcommon.utilsr r ZStatsNotifierDirr8 rD rE rH rF rG r/ rP rz r r r r �<module> s: $ 9