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
|
#!/usr/bin/python3
# -*- coding: utf-8 -*-
# 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.
#
# Authors:
# @author: David Blaisonneau <david.blaisonneau@orange.com>
# @author: Arnaud Morin <arnaud1.morin@orange.com>
import base64
from string import Template
from opensteak.foreman_objects.item import ForemanItem
class ItemHost(ForemanItem):
"""
ItemHostsGroup class
Represent the content of a foreman hostgroup as a dict
"""
objName = 'hosts'
payloadObj = 'host'
def __init__(self, api, key, *args, **kwargs):
""" Function __init__
Represent the content of a foreman object as a dict
@param api: The foreman api
@param key: The object Key
@param *args, **kwargs: the dict representation
@return RETURN: Itself
"""
ForemanItem.__init__(self, api, key,
self.objName, self.payloadObj,
*args, **kwargs)
self.update({'puppetclass_ids':
self.api.list('{}/{}/puppetclass_ids'
.format(self.objName, key))})
self.update({'param_ids':
list(self.api.list('{}/{}/parameters'
.format(self.objName, key),
only_id=True)
.keys())})
def getStatus(self):
""" Function getStatus
Get the status of an host
@return RETURN: The host status
"""
return self.api.get('hosts', self.key, 'status')['status']
def powerOn(self):
""" Function powerOn
Power on a host
@return RETURN: The API result
"""
return self.api.set('hosts', self.key,
{"power_action": "start"},
'power', async=self.async)
def getParamFromEnv(self, var, default=''):
""" Function getParamFromEnv
Search a parameter in the host environment
@param var: the var name
@param hostgroup: the hostgroup item linked to this host
@param default: default value
@return RETURN: the value
"""
if self.getParam(var):
return self.getParam(var)
if self.hostgroup:
if self.hostgroup.getParam(var):
return self.hostgroup.getParam(var)
if self.domain.getParam('password'):
return self.domain.getParam('password')
else:
return default
def getUserData(self,
hostgroup,
domain,
defaultPwd='',
defaultSshKey='',
proxyHostname='',
tplFolder='templates_metadata/'):
""" Function getUserData
Generate a userdata script for metadata server from Foreman API
@param domain: the domain item linked to this host
@param hostgroup: the hostgroup item linked to this host
@param defaultPwd: the default password if no password is specified
in the host>hostgroup>domain params
@param defaultSshKey: the default ssh key if no password is specified
in the host>hostgroup>domain params
@param proxyHostname: hostname of the smartproxy
@param tplFolder: the templates folder
@return RETURN: the user data
"""
if 'user-data' in self.keys():
return self['user-data']
else:
self.hostgroup = hostgroup
self.domain = domain
if proxyHostname == '':
proxyHostname = 'foreman.' + domain
password = self.getParamFromEnv('password', defaultPwd)
sshauthkeys = self.getParamFromEnv('global_sshkey', defaultSshKey)
with open(tplFolder+'puppet.conf', 'rb') as puppet_file:
p = MyTemplate(puppet_file.read())
enc_puppet_file = base64.b64encode(p.substitute(
foremanHostname=proxyHostname))
with open(tplFolder+'cloud-init.tpl', 'r') as content_file:
s = MyTemplate(content_file.read())
if sshauthkeys:
sshauthkeys = ' - '+sshauthkeys
self.userdata = s.substitute(
password=password,
fqdn=self['name'],
sshauthkeys=sshauthkeys,
foremanurlbuilt="http://{}/unattended/built"
.format(proxyHostname),
puppet_conf_content=enc_puppet_file.decode('utf-8'))
return self.userdata
class MyTemplate(Template):
delimiter = '%'
idpattern = r'[a-z][_a-z0-9]*'
|
