File: //usr/lib/python3/dist-packages/landscape/sysinfo/__pycache__/deployment.cpython-38.pyc
U
y)�]m � @ s� d Z ddlZddlZddlmZmZ ddlmZ ddlm Z ddl
m Z m Z ddl
mZ ddlmZ dd lmZmZ d
d d d
dddgZG dd� de�Zddd�Zddd�Zddd�ZdS ) z%Deployment code for the sysinfo tool.� N)� getLogger� Formatter)�RotatingFileHandler)�
namedClass)�Deferred�
maybeDeferred)�VERSION)�BaseConfiguration)�SysInfoPluginRegistry�format_sysinfo�LoadZDiskZMemoryZ TemperatureZ ProcessesZ
LoggedInUsersZNetworkc sj e Zd ZdZeZdZe�� dkr2eej �
d�f7 ZdZ dZ � fdd�Z
� fd d
�Zd d � Zd
d� Z� ZS )�SysInfoConfigurationz9Specialized configuration for the Landscape sysinfo tool.)z/etc/landscape/client.confr z~/.landscape/sysinfo.confz/var/lib/landscape/client/�sysinfoc s t t| ��� d | jd< d S )N�config)�superr
�__init__Z_command_line_defaults)�self�� __class__� �>/usr/lib/python3/dist-packages/landscape/sysinfo/deployment.pyr s z SysInfoConfiguration.__init__c sB t t| ��� }|jdddd� |jdddd� dd�t� |_|S ) zg
Specialize L{Configuration.make_parser}, adding any
sysinfo-specific options.
z--sysinfo-pluginsZ PLUGIN_LISTzNComma-delimited list of sysinfo plugins to use. Default is to use all plugins.)�metavar�helpz--exclude-sysinfo-pluginszhComma-delimited list of sysinfo plugins to NOT use. This always take precedence over plugins to include.zDefault plugins: %sz, )r r
� make_parser�
add_option�join� ALL_PLUGINSZepilog)r �parserr r r r % s ��z SysInfoConfiguration.make_parserc C s dd� |� d�D �S )Nc S s g | ] }|� � �qS r )�strip��.0�xr r r �
<listcomp>9 s z9SysInfoConfiguration.get_plugin_names.<locals>.<listcomp>�,)�split)r Z plugin_specr r r �get_plugin_names8 s z%SysInfoConfiguration.get_plugin_namesc sX | j d krt}n | �| j �}| jd kr,g � n | �| j�� � fdd�|D �}dd� |D �S )Nc s g | ]}|� kr|�qS r r r ��excluder r r" D s z4SysInfoConfiguration.get_plugins.<locals>.<listcomp>c S s" g | ]}t d |�� |f �� �qS )zlandscape.sysinfo.%s.%s)r �lower)r Z plugin_namer r r r" E s �
�)Zsysinfo_pluginsr r% Zexclude_sysinfo_plugins)r �include�pluginsr r& r � get_plugins; s
�z SysInfoConfiguration.get_plugins)�__name__�
__module__� __qualname__�__doc__r �versionZdefault_config_filenames�os�getuid�path�
expanduserZdefault_data_dirZconfig_sectionr r r% r+ �
__classcell__r r r r r
s
� r
c C s* | dkr&t �� dkrd} n t j�d�} | S )zs
Work out the correct path to store logs in depending on the effective
user id of the current process.
Nr z/var/log/landscapez ~/.landscape)r1 r2 r3 r4 )�
landscape_dirr r r �get_landscape_log_directoryJ s
r7 c C sd t | �} td�}d|_tj�| �s,t�| � tj�| d�}t|ddd�}|� |� |�
t d�� d S )Nzlandscape-sysinfoFz sysinfo.logi � � )ZmaxBytesZ backupCountz'%(asctime)s %(levelname)-8s %(message)s) r7 r � propagater1 r3 �isdir�mkdirr r �
addHandler� setFormatterr )r6 �loggerZ log_filename�handlerr r r �
setup_loggingW s
�
r@ c
s� z
t � W n0 tk
r: } zt�d| � W 5 d}~X Y nX �dkrJt� �t� }|j| dd� |�� D ]}��|� qf�fdd����fdd���dk r�t � � ��
� �fd d
�� �fd d �}� � |� �� � n�� � � S )
zQ
@param reactor: The reactor to (optionally) run the sysinfo plugins in.
zUnable to setup logging. %sNT)Z!accept_nonexistent_default_configc s$ t t� �� � �� � �� dd�� d S )Nz )�indent)�printr Z get_headersZ get_notesZ
get_footnotes��result)r r r � show_outputv s �zrun.<locals>.show_outputc s �� � �� �S �N)�run� addCallbackr )rE r r r � run_sysinfoz s zrun.<locals>.run_sysinfoc s t ���� �S rF )r �
chainDeferredr )�donerI r r �<lambda>� � zrun.<locals>.<lambda>c s � � d� j� | S )Nr )� callLater�stoprC )�reactorr r � stop_reactor� s zrun.<locals>.stop_reactor)
r@ �IOError�sys�exitr
r
�loadr+ �addr �callWhenRunning�addBothrG )�argsrP r �er ZpluginrQ r )rK rP rI rE r r rG d s,
�
rG )N)N)NN)r/ r1 rS �loggingr r Zlogging.handlersr Ztwisted.python.reflectr Ztwisted.internet.deferr r Z landscaper Zlandscape.lib.configr Zlandscape.sysinfo.sysinfor
r r r
r7 r@ rG r r r r �<module> s"
�7