mirror of
https://github.com/ansible-collections/community.general.git
synced 2025-04-26 04:11:25 -07:00
71f46d69d6
* First bit of fixing temporary to have one source of truth * Fix pep8 * Remove explicit make_tmp_path() in copy The copy action plugin sets TRANSFER_FILES=True so it does not need to set the temporary directory explicitly; the base class's run() method will do that for us. * Fix for calling a module's run when a shell has already created a temp path. * Remember to inform the rest of the world when tempdir is removed * New strategy for how to warn on passing tmp Now we just warn when calling the parent class run() early. If the module does a late call to the parent run() and doesn't make use of the temporary directory, then we don't check for the possibility that the user mistakenly is sending tmp in. If we truly deprecate this (rather than ignoring it forever) then we might want to switch back to checking for someone passing a value in as tmp. * Remove tmp parameter from _execute_module as well * Port all action plugins to not send tmp explicitly This is now handled inside of _execute_module via the _connection._shell.tempdir attribute. Also update warnings and docs to tell people to set the attribute instead of using _execute_module's tmp parameter. * Always set local tempdir variable
241 lines
9.1 KiB
Python
241 lines
9.1 KiB
Python
# Copyright 2012, Tim Bielawa <tbielawa@redhat.com>
|
|
#
|
|
# This file is part of Ansible
|
|
#
|
|
# Ansible is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Ansible is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
import datetime
|
|
import signal
|
|
import termios
|
|
import time
|
|
import tty
|
|
|
|
from os import isatty
|
|
from ansible.errors import AnsibleError
|
|
from ansible.module_utils.six import PY3
|
|
from ansible.module_utils._text import to_text
|
|
from ansible.plugins.action import ActionBase
|
|
|
|
try:
|
|
from __main__ import display
|
|
except ImportError:
|
|
from ansible.utils.display import Display
|
|
display = Display()
|
|
|
|
|
|
class AnsibleTimeoutExceeded(Exception):
|
|
pass
|
|
|
|
|
|
def timeout_handler(signum, frame):
|
|
raise AnsibleTimeoutExceeded
|
|
|
|
|
|
class ActionModule(ActionBase):
|
|
''' pauses execution for a length or time, or until input is received '''
|
|
|
|
PAUSE_TYPES = ['seconds', 'minutes', 'prompt', 'echo', '']
|
|
BYPASS_HOST_LOOP = True
|
|
|
|
def run(self, tmp=None, task_vars=None):
|
|
''' run the pause action module '''
|
|
if task_vars is None:
|
|
task_vars = dict()
|
|
|
|
result = super(ActionModule, self).run(tmp, task_vars)
|
|
del tmp # tmp no longer has any effect
|
|
|
|
duration_unit = 'minutes'
|
|
prompt = None
|
|
seconds = None
|
|
echo = True
|
|
echo_prompt = ''
|
|
result.update(dict(
|
|
changed=False,
|
|
rc=0,
|
|
stderr='',
|
|
stdout='',
|
|
start=None,
|
|
stop=None,
|
|
delta=None,
|
|
echo=echo
|
|
))
|
|
|
|
if not set(self._task.args.keys()) <= set(self.PAUSE_TYPES):
|
|
result['failed'] = True
|
|
result['msg'] = "Invalid argument given. Must be one of: %s" % ", ".join(self.PAUSE_TYPES)
|
|
return result
|
|
|
|
# Should keystrokes be echoed to stdout?
|
|
if 'echo' in self._task.args:
|
|
echo = self._task.args['echo']
|
|
if not type(echo) == bool:
|
|
result['failed'] = True
|
|
result['msg'] = "'%s' is not a valid setting for 'echo'." % self._task.args['echo']
|
|
return result
|
|
|
|
# Add a note saying the output is hidden if echo is disabled
|
|
if not echo:
|
|
echo_prompt = ' (output is hidden)'
|
|
|
|
# Is 'prompt' a key in 'args'?
|
|
if 'prompt' in self._task.args:
|
|
prompt = "[%s]\n%s%s:" % (self._task.get_name().strip(), self._task.args['prompt'], echo_prompt)
|
|
else:
|
|
# If no custom prompt is specified, set a default prompt
|
|
prompt = "[%s]\n%s%s:" % (self._task.get_name().strip(), 'Press enter to continue', echo_prompt)
|
|
|
|
# Are 'minutes' or 'seconds' keys that exist in 'args'?
|
|
if 'minutes' in self._task.args or 'seconds' in self._task.args:
|
|
try:
|
|
if 'minutes' in self._task.args:
|
|
# The time() command operates in seconds so we need to
|
|
# recalculate for minutes=X values.
|
|
seconds = int(self._task.args['minutes']) * 60
|
|
else:
|
|
seconds = int(self._task.args['seconds'])
|
|
duration_unit = 'seconds'
|
|
|
|
except ValueError as e:
|
|
result['failed'] = True
|
|
result['msg'] = u"non-integer value given for prompt duration:\n%s" % to_text(e)
|
|
return result
|
|
|
|
########################################################################
|
|
# Begin the hard work!
|
|
|
|
start = time.time()
|
|
result['start'] = to_text(datetime.datetime.now())
|
|
result['user_input'] = b''
|
|
|
|
fd = None
|
|
old_settings = None
|
|
try:
|
|
if seconds is not None:
|
|
if seconds < 1:
|
|
seconds = 1
|
|
|
|
# setup the alarm handler
|
|
signal.signal(signal.SIGALRM, timeout_handler)
|
|
signal.alarm(seconds)
|
|
|
|
# show the timer and control prompts
|
|
display.display("Pausing for %d seconds%s" % (seconds, echo_prompt))
|
|
display.display("(ctrl+C then 'C' = continue early, ctrl+C then 'A' = abort)\r"),
|
|
|
|
# show the prompt specified in the task
|
|
if 'prompt' in self._task.args:
|
|
display.display(prompt)
|
|
|
|
else:
|
|
display.display(prompt)
|
|
|
|
# save the attributes on the existing (duped) stdin so
|
|
# that we can restore them later after we set raw mode
|
|
fd = None
|
|
try:
|
|
if PY3:
|
|
stdin = self._connection._new_stdin.buffer
|
|
else:
|
|
stdin = self._connection._new_stdin
|
|
fd = stdin.fileno()
|
|
except (ValueError, AttributeError):
|
|
# ValueError: someone is using a closed file descriptor as stdin
|
|
# AttributeError: someone is using a null file descriptor as stdin on windoez
|
|
stdin = None
|
|
if fd is not None:
|
|
if isatty(fd):
|
|
old_settings = termios.tcgetattr(fd)
|
|
tty.setraw(fd)
|
|
|
|
# Enable a few things turned off by tty.setraw()
|
|
# ICANON -> Allows characters to be deleted and hides things like ^M.
|
|
# ICRNL -> Makes the return key work when ICANON is enabled, otherwise
|
|
# you get stuck at the prompt with no way to get out of it.
|
|
# See man termios for details on these flags
|
|
if not seconds:
|
|
new_settings = termios.tcgetattr(fd)
|
|
new_settings[0] = new_settings[0] | termios.ICRNL
|
|
new_settings[3] = new_settings[3] | termios.ICANON
|
|
termios.tcsetattr(fd, termios.TCSANOW, new_settings)
|
|
|
|
if echo:
|
|
# Enable ECHO since tty.setraw() disables it
|
|
new_settings = termios.tcgetattr(fd)
|
|
new_settings[3] = new_settings[3] | termios.ECHO
|
|
termios.tcsetattr(fd, termios.TCSANOW, new_settings)
|
|
|
|
# flush the buffer to make sure no previous key presses
|
|
# are read in below
|
|
termios.tcflush(stdin, termios.TCIFLUSH)
|
|
while True:
|
|
try:
|
|
if fd is not None:
|
|
key_pressed = stdin.read(1)
|
|
|
|
if seconds:
|
|
if key_pressed == b'\x03':
|
|
raise KeyboardInterrupt
|
|
|
|
if not seconds:
|
|
if fd is None or not isatty(fd):
|
|
display.warning("Not waiting from prompt as stdin is not interactive")
|
|
break
|
|
# read key presses and act accordingly
|
|
if key_pressed in (b'\r', b'\n'):
|
|
break
|
|
else:
|
|
result['user_input'] += key_pressed
|
|
|
|
except KeyboardInterrupt:
|
|
if seconds is not None:
|
|
signal.alarm(0)
|
|
display.display("Press 'C' to continue the play or 'A' to abort \r"),
|
|
if self._c_or_a(stdin):
|
|
break
|
|
else:
|
|
raise AnsibleError('user requested abort!')
|
|
|
|
except AnsibleTimeoutExceeded:
|
|
# this is the exception we expect when the alarm signal
|
|
# fires, so we simply ignore it to move into the cleanup
|
|
pass
|
|
finally:
|
|
# cleanup and save some information
|
|
# restore the old settings for the duped stdin fd
|
|
if not(None in (fd, old_settings)) and isatty(fd):
|
|
termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
|
|
|
|
duration = time.time() - start
|
|
result['stop'] = to_text(datetime.datetime.now())
|
|
result['delta'] = int(duration)
|
|
|
|
if duration_unit == 'minutes':
|
|
duration = round(duration / 60.0, 2)
|
|
else:
|
|
duration = round(duration, 2)
|
|
result['stdout'] = "Paused for %s %s" % (duration, duration_unit)
|
|
|
|
result['user_input'] = to_text(result['user_input'], errors='surrogate_or_strict')
|
|
return result
|
|
|
|
def _c_or_a(self, stdin):
|
|
while True:
|
|
key_pressed = stdin.read(1)
|
|
if key_pressed.lower() == b'a':
|
|
return False
|
|
elif key_pressed.lower() == b'c':
|
|
return True
|