| 19 |
19 |
PUPPET_PROXY = 'Puppet Proxy'
|
| 20 |
20 |
PUPPET_CA_PROXY = 'Puppet CA Proxy'
|
| 21 |
21 |
CONTENT_SOURCE = 'Content Source'
|
|
22 |
PASSWORD = 'Password'
|
|
23 |
PUPPET_CLASSES = 'Puppet Classes'
|
| 22 |
24 |
|
| 23 |
25 |
def export
|
| 24 |
26 |
CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
|
| 25 |
27 |
csv << [NAME, PARENT, ORGANIZATIONS, LOCATIONS, SUBNET, DOMAIN, OPERATING_SYSTEM,
|
| 26 |
28 |
ENVIRONMENT, COMPUTE_PROFILE, PARTITION_TABLE, MEDIUM, ARCHITECTURE, REALM,
|
| 27 |
|
PUPPET_PROXY, PUPPET_CA_PROXY, CONTENT_SOURCE]
|
|
29 |
PUPPET_PROXY, PUPPET_CA_PROXY, CONTENT_SOURCE, PASSWORD, PUPPET_CLASSES]
|
| 28 |
30 |
search_options = {:per_page => 999999}
|
| 29 |
31 |
search_options['search'] = "organization=\"#{option_organization}\"" if option_organization
|
| 30 |
32 |
@api.resource(:hostgroups).call(:index, search_options)['results'].each do |hostgroup|
|
| ... | ... | |
| 45 |
47 |
realm = hostgroup['realm_name']
|
| 46 |
48 |
puppet_proxy = hostgroup['puppet_proxy_id'] ? foreman_host(:id => hostgroup['puppet_proxy_id']) : nil
|
| 47 |
49 |
puppet_ca_proxy = hostgroup['puppet_ca_proxy_id'] ? foreman_host(:id => hostgroup['puppet_ca_proxy_id']) : nil
|
| 48 |
|
# TODO: http://projects.theforeman.org/issues/7597
|
| 49 |
|
# content_source = hostgroup['content_source_id'] ? foreman_host(:id => hostgroup['content_source_id']) : nil
|
| 50 |
|
content_source = nil
|
|
50 |
content_source = hostgroup['content_source_id'] ? foreman_host(:id => hostgroup['content_source_id']) : nil
|
| 51 |
51 |
parent = hostgroup['ancestry'] ? foreman_hostgroup(:id => hostgroup['ancestry']) : nil
|
|
52 |
password = nil
|
|
53 |
puppet_classes = export_column(hostgroup, 'puppetclasses') do |puppet_class|
|
|
54 |
"#{puppet_class['module_name']}/#{puppet_class['name']}"
|
|
55 |
end
|
|
56 |
|
|
57 |
# TODO: http://projects.theforeman.org/issues/6273
|
|
58 |
# API call to get the smart class variable override values
|
| 52 |
59 |
|
| 53 |
60 |
csv << [name, parent, organizations, locations, subnet, domain, operating_system,
|
| 54 |
61 |
puppet_environment, compute_profile, partition_table, medium, architecture,
|
| 55 |
|
realm, puppet_proxy, puppet_ca_proxy, content_source]
|
|
62 |
realm, puppet_proxy, puppet_ca_proxy, content_source, password, puppet_classes]
|
| 56 |
63 |
end
|
| 57 |
64 |
end
|
| 58 |
65 |
end
|
| 59 |
66 |
|
| 60 |
67 |
def import
|
| 61 |
68 |
@existing = {}
|
| 62 |
|
@api.resource(:hostgroups).call(:index, {:per_page => 999999})['results'].each do |host|
|
| 63 |
|
@existing[host['name']] = host['id'] if host
|
|
69 |
@api.resource(:hostgroups).call(:index, {:per_page => 999999})['results'].each do |host_group|
|
|
70 |
@existing[host_group['name']] = host_group['id'] if host_group
|
| 64 |
71 |
end
|
| 65 |
72 |
|
| 66 |
73 |
thread_import do |line|
|
| 67 |
|
create_hosts_from_csv(line)
|
|
74 |
create_from_csv(line)
|
| 68 |
75 |
end
|
| 69 |
76 |
end
|
| 70 |
77 |
|
| 71 |
|
def create_hosts_from_csv(line)
|
|
78 |
def create_from_csv(line)
|
| 72 |
79 |
return if option_organization && !CSV.parse_line(line[ORGANIZATIONS], {:skip_blanks => true}).include?(option_organization)
|
| 73 |
80 |
|
|
81 |
params = {
|
|
82 |
'hostgroup' => {
|
|
83 |
'architecture_id' => foreman_architecture(:name => line[ARCHITECTURE]),
|
|
84 |
'operatingsystem_id' => foreman_operatingsystem(:name => line[OPERATING_SYSTEM]),
|
|
85 |
'medium_id' => foreman_medium(:name => line[MEDIUM]),
|
|
86 |
'ptable_id' => foreman_partitiontable(:name => line[PARTITION_TABLE]),
|
|
87 |
'root_pass' => line[PASSWORD],
|
|
88 |
'organization_ids' => collect_column(line[ORGANIZATIONS]) do |organization|
|
|
89 |
foreman_organization(:name => organization)
|
|
90 |
end,
|
|
91 |
'location_ids' => collect_column(line[LOCATIONS]) do |location|
|
|
92 |
foreman_location(:name => location)
|
|
93 |
end
|
|
94 |
}
|
|
95 |
}
|
|
96 |
|
| 74 |
97 |
count(line[COUNT]).times do |number|
|
| 75 |
98 |
name = namify(line[NAME], number)
|
|
99 |
params['hostgroup']['name'] = name
|
|
100 |
|
| 76 |
101 |
if !@existing.include? name
|
| 77 |
102 |
print "Creating host group '#{name}'..." if option_verbose?
|
| 78 |
|
# @api.resource(:hosts).call(:create, {
|
| 79 |
|
# 'host' => {
|
| 80 |
|
# 'name' => name,
|
| 81 |
|
# 'root_pass' => 'changeme',
|
| 82 |
|
# 'mac' => namify(line[MACADDRESS], number),
|
| 83 |
|
# 'organization_id' => foreman_organization(:name => line[ORGANIZATION]),
|
| 84 |
|
# 'location_id' => foreman_location(:name => line[LOCATION]),
|
| 85 |
|
# 'environment_id' => foreman_environment(:name => line[ENVIRONMENT]),
|
| 86 |
|
# 'operatingsystem_id' => foreman_operatingsystem(:name => line[OPERATINGSYSTEM]),
|
| 87 |
|
# 'architecture_id' => foreman_architecture(:name => line[ARCHITECTURE]),
|
| 88 |
|
# 'domain_id' => foreman_domain(:name => line[DOMAIN]),
|
| 89 |
|
# 'ptable_id' => foreman_partitiontable(:name => line[PARTITIONTABLE])
|
| 90 |
|
# }
|
| 91 |
|
# })
|
|
103 |
hostgroup = @api.resource(:hostgroups).call(:create, params)
|
|
104 |
@existing[name] = hostgroup['id']
|
| 92 |
105 |
else
|
| 93 |
106 |
print "Updating host '#{name}'..." if option_verbose?
|
| 94 |
|
# @api.resource(:hosts).call(:update, {
|
| 95 |
|
# 'id' => @existing[name],
|
| 96 |
|
# 'host' => {
|
| 97 |
|
# 'name' => name,
|
| 98 |
|
# 'mac' => namify(line[MACADDRESS], number),
|
| 99 |
|
# 'organization_id' => foreman_organization(:name => line[ORGANIZATION]),
|
| 100 |
|
# 'environment_id' => foreman_environment(:name => line[ENVIRONMENT]),
|
| 101 |
|
# 'operatingsystem_id' => foreman_operatingsystem(:name => line[OPERATINGSYSTEM]),
|
| 102 |
|
# 'architecture_id' => foreman_architecture(:name => line[ARCHITECTURE]),
|
| 103 |
|
# 'domain_id' => foreman_domain(:name => line[DOMAIN]),
|
| 104 |
|
# 'ptable_id' => foreman_partitiontable(:name => line[PARTITIONTABLE])
|
| 105 |
|
# }
|
| 106 |
|
# })
|
|
107 |
params['id'] = @existing[name]
|
|
108 |
hostgroup = @api.resource(:hostgroups).call(:update, params)
|
| 107 |
109 |
end
|
|
110 |
|
|
111 |
# TODO: puppet classes
|
|
112 |
puppetclass_ids = collect_column(line[PUPPET_CLASSES]) do |puppet_class|
|
|
113 |
module_name, name = puppet_class.split('/')
|
|
114 |
foreman_puppet_class(:name => name)
|
|
115 |
end
|
|
116 |
existing_ids = hostgroup['puppet_classes'].collect { |puppet_class| puppet_class['id'] }
|
|
117 |
# DELETE existing_ids - puppetclass_ids
|
|
118 |
# POST puppetclass_ids - existing_ids
|
|
119 |
|
| 108 |
120 |
print "done\n" if option_verbose?
|
| 109 |
121 |
end
|
| 110 |
122 |
rescue RuntimeError => e
|
Added by Thomas McKay almost 7 years ago
Added by
fixes #14389 - import/export host groups