|
module HammerCLICsv
|
|
class CsvCommand
|
|
class PuppetReportsCommand < BaseCommand
|
|
command_name 'puppet-reports'
|
|
desc 'import or export puppet reports'
|
|
|
|
ORGANIZATION = 'Organization'
|
|
ENVIRONMENT = 'Environment'
|
|
CONTENTVIEW = 'Content View'
|
|
SYSTEMGROUPS = 'System Groups'
|
|
VIRTUAL = 'Virtual'
|
|
HOST = 'Host'
|
|
OPERATINGSYSTEM = 'OS'
|
|
ARCHITECTURE = 'Arch'
|
|
SOCKETS = 'Sockets'
|
|
RAM = 'RAM'
|
|
CORES = 'Cores'
|
|
SLA = 'SLA'
|
|
PRODUCTS = 'Products'
|
|
SUBSCRIPTIONS = 'Subscriptions'
|
|
|
|
def export(csv)
|
|
csv << [NAME, ORGANIZATION, ENVIRONMENT, CONTENTVIEW, SYSTEMGROUPS, VIRTUAL, HOST,
|
|
OPERATINGSYSTEM, ARCHITECTURE, SOCKETS, RAM, CORES, SLA, PRODUCTS, SUBSCRIPTIONS]
|
|
@api.resource(:organizations).call(:index, {
|
|
:per_page => 999999
|
|
})['results'].each do |organization|
|
|
@api.resource(:systems).call(:index, {
|
|
'per_page' => 999999,
|
|
'organization_id' => organization['id']
|
|
})['results'].each do |system|
|
|
system = @api.resource(:systems).call(:show, {
|
|
'id' => system['uuid'],
|
|
'fields' => 'full'
|
|
})
|
|
|
|
name = system['name']
|
|
organization_label = organization['label']
|
|
environment = system['environment']['label']
|
|
contentview = system['content_view']['name']
|
|
hostcollections = CSV.generate do |column|
|
|
column << system['systemGroups'].collect do |hostcollection|
|
|
hostcollection['name']
|
|
end
|
|
end
|
|
hostcollections.delete!("\n")
|
|
virtual = system['facts']['virt.is_guest'] == 'true' ? 'Yes' : 'No'
|
|
host = system['host']
|
|
operatingsystem = "#{system['facts']['distribution.name']} " if system['facts']['distribution.name']
|
|
operatingsystem += system['facts']['distribution.version'] if system['facts']['distribution.version']
|
|
architecture = system['facts']['uname.machine']
|
|
sockets = system['facts']['cpu.cpu_socket(s)']
|
|
ram = system['facts']['memory.memtotal']
|
|
cores = system['facts']['cpu.core(s)_per_socket']
|
|
sla = ''
|
|
products = CSV.generate do |column|
|
|
column << system['installedProducts'].collect do |product|
|
|
"#{product['productId']}|#{product['productName']}"
|
|
end
|
|
end
|
|
products.delete!("\n")
|
|
subscriptions = CSV.generate do |column|
|
|
column << @api.resource(:subscriptions).call(:index, {
|
|
'system_id' => system['uuid']
|
|
})['results'].collect do |subscription|
|
|
"#{subscription['product_id']}|#{subscription['product_name']}"
|
|
end
|
|
end
|
|
subscriptions.delete!("\n")
|
|
csv << [name, organization_label, environment, contentview, hostcollections, virtual, host,
|
|
operatingsystem, architecture, sockets, ram, cores, sla, products, subscriptions]
|
|
end
|
|
end
|
|
end
|
|
|
|
def import
|
|
@existing = {}
|
|
@host_guests = {}
|
|
|
|
thread_import do |line|
|
|
create_systems_from_csv(line)
|
|
end
|
|
|
|
print 'Updating host and guest associations...' if option_verbose?
|
|
@host_guests.each do |host_id, guest_ids|
|
|
@api.resource(:systems).call(:update, {
|
|
'id' => host_id,
|
|
'guest_ids' => guest_ids
|
|
})
|
|
end
|
|
puts 'done' if option_verbose?
|
|
end
|
|
|
|
def create_systems_from_csv(line)
|
|
if !@existing[line[ORGANIZATION]]
|
|
@existing[line[ORGANIZATION]] = {}
|
|
@api.resource(:systems).call(:index, {
|
|
'organization_id' => line[ORGANIZATION],
|
|
'per_page' => 999999
|
|
})['results'].each do |system|
|
|
@existing[line[ORGANIZATION]][system['name']] = system['uuid'] if system
|
|
end
|
|
end
|
|
|
|
count(line[COUNT]).times do |number|
|
|
name = namify(line[NAME], number)
|
|
|
|
# TODO: w/ @daviddavis p-r
|
|
#subscriptions(line).each do |subscription|
|
|
# get_subscription(line[ORGANIZATION], :name => subscription[:number])
|
|
#end
|
|
|
|
if !@existing[line[ORGANIZATION]].include? name
|
|
print "Creating system '#{name}'..." if option_verbose?
|
|
system_id = @api.resource(:systems).call(:create, {
|
|
'name' => name,
|
|
'organization_id' => line[ORGANIZATION],
|
|
'environment_id' => lifecycle_environment(line[ORGANIZATION], :name => line[ENVIRONMENT]),
|
|
'content_view_id' => lifecycle_contentview(line[ORGANIZATION], :name => line[CONTENTVIEW]),
|
|
'facts' => facts(line),
|
|
'installed_products' => products(line),
|
|
'type' => 'system'
|
|
})['uuid']
|
|
@existing[line[ORGANIZATION]][name] = system_id
|
|
else
|
|
print "Updating system '#{name}'..." if option_verbose?
|
|
puts line
|
|
system_id = @api.resource(:systems).call(:update, {
|
|
'id' => @existing[line[ORGANIZATION]][name],
|
|
'name' => name,
|
|
'environment_id' => katello_environment(line[ORGANIZATION], :name => line[ENVIRONMENT]),
|
|
'content_view_id' => katello_contentview(line[ORGANIZATION], :name => line[CONTENTVIEW]),
|
|
'facts' => facts(line),
|
|
'installed_products' => products(line)
|
|
})['uuid']
|
|
end
|
|
|
|
if line[VIRTUAL] == 'Yes' && line[HOST]
|
|
raise "Host system '#{line[HOST]}' not found" if !@existing[line[ORGANIZATION]][line[HOST]]
|
|
@host_guests[@existing[line[ORGANIZATION]][line[HOST]]] ||= []
|
|
@host_guests[@existing[line[ORGANIZATION]][line[HOST]]] << system_id
|
|
end
|
|
|
|
set_host_collections(system_id, line)
|
|
|
|
puts 'done' if option_verbose?
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def facts(line)
|
|
facts = {}
|
|
facts['cpu.core(s)_per_socket'] = line[CORES]
|
|
facts['cpu.cpu_socket(s)'] = line[SOCKETS]
|
|
facts['memory.memtotal'] = line[RAM]
|
|
facts['uname.machine'] = line[ARCHITECTURE]
|
|
if line[OPERATINGSYSTEM].index(' ')
|
|
(facts['distribution.name'], facts['distribution.version']) = line[OPERATINGSYSTEM].split(' ')
|
|
else
|
|
(facts['distribution.name'], facts['distribution.version']) = ['RHEL', line[OPERATINGSYSTEM]]
|
|
end
|
|
facts['virt.is_guest'] = line[VIRTUAL] == 'Yes' ? true : false
|
|
facts
|
|
end
|
|
|
|
def set_host_collections(system_id, line)
|
|
CSV.parse_line(line[SYSTEMGROUPS]).each do |hostcollection_name|
|
|
@api.resource(:hostcollections).call(:add_systems, {
|
|
'id' => katello_hostcollection(line[ORGANIZATION], :name => hostcollection_name),
|
|
'system_ids' => [system_id]
|
|
})
|
|
end
|
|
end
|
|
|
|
def products(line)
|
|
products = CSV.parse_line(line[PRODUCTS]).collect do |product_details|
|
|
product = {}
|
|
# TODO: these get passed straight through to candlepin; probably would be better to process in server
|
|
# to allow underscore product_id here
|
|
(product['productId'], product['productName']) = product_details.split('|')
|
|
product
|
|
end
|
|
products
|
|
end
|
|
|
|
def subscriptions(line)
|
|
subscriptions = CSV.parse_line(line[SUBSCRIPTIONS]).collect do |subscription_details|
|
|
subscription = {}
|
|
(subscription[:number], subscription[:name]) = subscription_details.split('|')
|
|
subscription
|
|
end
|
|
subscriptions
|
|
end
|
|
end
|
|
end
|
|
end
|