File: //lib/python3/dist-packages/cloudinit/reporting/__pycache__/events.cpython-310.pyc
o
�Ad�# � @ s� d Z ddlZddlZddlZddlmZ ddlmZm Z ddl
m Z dZ dZ
dZG d d
� d
e�Zed �ZG d d
� d
�ZG dd� de�Zddd�Zejdfdd�Zdd� ZG dd� d�Zdd� ZdS )z�
events for reporting.
The events here are designed to be used with reporting.
They can be published to registered handlers with report_event.
� N)�List)�available_handlers� instantiated_handler_registry)�ReportingHandler�finish�start� cloudinitc @ s e Zd Zdd� ZdS )�_namesetc C s || v r|S t d| ��)Nz%s not a valid value)�AttributeError)�self�name� r
�</usr/lib/python3/dist-packages/cloudinit/reporting/events.py� __getattr__ s z_nameset.__getattr__N)�__name__�
__module__� __qualname__r r
r
r
r r s r )�SUCCESS�WARN�FAILc @ s. e Zd ZdZedfdd�Zdd� Zdd� ZdS ) �ReportingEventz"Encapsulation of event formatting.Nc C s2 || _ || _|| _|| _|d u rt�� }|| _d S �N)�
event_typer � description�origin�time� timestamp)r r r r r r r
r
r �__init__) s
zReportingEvent.__init__c C s d� | j| j| j�S )z"The event represented as a string.z
{0}: {1}: {2})�formatr r r �r r
r
r � as_string9 s �zReportingEvent.as_stringc C s | j | j| j| j| jd�S )z&The event represented as a dictionary.�r r r r r r! r r
r
r �as_dict? s �zReportingEvent.as_dict)r r r �__doc__�DEFAULT_EVENT_ORIGINr r r" r
r
r
r r & s
� r c s8 e Zd Zejdf� fdd� Zdd� Z� fdd�Z� ZS )�FinishReportingEventNc sD t t| ��t||� || _|d u rg }|| _|tvr td| ��d S )NzInvalid result: %s)�superr% r �FINISH_EVENT_TYPE�result�
post_files�status�
ValueError)r r r r( r) �� __class__r
r r K s
� �z FinishReportingEvent.__init__c C s d� | j| j| j| j�S )Nz{0}: {1}: {2}: {3})r r r r( r r r
r
r r X s �z FinishReportingEvent.as_stringc s0 t t| ��� }| j|d< | jrt| j�|d< |S )z'The event represented as json friendly.r( �files)r& r% r" r( r) �_collect_file_info)r �datar, r
r r" ] s
z FinishReportingEvent.as_dict) r r r r* r r r r" �
__classcell__r
r
r, r r% J s
�
r% c sT � si � � fdd�t j�� D �}ttj�� �}|D ]}t|�|v r"q|�| � qdS )a� Report an event to all registered event handlers
except those whose type is in excluded_handler_types.
This should generally be called via one of the other functions in
the reporting module.
:param excluded_handler_types:
List of handlers types to exclude from reporting the event to.
:param event_type:
The type of the event; this should be a constant from the
reporting module.
c s h | ]
\}}|� v r|�qS r
r
)�.0� hndl_type�hndl_cls��excluded_handler_typesr
r � <setcomp>v s ��z report_event.<locals>.<setcomp>N)r �registered_items�items�listr �values�type�
publish_event)�eventr6 �excluded_handler_classes�handlers�handlerr
r5 r � report_eventf s
�� �rB c C s t | |||d�}t|�S )zVReport a "finish" event.
See :py:func:`.report_event` for parameter details.
�r) )r% rB )�
event_name�event_descriptionr( r) r> r
r
r �report_finish_event� s �rF c C s t t| |�}t|�S )a' Report a "start" event.
:param event_name:
The name of the event; this should be a topic which events would
share (e.g. it will be the same for start and finish events).
:param event_description:
A human-readable description of the event that has occurred.
)r �START_EVENT_TYPErB )rD rE r> r
r
r �report_start_event� s
rH c @ s� e Zd ZdZdddejdfdd�Zdd� Zdd� Zd d
� Z e
d d � �Z e j d
d � �Z e
dd� �Z
e
j dd� �Z
dd� Zdd� ZdS )�ReportEventStackaK Context Manager for using :py:func:`report_event`
This enables calling :py:func:`report_start_event` and
:py:func:`report_finish_event` through a context manager.
:param name:
the name of the event
:param description:
the event's description, passed on to :py:func:`report_start_event`
:param message:
the description to use for the finish event. defaults to
:param:description.
:param parent:
:type parent: :py:class:ReportEventStack or None
The parent of this event. The parent is populated with
results of all its children. The name used in reporting
is <parent.name>/<name>
:param reporting_enabled:
Indicates if reporting events should be generated.
If not provided, defaults to the parent's value, or True if no parent
is provided.
:param result_on_exception:
The result value to set if an exception is caught. default
value is FAIL.
:param post_files:
Can hold filepaths of files that are to get posted/created
regarding a given event. Something like success or failure information
in a given log file. For each filepath, if it's a valid regular file
it will get: read & encoded as base64 at the close of the event.
Default value, if None, is an empty list.
Nc C s� || _ || _|| _|| _|| _tj| _|d u rg }|| _|d u r(|r&|j }nd}|| _ |r7d�
|j |f�| _ n| j| _ i | _ d S )NT�/)
�parentr r �message�result_on_exceptionr* r r( r) �reporting_enabled�join�fullname�children)r r r rL rK rN rM r) r
r
r r � s,
��
zReportEventStack.__init__c C s d| j | j| jf S )Nz.ReportEventStack(%s, %s, reporting_enabled=%s))r r rN r r
r
r �__repr__� s
�zReportEventStack.__repr__c C s4 t j| _| jrt| j| j� | jrd| jj| j < | S )N)NN)
r* r r( rN rH rP r rK rQ r r r
r
r � __enter__� s zReportEventStack.__enter__c C sP t jt jfD ]}| j�� D ]\}\}}||kr || jf S q
q| j| jfS r )r* r r rQ r9 rL r( )r � cand_result�_name�value�_msgr
r
r �_childrens_finish_info� s �� z'ReportEventStack._childrens_finish_infoc C s | j S r )�_resultr r
r
r r( s zReportEventStack.resultc C s |t vr
td| ��|| _d S )Nz'%s' not a valid result)r* r+ rY �r rV r
r
r r( s
c C s | j d ur| j S | jS r )�_messager r r
r
r rL s
zReportEventStack.messagec C s
|| _ d S r )r[ rZ r
r
r rL s
c C s |r| j | jfS | �� S r )rM rL rX )r �excr
r
r � _finish_info s z ReportEventStack._finish_infoc C sH | � |�\}}| jr||f| jj| j< | jr"t| j||| jd� d S d S )NrC )r] rK rQ r rN rF rP r) )r �exc_type� exc_value� tracebackr( �msgr
r
r �__exit__ s
��zReportEventStack.__exit__)r r r r# r* r r rR rS rX �propertyr( �setterrL r] rb r
r
r
r rI � s* *
�'
rI c C sz | sd S g }| D ]2}t j�|�sd }n t|d��}t�|�� ��� }W d � n1 s,w Y |�||dd�� q|S )N�rb�base64)�path�content�encoding) �osrg �isfile�openrf � b64encode�read�decode�append)r. �ret�fnamerh �fpr
r
r r/ ( s �r/ r )r# rf �os.pathrj r �typingr �cloudinit.reportingr r � cloudinit.reporting.handlersr r' rG r$ �setr r* r r% rB r rF rH rI r/ r
r
r
r �<module> s* $
�