File: //lib/python3/dist-packages/cloudinit/config/__pycache__/cc_write_files.cpython-38.pyc
U
Ӈg� � @ s� U d Z ddlZddlZddlZddlmZ ddlmZmZ ddl m
Z
ddl m Z ddl
mZ ddlmZ d Zd
Zd Ze�e�Zd d
gedgd�Zeed< ee e
edd�dd�Zdd� Z d eee d�dd�Z dd� Z dd� Z d d � Z!dS ) z"Write Files: write arbitrary files� N)�Optional)�
url_helper�util)�Cloud)�Config)�
MetaSchema)� PER_INSTANCEi� Fz
text/plainZcc_write_files�all� write_files)�idZdistrosZ frequencyZactivate_by_schema_keys�meta)�name�cfg�cloud�args�returnc C sP |� dg �}dd� |D �}|s.t�d| � d S t�|j�}t| ||jj|� d S )Nr
c S s g | ]}t �|d t�s|�qS )Zdefer)r �get_cfg_option_bool�
DEFAULT_DEFER)�.0�f� r �A/usr/lib/python3/dist-packages/cloudinit/config/cc_write_files.py�
<listcomp>$ s �zhandle.<locals>.<listcomp>zESkipping module named %s, no/empty 'write_files' key in configuration) �get�LOG�debugr Zfetch_ssl_details�pathsr
ZdistroZ
default_owner)r
r r r Z file_listZfiltered_files� ssl_detailsr r r �handle" s �� r c C sf | sd} | � � �� } | dkr"dgS | dkr2ddgS | dkr@dgS | tkrNtgS | r`t�d| t� tgS )N� )ZgzZgzip�application/x-gzip)z gz+base64z gzip+base64zgz+b64zgzip+b64�application/base64)Zb64�base64z%Unknown encoding type %s, assuming %s)�lower�strip�TEXT_PLAIN_ENCr �warning)Z
encoding_typer r r �canonicalize_extraction4 s$ �r' )�ownerr c C s� |sd S t |�D ]�\}}|�d�}|s:t�d|d | � qtj�|�}t|�dd �||�dd �|�dd ��}|d kr�t�d|d | � qt� |�d|��\}} t
|�d �t �}
t� |d
�r�d nd } tj
||| |
|| d
� t�||| � qd S )N�pathz3No path provided to write for entry %s in module %s� �source�content�encodingz>No content could be loaded for entry %s in module %s; skippingr( Z permissions�appendZab�wb)�omode�mode�user�group)� enumerater r r& �osr) �abspath�read_url_or_decoder Zextract_usergroup� decode_perms�
DEFAULT_PERMSr Z
write_fileZ chownbyname) r
�filesr( r �iZf_infor) �contents�u�gZpermsr0 r r r r
M sJ
�
�� �c C s� | d kr |S z,t | ttf�r&t| �W S tt| �d�W S W nr ttfk
r� g }| |fD ]:}z|�d| � W qX tk
r� |�d| � Y qXX qXtjd|�� | Y S X d S )N� z%oz%r�0Undecodable permissions %s, returning default %s)r@ ) �
isinstance�int�float�str� TypeError�
ValueErrorr. r r& )Zperm�defaultZreps�rr r r r8 u s
r8 c C s� | d kr d n
| � dd �}t|�}|d kr0|s0dS d }|r�z"tj|| � dd �dd|d�j}W n& tk
r� t�td|� d}Y nX |d k r�|s�t |�}t
||�}|S )NZurir �headers� )rI ZretriesZ sec_betweenr zVFailed to retrieve contents from source "%s"; falling back to data from "contents" keyF) r �boolr Zread_file_or_urlr< � Exceptionr Zlogexcr r' �extract_contents)r+ r r, r- ZurlZuse_url�resultZ extractionsr r r r7 � s2
� �
r7 c C sH | }|D ]:}|dkr&t j|ddd�}q|dkr:t�|�}q|tkrq|S )Nr F)�quiet�decoder! )r Z decomp_gzipr" Z b64decoder% )r< Zextraction_typesrN �tr r r rM � s rM )N)"�__doc__r" Zloggingr5 �typingr Z cloudinitr r Zcloudinit.cloudr Zcloudinit.configr Zcloudinit.config.schemar Zcloudinit.settingsr r9 r r% Z getLogger�__name__r r �__annotations__rD �listr r' �dictr
r8 r7 rM r r r r �<module> s0
�(%