File: //lib/python3.6/site-packages/cloudinit/cmd/devel/__pycache__/make_mime.cpython-36.opt-1.pyc
3
\�meZ � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z dZ
ej e �Z
dZdd � Zd
d � Zdd d
�Zddd�Zdd� Zdd� Ze dkr�eje� � dS )z0Generate multi-part mime messages for user-data.� N)�
MIMEMultipart)�MIMEText)�INCLUSION_TYPES_MAPz make-mimezUExample: make-mime -a config.yaml:cloud-config -a script.sh:x-shellscript > user-datac C s� g }g }x|t | �D ]p\}\}}}|j� }t||tj� �}|jdd| � |j� j� } | t� krxd| |d f }
|j |
� |j |� qW t
� } x|D ]}
| j |
� q�W | |fS )NzContent-Dispositionzattachment; filename="%s"z3content type %r for attachment %s may be incorrect!� ) � enumerate�readr �sys�getdefaultencodingZ
add_headerZget_content_type�lower�get_content_types�appendr Zattach) �filesZ sub_messages�errors�iZfh�filenameZ format_type�contentsZ sub_message� content_type�msg�combined_message� r � /usr/lib/python3.6/make_mime.py�create_mime_message s$
r c C s^ y$| j dd�\}}t|d�||j� fS tk
rX } ztj| d| �|�W Y dd}~X nX dS )z@Return file content type by reading the first line of the input.�:r �rzInvalid value for %rN)�split�open�strip�
ValueError�argparseZ
ArgumentError)�textr r �er r r �file_content_type. s
r! c
C s^ | s t j� } t| _t| _| jdddtdg ddd� | jdd d
d d d
� | jddd
d dd
� | S )a Build or extend and arg parser for make-mime utility.
@param parser: Optional existing ArgumentParser instance representing the
subcommand which will be extended to support the args of this utility.
@returns: ArgumentParser with proper argument configuration.
z-az--attachr
r z<file>:<content-type>z3attach the given file as the specified content-type)�dest�type�action�default�metavar�helpz-lz --list-types�
store_trueFz&List support cloud-init content types.)r$ r% r' z-fz--forcez$Ignore unknown content-type warnings)r �ArgumentParser�__doc__� description�EPILOGZepilog� add_argumentr! )�parserr r r �
get_parser9 s4 r/ Fc s t � fdd�tj� D ��S )z�Return a list of cloud-init supported content types. Optionally
strip out the leading 'text/' of the type if strip_prefix=True.
c s g | ]}� r|j d d�n|�qS )ztext/� )�replace)�.0Zctype)� strip_prefixr r �
<listcomp>g s z%get_content_types.<locals>.<listcomp>)�sortedr �values)r3 r )r3 r r a s
r c C s� |j r tdjtdd��� dS t|j�\}}|rz|jr:dnd}x$|D ] }tjj |� d|� d�� qDW tjj d� |jszd S t|� dS )
a< Create a multi-part MIME archive for use as user-data. Optionally
print out the list of supported content types of cloud-init.
Also setup CLI log handlers to report to stderr since this is a development
utility which should be run by a human on the CLI.
@return 0 on success, 1 on failure.
�
T)r3 r ZWARNINGZERRORz: z-Invalid content-types, override with --force
r )
Z
list_types�print�joinr r r
�forcer �stderr�write)�name�argsr r �level�errorr r r � handle_argsm s
rA c C s t � j� } tt| �S )N)r/ �
parse_argsrA �NAME)r> r r r �main� s
rD �__main__)N)F)r* r Zloggingr Zemail.mime.multipartr Zemail.mime.textr Zcloudinit.handlersr rC Z getLogger�__name__ZLOGr, r r! r/ r rA rD �exitr r r r �<module> s"
(