�
܍~c� � �� � d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z dZ
ej e
� � Zd Zd
� Zd � Zdd �Zdd�Zd� Zd� Zedk r ej e� � � � dS dS )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 �
� g }g }t | � � D ]�\ }\ }}}|� � � }t ||t j � � � � }|� dd|z � � |� � � � � � } | t � � vr!d| �d|dz �d�}
|� |
� � |� |� � ��t � � } |D ]}
| � |
� � �| |fS )NzContent-Dispositionzattachment; filename="%s"z
content type z for attachment � z 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_messages �?/usr/lib/python3/dist-packages/cloudinit/cmd/devel/make_mime.py�create_mime_messager! s- � ��L�
�F�*3�E�*:�*:� )� )�&�� &�B��+��7�7�9�9�� �x��c�6L�6N�6N�O�O� �� � � !�#>�(�#K�
�
�
� #�3�3�5�5�;�;�=�=� � � 0� 2� 2� 2� 2� 2� � � �q�1�u�u�u�&�C�
�M�M�#� � � ����K�(�(�(�(�$����� %� %��� � ��$�$�$�$� �f� %�%� c �� � | � dd� � \ }}t |d� � ||� � � fS # t $ r }t j | d| z � � |�d}~ww xY 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)�textr r �es r �file_content_typer. . s� � ��!%���C��!3�!3� ��,��X�s�#�#�X�|�/A�/A�/C�/C�D�D�� � � � ��$� �(�D�1�
�
�� ��������s �=A �
A(�
A#�#A(c
� � | st j � � } t | _ t | _ | � dddt dg dd�� � | � dd d
d d �
� � | � ddd
d d�
� � | 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.)r2 r3 r5 z-fz--forcez$Ignore unknown content-type warnings)r* �ArgumentParser�__doc__� description�EPILOG�epilog� add_argumentr. )�parsers r �
get_parserr> 9 s� � � � +��(�*�*�� �F���F�M�
��� ��
�
���'�
B� � � � � ��� ����
5� � � � � ��� ����
3� � � � � �Mr" Fc �\ � � t � fd�t j � � 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 �D �� g | ] }�r|� d d� � n|�� S )ztext/� )�replace)�.0�ctype� strip_prefixs �r �
<listcomp>z%get_content_types.<locals>.<listcomp>f sA �� �
�
�
�� +7� A�E�M�M�'�2� &� &� &�E�
�
�
r" )�sortedr �values�rE s `r r r a sF �� � �
�
�
�
�,�3�5�5�
�
�
� � � r" c �� � t t |j r t j n t j � � |j r2t d� t d�� � � � � � dS t |j � � \ }}|r]|j rdnd}|D ]'}t j � |� d|� d�� � �(t j � d� � |j sd 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.
�
TrI r �WARNING�ERRORz: z-Invalid content-types, override with --force
r )r �LOG�debugr �DEBUGrL �
list_types�print�joinr r! r �forcer �stderr�write)�name�argsr r �level�errors r � handle_argsr[ m s� � � �S�t�z�B�#�)�)�s�{�C�C�C� �� �
�d�i�i�)�t�<�<�<�=�=�>�>�>��q� 2�4�:� >� >� ��f�
� � !�Z�4� � �W��� 4� 4�E� �J� � �� 2� 2�� 2� 2� 2� 3� 3� 3� 3� �
���I�J�J�J��z� ��1� �
���� �1r" c �l � t � � � � � } t t | � � S �N)r> �
parse_argsr[ �NAME)rX s r �mainr` � s' � � �<�<� "� "� $� $�D� �t�T� "� "�"r" �__main__r] )F)r8 r* r �email.mime.multipartr �email.mime.textr � cloudinitr �cloudinit.cmd.develr �cloudinit.handlersr r_ � getLoggerrN r: r! r. r> r r[ r` �__name__�exit� r" r �<module>rk s1 ��
7� 6� ����
�
�
�
� .� .� .� .� .� .� $� $� $� $� $� $� � � � � � � 0� 0� 0� 0� 0� 0� 2� 2� 2� 2� 2� 2����c�m�D����-� � &� &� &�,� � �%� %� %� %�P � � � �
�
�
�4#� #� #�
�z��� �C�H�T�T�V�V������ �r" |