File: //lib/python3.6/site-packages/cloudinit/cmd/devel/__pycache__/hotplug_hook.cpython-36.pyc
3
\�me�# � @ s0 d Z ddlZddlZddlZddlZddlZddlZddlmZm Z m
Z
ddl m Z m
Z
ddlmZ ddlmZ ddlmZ ddlmZmZ dd lmZ eje�Zd
Z d d d �Z G d
d� dej �Z G dd� de �Z de e j!fiZ"dd� Z#ee$d�dd�Z%ed�dd�Z&dd� Z'ed k�r,e � j(� Z)e'e e)� dS ) z)Handle reconfiguration on hotplug events.� N)�log� reporting�stages)�
EventScope� EventType)�read_sys_net_safe)�parse_net_config_data)�events)�
DataSource�DataSourceNotFoundException)�Initz hotplug-hookc C s� | st jttd�} t| _| jdddddgd� | jdd d
�}d|_|jd d d
� |jddd
�}|jdddddd� |jddddddgd� | S )a Build or extend an arg parser for hotplug-hook 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.
)�prog� descriptionz-sz --subsystemTzsubsystem to act on�net)�required�help�choiceszHotplug Action�hotplug_action)�title�dest�queryz0Query if hotplug is enabled for given subsystem.)r ZhandlezHandle the hotplug event.z-dz --devpath�PATHz Sysfs path to hotplugged device)r �metavarr z-uz --udevactionzSpecify action to take.�add�remove) �argparse�ArgumentParser�NAME�__doc__r � add_argumentZadd_subparsersr Z
add_parser)�parserZ
subparsersZ
parser_handle� r! �"/usr/lib/python3.6/hotplug_hook.py�
get_parser s:
r# c @ s` e Zd Zdd� Zejdd� �Zeejdd� ��Zeje d�dd ��Z
d
d � Z d d
� Z dd� Z
dS )�
UeventHandlerc C s" || _ || _|| _|| _|| _d S )N)�id�
datasource�devpath�action�
success_fn)�selfr% r& r' r( r) r! r! r" �__init__K s
zUeventHandler.__init__c C s
t � �d S )N)�NotImplementedError)r* r! r! r" �applyR s zUeventHandler.applyc C s
t � �d S )N)r, )r* r! r! r" �configV s zUeventHandler.config)�returnc C s
t � �d S )N)r, )r* r! r! r" �device_detected[ s z UeventHandler.device_detectedc C sP d }| j dkrd}n | j dkr$d}ntd| j ��|| j� krLtd| j ��d S )Nr Tr FzUnknown action: %sz'Failed to detect %s in updated metadata)r( �
ValueErrorr0 � RuntimeErrorr% )r* Zdetect_presencer! r! r" �detect_hotplugged_device_ s
z&UeventHandler.detect_hotplugged_devicec C s | j � S )N)r) )r* r! r! r" �successm s zUeventHandler.successc C s, | j jtjg�}|s(td| j tjf ��|S )Nz&Datasource %s not updated for event %s)r& Z update_metadata_if_supportedr �HOTPLUGr2 )r* �resultr! r! r" �update_metadatap s
z UeventHandler.update_metadataN)�__name__�
__module__� __qualname__r+ �abc�abstractmethodr- �propertyr. �boolr0 r3 r4 r7 r! r! r! r" r$ J s r$ c s> e Zd Z� fdd�Zdd� Zedd� �Zed�dd �Z� Z S )
�
NetHandlerc s* t tjj|�d�}t� j|||||� d S )NZaddress)r �os�path�basename�superr+ )r* r& r' r( r) r% )� __class__r! r" r+ } s zNetHandler.__init__c C s| | j jj| jdd� tjj| j�}| j jj� }| j dkrT|j
|�sxt dj | j���n$| j dkrx|j
|�sxt dj | j���d S )NF)Zbring_upr z Failed to bring up device: {}r z Failed to bring down device: {})r& ZdistroZapply_network_configr. r@ rA rB r' Znetwork_activatorr( Zbring_up_interfacer2 �formatZbring_down_interface)r* Zinterface_nameZ activatorr! r! r" r- � s
zNetHandler.applyc C s | j jS )N)r& Znetwork_config)r* r! r! r" r. � s zNetHandler.config)r/ c s< t � j�}� fdd�|j� D �}tjd� j|� t|�dkS )Nc s g | ]}|j d �� jkr|�qS )Z mac_address)�getr% )�.0Ziface)r* r! r" �
<listcomp>� s z.NetHandler.device_detected.<locals>.<listcomp>zIfaces with ID=%s : %sr )r r. Ziter_interfaces�LOG�debugr% �len)r* Znetstate�foundr! )r* r" r0 � s
zNetHandler.device_detected)
r8 r9 r: r+ r- r= r. r> r0 �
__classcell__r! r! )rD r" r? | s r? r c C s\ yt | d }W n2 tk
rB } ztdj|��|�W Y d d }~X nX tj| j| jtj |d�S )N� z4hotplug-hook: cannot handle events for subsystem: {})r& �cfgZevent_source_type�scope)
�SUBSYSTEM_PROPERTES_MAP�KeyErrorr2 rE r Zupdate_event_enabledr& rO r r5 )� hotplug_init� subsystemrP �er! r! r" �
is_enabled� s rV )rS rT c C sR t jd� | jdd�}|jtjg�s4t jd|� d S t| |�sNt jd|� d S |S )NzFetching datasourceZtrust)Zexistingz*hotplug not supported for event of type %sz(hotplug not enabled for event of type %s)rI rJ ZfetchZget_supported_eventsr r5 rV )rS rT r& r! r! r" �initialize_datasource� s
rW )rS c C s t | |�}|sd S t| d }tjd|� ||||| jd�}dddddg}td �}x�t|�D ]�\} }
tjd
|| t|�� yTtjd � |j� |j s�tjd � |j
� tjd
� |j � tjd� |j � P W q\ tk
�r
} z tjd| � t
j|
� | }W Y d d } ~ X q\X q\W |�d S )Nr zCreating %s event handler)r& r' r( r) rN � � �
� z#Bug while processing hotplug event.z!subsystem=%s update attempt %s/%szRefreshing metadataz$Detecting device in updated metadatazApplying config changezUpdating cachez,Exception while processing hotplug event. %s)rW rQ rI rJ Z_write_to_cache� Exception� enumeraterK r7 Zskip_hotplug_detectr3 r- r4 �timeZsleep) rS r' rT �
udevactionr& Z handler_clsZ
event_handlerZ
wait_timesZlast_exceptionZattempt�waitrU r! r! r" �handle_hotplug� sD
ra c C s8 t j| tdd�}tg |d�}|j� tj|j� d|jkrLtj |jj
d�� t j d| |j
|jd|krj|jnd d|krz|jnd � |�� yn|j
dkr�yt||j�}W n& tk
r� td � tjd
� Y nX t|r�d nd � nt||j|j|jd
� W n" tk
�r t jd� � Y nX W d Q R X t j d� tj� d S )NT)Zreporting_enabled)Zds_depsZreporterr zh%s called with the following arguments: {hotplug_action: %s, subsystem: %s, udevaction: %s, devpath: %s}r_ r' r z9Unable to determine hotplug state. No datasource detectedrN ZenabledZdisabled)rS r' rT r_ z*Received fatal exception handling hotplug!zExiting hotplug handler)r ZReportEventStackr r Zread_cfgr Z
setup_loggingrO r Zupdate_configurationrF rI rJ r rT r_ r' rW r �print�sys�exitra r\ Z exceptionZ flush_events)�name�argsZhotplug_reporterrS r& r! r! r" � handle_args� sF
rg �__main__)N)*r r; r Zloggingr@ rc r^ Z cloudinitr r r Zcloudinit.eventr r Z
cloudinit.netr Zcloudinit.net.network_stater Zcloudinit.reportingr Zcloudinit.sourcesr
r Zcloudinit.stagesr Z getLoggerr8 rI r r# �ABCr$ r? ZNETWORKrQ rV �strrW ra rg �
parse_argsrf r! r! r! r" �<module> s4
22((5