�
�~�e� � �� � d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm
Z
ddl m Z dd l
mZ dd
lmZ dd lmZ dd lmZ g d
�Z G d� de� Zdd�Zdd�Zdd�Zy)z%Deployment code for the sysinfo tool.� N)� Formatter)� getLogger)�RotatingFileHandler)�Deferred)�
maybeDeferred)�
namedClass)�VERSION)�BaseConfiguration)�format_sysinfo)�SysInfoPluginRegistry)�Load�Disk�Memory� Temperature� Processes�
LoggedInUsers�Networkc � � � e Zd ZdZeZdZ ej � dk7 r eej j d� fz
ZdZ dZ � fd�Z
� fd�Zd � Zd
� Z� xZS ) �SysInfoConfigurationz9Specialized configuration for the Landscape sysinfo tool.)z/etc/landscape/client.confr z~/.landscape/sysinfo.confz/var/lib/landscape/client/�sysinfoc �@ �� t �| � � d | j d< y )N�config)�super�__init__�_command_line_defaults)�self� __class__s ��>/usr/lib/python3/dist-packages/landscape/sysinfo/deployment.pyr z SysInfoConfiguration.__init__+ s �� �
����04��#�#�H�-� c � �� t �| � � }|j ddd�� |j ddd�� |j dt dd �
� d j d j t � � |_ |S )
zg
Specialize L{Configuration.make_parser}, adding any
sysinfo-specific options.
z--sysinfo-plugins� 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.z--width�P z(Maximum width for each column of output.)�type�defaultr# zDefault plugins: {}z, )r � make_parser�
add_option�int�format�join� ALL_PLUGINS�epilog)r �parserr s �r r' z SysInfoConfiguration.make_parser0 s� �� �
��$�&����� �!�2� �
� ��� '�!�"� �
� ��� ���;� �
� .�4�4�T�Y�Y�{�5K�L��
��
r c �f � |j d� D �cg c] }|j � �� c}S c c}w )N�,)�split�strip)r � plugin_spec�xs r �get_plugin_namesz%SysInfoConfiguration.get_plugin_namesP s' � �#.�#4�#4�S�#9�:�a���� �:�:��:s �.c
�R � | j �t }n| j | j � }| j �g }n| j | j � }|D �cg c] }||vs�|�� }}|D �cg c]& } t d|j � � d|� �� � ��( c}S c c}w c c}w )Nzlandscape.sysinfo.�.)�sysinfo_pluginsr, r5 �exclude_sysinfo_pluginsr �lower)r �include�excluder4 �plugins� plugin_names r � get_pluginsz SysInfoConfiguration.get_pluginsS s� � � � � � '�!�G��+�+�D�,@�,@�A�G� � '� '� /��G��+�+�D�,H�,H�I�G� %�:���'�)9�1�:��:�
'�
� �
�J�$�[�%6�%6�%8�$9��;�-�H�
�
�
�
�� ;��
s � B �'B �1+B$)�__name__�
__module__� __qualname__�__doc__r �version�default_config_filenames�os�getuid�path�
expanduser�default_data_dir�config_sectionr r' r5 r? �
__classcell__)r s @r r r sf �� �C��G� >� ��r�y�y�{�a�� � �G�G� � � :� ;�%
�
� � 4�� �N�5�
�@;�
r r c �~ � | �:t j � dk( rd} | S t j j d� } | S )zs
Work out the correct path to store logs in depending on the effective
user id of the current process.
r z/var/log/landscapez ~/.landscape)rF rG rH rI )�
landscape_dirs r �get_landscape_log_directoryrO e sA � �
� �
�9�9�;�!� � 0�M� �� �G�G� .� .�~� >�M� �r c �X � t | � } t d� }d|_ t j j | � st j | � t j j | d� }t |dd�� }|j |� |j t d� � y )Nzlandscape-sysinfoFz sysinfo.logi � � )�maxBytes� backupCountz'%(asctime)s %(levelname)-8s %(message)s) rO r � propagaterF rH �isdir�mkdirr+ r �
addHandler� setFormatterr )rN �logger� log_filename�handlers r �
setup_loggingr\ r s� � �/�
�>�M�
�*�
+�F� �F��
�7�7�=�=� � '�
��� � ��7�7�<�<�
�}�=�L�!�����G�
���g� � ����#L�M�Nr c �� ������ � t � ��
t � �t � ��j
| d�� �j � D ] }�j |� � ��fd�� � �fd����Gt � ��j ��fd�� �fd�}�j |� �j � �S �� ��S # t $ r"}t j d|� �� Y d}~��d}~ww xY w) zQ
@param reactor: The reactor to (optionally) run the sysinfo plugins in.
zUnable to setup logging. NT)�!accept_nonexistent_default_configc � �� t t �j � �j � �j � �j
d�� � y )Nz )�width�indent)�printr � get_headers� get_notes�
get_footnotesr` )�resultr r s ��r � show_outputzrun.<locals>.show_output� sA �� �
� ��#�#�%��!�!�#��%�%�'� �l�l��
�
r c �B �� �j � j � � S �N)�run� addCallback)rg r s ��r � run_sysinfozrun.<locals>.run_sysinfo� s �� ��{�{�}�(�(��5�5r c �8 �� t �� j � � S ri )r �
chainDeferred)�donerl s ��r �<lambda>zrun.<locals>.<lambda>� s �� �M�+�.�<�<�T�B� r c �@ �� �j d�j � | S )Nr )� callLater�stop)rf �reactors �r � stop_reactorzrun.<locals>.stop_reactor� s �� �
� � �a�� � � .��Mr )
r\ �OSError�sys�exitr r �loadr? �addr �callWhenRunning�addBothrj )
�argsrt r �e�pluginru r ro rl rg s
`` @@@@r rj rj � s� �� �2��� ��'�)��
!�
#�F� �K�K���K�=��$�$�&� ��� � �F�� �
�6� �� �z��� � � B�
� �
� � �\�"�� � �
� �K� �}�� �K��[ � 2� ���,�Q�C�0�1�1��2�s �
B? �? C*�C%�%C*ri )NN)rC rF rw �loggingr r �logging.handlersr �twisted.internet.deferr r �twisted.python.reflectr � landscaper �landscape.lib.configr
�landscape.sysinfo.sysinfor r r, r rO r\ rj � r r �<module>r� sS �� +� �
� � � 0� +� 0� -� � 2� 4� ;�� �E
�,� E
�P
�
O� 3r |