U
    lHJeH  ã                
   @   sª   d dl Z d dlZd dlmZ d dlmZmZ e  e e	¡¡Z
zd dlZW nD ek
rŒ Z z&e
 e¡ eeejd e d¡ W 5 dZ[X Y nX dd„ Zd
dd	„ZejZdS )é    N)Úutil)ÚBROKEN_YAML_MODULEÚMISSING_YAML_MODULE©Úfileé   c              
   C   s`   zt  | ¡W S  tk
rZ } z0t |¡ ttjt jdt	j
d t	 d¡ W 5 d }~X Y nX d S ©N)Úpathr   r   )ÚyamlÚ	safe_loadÚAttributeErrorÚLOGÚ	exceptionÚprintr   ÚformatÚ__path__ÚsysÚstderrÚexit)ÚstreamÚe© r   ú//usr/lib/python3/dist-packages/uaclient/yaml.pyr      s    
r   c              
   K   sf   zt j| |f|ŽW S  tk
r` } z0t |¡ ttjt jdt	j
d t	 d¡ W 5 d }~X Y nX d S r   )r
   Ú	safe_dumpr   r   r   r   r   r   r   r   r   r   )Údatar   Úkwargsr   r   r   r   r      s    
r   )N)Zloggingr   Zuaclientr   Zuaclient.messagesr   r   Z	getLoggerZreplace_top_level_logger_nameÚ__name__r   r
   ÚImportErrorr   r   r   r   r   r   r   Úparserr   r   r   r   Ú<module>   s   
	
	