3

-�_g7�	@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�charmed-postgresql)
�returnc

Cs
d
t��S)Nz--config-file )
�PATRONI_CONFIG_FILE)
�self�r�(/usr/lib/python3.6/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_port)
rrrr�	psql_args-szCharmedPostgreSQL.psql_argsc
Cs |jt
d
�d�t
d�d�g�

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

|jd�

|jd
dd�
yvttdd��}
tj|
�
}WdQRX|d|_	|d}|dj
d�
}|\|_|_|d}|d}|d|_
|d|_WnFttjttttfk
�
rH
}
z|jdt�d|���

dSd}~XnX|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�	superuserZusernameZpasswordzVCan 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_configrZ	host_portrr�errorrrr�setup4sp








































zCharmedPostgreSQL.setupc

Cs^d
tj
krtj
d
=|jtd�d�dd�
|jtd�d�dd	�
|jtd
�d�dd�
dS)
Nrrz/*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)
r
)�__name__�
__module__�__qualname__Z
short_descZplugin_nameZpackages�property�strrrr0r1rrrrr#s



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