87 lines
3.4 KiB
Python
Executable File
87 lines
3.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# ansible.py - PVC Cluster Auto-bootstrap Ansible libraries
|
|
# Part of the Parallel Virtual Cluster (PVC) system
|
|
#
|
|
# Copyright (C) 2018-2021 Joshua M. Boniface <joshua@boniface.me>
|
|
#
|
|
# This program 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, version 3.
|
|
#
|
|
# This program 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 this program. If not, see <https://www.gnu.org/licenses/>.
|
|
#
|
|
###############################################################################
|
|
|
|
import pvcbootstrapd.lib.notifications as notifications
|
|
import pvcbootstrapd.lib.git as git
|
|
|
|
import ansible_runner
|
|
import tempfile
|
|
|
|
from time import sleep
|
|
from celery.utils.log import get_task_logger
|
|
|
|
|
|
logger = get_task_logger(__name__)
|
|
|
|
|
|
def run_bootstrap(config, cspec, cluster, nodes):
|
|
"""
|
|
Run an Ansible bootstrap against a cluster
|
|
"""
|
|
logger.debug(nodes)
|
|
|
|
# Construct our temporary INI inventory string
|
|
logger.info("Constructing virtual Ansible inventory")
|
|
base_yaml = cspec["clusters"][cluster.name]["base_yaml"]
|
|
local_domain = base_yaml.get("local_domain")
|
|
inventory = [f"""[{cluster.name}]"""]
|
|
for node in nodes:
|
|
inventory.append(
|
|
f"""{node.name}.{local_domain} ansible_host={node.host_ipaddr}"""
|
|
)
|
|
inventory = "\n".join(inventory)
|
|
logger.debug(inventory)
|
|
|
|
# Waiting 30 seconds to ensure everything is booted an stabilized
|
|
logger.info("Waiting 60s before starting Ansible bootstrap.")
|
|
sleep(60)
|
|
|
|
logger.info(f"Starting Ansible bootstrap of cluster {cluster.name}")
|
|
notifications.send_webhook(config, "begin", f"Cluster {cluster.name}: Starting Ansible bootstrap")
|
|
|
|
# Run the Ansible playbooks
|
|
with tempfile.TemporaryDirectory(prefix="pvc-ansible-bootstrap_") as pdir:
|
|
try:
|
|
r = ansible_runner.run(
|
|
private_data_dir=f"{pdir}",
|
|
inventory=inventory,
|
|
limit=f"{cluster.name}",
|
|
playbook=f"{config['ansible_path']}/pvc.yml",
|
|
extravars={
|
|
"ansible_ssh_private_key_file": config["ansible_key_file"],
|
|
"do_bootstrap": "yes",
|
|
},
|
|
forks=len(nodes),
|
|
verbosity=2,
|
|
)
|
|
logger.info("Final status:")
|
|
logger.info("{}: {}".format(r.status, r.rc))
|
|
logger.info(r.stats)
|
|
if r.rc == 0:
|
|
git.commit_repository(config, f"Generated files for cluster '{cluster.name}'")
|
|
git.push_repository(config)
|
|
notifications.send_webhook(config, "success", f"Cluster {cluster.name}: Completed Ansible bootstrap")
|
|
else:
|
|
notifications.send_webhook(config, "failure", f"Cluster {cluster.name}: Failed Ansible bootstrap; check pvcbootstrapd logs")
|
|
except Exception as e:
|
|
logger.warning(f"Error: {e}")
|
|
notifications.send_webhook(config, "failure", f"Cluster {cluster.name}: Failed Ansible bootstrap with error '{e}'; check pvcbootstrapd logs")
|