112 lines
4.1 KiB
Python
112 lines
4.1 KiB
Python
|
# Copyright 2022 Red Hat, Inc.
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
||
|
|
# not use this file except in compliance with the License. You may obtain
|
||
|
|
# a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||
|
|
# License for the specific language governing permissions and limitations
|
||
|
|
# under the License.
|
||
|
|
#
|
||
|
|
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
|
||
|
|
from osc_lib.command import command
|
||
|
|
from osc_lib.i18n import _
|
||
|
|
|
||
|
|
from observabilityclient.utils import runner
|
||
|
|
from observabilityclient.utils import shell
|
||
|
|
|
||
|
|
|
||
|
|
OBSLIBDIR = shell.file_check('/usr/share/osp-observability', 'directory')
|
||
|
|
OBSWRKDIR = shell.file_check(
|
||
|
|
os.path.expanduser('~/.osp-observability'), 'directory'
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class ObservabilityBaseCommand(command.Command):
|
||
|
|
"""Base class for observability commands."""
|
||
|
|
|
||
|
|
def get_parser(self, prog_name):
|
||
|
|
parser = super().get_parser(prog_name)
|
||
|
|
parser.add_argument(
|
||
|
|
'--dev',
|
||
|
|
action='store_true',
|
||
|
|
help=_("Enable development output.")
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'--messy',
|
||
|
|
action='store_true',
|
||
|
|
help=_("Disable cleanup of temporary files.")
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'--workdir',
|
||
|
|
default=OBSWRKDIR,
|
||
|
|
help=_("Working directory for observability commands.")
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'--moduledir',
|
||
|
|
default=None,
|
||
|
|
help=_("Directory with additional Ansible modules.")
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'--ssh-user',
|
||
|
|
default='heat-admin',
|
||
|
|
help=_("Username to be used for SSH connection.")
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'--ssh-key',
|
||
|
|
default='/home/stack/.ssh/id_rsa',
|
||
|
|
help=_("SSH private key to be used for SSH connection.")
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'--ansible-cfg',
|
||
|
|
default=os.path.join(OBSWRKDIR, 'ansible.cfg'),
|
||
|
|
help=_("Path to Ansible configuration.")
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'--config',
|
||
|
|
default=None,
|
||
|
|
help=_("Path to playbook configuration file.")
|
||
|
|
)
|
||
|
|
return parser
|
||
|
|
|
||
|
|
def _run_playbook(self, playbook, inventory, parsed_args):
|
||
|
|
"""Run Ansible raw playbook"""
|
||
|
|
playbook = os.path.join(OBSLIBDIR, 'playbooks', playbook)
|
||
|
|
with shell.tempdir(parsed_args.workdir,
|
||
|
|
prefix=os.path.splitext(playbook)[0],
|
||
|
|
clear=not parsed_args.messy) as tmpdir:
|
||
|
|
# copy extravars file for the playbook run
|
||
|
|
if parsed_args.config:
|
||
|
|
envdir = shell.file_check(os.path.join(tmpdir, 'env'),
|
||
|
|
'directory')
|
||
|
|
shutil.copy(parsed_args.config,
|
||
|
|
os.path.join(envdir, 'extravars'))
|
||
|
|
# copy inventory file for the playbook run
|
||
|
|
shutil.copy(inventory, os.path.join(tmpdir, 'inventory'))
|
||
|
|
# run playbook
|
||
|
|
rnr = runner.AnsibleRunner(tmpdir,
|
||
|
|
moduledir=parsed_args.moduledir,
|
||
|
|
ssh_user=parsed_args.ssh_user,
|
||
|
|
ssh_key=parsed_args.ssh_key,
|
||
|
|
ansible_cfg=parsed_args.ansible_cfg)
|
||
|
|
if parsed_args.messy:
|
||
|
|
print("Running playbook %s" % playbook)
|
||
|
|
rnr.run(playbook, debug=parsed_args.dev)
|
||
|
|
rnr.destroy(clear=not parsed_args.messy)
|
||
|
|
|
||
|
|
def _execute(self, command, parsed_args):
|
||
|
|
"""Execute local command"""
|
||
|
|
with shell.tempdir(parsed_args.workdir, prefix='exec',
|
||
|
|
clear=not parsed_args.messy) as tmpdir:
|
||
|
|
rc, out, err = shell.execute(command, workdir=tmpdir,
|
||
|
|
can_fail=parsed_args.dev,
|
||
|
|
use_shell=True)
|
||
|
|
return rc, out, err
|