File: //lib/python3/dist-packages/landscape/sysinfo/__pycache__/deployment.cpython-38.pyc
U
���cm � @ 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_sysinfoZLoadZDiskZMemoryZ 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_parserZ
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 �Zexcluder 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 ZincludeZpluginsr 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)r, r- r. r/ )�
landscape_dirr r r �get_landscape_log_directoryJ s
r2 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) r2 r Z propagater, r. �isdir�mkdirr r Z
addHandlerZ setFormatterr )r1 ZloggerZ log_filenameZhandlerr r r �
setup_loggingW s
�
r6 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)�runZ addCallbackr )r; r
r r � run_sysinfoz s zrun.<locals>.run_sysinfoc s t ���� �S r< )r Z
chainDeferredr )�doner> r r �<lambda>� � zrun.<locals>.<lambda>c s � � d� j� | S )Nr )Z callLater�stopr9 )�reactorr r � stop_reactor� s zrun.<locals>.stop_reactor)
r6 �IOError�sys�exitr
r �loadr& �addr ZcallWhenRunningZaddBothr= )�argsrC r
�er ZpluginrD r )r? rC r> r; r
r r= d s,
�
r= )N)N)NN)r* r, rF Zloggingr r Zlogging.handlersr Ztwisted.python.reflectr Ztwisted.internet.deferr r Z landscaper Zlandscape.lib.configr Zlandscape.sysinfo.sysinfor
r r r r2 r6 r= r r r r �<module> s"
�7