3
�f! � @ s d Z y ddlZW n ek
r, ddlZY nX G dd� de�ZG dd� de�ZG dd� de�Zd d
� ZG d d � d e�Z G d
d� de�Z
G dd� de�Z G dd� de�Z G dd� de�Z
G dd� de�Zdd� ZG dd� de�ZG dd � d e�ZG d d � d e�ZdS ) zR This provides a restricted tag language to define the sosreport
index/report
� Nc @ s e Zd Zdd� Zdd� ZdS )�Nodec C s t j| j�S )N)�json�dumps�data)�self� r � /usr/lib/python3.6/reporting.py�__str__ s z Node.__str__c C s dS )NFr )r �noder r r �can_add s z Node.can_addN)�__name__�
__module__� __qualname__r r r r r r r s r c @ s e Zd ZdZdS )�Leafz0Marker class that can be added to a Section nodeN)r r
r �__doc__r r r r r s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �Reportz?The root element of a report. This is a container for sections.c C s
i | _ d S )N)r )r r r r �__init__' s zReport.__init__c C s
t |t�S )N)�
isinstance�Section)r r
r r r r * s zReport.can_addc G s* x$|D ] }| j |�r|j| j|j< qW d S )N)r r �name)r �nodesr
r r r �add- s
z
Report.addN)r r
r r r r r r r r r r $ s r c C s t | t�r| S | jdd�S )z3returns a string text for a given unicode/str input�utf8�ignore)r �str�decode)�sr r r �_decode3 s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r zeA section is a container for leaf elements. Sections may be nested
inside of Report objects only.c C s t |�| _i | _d S )N)r r r )r r r r r r < s
zSection.__init__c C s
t |t�S )N)r r )r r
r r r r @ s zSection.can_addc G s4 x.|D ]&}| j |�r| jj|jg �j|j� qW d S )N)r r �
setdefault�ADDS_TO�append)r r r
r r r r C s
z Section.addN)r r
r r r r r r r r r r 8 s r c @ s e Zd ZdZdd� ZdS )�CommandZcommandsc C s t |�|t |�d�| _d S )N)r � return_code�href)r r )r r r" r# r r r r M s zCommand.__init__N)r r
r r r r r r r r! I s r! c @ s e Zd ZdZdd� ZdS )�
CopiedFileZ copied_filesc C s t |�t |�d�| _d S )N)r r# )r r )r r r# r r r r W s zCopiedFile.__init__N)r r
r r r r r r r r$ S s r$ c @ s e Zd ZdZdd� ZdS )� CreatedFileZ
created_filesc C s t |�t |�d�| _d S )N)r r# )r r )r r r# r r r r ` s zCreatedFile.__init__N)r r
r r r r r r r r% \ s r% c @ s e Zd ZdZdd� ZdS )�AlertZalertsc C s t |�| _d S )N)r r )r �contentr r r r i s zAlert.__init__N)r r
r r r r r r r r& e s r& c @ s e Zd ZdZdd� ZdS )�NoteZnotesc C s t |�| _d S )N)r r )r r' r r r r q s z
Note.__init__N)r r
r r r r r r r r( m s r( c C s
| j d�S )a Return True if 'string' ends with a backslash, and False otherwise.
Define this as a named function for no other reason than that pep8
now forbids binding of a lambda expression to a name:
'E731 do not assign a lambda expression, use a def'
�\)�endswith)�stringr r r �ends_bsu s r, c @ s� e Zd ZdZdZdZdZdZdZdZ dZ
dZ d Z dZ
d
Zd d Zeed
dfeeddfeeddfeeddfeeddffZg Zdd� Zdd� Zdd� ZdS )�PlainTextReportz@Will generate a plain text report from a top_level Report object� z * %(name)sz ! %sz * %szLoaded Plugins:z {name}�
� z{name}�=�H z- commands executed:z- files copied:z- files created:z
- alerts:z - notes:c C s t tj|j��| _d S )N)�sorted�dict�itemsr � report_data)r Z report_noder r r r � s zPlainTextReport.__init__c
C s, g | _ }t| j�dkr$|j| j� |j| j� d}d}t| j�}xH| jD ]>\}}|d|� �7 }|d7 }|| j dkrJ||k rJ|| j7 }qJW || j7 }|j|� xV| jD ]L\}}|j| j � |j|� � x*| j
D ] \}} }
} | j ||j |
| | � q�W q�W t| j
�dk�r|j| j
� djtdd� |��} | S )Nr r. z � r/ c S s t | t�r| S | jdd�S )Nr r )r r r )�ir r r �<lambda>� s z)PlainTextReport.unicode.<locals>.<lambda>)�line_buf�len�HEADERr �PLUGLISTHEADERr6 �PLUGLISTMAXITEMS� PLUGLISTSEP�PLUGLISTFOOTER� PLUGDIVIDER� subsections�process_subsectionr �FOOTER�join�map)
r r: �liner8 Z plugcountZ section_name�_Zsection_contentsZtype_�format_�header�footer�outputr r r �unicode� s2
zPlainTextReport.unicodec C s` ||kr\| j j|� x.t|j|�dd� d�D ]}| j j|| � q,W t|�dkr\| j j|� d S )Nc S s t | t�r| d S dS )Nr r. )r r4 )�xr r r r9 � s z4PlainTextReport.process_subsection.<locals>.<lambda>)�keyr )r: r r3 �getr; )r ZsectionrO rJ rI rK �itemr r r rC � s z"PlainTextReport.process_subsectionN)r r
r r r< rD �LEAF�ALERT�NOTEr= � PLUGLISTITEMr? r> r@ � PLUGINFORMATrA r! r$ r% r& r( rB r: r rM rC r r r r r- � s,
"r- c @ sv e Zd ZdZdZdZdZdZdZdZ dZ
dZ d Z d
Z
d Zd Zeed
dfeeddfeeddfeeddfeeddffZdS )�
HTMLReportz:Will generate a HTML report from a top_level Report objecta� <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN"
"http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
<html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en">
<head>
<meta http-equiv="Content-Type" content="text/html;
charset=utf-8" />
<title>Sos System Report</title>
<style type="text/css">
td {
padding: 0 5px;
}
</style>
</head>
<body>
z</body></html>z(<li><a href="%(href)s">%(name)s</a></li>z <li>%s</li>z#<h3>Loaded Plugins:</h3><table><tr>z&<td><a href="#{name}">{name}</a></td>
z
</tr>
<tr>r0 z
</tr></table>z+<h2 id="{name}">Plugin <em>{name}</em></h2>z<hr/>
z <p>Commands executed:</p><ul>z</ul>z<p>Files copied:</p><ul>z<p>Files created:</p><ul>z<p>Alerts:</p><ul>z<p>Notes:</p><ul>N)r r
r r r< rD rR rS rT r= rU r? r> r@ rV rA r! r$ r% r& r( rB r r r r rW � s$
rW c @ s e Zd ZdZdd� ZdS )�
JSONReportz:Will generate a JSON report from a top_level Report objectc C s t j| jddd�}|S )N� F)�indentZ ensure_ascii)r r r6 )r rL r r r rM � s zJSONReport.unicodeN)r r
r r rM r r r r rX � s rX )r r � ImportErrorZ
simplejson�objectr r r r r r! r$ r% r& r( r, r- rW rX r r r r �<module> s$
K&