File: //lib/python3.6/site-packages/sos/report/plugins/__pycache__/charmed_mysql.cpython-36.pyc
3
ܓ�i�& � @ sF d dl Z d dlZd dlmZmZmZ d dlmZ G dd� dee�ZdS )� N)�Plugin� PluginOpt� UbuntuPlugin)�
is_executablec @ s� e Zd ZdZdZdZdddddd gZd
Zd Zd Z d
Z
dZ dddddd�Z e
ddedd�e
ddedd�e
d d ed d�e
d d ed!d�gZd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4S )5� CharmedMySQLa�
The Charmed MySQL plugin is used to collect MySQL configuration and logs
from the Charmed MySQL snap package or K8s deployment.
It also collects MySQL Router and MySQL Shell configuration and logs
where available, journal logs for the snap, and snap info.
If the `dumpdbs` option is set to `True`, the plugin will also try and
collect the names of the databases that the user has access to. The
`mysql` user is used by default, but that can be set using the `dbuser`
option. When using the `dumpdbs` option, you must then provide the
password for the user using the `dbpass` option or the `MYSQL_PWD`
environment variable.
z
Charmed MySQLZ
charmed_mysqlzshow databases;z*select * from mysql.juju_units_operations;zPselect * from performance_schema.replication_group_members order by MEMBER_HOST;z"show global status like "%conne%";z,select * from performance_schema.host_cache;zshow ENGINE InnoDB STATUS;z
charmed-mysqlz /var/snap/charmed-mysql/commonz /var/snap/charmed-mysql/currentZkubectlz app.kubernetes.io/name=mysql-k8sz
/etc/mysqlz/var/log/mysqlz/etc/mysqlrouterz/var/log/mysqlrouterz/var/log/mysqlsh)�
MYSQL_CONF�
MYSQL_LOGS�MYSQL_ROUTER_CONF�MYSQL_ROUTER_LOGS�MYSQL_SHELL_LOGS�dbuser�mysqlz%Username for database dump collection)�defaultZval_typeZdesc�dbpass� z%Password for database dump collection�dumpdbsFzGet name of all databases�
logs_sinceZ48hz9How far back to fetch logs with kubectl --since, K8s onlyc C sX | j d�}| j d�}dtjkr<| r<| jjd� tjd }n|sP| jjd� dS ||fS )Nr r � MYSQL_PWDzWMYSQL_PWD present: Using MYSQL_PWD environment variable, user did not provide password.zHdumpdbs_error: option is set, but username and password are not provided)NN)�
get_option�os�environZsoslog�infoZwarning)�self�db_user�db_pass� r �#/usr/lib/python3.6/charmed_mysql.py�_get_db_credentials[ s
z CharmedMySQL._get_db_credentialsc C sB | j | j� d| j� d��}|d dkr>tt|d j� j� ��S g S )Nz get pods -A -l z- -o jsonpath='{.items[*].metadata.namespace}'�statusr �output)�exec_cmd�kube_cmd�selector�list�set�strip�split)r �
namespacesr r r �_determine_namespacesn s
z"CharmedMySQL._determine_namespacesc C s@ | j | j� d|� d| j� d��}|d dkr<|d j� j� S g S )Nz -n z
get pods -l z( -o jsonpath='{.items[*].metadata.name}'r r r )r r! r" r% r& )r � namespace�podsr r r �_get_pod_namesw s
zCharmedMySQL._get_pod_namesc C sB | j � d|� �}| j|� d| j� d�|� d| j� �g� | j|� d| j� d|� d�� d}| j|�}| j� }xx| jj� D ]j\}}x`|D ]X} tj|� d | � d |� �d
d � |� d |� d
| � d|� d
|� d | � d |� �
}
| j |
� q�W qxW | j
|� d�|� d�g� | j d��r>| j � \} } | �s(| �r,d S d| � �}
d|
� d�}d
j
dd� | jD ��}tj� ��}d}|d | }t|ddd��}|j| � W d Q R X x�|D ]�} |� d|� d
| � d |� �}| j |� |� d |� d
|� d
| � d|� � }
| j |
� |� d|� d
| � d |� d |� d |� d |� �
}| j|| � d!�d"� �q�W W d Q R X d S )#Nz -n z get all -l z -o widez describe pods -l z logs -l z --since=z* --all-containers=true --prefix --all-podsr
�/T)�exist_okz cp -c � �:z/*/MYSQL_CONF/*.pemz/*/MYSQL_CONF/*.keyr z-h 127.0.0.1 -uzmysql z -ec s s | ]}|j d d�V qdS )�"z\"N)�replace)�.0�queryr r r � <genexpr>� s z6CharmedMySQL._collect_per_namespace.<locals>.<genexpr>Z mysql_pwd�w�utf8)�encodingz exec -c z
-- mkdir -p z -- sh -lc 'MYSQL_PWD=$(cat z) z "z
"' && rm -rf z_dbs.txt)Zsuggest_filename)r! �add_cmd_outputr" r+ Zget_cmd_output_path�
conf_paths�itemsr �makedirsr �add_forbidden_pathr r �join�
mysql_queries�tempfileZTemporaryDirectory�open�write)r r) r r! Z
mysql_contr* Zdump_files_path�name�pathZpodZcopy_cmdr r �opts�sql_cmdZqueriesZtmpdirZpwd_fileZpwd_path�fZ mkdir_cmdZ queries_cmdr r r �_collect_per_namespace� sP
,
,z#CharmedMySQL._collect_per_namespacec G s dd� |D �}t jj|f|�� S )Nc S s g | ]}|j tjj��qS r )�lstripr rC �sep)r2 �pr r r �
<listcomp>� s z0CharmedMySQL._join_conf_path.<locals>.<listcomp>)r rC r= )r �base�partsZstripped_partsr r r �_join_conf_path� s z CharmedMySQL._join_conf_pathc C s0 | j d�}| j� }x|D ]}| j||� qW d S )Nr )r r( rG )r r r' r) r r r � _process_k8s� s
zCharmedMySQL._process_k8sc s | j | j| j| jd d�| j| j| jd d�g� | j| j| j| jd �| j| j| jd �| j| j| jd �| j| j| jd �| j| j| jd �g� | jd� | jd � | jd
��r| j � \}}| s�| r�d S d |i}d |� �}d
|� d�� | j� fdd�| j
D �|d� d S )Nr z*.pemz*.keyr r r
r zsnap.charmed-mysql.*zsnap info charmed-mysqlr r z-h 127.0.0.1 -uzmysql z -ec s g | ]}� � d |� d��qS )z '�'r )r2 r3 )rE r r rK s z.CharmedMySQL._process_snap.<locals>.<listcomp>)�env) r< rN �snap_path_currentr9 Z
add_copy_spec�snap_path_commonZ add_journalr8 r r r> )r r r Z mysql_envrD r )rE r �
_process_snap� sJ
zCharmedMySQL._process_snapc C s* | j | j�rdS t| j| j�o(t| j� �S )NT)� is_installed� snap_packager r! �sysroot�boolr( )r r r r �
check_enabled s zCharmedMySQL.check_enabledc C s. | j | j�r| j� t| j| j�r*| j� d S )N)rU rV rT r r! rW rO )r r r r �setup s zCharmedMySQL.setupN)�__name__�
__module__� __qualname__�__doc__Z
short_descZ plugin_namer> rV rS rR r! r" r9 r �strrX Z option_listr r( r+ rG rN rO rT rY rZ r r r r r sP
N< r ) r r? Zsos.report.pluginsr r r Z
sos.utilitiesr r r r r r �<module> s