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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
#
# Common code for the EC2 configuration files in Ubuntu
# Copyright (C) 2008-2010 Canonical Ltd.
#
# Author: Chuck Short <chuck.short@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 yaml
import re
import ec2init
import subprocess
import os
per_instance="once-per-instance"
class CloudConfig():
cfgfile = None
handlers = { }
cfg = None
def __init__(self,cfgfile):
print "reading %s" % cfgfile
self.cfg=read_conf(cfgfile)
import pprint; pprint.pprint(self.cfg)
self.cloud = ec2init.EC2Init()
self.cloud.get_data_source()
self.add_handler('apt-update-upgrade', self.h_apt_update_upgrade)
self.add_handler('config-ssh')
def add_handler(self, name, handler=None, freq=None):
if handler is None:
try:
handler=getattr(self,'h_%s' % name.replace('-','_'))
except:
raise Exception("Unknown hander for name %s" %name)
if freq is None:
freq = per_instance
self.handlers[name]= { 'handler': handler, 'freq': freq }
def get_handler_info(self, name):
return(self.handlers[name]['handler'], self.handlers[name]['freq'])
def check_for_updates(self):
value = self.cfg['apt_update']
return value
def check_for_upgrade(self):
value = self.cfg['apt_upgrade']
return value
def parse_ssh_keys(self):
disableRoot = self.cfg['disable_root']
if disableRoot == 'true':
value = 'disabled_root'
return value
else:
ec2Key = self.cfg['ec2_fetch_key']
if ec2Key != 'none':
value = 'default_key'
return value
else:
return ec2Key
def add_ppa(self):
#value = self.cfg['apt_sources']
for ent in self.cfg['apt_sources']:
ppa = ent['source']
where = ppa.find('ppa:')
if where != -1:
return ppa
def add_custom_repo(self):
sources = []
value = self.cfg['apt_sources']
for ent in self.cfg['apt_sources']:
if ent.has_key('keyserver'):
keyserver = ent['keyserver']
if ent.has_key('keyid'):
keyid = ent['keyid']
if ent.has_key('filename'):
filename = ent['filename']
source = ent['source']
if source.startswith("deb"):
sources.append(source)
return (keyserver,sources,keyid,filename)
def handle(self, name, args):
handler = None
freq = None
try:
(handler, freq) = self.get_handler_info(name)
except:
raise Exception("Unknown config key %s\n", name)
self.cloud.sem_and_run(name, freq, handler, [ name, args ])
def h_apt_update_upgrade(self,name,args):
update = get_cfg_option_bool(self.cfg, 'apt_update', False)
upgrade = get_cfg_option_bool(self.cfg, 'apt_upgrade', False)
print "update = %s , upgrade = %s\n" % (update,upgrade)
if update or upgrade:
#retcode = subprocess.call(list)
subprocess.Popen(['apt-get', 'update']).communicate()
if upgrade:
e=os.environ.copy()
e['DEBIAN_FRONTEND']='noninteractive'
subprocess.Popen(['apt-get', 'upgrade', '--assume-yes'], env=e).communicate()
return(True)
def h_config_ssh(self,name,args):
print "Warning, not doing anything for config %s" % name
def h_config_ec2_ebs_mounts(self,name,args):
print "Warning, not doing anything for config %s" % name
def h_config_setup_raid(self,name,args):
print "Warning, not doing anything for config %s" % name
def h_config_runurl(self,name,args):
print "Warning, not doing anything for config %s" % name
def get_cfg_option_bool(yobj, key, default=False):
print "searching for %s" % key
if not yobj.has_key(key): return default
val = yobj[key]
if yobj[key] in [ True, '1', 'on', 'yes', 'true']:
return True
return False
def get_cfg_option_str(yobj, key, default=None):
if not yobj.has_key(key): return default
return yobj[key]
def read_conf(fname):
stream = file(fname)
conf = yaml.load(stream)
stream.close()
return conf
|
