3
\meX @ s d dl Z d dlmZ d dlmZ d dlmZ d dlmZmZ d dl m
Z
m Z e j e
Zeej ddd Zeed
d d Ze
jed
ddZedddZdS ) N)datetime)Optional)version)handlers instantiated_handler_registry)errorsidentity)returnc C s t jjd} t| tjs dS | S )z'Get instantiated KVP telemetry handler.Z telemetryN)r Zregistered_itemsget
isinstancer HyperVKvpReportingHandler) kvp_handler r /usr/lib/python3.6/kvp.pyget_kvp_handler s
r )reportr c C s, t }|dkr tjd dS |jd| dS )z/Report to host via PROVISIONING_REPORT KVP key.Nz.KVP handler not enabled, skipping host report.FZPROVISIONING_REPORTT)r LOGdebugZ write_key)r r
r r r report_via_kvp s
r )errorr c C s t | j S )N)r Zas_encoded_report)r r r r report_failure_to_host& s r c C sr y t j } W n, tk
r8 } zd|} W Y d d }~X nX tjddtj dtj 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 ZutcnowZ isoformatr )Zvm_idZid_errorr r r r report_success_to_host* s r )Zloggingr Ztypingr Z cloudinitr Zcloudinit.reportingr r Zcloudinit.sources.azurer r Z getLogger__name__r r r strboolr ZReportableErrorr r r r r r s