o w7e@sdZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z dZ ee Zd Zd d Zd d ZdddZdddZddZddZedkrWeedSdS)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 Csg}g}t|D]8\}\}}}|}t||t}|dd||} | tvr;d| |df} | | | |qt } |D]} | | qF| |fS)NzContent-Dispositionzattachment; filename="%s"z3content type %r for attachment %s may be incorrect!) enumeratereadrsysgetdefaultencoding add_headerget_content_typelowerget_content_typesappendrattach) files sub_messageserrorsifhfilename format_typecontents sub_message content_typemsgcombined_messager?/usr/lib/python3/dist-packages/cloudinit/cmd/devel/make_mime.pycreate_mime_messages(     r c CsRz|dd\}}t|d||fWSty(}z t|d||d}~ww)z@Return file content type by reading the first line of the input.:rrzInvalid value for %rN)splitopenstrip ValueErrorargparse ArgumentError)textrrerrrfile_content_type/sr+c Cs^|st}t|_t|_|jdddtdgddd|jdd d d d d |jddd d dd |S)aBuild 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:z3attach the given file as the specified content-type)desttypeactiondefaultmetavarhelpz-lz --list-types store_trueFz&List support cloud-init content types.)r.r/r1z-fz--forcez$Ignore unknown content-type warnings)r'ArgumentParser__doc__ descriptionEPILOGepilog add_argumentr+)parserrrr get_parser:s: r:FcstfddtDS)zReturn a list of cloud-init supported content types. Optionally strip out the leading 'text/' of the type if strip_prefix=True. cs g|] }r |ddn|qS)ztext/)replace).0ctype strip_prefixrr gsz%get_content_types..)sortedrvaluesr?rr?rrbs  rcCstt|jrtjntj|jrtdt dddSt |j \}}|rG|j r)dnd}|D]}t j|d|dq-t jd|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?rWARNINGERRORz: z-Invalid content-types, override with --force r)rLOGdebugrDEBUGrE list_typesprintjoinrr rforcer stderrwrite)nameargsrrlevelerrorrrr handle_argsns  rTcCst}tt|SN)r: parse_argsrTNAME)rQrrrmains  rX__main__rU)F)r4r'r email.mime.multipartremail.mime.textr cloudinitrcloudinit.cmd.develrcloudinit.handlersrrW getLoggerrGr6r r+r:rrTrX__name__exitrrrrs*       (