"T` @sdZddlZddlZddlZddlmZmZdZdZ dZ Gdd d e Z e dZ Gd ddeZGdddeZdddZe jdddZddZGdddeZddZdS)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)instantiated_handler_registryavailable_handlersfinishstart cloudinitc@seZdZddZdS)_namesetcCs$||kr|Std|dS)Nz%s not a valid value)AttributeError)selfnamer t|j|d<|S)z'The event represented as json friendly.r$files)r"r!rr$r%_collect_file_info)r data)r(r r rKs   zFinishReportingEvent.as_dict)rrrr&rrrrr r )r(r r!:s  r!cszs ifddtjjD}tjj}x6|D].\}}t||kreqD|j|qDWdS)aReport 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. cs(h|]\}}|kr|qSr r ).0Z hndl_typeZhndl_cls)excluded_handler_typesr r ds  zreport_event..N)rregistered_itemsitemsrtype publish_event)eventr-Zexcluded_handler_classeshandlers_handlerr )r-r report_eventTsr7cCs"t|||d|}t|S)zVReport a "finish" event. See :py:func:`.report_event` for parameter details. r%)r!r7) event_nameevent_descriptionr$r%r3r r r report_finish_eventqs r:cCstt||}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. )rSTART_EVENT_TYPEr7)r8r9r3r r r report_start_event|s r<c@seZdZdZdddejdddZddZddZd d Z e d d Z e j d d Z e ddZ e j ddZ ddZddZdS)ReportEventStackaContext 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 / :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. NcCs||_||_||_||_||_tj|_|dkrKg}||_|dkrx|rr|j }nd}||_ |rdj |j |f|_ n |j|_ i|_ dS)NT/) parentr rmessageresult_on_exceptionr&rr$r%reporting_enabledjoinfullnamechildren)r r rr@r?rBrAr%r r r rs$            zReportEventStack.__init__cCsd|j|j|jfS)Nz.ReportEventStack(%s, %s, reporting_enabled=%s))r rrB)r r r r __repr__szReportEventStack.__repr__cCsHtj|_|jr(t|j|j|jrDd|jj|j <|S)N)NN) r&rr$rBr<rDrr?rEr )r r r r __enter__s    zReportEventStack.__enter__cCslxYtjtjfD]E}x<|jjD]+\}\}}||kr)||jfSq)WqW|j|jfS)N)r&rrrEr0r@r$)r Z cand_result_namevalueZ_msgr r r _childrens_finish_infos " z'ReportEventStack._childrens_finish_infocCs|jS)N)_result)r r r r r$szReportEventStack.resultcCs)|tkrtd|||_dS)Nz'%s' not a valid result)r&r'rK)r rIr r r r$s cCs|jdk r|jS|jS)N)_messager)r r r r r@szReportEventStack.messagecCs ||_dS)N)rL)r rIr r r r@scCs |r|j|jfS|jS)N)rAr@rJ)r excr r r _finish_infoszReportEventStack._finish_infocCs`|j|\}}|jr7||f|jj|j<|jr\t|j||d|jdS)Nr%)rNr?rEr rBr:rDr%)r exc_type exc_value tracebackr$msgr r r __exit__s   zReportEventStack.__exit__)rrrrr&rrrFrGrJpropertyr$setterr@rNrSr r r r r=s      r=cCs|s dSg}x||D]t}tjj|s8d}n4t|d"}tj|jj}WdQRX|jd|d|ddiqW|S)Nrbpathcontentencodingbase64) osrWisfileopenrZ b64encodereaddecodeappend)r)retfnamerXfpr r r r*s  "r*)zSUCCESSzWARNzFAIL)rrZZos.pathr[rrrr#r;r setrr&objectrr!r7rr:r<r=r*r r r r  s      p