3
\meZ @ s d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z dZ
ej e Z
dZdd Zd
d Zdd d
ZdddZdd Zdd Ze dkreje 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 |
qW | |fS )NzContent-Dispositionzattachment; filename="%s"z3content type %r for attachment %s may be incorrect! ) enumeratereadr sysgetdefaultencodingZ
add_headerZget_content_typelowerget_content_typesappendr Zattach) filesZ sub_messageserrorsiZfhfilenameZ format_typecontentsZ sub_message content_typemsgcombined_message r /usr/lib/python3.6/make_mime.pycreate_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)splitopenstrip
ValueErrorargparseZ
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:z3attach the given file as the specified content-type)desttypeactiondefaultmetavarhelpz-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__ descriptionEPILOGZepilog add_argumentr! )parserr r r
get_parser9 s4 r/ Fc s t fddtj D S )zReturn 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 dn|qS )ztext/ )replace).0Zctype) strip_prefixr r
g s z%get_content_types..)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_typesprintjoinr r r
forcer stderrwrite)nameargsr r levelerrorr 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 s"
(