a

\��h7�	@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.yamlZsnap_daemonzcharmed-postgresql.psqlzcharmed-postgresql.patronictlc@sLeZ
dZd
ZdZdZeed�
dd��
Zeed�
dd��
Z	d	d
�Z
dd�Zd
S)�CharmedPostgreSQLzCharmed PostgreSQLZcharmed_postgresql)
zcharmed-postgresql)
�returnc

Cs
d
t��S)Nz--config-file )
�PATRONI_CONFIG_FILE�
�self�r�I/usr/lib/python3.9/site-packages/sos/report/plugins/charmed_postgresql.py�patronictl_args)sz!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

Cs:|�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�
z�ttdd��}
t�|
�
}Wd�
n1s�0


Y
|d|_	|d}|d�
d�
}|\|_|_|d}|d}|d|_
|d|_WnLttjttttf�
yb
}
z$|�dt�d|���

WYd}~dSd}~00|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)
�encodingZscope�
postgresqlZconnect_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)rZrunasz	 history zpatroni-historyz
 show-config zpatroni-dcs-config�
PGPASSWORDz	 -c '\l+'zpostgresql-databasesz -c '\duS+'zpostgresql-usersz -c '\dtS+'zpostgresql-tables)Zadd_forbidden_path�PATHSZ
add_copy_specZadd_journalZadd_cmd_output�openr�yamlZ	safe_loadZpatroni_cluster_name�splitrrrZpostgresql_password�OSErrorZ	YAMLError�	TypeError�
ValueError�KeyError�AttributeErrorZ	_log_warn�
PATRONICTLr�RUNAS�os�environ�PSQLr)r�
fZpatroni_configr�	host_portrr�errorrrr�setup4s�

�







�


�

(












���

�
�	

�
�	

�
�

�	

�	

�zCharmedPostgreSQL.setupc

Cs^d
tj
vrtj
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.Zdo_path_regex_subr"rrrr�postproc�s"




�


�


�zCharmedPostgreSQL.postprocN)�__name__�
__module__�__qualname__Z
short_descZplugin_nameZpackages�property�strrrr3r4rrrrr#s



nr)
r-r$Zsos.report.pluginsrrZSNAP_COMMON_PATHZSNAP_CURRENT_PATHr"rr,r/r+rrrrr�<module>	s$








�