File: //proc/self/root/lib/python3/dist-packages/cloudinit/handlers/__pycache__/__init__.cpython-38.pyc
U
Ӈg}& �
@ s d dl Z d dlZd dlZd dlmZmZmZ d dlmZm Z m
Z
e� e �Z
dZdZdZdZeegZdgZd Zd
Zd d d
dddddddddd� Zeee�� �dd� d�ZG dd� de jd �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-d.�Z%d1d/d0�Z&dS )2� N)�importer�
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__C s zHandler.__init__c C s dt �| �| �� f S )Nz%s: [%s])r Zobj_name�
list_types�r r
r
r �__repr__G 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>O s �z&Handler.list_types.<locals>.<listcomp>)�hasattr�NotImplementedError�getattrr r
r
r r J s
�zHandler.list_typesc O s
t � �d S r )r# )r �args�kwargsr
r
r � handle_partT s zHandler.handle_partN)r ) �__name__�
__module__� __qualname__r r r �abc�abstractmethodr' r
r
r
r r B 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_partY s^ ���
� ��r? c C s dt i}t| |d d ||� d S �Nr/ )�
CONTENT_STARTr? �r9 r: r r= r
r
r �
call_begin� s �rC c C s dt i}t| |d d ||� d S r@ )� CONTENT_ENDr? rB r
r
r �call_end� s �rE 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_modulerC �registerr7 r8 r5 )
ZpdataZ_ctypeZ _filenamer< Zcurcount�modnamer ZmodfnamerF r9 r
r
r �walker_handle_handler� s(
�rP 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� LookupErrorr2 r4 )�textr
r
r �_escape_string� s
r` 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 ) Nr/ Zexcludedz content_type "%s" is excludedrF r: r � z'%s...'z)Unhandled non-multipart (%s) userdata: %sz0Unhandled unknown content-type (%s) userdata: %szEmpty payload of type %s)
�getr5 r6 �PART_CONTENT_TYPESrP r? r\ r` �NOT_MULTIPART_TYPE�warning)r: r; r< r= r> rF r[ Zdetailsr
r
r �walker_callback� s@ �
��rf c C s� d}| � � D ]t}|�� dkr q |�� }|d kr2t}|�� }|sFt| }t|�}t�|� ||d< t �
|�}|||||� |d }q d S )Nr Z multipartr/ r. ) �walkZget_content_maintypeZget_content_type�
OCTET_TYPE� get_filename� PART_FN_TPL�dictr5 r6 r Zfully_decoded_payload) �msg�callbackr: Zpartnum�partZctyper; r= r< r
r
r rg s
rg 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 r5 re )r9 Zdef_freqZfreqr
r
r rL s
rL c C sX zt �| ��� }W n tk
r* | Y S X |�� }tD ]}|�|�r8t| S q8|S r )r Z
decode_binary�lowerrY �lstrip�INCLUSION_SRCH�
startswithr )r< �defaultZ
payload_lcr_ r
r
r �type_from_starts_with# s
ru )N)'r+ ZloggingrH Z cloudinitr r r Zcloudinit.settingsr r r Z getLoggerr( r5 rd rh rD rA ZCONTENT_SIGNALSrc rG rj r �sorted�list�keysrr �ABCMetar r? rC rE rP r\ r` rf rg rL ru r
r
r
r �<module> sR
�
�0)