File: //snap/core20/2599/lib/python3/dist-packages/cloudinit/sources/azure/__pycache__/kvp.cpython-38.pyc
U
Ӈgk � @ s� d dl Z d dlmZmZ d dlmZ d dlmZ d dlmZm Z d dl
m Z m Z e �
e�Zeej d�dd �Zeed
�d d �Ze jed
�dd�Zed�dd�ZdS )� N)�datetime�timezone)�Optional)�version)�handlers� instantiated_handler_registry)�errors�identity)�returnc C s t j�d�} t| tj�s dS | S )z'Get instantiated KVP telemetry handler.Z telemetryN)r Zregistered_items�get�
isinstancer �HyperVKvpReportingHandler)� kvp_handler� r �=/usr/lib/python3/dist-packages/cloudinit/sources/azure/kvp.py�get_kvp_handler s � r )�reportr
c C s, t � }|dkr t�d� dS |�d| � dS )z/Report to host via PROVISIONING_REPORT KVP key.Nz.KVP handler not enabled, skipping host report.FZPROVISIONING_REPORTT)r �LOG�debugZ write_key)r r r r r �report_via_kvp s
r )�errorr
c C s t | �� �S )N)r Zas_encoded_report)r r r r �report_failure_to_host& s r c
C sv z t �� } W n, tk
r8 } zd|��} W 5 d }~X Y nX t�ddt�� � �dt�t j
�� � � �d| � �g�}t |�S )Nzfailed to read vm id: zresult=successzagent=Cloud-Init/z
timestamp=zvm_id=)
r Z query_vm_id� Exceptionr Z
encode_reportr Zversion_stringr Znowr ZutcZ isoformatr )Zvm_idZid_errorr r r r �report_success_to_host* s �� r )Zloggingr r �typingr Z cloudinitr Zcloudinit.reportingr r Zcloudinit.sources.azurer r Z getLogger�__name__r r
r �str�boolr ZReportableErrorr r r r r r �<module> s