3
\�me� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZ dZej e
�Z dd� Z dd� Z
edd fd
d �Zd d
� Zdd� Zefdd�ZdS )� N)� b64decode� b64encodei� c C s, t | �}y
|jd�S tk
r& |S X d S )Nzutf-8)r �decode�UnicodeDecodeError)�sourceZdecoded� r �#/usr/lib/python3.6/atomic_helper.py�b64d s
r c C s" t | t�s| jd�} t| �jd�S )Nzutf-8)�
isinstance�bytes�encoder r )r r r r �b64e s
r
�wbFc C s� |r4yt j| �}tj|j�}W n tk
r2 Y nX d }ydtjt jj| �d|d�}t j
d| |j ||t |�� |j
|� |j� t j|j |� t j|j | � W n: tk
r� } z |d k r�t j|j � |�W Y d d }~X nX d S )NF)�dir�delete�modezOAtomically writing to file %s (via temporary file %s) - %s: [%o] %d bytes/chars)�os�stat�S_IMODE�st_mode�OSError�tempfileZNamedTemporaryFile�path�dirname�LOG�debug�name�len�write�close�chmod�rename� Exception�unlink)�filenameZcontentr �omodeZ
preserve_modeZ file_statZtf�er r r �
write_file s2
r' c C s2 ydj t| ��S tk
r, dj t| ��S X dS )z1Handler for types which aren't json serializable.z
ci-b64:{0}z)Warning: redacted unserializable type {0}N)�formatr
�AttributeError�type)Z_objr r r �json_serialize_defaultD s r+ c C s t j| dddtd�S )z%Return data in nicely formatted json.� T�,�: )�indentZ sort_keysZ
separators�default)r- r. )�json�dumpsr+ )�datar r r �
json_dumpsL s r4 c C s t | t|�d d|d�S )N�
�w)r% r )r' r4 )r$ r3 r r r r �
write_jsonW s
r7 )r1 Zloggingr r r �base64r r Z
_DEF_PERMSZ getLogger�__name__r r r
r' r+ r4 r7 r r r r �<module> s
#