o

([Qh7�	@s�dd
lZdd
l
Z
ddlmZmZ
dZdZededededed	ed
ededd
�Zed�d�ZdZ	dZ
dZGdd�dee�Zd
S)�N)�Plugin�UbuntuPluginz#/var/snap/charmed-postgresql/commonz$/var/snap/charmed-postgresql/currentz/var/lib/postgresqlz/var/log/postgresqlz/etc/patroniz/var/log/patroniz/etc/pgbackrestz/var/log/pgbackrestz/etc/pgbouncerz/var/log/pgbouncer)�POSTGRESQL_CONF�POSTGRESQL_LOGS�PATRONI_CONF�PATRONI_LOGS�PGBACKREST_CONF�PGBACKREST_LOGS�PGBOUNCER_CONF�PGBOUNCER_LOGSrz
/patroni.yaml�snap_daemonzcharmed-postgresql.psqlzcharmed-postgresql.patronictlc@sLeZ
dZd
ZdZdZedefdd��
Zedefdd��
Z	d	d
�Z
dd�Zd
S)�CharmedPostgreSQLzCharmed PostgreSQL�charmed_postgresql)
zcharmed-postgresql�returnc

Cs
d
t��S)Nz--config-file )
�PATRONI_CONFIG_FILE�
�self�r�G/usr/lib/python3/dist-packages/sos/report/plugins/charmed_postgresql.py�patronictl_args)s
z!CharmedPostgreSQL.patronictl_argsc

Csd
|j�d|j
�d|j�d�S)Nz-U z -h z -p z -d postgres -P pager=off)�postgresql_username�postgresql_host�postgresql_portrrrr�	psql_args-s


��zCharmedPostgreSQL.psql_argsc

Cs8|�t
d
�d�t
d�d�g�

|�t
d�d�t
d�t
d
�d�t
d�t
d	�t
d
�t
d�t
d�g�

|�d�

|jd
dd�
zEttdd��
}
t�|
�
}Wd�
n1s]w



Y
|d|_	|d}|d�
d�
}|\|_|_|d}|d}|d|_
|d|_Wn%ttjttttfy�
}
z|�dt�d|���

WYd}~dSd}~w
w|jt�d|j�d|j	��dtd�
|jt�d|j�d |j	��d!td�
|jt�d|j�d"|j	��d#td�
|jtjd$<|jt�d|j�d%�d&td�
|jt�d|j�d'�d(td�
|jt�d|j�d)�d*td�
dS)+Nrz/*.pemr
rz/*.conf*rz/*.y*mlrrr	rzsnap.charmed-postgresql.*zsnap info charmed-postgresqlz	snap-info)
�suggest_filenamezutf-8)
�encoding�scope�
postgresql�connect_address�
:�authentication�	superuser�username�passwordzVCan not run additional commands due to an error on opening or parsing the config file z: �
 z
 topology zpatroni-topology)r�runasz	 history zpatroni-historyz
 show-config zpatroni-dcs-config�
PGPASSWORDz	 -c '\l+'zpostgresql-databasesz -c '\duS+'zpostgresql-usersz -c '\dtS+'zpostgresql-tables)�add_forbidden_path�PATHS�
add_copy_spec�add_journal�add_cmd_output�openr�yaml�	safe_load�patroni_cluster_name�splitrrr�postgresql_password�OSError�	YAMLError�	TypeError�
ValueError�KeyError�AttributeError�	_log_warn�
PATRONICTLr�RUNAS�os�environ�PSQLr)r�
f�patroni_configr�	host_portr r!�errorrrr�setup4s�

�







�



�

�











�����


�
�	

�
�	

�
�

�	

�	


�zCharmedPostgreSQL.setupc

Cs^d
tj
vr	tj
d
=|�td�d�dd�
|�td�d�dd	�
|�td
�d�dd�
dS)
Nr&rz/*z(password: )"?.*"?z
\1"*********"rz/pgbackrest.confz(.*s3-key.*=).*z\1*********r
z/pgbouncer/userlist.txtz(".*" )".*")r;r<�do_path_regex_subr(rrrr�postproc�s"




�


�


�zCharmedPostgreSQL.postprocN)�__name__�
__module__�__qualname__�
short_desc�plugin_name�packages�property�strrrrBrDrrrrr
#s



nr
)
r;r-�sos.report.pluginsrr�SNAP_COMMON_PATH�SNAP_CURRENT_PATHr(rr:r=r9r
rrrr�<module>
s&








�