o

�Ad��@s�dZd
dl
Z
d
dlZd
dlmZ
d
dlmZ
d
dlmZ
d
dl	m
Z

d
dlmZ
dZ
e�e
�
Zd	Zd
d�Zdd
�Zddd�
Zddd�
Zdd�Zdd�ZedkrWe�e��

dSdS)z0Generate multi-part mime messages for user-data.�N)
�
MIMEMultipart)
�MIMEText)
�log)
�addLogHandlerCLI)
�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&






�



�





r c

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/r1z-fz--forcez$Ignore unknown content-type warnings)r'�ArgumentParser�__doc__�description�EPILOG�epilog�add_argumentr+)
�parserrrr�
get_parser9s:










�





�




�r:Fc

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?rras


��rcCs�tt
|
jrtjntj�
|
jrtd
�t	dd�
�
�

dSt
|
j�
\}}|rG|
jr)dn
d}|D]}t
j�|�d|�d
��

q-t
j�d�

|
jsGd	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)r�LOG�debugr�DEBUGrE�
list_types�print�joinrr r�forcer
�stderr�write)�name�argsrr�level�errorrrr�handle_argsms	











rTc
Cst��
�}tt|�S�
N)r:�
parse_argsrT�NAME)
rQrrr�main�s



rX�__main__rU)
F)r4r'r
�email.mime.multipartr�email.mime.textr�	cloudinitr�cloudinit.cmd.develr�cloudinit.handlersrrW�	getLoggerrGr6r r+r:rrTrX�__name__�exitrrrr�<module>
s*






�

(
�