|
module HammerCLICsv
|
|
class CsvCommand
|
|
class ActivationKeysCommand < BaseCommand
|
|
command_name 'activation-keys'
|
|
desc _('import or export activation keys')
|
|
|
|
ORGANIZATION = 'Organization'
|
|
DESCRIPTION = 'Description'
|
|
LIMIT = 'Limit'
|
|
ENVIRONMENT = 'Environment'
|
|
CONTENTVIEW = 'Content View'
|
|
HOSTCOLLECTIONS = 'Host Collections'
|
|
SERVICELEVEL = "Service Level"
|
|
RELEASEVER = "Release Version"
|
|
AUTOATTACH = "Auto-Attach"
|
|
SUBSCRIPTIONS = 'Subscriptions'
|
|
|
|
def export(csv)
|
|
csv << [NAME, ORGANIZATION, DESCRIPTION, LIMIT, ENVIRONMENT, CONTENTVIEW,
|
|
HOSTCOLLECTIONS, AUTOATTACH, SERVICELEVEL, RELEASEVER, SUBSCRIPTIONS]
|
|
@api.resource(:organizations).call(:index, {
|
|
:per_page => 999999
|
|
})['results'].each do |organization|
|
|
next if option_organization && organization['name'] != option_organization
|
|
|
|
@api.resource(:activation_keys).call(:index, {
|
|
'per_page' => 999999,
|
|
'organization_id' => organization['id']
|
|
})['results'].each do |activationkey|
|
|
name = namify(activationkey['name'])
|
|
count = 1
|
|
description = activationkey['description']
|
|
limit = activationkey['unlimited_content_hosts'] ? 'Unlimited' : activationkey['max_content_hosts']
|
|
environment = activationkey['environment']['label']
|
|
contentview = activationkey['content_view']['name']
|
|
hostcollections = export_column(activationkey, 'host_collections', 'name')
|
|
autoattach = activationkey['auto_attach'] ? 'Yes' : 'No'
|
|
servicelevel = activationkey['service_level']
|
|
releasever = activationkey['release_version']
|
|
subscriptions = CSV.generate do |column|
|
|
column << @api.resource(:subscriptions).call(:index, {
|
|
'organization_id' => organization['id'],
|
|
'activation_key_id' => activationkey['id']
|
|
})['results'].collect do |subscription|
|
|
amount = subscription['amount'] == 0 ? 'Automatic' : subscription['amount']
|
|
sku = subscription['product_id'].match(/\A[0-9]/) ? 'Custom' : subscription['product_id']
|
|
"#{amount}|#{sku}|#{subscription['product_name']}"
|
|
end
|
|
end
|
|
subscriptions.delete!("\n")
|
|
csv << [name, count, organization['name'], description, limit, environment, contentview,
|
|
hostcollections, servicelevel, releasever, autoattach, subscriptions]
|
|
end
|
|
end
|
|
end
|
|
|
|
def import
|
|
@existing = {}
|
|
|
|
thread_import do |line|
|
|
create_activationkeys_from_csv(line)
|
|
end
|
|
end
|
|
|
|
def create_activationkeys_from_csv(line)
|
|
return if option_organization && line[ORGANIZATION] != option_organization
|
|
|
|
if !@existing[line[ORGANIZATION]]
|
|
@existing[line[ORGANIZATION]] = {}
|
|
@api.resource(:activation_keys).call(:index, {
|
|
'per_page' => 999999,
|
|
'organization_id' => foreman_organization(:name => line[ORGANIZATION])
|
|
})['results'].each do |activationkey|
|
|
@existing[line[ORGANIZATION]][activationkey['name']] = activationkey['id'] if activationkey
|
|
end
|
|
end
|
|
|
|
count(line[COUNT]).times do |number|
|
|
name = namify(line[NAME], number)
|
|
|
|
params = {
|
|
'organization_id' => foreman_organization(:name => line[ORGANIZATION]),
|
|
'name' => name,
|
|
'environment_id' => lifecycle_environment(line[ORGANIZATION],
|
|
:name => line[ENVIRONMENT]),
|
|
'content_view_id' => katello_contentview(line[ORGANIZATION],
|
|
:name => line[CONTENTVIEW]),
|
|
'description' => line[DESCRIPTION],
|
|
'unlimited_content_hosts' => (line[LIMIT] == 'Unlimited') ? true : false,
|
|
'max_content_hosts' => (line[LIMIT] == 'Unlimited') ? nil : line[LIMIT].to_i
|
|
}
|
|
params['auto_attach'] = (line[AUTOATTACH] == 'Yes' ? true : false) if params['auto_attach']
|
|
params['service_level'] = line[SERVICELEVEL].nil? || line[SERVICELEVEL].empty? ? nil : line[SERVICELEVEL]
|
|
params['release_version'] = line[RELEASEVER].nil? || line[RELEASEVER].empty? ? nil : line[RELEASEVER]
|
|
if !@existing[line[ORGANIZATION]].include? name
|
|
print _("Creating activation key '%{name}'...") % {:name => name} if option_verbose?
|
|
activationkey = @api.resource(:activation_keys).call(:create, params)
|
|
@existing[line[ORGANIZATION]][activationkey['name']] = activationkey['id']
|
|
else
|
|
print _("Updating activation key '%{name}'...") % {:name => name} if option_verbose?
|
|
params['id'] = @existing[line[ORGANIZATION]][name]
|
|
activationkey = @api.resource(:activation_keys).call(:update, params)
|
|
end
|
|
|
|
update_subscriptions(activationkey, line)
|
|
update_groups(activationkey, line)
|
|
|
|
puts _('done') if option_verbose?
|
|
end
|
|
end
|
|
|
|
def update_groups(activationkey, line)
|
|
if line[HOSTCOLLECTIONS] && line[HOSTCOLLECTIONS] != ''
|
|
# TODO: note that existing system groups are not removed
|
|
CSV.parse_line(line[HOSTCOLLECTIONS], {:skip_blanks => true}).each do |name|
|
|
@api.resource(:activation_keys).call(:add_host_collections, {
|
|
'id' => activationkey['id'],
|
|
'host_collection_ids' => [katello_hostcollection(line[ORGANIZATION], :name => name)]
|
|
})
|
|
end
|
|
end
|
|
end
|
|
|
|
def update_subscriptions(activationkey, line)
|
|
if line[SUBSCRIPTIONS] && line[SUBSCRIPTIONS] != ''
|
|
subscriptions = CSV.parse_line(line[SUBSCRIPTIONS], {:skip_blanks => true}).collect do |subscription_details|
|
|
(amount, sku, name) = subscription_details.split('|')
|
|
{
|
|
:id => katello_subscription(line[ORGANIZATION], :name => name),
|
|
:quantity => (amount.nil? || amount == 'Automatic') ? 0 : amount
|
|
}
|
|
end
|
|
|
|
existing_subscriptions = @api.resource(:subscriptions).call(:index, {
|
|
'organization_id' => foreman_organization(:name => line[ORGANIZATION]),
|
|
'per_page' => 999999,
|
|
'activation_key_id' => activationkey['id']
|
|
})['results']
|
|
if existing_subscriptions.length > 0
|
|
@api.resource(:activation_keys).call(:remove_subscriptions, {
|
|
'id' => activationkey['id'],
|
|
'subscriptions' => existing_subscriptions
|
|
})
|
|
end
|
|
|
|
@api.resource(:activation_keys).call(:add_subscriptions, {
|
|
'id' => activationkey['id'],
|
|
'subscriptions' => subscriptions
|
|
})
|
|
end
|
|
end
|
|
|
|
def usage_limit(limit)
|
|
Integer(limit) rescue -1
|
|
end
|
|
end
|
|
end
|
|
end
|
