fuel-octane/octane/util/puppet.py

63 lines
2.2 KiB
Python

# 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 logging
import os.path
from octane import magic_consts
from octane.util import patch
from octane.util import subprocess
LOG = logging.getLogger(__name__)
def apply_task(task):
filename = '{0}.pp'.format(task)
path = os.path.join(magic_consts.PUPPET_TASKS_DIR, filename)
cmd = ['puppet', 'apply', '-d', '-v', "--color", "false",
'--detailed-exitcodes', path]
try:
subprocess.call(cmd)
except subprocess.CalledProcessError as exc:
# NOTE(akscram): Detailed exit codes of puppet apply:
# 0: The run succeeded with no changes or failures; the system
# was already in the desired state.
# 1: The run failed, or wasn't attempted due to another run
# already in progress.
# 2: The run succeeded, and some resources were changed.
# 4: The run succeeded, and some resources failed.
# 6: The run succeeded, and included both changes and failures.
if exc.returncode != 2:
LOG.error("Cannot apply the Puppet task: %s, %s",
task, exc.message)
raise
def apply_all_tasks():
try:
subprocess.call([magic_consts.PUPPET_APPLY_TASKS_SCRIPT])
except subprocess.CalledProcessError as exc:
LOG.error("Cannot apply Puppet state on host: %s",
exc)
raise
def patch_modules(revert=False):
puppet_patch_dir = os.path.join(magic_consts.CWD, "patches", "puppet")
patches = []
for d in os.listdir(puppet_patch_dir):
d = os.path.join(puppet_patch_dir, d)
if not os.path.isdir(d):
continue
patches.append(os.path.join(d, "patch"))
patch.patch_apply(magic_consts.PUPPET_DIR, patches, revert=revert)