← Back to branch summary

~cloud-init-dev/cloud-init/trunk

« back to all changes in this revision

Viewing changes to cloudinit/handlers/upstart_job.py

  • Committer: Scott Moser
  • Date: 2016-08-10 15:06:15 UTC
  • Revision ID: smoser@ubuntu.com-20160810150615-ma2fv107w3suy1ma
README: Mention move of revision control to git.

cloud-init development has moved its revision control to git.
It is available at 
  https://code.launchpad.net/cloud-init

Clone with 
  git clone https://git.launchpad.net/cloud-init
or
  git clone git+ssh://git.launchpad.net/cloud-init

For more information see
  https://git.launchpad.net/cloud-init/tree/HACKING.rst

Show diffs side-by-side

added added

removed removed

Lines of Context:
cloudinit/handlers/upstart_job.py
1
 
# vi: ts=4 expandtab
2
 
#
3
 
#    Copyright (C) 2012 Canonical Ltd.
4
 
#    Copyright (C) 2012 Hewlett-Packard Development Company, L.P.
5
 
#    Copyright (C) 2012 Yahoo! Inc.
6
 
#
7
 
#    Author: Scott Moser <scott.moser@canonical.com>
8
 
#    Author: Juerg Haefliger <juerg.haefliger@hp.com>
9
 
#    Author: Joshua Harlow <harlowja@yahoo-inc.com>
10
 
#
11
 
#    This program is free software: you can redistribute it and/or modify
12
 
#    it under the terms of the GNU General Public License version 3, as
13
 
#    published by the Free Software Foundation.
14
 
#
15
 
#    This program is distributed in the hope that it will be useful,
16
 
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
17
 
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18
 
#    GNU General Public License for more details.
19
 
#
20
 
#    You should have received a copy of the GNU General Public License
21
 
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
22
 
 
23
 
 
24
 
import os
25
 
import re
26
 
 
27
 
from cloudinit import handlers
28
 
from cloudinit import log as logging
29
 
from cloudinit import util
30
 
 
31
 
from cloudinit.settings import (PER_INSTANCE)
32
 
 
33
 
LOG = logging.getLogger(__name__)
34
 
UPSTART_PREFIX = "#upstart-job"
35
 
 
36
 
 
37
 
class UpstartJobPartHandler(handlers.Handler):
38
 
    def __init__(self, paths, **_kwargs):
39
 
        handlers.Handler.__init__(self, PER_INSTANCE)
40
 
        self.upstart_dir = paths.upstart_conf_d
41
 
 
42
 
    def list_types(self):
43
 
        return [
44
 
            handlers.type_from_starts_with(UPSTART_PREFIX),
45
 
        ]
46
 
 
47
 
    def handle_part(self, data, ctype, filename, payload, frequency):
48
 
        if ctype in handlers.CONTENT_SIGNALS:
49
 
            return
50
 
 
51
 
        # See: https://bugs.launchpad.net/bugs/819507
52
 
        if frequency != PER_INSTANCE:
53
 
            return
54
 
 
55
 
        if not self.upstart_dir:
56
 
            return
57
 
 
58
 
        filename = util.clean_filename(filename)
59
 
        (_name, ext) = os.path.splitext(filename)
60
 
        if not ext:
61
 
            ext = ''
62
 
        ext = ext.lower()
63
 
        if ext != ".conf":
64
 
            filename = filename + ".conf"
65
 
 
66
 
        payload = util.dos2unix(payload)
67
 
        path = os.path.join(self.upstart_dir, filename)
68
 
        util.write_file(path, payload, 0o644)
69
 
 
70
 
        if SUITABLE_UPSTART:
71
 
            util.subp(["initctl", "reload-configuration"], capture=False)
72
 
 
73
 
 
74
 
def _has_suitable_upstart():
75
 
    # (LP: #1124384)
76
 
    # a bug in upstart means that invoking reload-configuration
77
 
    # at this stage in boot causes havoc.  So, try to determine if upstart
78
 
    # is installed, and reloading configuration is OK.
79
 
    if not os.path.exists("/sbin/initctl"):
80
 
        return False
81
 
    try:
82
 
        (version_out, _err) = util.subp(["initctl", "version"])
83
 
    except Exception:
84
 
        util.logexc(LOG, "initctl version failed")
85
 
        return False
86
 
 
87
 
    # expecting 'initctl version' to output something like: init (upstart X.Y)
88
 
    if re.match("upstart 1.[0-7][)]", version_out):
89
 
        return False
90
 
    if "upstart 0." in version_out:
91
 
        return False
92
 
    elif "upstart 1.8" in version_out:
93
 
        if not os.path.exists("/usr/bin/dpkg-query"):
94
 
            return False
95
 
        try:
96
 
            (dpkg_ver, _err) = util.subp(["dpkg-query",
97
 
                                          "--showformat=${Version}",
98
 
                                          "--show", "upstart"], rcs=[0, 1])
99
 
        except Exception:
100
 
            util.logexc(LOG, "dpkg-query failed")
101
 
            return False
102
 
 
103
 
        try:
104
 
            good = "1.8-0ubuntu1.2"
105
 
            util.subp(["dpkg", "--compare-versions", dpkg_ver, "ge", good])
106
 
            return True
107
 
        except util.ProcessExecutionError as e:
108
 
            if e.exit_code is 1:
109
 
                pass
110
 
            else:
111
 
                util.logexc(LOG, "dpkg --compare-versions failed [%s]",
112
 
                            e.exit_code)
113
 
        except Exception as e:
114
 
            util.logexc(LOG, "dpkg --compare-versions failed")
115
 
        return False
116
 
    else:
117
 
        return True
118
 
 
119
 
SUITABLE_UPSTART = _has_suitable_upstart()