Your IP : 216.73.217.13
U
�Ad�& �
@ s d dl Z d dlZd dlmZ d dlmZ d dlmZmZ d dlm Z m
Z
m Z e� e
�ZdZdZdZd ZeegZd
gZd Zd Zd
dddd
dddddddd� Zeee�� �dd� d�ZG d d � d e j d �Z d d � Z d!d"� Z d#d$� Z d%d&� Z!d'd(� Z"d)d*� Z#d+d,� Z$d-d.� Z%e fd/d0�Z&d3d1d2�Z'dS )4� N)�importer)�log)�
type_utils�util)� FREQUENCIES�
PER_ALWAYS� PER_INSTANCEztext/x-not-multipartzapplication/octet-streamZ__end__Z __begin__ztext/part-handlerzpart-handler-%03dz part-%03dztext/x-include-urlztext/x-include-once-urlztext/x-shellscriptztext/cloud-configztext/cloud-boothookztext/cloud-config-archiveztext/cloud-config-jsonpz text/jinja2�text/x-shellscript-per-boot� text/x-shellscript-per-instance�text/x-shellscript-per-once) z#includez
#include-oncez#!z
#cloud-configz
#part-handlerz#cloud-boothookz#cloud-config-archivez#cloud-config-jsonpz## template: jinjar r
r c C s dt | � S )Nr )�len)�e� r �=/usr/lib/python3/dist-packages/cloudinit/handlers/__init__.py�<lambda>? � r )�keyc @ s4 e Zd Zd dd�Zdd� Zdd� Zejdd � �Zd
S ) �Handler� c C s || _ || _d S �N)�handler_version� frequency)�selfr �versionr r r �__init__D s zHandler.__init__c C s dt �| �| �� f S )Nz%s: [%s])r Zobj_name�
list_types�r r r r �__repr__H s zHandler.__repr__c C s, t | d�std��ndd� t| d�D �S d S )N�prefixesz#Missing prefixes subclass attributec S s g | ] }t | �qS r )�INCLUSION_TYPES_MAP)�.0�prefixr r r �
<listcomp>P s �z&Handler.list_types.<locals>.<listcomp>)�hasattr�NotImplementedError�getattrr r r r r K s
�zHandler.list_typesc O s
t � �d S r )r$ )r �args�kwargsr r r � handle_partU s zHandler.handle_partN)r ) �__name__�
__module__� __qualname__r r r �abc�abstractmethodr( r r r r r C s
r )� metaclassc C s | j }|tks"|tkr |tks"d S z| j}t|�}W n tttfk
rR d}Y nX |d }zxt� d| ||||� |dkr�| �
||||||� nB|dkr�| �
|||||� n&|dkr�| �
||||� n td| ��W n* t k
r� t �
td| ||||� Y nX d S )N� � Content-Typez1Calling handler %s (%s, %s, %s) with frequency %s� r zUnknown module version %sz8Failed calling handler %s (%s, %s, %s) with frequency %s)r r r r �int� TypeError�
ValueError�AttributeError�LOG�debugr( � Exceptionr �logexc) �mod�data�filename�payloadr �headersZmod_freqZmod_ver� content_typer r r �run_partZ s^ ���
� ��r@ c C s dt i}t| |d d ||� d S �Nr0 )�
CONTENT_STARTr@ �r: r; r r> r r r �
call_begin� s �rD c C s dt i}t| |d d ||� d S rA )� CONTENT_ENDr@ rC r r r �call_end� s �rF c
C s� | d }t | }| d }tj�| d d| �}|�d�s@d| }t�||d� | d }z<tt� |��} t
| | d |� |j | d
d � |d | d< W n$ t k
r� t�
td
||� Y nX d S )NZ handlercountr Z
handlerdirz%sz.pyz%s.pyi� �handlersr; T)Z initializedr/ z7Failed at registering python file: %s (part handler %s))�PART_HANDLER_FN_TMPL�os�path�join�endswithr Z
write_file�
fixup_handlerr �
import_modulerD �registerr8 r9 r6 )
ZpdataZ_ctypeZ _filenamer= Zcurcount�modnamer ZmodfnamerG r: r r r �walker_handle_handler� s(
�rQ c C s� z\t | t�r | �dd�d }n$| d d| � jdd��dd�d }t|�|krZ|d |� }W n tk
r| | d|� }Y nX |S )N�
r/ r � �ignore)�errors)�
isinstance�str�split�decoder �UnicodeDecodeError)Zblob�size�startr r r �_extract_first_or_bytes� s
$ r] c C sb z | � d�W S ttfk
rJ z| � d�W Y S tk
rD Y nX Y n tk
r\ Y nX | S )NZ
string_escapeZunicode_escape)�encode� LookupErrorr3 r5 )�textr r r �_escape_string� s
ra c C s� |d }|| � d�kr&t�d|� d S |tkr@t| |||� d S | d }||krpt|| | d ||| d |� nN|r�t|d�}dt|� }|tkr�t� d ||� q�t� d
||� n t�d |� d S ) Nr0 Zexcludedz content_type "%s" is excludedrG r; r � z'%s...'z)Unhandled non-multipart (%s) userdata: %sz0Unhandled unknown content-type (%s) userdata: %szEmpty payload of type %s)
�getr6 r7 �PART_CONTENT_TYPESrQ r@ r] ra �NOT_MULTIPART_TYPE�warning)r; r< r= r> r? rG r\ Zdetailsr r r �walker_callback� s@ �
��rg c C s� d}| � � D ]t}|�� dkr q |�� }|d kr2t}|�� }|sFt| }t|�}t�|� ||d< t �
|�}|||||� |d }q d S )Nr Z multipartr0 r/ ) �walkZget_content_maintypeZget_content_type�
OCTET_TYPE� get_filename� PART_FN_TPL�dictr6 r7 r Zfully_decoded_payload) �msg�callbackr; Zpartnum�partZctyper< r> r= r r r rh s
rh c C sR t | d�st| dd� t | d�s.t| d|� n | j}|rN|tkrNt�d| |� | S )Nr r/ r z&Handler %s has an unknown frequency %s)r# �setattrr r r6 rf )r: Zdef_freqZfreqr r r rM s
rM c C sX zt �| ��� }W n tk
r* | Y S X |�� }tD ]}|�|�r8t| S q8|S r )r Z
decode_binary�lowerrZ �lstrip�INCLUSION_SRCH�
startswithr )r= �defaultZ
payload_lcr` r r r �type_from_starts_with$ s
rv )N)(r, rI Z cloudinitr r Zloggingr r Zcloudinit.settingsr r r Z getLoggerr) r6 re ri rE rB ZCONTENT_SIGNALSrd rH rk r �sorted�list�keysrs �ABCMetar r@ rD rF rQ r] ra rg rh rM rv r r r r �<module> sT
�
�0)