Edit File: db_governor_lib.cpython-37.pyc
B 5/9e�< � @ s d dl mZ d dlZd dlZd dlmZ d dlmZ ddlm Z m Z m Z m Z d dl mZ d dlmZ G d d � d e�Zd d d d d dddd�d d dd�d dd�d dd�d� ZdZdd� Zd.dd�Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd/d&d'�Zd(d)� Z d*d+� Z d,d-� Z dS )0� )�absolute_importN)�etree)�Optional� )�str_to_boolean�time_convertor_to_dict�boolean_to_on_off�time_unit_to_letter)�FormattedException)� MySQLGovernorc @ s e Zd ZdS )�DBGovernorExceptionN)�__name__� __module__� __qualname__� r r � py/clconfig/db_governor_lib.pyr s r ZoffZon� � �limitZ60s)�mode�unlimit)�kill�logPath�timeoutZLONG)r �formatZERROR) �modeOfOperation�gatherDataForDetailedStatsZsave_uid�logRestrictedUsersQueries� scriptPath�userMaxConnections� restrictType� slowQueries� restrictLog�errorLogz!/etc/container/mysql-governor.xmlc C s t | �S )N)�open)� file_namer r r �_open9 s r& c C sH y*dt |� d t |� }| �|�d }W n ttfk rB |S X |S )a Gets data from specified node and attribute :param x: Parent node object :param node: Node name to extract data :param attrib: Attribute name to extract data :param default: Default value to return if specified node/attribute not found :return: Value z//z/@r )�str�xpath�AttributeError� IndexError)�x�nodeZattrib�default�path�datar r r �get_data_from_xml= s r0 c C s� t | ��� } | �d�r,t| dd� �d }np| �d�rPt| dd� �d d }nL| �d�rxt| dd� �d d d }n$| �d�r�t| dd� �}nt| �}|S ) z~ Convert value in minutes, hours, days to seconds :param value: value of slow limit :return: value in seconds �mN����< �h�d� �s)r' �lower�endswith�int)�value�resultr r r � convert_time_value_to_secondsN s r= c C s> y | � d�d }t|�}t|�S tttfk r8 |S X dS )z� Gets slow queries timeout :param x: Root node :param default: Default value to return if timeout not found :return: Timeout z$default/limit[@name='slow']/@currentr N)r( r= r: r) r* � ValueError)r+ r- r/ r r r �get_slow_queries_timeoutc s r? c C s� d}| � |�}t|�dkrt| � d�}t|�dkrDt�d�}| �|� | � d�d }t�d�}|�|� |�dd� n|d }|�dt|�� dS ) z{ Sets slow queries timeout :param node: Root node :param slow_timeout: new timeout value :return: None zdefault/limit[@name='slow']r r- r �nameZslowZcurrentN)r( �lenr �Element�append�setr' )r, Z slow_timeoutr. �tag_listZ node_list�new_nodeZ default_nodeZ tag_to_setr r r �set_slow_queries_timeoutr s rG c C sX dt |� }| �|�}t|�dkr@t�|�}| �|� | �|�}|d }|�||� dS )z� Sets value to specified node/attribute :param x: Parent node :param node_name: Node name :param attribute_name: Attribute name :param value: Value to set :return: None z//r N)r' r( rA r rB rC rD )r+ Z node_nameZattribute_namer; r. rE rF �tagr r r �set_data_to_xml� s rI c C s: t | ||�}yt|�S ttd||f ��Y nX dS )z� Get time from specified tag/attribute as seconds :param x: XML object :param tag_name: XML tag name :param attr_name: XML object name :return: Number of seconds. Throws DBGovernorException if any error z% has invalid value in governer/%s/@%sN)r0 r r � DbGovConfig)r+ Ztag_nameZ attr_namer; r r r �_get_timedict_from_xml� s rK c C sF t j�t�sdS | dkr0tt�}t�|��� }n| }t|ddt d �S )z� Get MySQL Governor mode. We return None if governor config isn't exists All exceptions will be catched in top level functions :param parsed_xml_tree: parsed governor config N�lve�user ) �osr. �existsrJ r& r �parse�getrootr0 � DEFAULTS_DICT)�parsed_xml_tree�fr+ r r r �get_gov_mode_operation� s rU c C s �y�t j�t�si S tt�} t�| ��� }t� }t |d�|d< t t |ddt d ��|d< t t |ddt d ��|d< t |d d t d �|d < t t |d d t d ��|d < t|d d�t|d d�t|d d�t|d d�t|d d�d�|d< t |ddt d d �tt |ddt d d ��d�|d< t t |ddt d d ��t |dd t d d �t|t d d �d �|d< t |d d t d d �t |d d!t d d! �d"�|d < t |d d#t d$ d �t |d dt d$ d! �d%�|d$< |�t� � d&|iS tttjtjtjtfk �r } zttd' t|� ��W d(d(}~X Y nX d(S ))zL Get MySQL Governer config :return: Dictionary with config data )rS r � statisticr r � logqueriesrM r �restrict�scriptr �user_max_connectionsr �level1�level2�level3�level4r )r[ r\ r] r^ r ZrestrictedTimePeriods� restrict_moder r )r r � slow_queries�runr! r �logr )r r r r" r )r r �filer# )r �levelZmySQLGovSettingsz read/parse error: N)rN r. rO rJ r& r rP rQ �dictrU r r0 rR r: rK r r? �update�get_db_gov_stat_info�OSError�IOError�XMLSyntaxError�XMLSchemaParseError�XMLSchemaErrorr r' )rT r+ Z result_dict�er r r �get_db_gov_config� sH rn c C s@ �y�t j�t�sdS tt�}t�|��� }|�� dddt fdddt fdd d t fd dd t fd ddt fdddt fdddt fdddt fddd t fdddt fdddt fdddt fg }xP|D ]H}y.|\}}}}t d | �} t ||||| �� W q� t k r� Y q�X q�W d d d d d!d"g} xh| D ]`}|\} } }}y6t d | �} tt d | ��}t |||d#| |f � W n t k �rr Y nX �qW y| d$ d% }t||� W n t k �r� Y nX t�|�}ttd&��}|�d'| � W dQ R X W nN tttjtjtjfk �r* } z td(t |� d)tid*���W dd}~X Y nX tjd+d,d-d.� dS )/a% Set MySQL governer config :param parameters_dict: Parameters to set dictionary. For example: { u'modeOfOperation': u'abusers', u'scriptPath': u'', u'gatherDataForDetailedStats': True, u'userMaxConnections': 30, u'logRestrictedUsersQueries': False, u'restrictLog': {u'logPath': u'/var/log/dbgovernor-restrict.log', u'format': u'LONG'}, u'restrictedTimePeriods': {u'level1': {u'unitOfTime': u'seconds', u'period': 60}, u'level2': {u'unitOfTime': u'minutes', u'period': 15}, u'level3': {u'unitOfTime': u'hours', u'period': 1}, u'level4': {u'unitOfTime': u'days', u'period': 1}, u'timeout': {u'unitOfTime': u'hours', u'period': 1}}, u'errorLog': {u'logPath': u'/var/log/dbgovernor-error-my.log', u'level': u'DEBUG'}, u'slowQueries': {u'logPath': u'/var/log/dbgovernor-kill.log', u'kill': True, u'timeout': 30}, u'restrictType': {u'mode': u'limit', u'unlimit': {u'unitOfTime': u'seconds', u'period': 60}} } Any key in dictionary can absent, appropriate MySQL Governor parameter will not changed :return: None. Throws DBGovernorException if error Nz['modeOfOperation']rL rM z['scriptPath']rX rY z ['gatherDataForDetailedStats']rV r z['userMaxConnections']rZ z ['logRestrictedUsersQueries']rW z['restrictLog']['logPath']rb z['restrictLog']['format']r z['errorLog']['logPath']rc z['errorLog']['level']z['slowQueries']['logPath']r` z['slowQueries']['kill']ra z['restrictType']['mode']r_ zparameters_dict%s)z%['restrictType']['unlimit']['period']z)['restrictType']['unlimit']['unitOfTime']r_ r )z-['restrictedTimePeriods']['level1']['period']z1['restrictedTimePeriods']['level1']['unitOfTime']rX r[ )z-['restrictedTimePeriods']['level2']['period']z1['restrictedTimePeriods']['level2']['unitOfTime']rX r\ )z-['restrictedTimePeriods']['level3']['period']z1['restrictedTimePeriods']['level3']['unitOfTime']rX r] )z-['restrictedTimePeriods']['level4']['period']z1['restrictedTimePeriods']['level4']['unitOfTime']rX r^ )z.['restrictedTimePeriods']['timeout']['period']z2['restrictedTimePeriods']['timeout']['unitOfTime']rX r z%s%sr! r �wbs <?xml version="1.0" ?> %s z%(dbgov_cfg)s error: Z dbgov_cfg)�message�contextz//sbin/service db_governor restart &>/dev/null &Tz /bin/bash)�shell� executable)rN r. rO rJ r& r rP rQ �closer' r �evalrI �KeyErrorr rG Ztostringr$ �writerh ri rj rk rl r � subprocessra )Zparameters_dictrT r+ Zcontrol_data_listZcontrol_data_itemZinput_dict_keysZ xml_tag_nameZxml_attribute_nameZfunc_converter�valZcontrol_data_pairs_listZperiod_indexesZperiod_unit_of_time_indexesZ period_valZperiod_unit_of_timeZslow_queries_timeoutZs_xmlrm r r r �set_db_gov_config� sf " rz c C s& t � } | �� \}}| �� |t|�d�S )z� Get db_governor status and version info :return: dict( 'dbGovVersion': <version>, 'dbGovStatus': enabled|error|disabled, 'dbGovError': error details ) )Z dbGovVersionZ dbGovStatusZ dbGovError)r Zget_governor_statusZget_governor_versionr' )Zdb_govZstatus�errorr r r rg e s rg )N)N) Z __future__r rN rx Zlxmlr �typingr Zclconfig_utilsr r r r Zclcommon.clexceptionr Z clcommon.lib.mysql_governor_libr r rR rJ r& r0 r= r? rG rI rK rU rn rz rg r r r r �<module> sF 6k