← Back to branch summary

~aluria/+junk/charm-collectd-openstack

« back to all changes in this revision

Viewing changes to hooks/charmhelpers/payload/execd.py

  • Committer: Alvaro Uría
  • Date: 2015-10-21 09:28:27 UTC
  • Revision ID: alvaro.uria@canonical.com-20151021092827-3eyl5elu92ftpov9
charm structure and simple config set-up

Show diffs side-by-side

added added

removed removed

Lines of Context:
hooks/charmhelpers/payload/execd.py
 
1
#!/usr/bin/env python
 
2
 
 
3
# Copyright 2014-2015 Canonical Limited.
 
4
#
 
5
# This file is part of charm-helpers.
 
6
#
 
7
# charm-helpers is free software: you can redistribute it and/or modify
 
8
# it under the terms of the GNU Lesser General Public License version 3 as
 
9
# published by the Free Software Foundation.
 
10
#
 
11
# charm-helpers is distributed in the hope that it will be useful,
 
12
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
13
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
14
# GNU Lesser General Public License for more details.
 
15
#
 
16
# You should have received a copy of the GNU Lesser General Public License
 
17
# along with charm-helpers.  If not, see <http://www.gnu.org/licenses/>.
 
18
 
 
19
import os
 
20
import sys
 
21
import subprocess
 
22
from charmhelpers.core import hookenv
 
23
 
 
24
 
 
25
def default_execd_dir():
 
26
    return os.path.join(os.environ['CHARM_DIR'], 'exec.d')
 
27
 
 
28
 
 
29
def execd_module_paths(execd_dir=None):
 
30
    """Generate a list of full paths to modules within execd_dir."""
 
31
    if not execd_dir:
 
32
        execd_dir = default_execd_dir()
 
33
 
 
34
    if not os.path.exists(execd_dir):
 
35
        return
 
36
 
 
37
    for subpath in os.listdir(execd_dir):
 
38
        module = os.path.join(execd_dir, subpath)
 
39
        if os.path.isdir(module):
 
40
            yield module
 
41
 
 
42
 
 
43
def execd_submodule_paths(command, execd_dir=None):
 
44
    """Generate a list of full paths to the specified command within exec_dir.
 
45
    """
 
46
    for module_path in execd_module_paths(execd_dir):
 
47
        path = os.path.join(module_path, command)
 
48
        if os.access(path, os.X_OK) and os.path.isfile(path):
 
49
            yield path
 
50
 
 
51
 
 
52
def execd_run(command, execd_dir=None, die_on_error=False, stderr=None):
 
53
    """Run command for each module within execd_dir which defines it."""
 
54
    for submodule_path in execd_submodule_paths(command, execd_dir):
 
55
        try:
 
56
            subprocess.check_call(submodule_path, shell=True, stderr=stderr)
 
57
        except subprocess.CalledProcessError as e:
 
58
            hookenv.log("Error ({}) running  {}. Output: {}".format(
 
59
                e.returncode, e.cmd, e.output))
 
60
            if die_on_error:
 
61
                sys.exit(e.returncode)
 
62
 
 
63
 
 
64
def execd_preinstall(execd_dir=None):
 
65
    """Run charm-pre-install for each module within execd_dir."""
 
66
    execd_run('charm-pre-install', execd_dir=execd_dir)