관리-도구
편집 파일: make_mime.cpython-39.pyc
a '�DgZ � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z dZ e� e �Z dZdd � Zd d � Zdd d �Zddd�Zdd� Zdd� Ze dkr�e�e� � 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 }t | �D ]p\}\}}}|�� }t||t�� �}|�dd| � |�� �� } | t� vrvd| |d f } |� | � |� |� qt � } |D ]} | � | � q�| |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 �A/usr/lib/python3.9/site-packages/cloudinit/cmd/devel/make_mime.py�create_mime_message s( � � r c C sb z&| � dd�\}}t|d�||�� fW S ty\ } z t�| d| �|�W Y d}~n d}~0 0 dS )z@Return file content type by reading the first line of the input.�:r �rzInvalid value for %rN)�split�open�strip� ValueError�argparse� ArgumentError)�textr r �er r r �file_content_type. s ��r" c C s^ | s t �� } 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�EPILOG�epilog� add_argumentr" )�parserr r r � get_parser9 s: � ��r1 Fc s t � fdd�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|� d d�n|�qS )ztext/� )�replace)�.0Zctype�Z strip_prefixr r � <listcomp>f s �z%get_content_types.<locals>.<listcomp>)�sortedr �valuesr5 r r5 r r a s ��r c C s� |j r td�tdd��� dS t|j�\}}|rv|jr:dnd}|D ] }tj� |� d|� d�� qBtj� d� |jsvd 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. � Tr5 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 rC c C s t � �� } tt| �S )N)r1 � parse_argsrC �NAME)r@ r r r �main� s rF �__main__)N)F)r+ r Zloggingr Zemail.mime.multipartr Zemail.mime.textr Zcloudinit.handlersr rE Z getLogger�__name__ZLOGr- r r" r1 r rC rF �exitr r r r �<module> s$ � (