1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
#!/usr/bin/python
# vi: ts=4 expandtab
#
# Copyright (C) 2009-2010 Canonical Ltd.
#
# Author: Scott Moser <scott.moser@canonical.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 3, as
# published by the Free Software Foundation.
#
# 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 <http://www.gnu.org/licenses/>.
import subprocess
import sys
import cloudinit
import cloudinit.util as util
import time
import logging
def warn(str):
sys.stderr.write(str)
def main():
now = time.strftime("%a, %d %b %Y %H:%M:%S %z")
try:
uptimef=open("/proc/uptime")
uptime=uptimef.read().split(" ")[0]
uptimef.close()
except IOError as e:
warn("unable to open /proc/uptime\n")
uptime = "na"
msg = "cloud-init running: %s. up %s seconds" % (now, uptime)
sys.stderr.write(msg + "\n")
sys.stderr.flush()
cloudinit.logging_set_from_cfg_file()
log = logging.getLogger()
log.info(msg)
# cache is not instance specific, so it has to be purged
cloudinit.purge_cache()
cloud = cloudinit.CloudInit()
try:
cloud.get_data_source()
except Exception as e:
print e
sys.stderr.write("Failed to get instance data\n")
sys.exit(1)
# store the metadata
cloud.update_cache()
# parse the user data (ec2-run-userdata.py)
try:
cloud.sem_and_run("consume_userdata", "once-per-instance",
cloud.consume_userdata,[],False)
except:
warn("consuming user data failed!\n")
raise
try:
hostname = cloud.get_hostname()
cloud.sem_and_run("set_hostname", "once-per-instance",
set_hostname, [ hostname ], False)
except:
warn("failed to set hostname\n")
#print "user data is:" + cloud.get_user_data()
# set the defaults (like what ec2-set-defaults.py did)
try:
cloud.sem_and_run("set_defaults", "once-per-instance",
set_defaults,[ cloud ],False)
except:
warn("failed to set defaults\n")
# finish, send the cloud-config event
cloud.initctl_emit()
sys.exit(0)
def set_defaults(cloud):
apply_locale(cloud.get_locale())
def apply_locale(locale):
subprocess.Popen(['locale-gen', locale]).communicate()
subprocess.Popen(['update-locale', locale]).communicate()
util.render_to_file('default-locale', '/etc/default/locale', \
{ 'locale' : locale })
def set_hostname(hostname):
subprocess.Popen(['hostname', hostname]).communicate()
f=open("/etc/hostname","wb")
f.write("%s\n" % hostname)
f.close()
if __name__ == '__main__':
main()
|
