File: //lib/python3.9/site-packages/ansible/plugins/action/__pycache__/reboot.cpython-39.pyc
a
�)gXR � @ s� d dl mZmZmZ eZd dlZd dlZd dlmZm Z d dl
m Z m Z d dl
mZmZ d dlmZmZ d dlmZ d dlmZ e� ZG d d
� d
e�ZG d d � d e�ZdS )
� )�absolute_import�division�print_functionN)�datetime� timedelta)� AnsibleError�AnsibleConnectionFailure)� to_native�to_text)�check_type_list�check_type_str)�
ActionBase)�Displayc @ s e Zd ZdS )�TimedOutExceptionN)�__name__�
__module__� __qualname__� r r �A/usr/lib/python3.9/site-packages/ansible/plugins/action/reboot.pyr s r c s e Zd ZdZed�ZdZdZdZdZ dZ
dZ dZ d Z
d
Zd Zi Zd d d
d
d
dd
d�Zddd�Zdddeddddddd�
Zddd�Z� fdd �Zed d � �Zed d � �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Z d)d*� Z d+d,� Z d-d.� Z d/d0� Z d1d2� Z!d;d3d4�Z"d5d6� Z#d<d7d8�Z$d=� fd9d:� Z%� Z&S )>� ActionModuleF) �boot_time_command�connect_timeout�msg�post_reboot_delay�pre_reboot_delay�reboot_command�reboot_timeout� search_paths� test_commandiX Nr Zwhoamiz#cat /proc/sys/kernel/random/boot_idzReboot initiated by AnsibleZshutdownz-r {delay_min} "{message}"Tz/sbin/sysctl kern.boottimezwho -bz/grep booted /var/log/vmksummary.log | tail -n 1)�freebsd�openbsd�macosx�solaris�sunos�vmkernel�aix�reboot)�alpiner$ � z-r +{delay_min} "{message}"z -r +{delay_sec}s "{message}"z"-y -g {delay_sec} -i 6 "{message}"z-d {delay_sec}z-Fr)
r' Zvoidr Zlinuxr! r r" r# r$ r% Zwho)r" r$ c s t t| �j|i |�� d S )N)�superr �__init__)�self�args�kwargs�� __class__r r r* V s zActionModule.__init__c C s | � d| j�S )Nr )� _check_delay�DEFAULT_PRE_REBOOT_DELAY�r+ r r r r Y s z ActionModule.pre_reboot_delayc C s | � d| j�S )Nr )r0 �DEFAULT_POST_REBOOT_DELAYr2 r r r r ] s z ActionModule.post_reboot_delayc C s4 t | jj�|| jj�|d |���}|dk r0d}|S )z)Ensure that the value is positive or zeroZ_secr )�int�_taskr, �get)r+ �key�default�valuer r r r0 a s $zActionModule._check_delayc C sD t | |�}|�|d |d |�|d |�|d t | |����}|S )zXGet dist+version specific args first, then distribution, then family, lastly use default�name�version�family)�getattrr6 )r+ Z
variable_name� distributionZ
default_value�attrr9 r r r �_get_value_from_factsh s
���z"ActionModule._get_value_from_factsc
C s� | j j�d�}|d ur�zt|dd�}W n4 tyZ } z tdt|� ��W Y d }~n
d }~0 0 z|�dd�d W S ty� Y dS 0 n<| � d|d �}| j
d
}| j j�d | j �}|j | j
||d �S d S )
Nr F�Zallow_conversion�-Invalid value given for 'reboot_command': %s.� � r( �SHUTDOWN_COMMAND_ARGS� DEFAULT_SHUTDOWN_COMMAND_ARGS�< r )Z delay_sec� delay_min�message)
r5 r, r6 r � TypeErrorr r �split�
IndexErrorr@ r �DEFAULT_REBOOT_MESSAGE�format)r+ r> r �er, rH Zreboot_messager r r �get_shutdown_command_argst s &
z&ActionModule.get_shutdown_command_argsc
C s
i }t �dj| jjd�� | j|dddid�}z�|�dd�rdtd �t|d
�� � t|d �� � ���|d d
�
� |d< t |d d � d�d �|d< t |d d �
� �|d< t �dj| jj|d�� |W S t
�y } z td�|jd ���W Y d }~n
d }~0 0 d S )Nz2{action}: running setup module to get distribution��actionzansible.legacy.setupZ
gather_subset�min�� task_varsZ module_nameZ module_args�failedFz1Failed to determine system distribution. {0}, {1}Z
module_stdoutZ
module_stderrZ
ansible_factsZansible_distributionr: Z ansible_distribution_version�.r r; Zansible_os_familyr< z {action}: distribution: {dist})rR Zdistz@Failed to get distribution information. Missing "{0}" in output.)�display�debugrN r5 rR �_execute_moduler6 r r �strip�lowerr
rK �KeyErrorr, )r+ rU r> Z
module_outputZker r r �get_distribution� s( � � z ActionModule.get_distributionc
C s> | j j�d�}|d urnzt|dd�}W n4 tyZ } z tdt|� ��W Y d }~n
d }~0 0 |�dd�d }n| �d|d �}|d d
kr�|S g d �}| j j�d |�}z t |�}W n$ ty� d
}t|�
|���Y n0 t � dj
| j j
||d�� | j|d||gdd�d�} dd� | d D �}
|
�s2td�
||���|
d S d S )Nr FrA rB rC rD r �SHUTDOWN_COMMANDS�DEFAULT_SHUTDOWN_COMMAND�/)z/sbinz/binz /usr/sbinz/usr/binz/usr/local/sbinr z@'search_paths' must be a string or flat list of strings, got {0}zL{action}: running find module looking in {paths} to get path for "{command}")rR �command�pathszansible.legacy.find�any)rc �patternsZ file_typerT c S s g | ] }|d �qS )�pathr )�.0�xr r r �
<listcomp>� � z5ActionModule.get_shutdown_command.<locals>.<listcomp>�filesz1Unable to find command "{0}" in search paths: {1})r5 r, r6 r rJ r r rK r@ r rN rX rY rR rZ ) r+ rU r> r rO Z shutdown_binZdefault_search_pathsr �err_msgZ find_result� full_pathr r r �get_shutdown_command� sD & ��� z!ActionModule.get_shutdown_commandc C sD | j �� D ]4\}}| jj�|�d ur
t�dj||| jjd�� q
d S )NzGSince Ansible {version}, {arg} is no longer a valid option for {action})r; �argrR ) �DEPRECATED_ARGS�itemsr5 r, r6 rX �warningrN rR )r+ ro r; r r r �deprecated_args� s �z ActionModule.deprecated_argsc
C s | � d|d�}| jj�d�rp| jj�d�}zt|dd� W n4 tyn } z tdt|� ��W Y d }~n
d }~0 0 t� dj
| jj |d�� | j || j
d �}|d
d kr�|d }|d
}tdj
| jj |d
t|�t|�d���t� dj
| jj |d �� d�� |d �� S )N�BOOT_TIME_COMMANDS�DEFAULT_BOOT_TIME_COMMANDr FrA z0Invalid value given for 'boot_time_command': %s.z5{action}: getting boot time with command: '{command}'�rR rb �Zsudoable�rcr �stdout�stderrzS{action}: failed to get host boot time info, rc: {rc}, stdout: {out}, stderr: {err})rR rx �out�errz {action}: last boot time: {boot})rR Zboot)r@ r5 r, r6 r rJ r r rX rY rN rR �_low_level_execute_command�DEFAULT_SUDOABLEr[ )r+ r> r rO �command_resultry rz r r r �get_system_boot_time� s( & � z!ActionModule.get_system_boot_timec
C s� t �dj| jjd�� | jj�d| jj�d| j��}|r�z4t �dj| jj|d�� | j �
d|� | j � � W n t y� t �
d� Y n0 z| �|�}W n( ty� } z|�W Y d }~n
d }~0 0 t|�d ks�||kr�td
��d S ) Nz,{action}: attempting to get system boot timerQ r Zconnect_timeout_secz,{action}: setting connect_timeout to {value}�rR r9 �connection_timeoutzHConnection plugin does not allow the connection timeout to be overriddenr zboot time has not changed)rX �vvvrN r5 rR r, r6 �DEFAULT_CONNECT_TIMEOUTrY � _connection�
set_option�reset�AttributeErrorrr r� � Exception�len�
ValueError)r+ r> �previous_boot_timer Zcurrent_boot_timerO r r r �check_boot_time� s z ActionModule.check_boot_timec K s� | j j�d| �d|d��}t�dj| j jd�� t�dj| j j|d�� z| j || j
d�}W n6 t y� z| j �
� W n ty� Y n0 � Y n0 |d d
kr�d jt|d �t|d
�d�}t|��t�dj| j jd�� d S )Nr �
TEST_COMMANDS�DEFAULT_TEST_COMMANDz-{action}: attempting post-reboot test commandrQ z9{action}: attempting post-reboot test command '{command}'rv rw rx r z Test command failed: {err} {out}rz ry )r| r{ z&{action}: system successfully rebooted)r5 r, r6 r@ rX r� rN rR rY r} r~ r� r� r� r� r � RuntimeError)r+ r> r- r r r r r r �run_test_command s$
�z ActionModule.run_test_commandc
C sx t �� t|d� }|d u r i }d}d}t �� |k �rbz4|f d|i|�� |rdt�dj| jj|d�� W d S t�y^ } z�t | t
�r�z| j � � W n t
y� Y n0 t
�dd�d }
d| |
} | |kr�||
} |�r8zt| ��� d } W n. t�y } zt| �} W Y d } ~ n
d } ~ 0 0 t�d
j| jj|| | d �� |d 7 }t�| � W Y d } ~ q&d } ~ 0 0 q&td
j||d���d S )N)�secondsr � r> z{action}: {desc} success)rR �desci� � ���z@{action}: {desc} fail '{err}', retrying in {sleep:.4} seconds...)rR r� r| �sleeprD z0Timed out waiting for {desc} (timeout={timeout}))r� �timeout)r �utcnowr rX rY rN r5 rR r� �
isinstancer r� r� �randomZrandintr
�
splitlinesrL �timer� r )
r+ rR r � action_descr> �
action_kwargsZ max_end_timeZ
fail_countZmax_fail_sleeprO Z
random_intZ
fail_sleep�errorr r r �do_until_success_or_timeout sF
�"z(ActionModule.do_until_success_or_timeoutc
C s i }i }| � ||�}| �|�}d�||�}zBt�dj| jjd�� t�dj| jj|d�� | j|| j d�}W nH t
y� } z0t�dj| jjt |�d�� d |d
< W Y d }~n
d }~0 0 t �
� |d < |d
d k�rd |d
< d|d< djt|d �� �t|d �� �d�|d< |S d|d
< |S )Nz{0} {1}z {action}: rebooting server...rQ z3{action}: rebooting server with command '{command}'rv rw z>{action}: AnsibleConnectionFailure caught and handled: {error}�rR r� r rx �startTrV F�rebootedz6Reboot command failed. Error was: '{stdout}, {stderr}'ry rz )ry rz r )rn rP rN rX r� r5 rR rY r} r~ r r
r r� r r[ ) r+ rU r> �result�
reboot_resultZshutdown_commandZshutdown_command_argsr rO r r r �perform_rebootC s.
�
zActionModule.perform_rebootc
C sp t �dj| jjd�� i }�zt| jj�d| jj�d| j���}| j | j
d|||d� z| j � d�}W n t
yx Y n�0 ||kr�z4t �dj| jj|d �� | j �d|� | j �� W nD ttfy� } z(t �d
j| jjt|�d �� W Y d }~n
d }~0 0 | j | jd |||d� d
|d< d
|d< W nF t�yj } z,d
|d< d
|d< t|�|d< |W Y d }~S d }~0 0 |S )Nz{action}: validating rebootrQ r Zreboot_timeout_seczlast boot time check)rR r� r r> r� r� zC{action}: setting connect_timeout back to original value of {value}r� zE{action}: failed to reset connection_timeout back to default: {error}r� zpost-reboot test commandTr� �changedrV r )rX r� rN r5 rR r4 r, r6 �DEFAULT_REBOOT_TIMEOUTr� r� r� �
get_optionr] rY r� r� r r� r
r� r ) r+ r> �original_connection_timeoutr� r� r r rO Ztoexr r r �validate_reboot` sT "� �� � z ActionModule.validate_rebootc
s� d| _ d| _| jjdkr6d�| jj�}dddd|d�S | jjrJdddd�S |d u rVi }| � � t
t | �� ||�}|�
dd�s�|�
d d�r�|S | �|�}z| �|�}W nD ty� } z,d|d < d|d
< t|�|d < |W Y d }~S d }~0 0 d }z(| j�d �}t�d
j| jj|d�� W n* t�y@ t�dj| jjd�� Y n0 | �||�} | d �rz| }t�� | d }
|
j|d< |S | jdk�r�t�dj| jj| jd�� t�dj| jj| jd�� t � | j� | j ||d|id�}t�� | d }
|
j|d< |S )NT�localz@Running {0} with local connection would reboot the control node.Fr )r� �elapsedr� rV r )r� r� r� ZskippedrV r& r r� z6{action}: saving original connect_timeout of {timeout})rR r� z<{action}: connect_timeout connection option has not been setrQ r� r� z/{action}: waiting an additional {delay} seconds)rR Zdelayr� )r� ) Z_supports_check_modeZ_supports_asyncr� Z transportrN r5 rR Z
_play_contextZ
check_moders r) r �runr6 r^ r� r� r
r� rX rY r] r� r r� r� r r� r� r� r� ) r+ �tmprU r r� r> r� rO r� r� r� r. r r r� � sR
zActionModule.run)N)NN)NN)'r r r ZTRANSFERS_FILES� frozensetZ _VALID_ARGSr� r� r1 r3 r� ru rM r` rF r~ rp rt r_ rE r� r* �propertyr r r0 r@ rP r^ rn rs r� r� r� r� r� r� r� �
__classcell__r r r. r r sp � ���
-
)
5r )Z
__future__r r r �typeZ
__metaclass__r� r� r r Zansible.errorsr r Zansible.module_utils._textr r
Z&ansible.module_utils.common.validationr r Zansible.plugins.actionr
Zansible.utils.displayr rX r� r r r r r r �<module> s