U
    -_gE                     @   s.   d dl Z d dlmZmZ G dd deeZdS )    N)PluginUbuntuPluginc                   @   s4   e Zd ZdZdZdZdZdZdZdd Z	d	d
 Z
dS )Kafkaz
    This plugin collects log and configuration files, and also basic
    installation information (e.g., `snap info`) for Apache Kafka.
    zApache Kafka pluginkafka)services)zcharmed-kafka)r   c                 C   sJ   |  drdnd}| jr,| d d}d}nd}d}| || |g d S )	NZall_logsz*.log*z*.logzsnap info charmed-kafkaz-/var/snap/charmed-kafka/common/var/log/kafka/*/var/snap/charmed-kafka/current/etc/kafka/z/usr/local/kafka/logs//usr/local/kafka/config/)Z
get_optionis_snapZadd_cmd_outputZadd_copy_spec)selfZlog_file_patternZlog_pathconfig_path r   :/usr/lib/python3/dist-packages/sos/report/plugins/kafka.pysetup   s    
zKafka.setupc                 C   sd   ddg}| j rdnd}dd| d}t|D ],\}}}|D ]}| tj|||d q@q2d S )	NZpasswordZusernamer   r   z(^\s*(|z)\s*=\s*)(.*)z\1*********)r	   joinoswalkZdo_path_regex_subpath)r
   Zprotect_keysr   Zregexproot_filesfiler   r   r   postproc,   s    	 zKafka.postprocN)__name__
__module____qualname____doc__Z
short_descZplugin_nameZprofilesZpackagesr   r   r   r   r   r   r   r      s   r   )r   Zsos.report.pluginsr   r   r   r   r   r   r   <module>   s   