o

x[hZ�@s�dZd
dl
Z
d
dlZd
dlZd
dlmZ
d
dlmZ
d
dlm	Z	
dZ
e�e�
Z
dZdd	�Zd
d�Zddd
�
Zddd�
Zdd�Zdd�ZedkrOe�e��

dSdS)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
Cs�g}
g}t|�
D]8\}\}}}|�
�}t||t���}|�d
d|�
|����}	|	t�v
r;d|	|df}
|�	|
�

|
�	|�

qt
�}|
D]}
|�|
�

qF||fS)NzContent-Dispositionzattachment; filename="%s"z3content type %r for attachment %s may be incorrect!�
)�	enumerate�readr�sys�getdefaultencoding�
add_header�get_content_type�lower�get_content_types�appendr�attach)�files�sub_messages�errors�
i�fh�filename�format_type�contents�sub_message�content_type�msg�combined_message�r�?/usr/lib/python3/dist-packages/cloudinit/cmd/devel/make_mime.py�create_mime_messages(






�




�





rc

CsRz|�d
d�\}
}t
|
d�|
|��fWSty(
}
z	t�|d|�|�d}~w
w)z@Return file content type by reading the first line of the input.�
:r�
rzInvalid value for %rN)�split�open�strip�
ValueError�argparse�
ArgumentError)�textrr�
errr�file_content_type.s




����r)c


Cs^|st�
�}t|_t|_|jd
ddtdgddd�
|jdd	d
ddd
�
|jddd
ddd
�
|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--attachrrz<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�EPILOG�epilog�add_argumentr))
�parserrrr�
get_parser9s:










�





�




�r8Fc

st�f
d
d�t
��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|
�dd
�n
|
�qS)ztext/�)
�replace)�.0�ctype�
�strip_prefixrr�
<listcomp>fs��z%get_content_types.<locals>.<listcomp>)�sortedr�valuesr=rr=rr
as


��r
cCs�|
jrt
d
�tdd�
�
�

dSt|
j�
\}}|r;|
jrdn
d}|D]}tj�	|�d|�d
��

q!tj�	d�

|
js;d	St
|�

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.
    �

Tr=r
�WARNING�ERRORz: z-Invalid content-types, override with --force
r)
�
list_types�print�joinr
rr�forcer�stderr�write)�name�argsrr�level�errorrrr�handle_argsms	










rOc
Cst��
�}tt|�S�
N)r8�
parse_argsrO�NAME)
rLrrr�main�s



rS�__main__rP)
F)r2r%�loggingr�email.mime.multipartr�email.mime.textr�cloudinit.handlersrrR�	getLogger�__name__�LOGr4rr)r8r
rOrS�exitrrrr�<module>
s(





�

(
�