U
    lHJeA                      @   s   d Z ddlZddlZddlZddlmZmZmZmZm	Z	m
Z
 ddlmZ dZee
eeeef f  Zdadd ZejG dd	 d	ejZeeef eeef d
ddZG dd dZdS )a$  
This module is responsible for handling all events
that must be raised to the user somehow. The main idea
behind this module is to centralize all events that happens
during the execution of Pro commands and allows us to report
those events in real time or through a machine-readable format.
    N)AnyDictListOptionalSetUnion)	safe_dumpz0.1c                   C   s   t d krt a t S N)_event_loggerEventLogger r   r   7/usr/lib/python3/dist-packages/uaclient/event_logger.pyget_event_logger   s    r   c                   @   s"   e Zd ZdZe Ze Ze ZdS )EventLoggerModea  
    Defines event logger supported modes.
    Currently, we only support the cli and machine-readable mode. On cli mode,
    we will print to stdout/stderr any event that we receive. Otherwise, we
    will store those events and parse them for the specified format.
    N)__name__
__module____qualname____doc__objectCLIJSONYAMLr   r   r   r   r      s   r   )statusreturnc                 C   sh   ddl m} dd t|  D | d< | dd | dg  | dg D ]}d	|krL|d	 qL| S )
Nr   )get_pro_environmentc                 S   s   g | ]\}}||d qS ))namevaluer   ).0r   r   r   r   r   
<listcomp>/   s   z2format_machine_readable_output.<locals>.<listcomp>Zenvironment_varsorigin servicesZvariants)uaclient.utilr   sorteditemspop
setdefaultget)r   r   servicer   r   r   format_machine_readable_output,   s    
r)   c                
   @   sB  e Zd Zdd Zdd ZedddZedd	d
Ze	dddZ
d/eee dddZd0eee ee	eef  ee ee ee	eef  dddZd1eee ee ee ee	eef  dddZd2eee dddZedddZee ddd Zedd!d"Zed#d$d%Zd&d' Zd(d) Zd*d+ Zdd,d-d.ZdS )3r   c                 C   s:   g | _ g | _t | _t | _d| _d| _i | _tj	| _
d S )NFr    _error_events_warning_eventsset_processed_services_failed_services_needs_reboot_command_output_contentr   r   _event_logger_modeselfr   r   r   __init__F   s    zEventLogger.__init__c                 C   s:   g | _ g | _t | _t | _d| _d| _i | _tj	| _
dS )z/Reset the state of the event logger attributes.Fr    Nr*   r4   r   r   r   resetS   s    zEventLogger.reset)
event_modec                 C   s
   || _ dS )z_Set the event logger mode.

        We currently support the CLI, JSON and YAML modes.
        N)r3   )r5   r8   r   r   r   set_event_mode^   s    zEventLogger.set_event_mode)commandc                 C   s
   || _ dS )zSet the event logger command.

        The command will tell the process_events method which output method
        to use.
        N)r1   )r5   r:   r   r   r   set_commande   s    zEventLogger.set_command)output_contentc                 C   s
   || _ dS )zSet the event logger output content.

        The command will tell the process_events method which content
        to use.
        N)r2   )r5   r<   r   r   r   set_output_contentm   s    zEventLogger.set_output_contentN)info_msgendc                 C   s(   |s
t j}| jtjkr$t|||d dS )zL
        Print the info message if the event logger is on CLI mode.
        )filer?   N)sysstdoutr3   r   r   print)r5   r>   Z	file_typer?   r   r   r   infou   s    zEventLogger.infomsgr(   
event_dictcode
event_typeadditional_infoc                 C   s<   |d kr|rdnd}||||d}|r.||d< | | d S )Nr(   system)typer(   messageZmessage_coderJ   )append)r5   rF   r(   rG   rH   rI   rJ   Zevent_entryr   r   r   _record_dict_event   s    	zEventLogger._record_dict_event)	error_msg
error_coder(   
error_typerJ   c                 C   s(   | j tjkr$| j||| j|||d dS )z
        Store an error in the event logger.

        However, the error will only be stored if the event logger
        is not on CLI mode.
        rE   N)r3   r   r   rO   r+   )r5   rP   rQ   r(   rR   rJ   r   r   r   error   s    zEventLogger.error)warning_msgr(   c                 C   s"   | j tjkr| j||| jd dS )z
        Store a warning in the event logger.

        However, the warning will only be stored if the event logger
        is not on CLI mode.
        )rF   r(   rG   N)r3   r   r   rO   r,   )r5   rT   r(   r   r   r   warning   s    zEventLogger.warningr(   c                 C   s   | j | d S r	   )r.   addr5   r(   r   r   r   service_processed   s    zEventLogger.service_processed)r!   c                 C   s   | j | d S r	   )r/   update)r5   r!   r   r   r   services_failed   s    zEventLogger.services_failedc                 C   s   | j | d S r	   )r/   rW   rX   r   r   r   service_failed   s    zEventLogger.service_failed)reboot_requiredc                 C   s
   || _ d S r	   )r0   )r5   r]   r   r   r   needs_reboot   s    zEventLogger.needs_rebootc                 C   s"   dd | j D }tt| j|S )Nc                 S   s   h | ]}|d  r|d  qS rV   r   )r   rS   r   r   r   	<setcomp>   s   z8EventLogger._generate_failed_services.<locals>.<setcomp>)r+   listr-   unionr/   )r5   Zservices_with_errorr   r   r   _generate_failed_services   s    z%EventLogger._generate_failed_servicesc                 C   sV   t | jsdndt| jt|  | j| j| jd}ddlm} t	t
j||dd d S )Nsuccessfailure)Z_schema_versionresultZprocessed_servicesZfailed_serviceserrorswarningsr^   r   DatetimeAwareJSONEncoderTclsZ	sort_keys)JSON_SCHEMA_VERSIONr+   r#   r.   rb   r,   r0   r"   ri   rC   jsondumps)r5   Zresponseri   r   r   r   _process_events_services   s    

z$EventLogger._process_events_servicesc                 C   s~   t | j}| jsdnd|d< | j|d< | j|d< | jtjkr^ddlm} t	t
j||dd	 n| jtjkrzt	t|d
d d S )Nrc   rd   re   rf   rg   r   rh   Trj   F)Zdefault_flow_style)r)   r2   r+   r,   r3   r   r   r"   ri   rC   rm   rn   r   r   )r5   outputri   r   r   r   _process_events_status   s    


  z"EventLogger._process_events_status)r   c                 C   s,   | j tjkr(| jdkr |   n|   dS )z
        Creates a json response based on all of the
        events stored in the event logger.

        The json response will only be created if the event logger
        is not on CLI mode.
        r   N)r3   r   r   r1   rq   ro   r4   r   r   r   process_events   s    

zEventLogger.process_events)NN)NNN)NNNN)N)r   r   r   r6   r7   r   r9   strr;   r   r=   r   rD   r   EventFieldErrorTyperO   rS   rU   rY   r[   r\   boolr^   rb   ro   rq   rr   r   r   r   r   r   E   sJ          r   )r   enumrm   rA   typingr   r   r   r   r   r   Zuaclient.yamlr   rl   rs   rt   r
   r   uniqueEnumr   r)   r   r   r   r   r   <module>   s    	 