hammer_cli_csv 1.0.1 → 1.0.2

data/lib/hammer_cli_csv/compute_profiles.rb

@@ -1,20 +1,4 @@
-# Copyright 2013-2014 Red Hat, Inc.
-#
-# This software is licensed to you under the GNU General Public
-# License as published by the Free Software Foundation; either version
-# 2 of the License (GPLv2) or (at your option) any later version.
-# There is NO WARRANTY for this software, express or implied,
-# including the implied warranties of MERCHANTABILITY,
-# NON-INFRINGEMENT, or FITNESS FOR A PARTICULAR PURPOSE. You should
-# have received a copy of GPLv2 along with this software; if not, see
-# http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt.
-
-require 'hammer_cli'
-require 'json'
-require 'csv'
-
 # TODO: waiting for https://github.com/theforeman/foreman/pull/1326
-
 module HammerCLICsv
   class CsvCommand
     class ComputeProfilesCommand < BaseCommand
@@ -28,19 +12,18 @@ module HammerCLICsv
       URL = 'URL'
 
       def export
-        CSV.open(option_csv_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
-          csv << [NAME, COUNT, ORGANIZATIONS, LOCATIONS, DESCRIPTION, PROVIDER, URL]
+        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
+          csv << [NAME, ORGANIZATIONS, LOCATIONS, DESCRIPTION, PROVIDER, URL]
           @api.resource(:compute_profiles).call(:index, {:per_page => 999999})['results'].each do |compute_profile|
             puts compute_profile
             compute_profile = @api.resource(:compute_profiles).call(:show, {'id' => compute_profile['id']})
             name = compute_profile['name']
-            count = 1
             organizations = export_column(compute_profile, 'organizations', 'name')
             locations = export_column(compute_profile, 'locations', 'name')
             description = compute_profile['description']
             provider = compute_profile['provider']
             url = compute_profile['url']
-            csv << [name, count, organizations, locations, description, provider, url]
+            csv << [name, organizations, locations, description, provider, url]
           end
         end
       end
@@ -57,7 +40,7 @@ module HammerCLICsv
       end
 
       def create_compute_profiles_from_csv(line)
-        line[COUNT].to_i.times do |number|
+        count(line[COUNT]).times do |number|
           name = namify(line[NAME], number)
           if !@existing.include? name
             print "Creating compute profile '#{name}'..." if option_verbose?

data/lib/hammer_cli_csv/compute_resources.rb

@@ -1,14 +1,3 @@
-# Copyright 2013-2014 Red Hat, Inc.
-#
-# This software is licensed to you under the GNU General Public
-# License as published by the Free Software Foundation; either version
-# 2 of the License (GPLv2) or (at your option) any later version.
-# There is NO WARRANTY for this software, express or implied,
-# including the implied warranties of MERCHANTABILITY,
-# NON-INFRINGEMENT, or FITNESS FOR A PARTICULAR PURPOSE. You should
-# have received a copy of GPLv2 along with this software; if not, see
-# http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt.
-
 require 'hammer_cli'
 require 'json'
 require 'csv'
@@ -26,19 +15,18 @@ module HammerCLICsv
       URL = 'URL'
 
       def export
-        CSV.open(option_csv_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
-          csv << [NAME, COUNT, ORGANIZATIONS, LOCATIONS, DESCRIPTION, PROVIDER, URL]
+        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
+          csv << [NAME, ORGANIZATIONS, LOCATIONS, DESCRIPTION, PROVIDER, URL]
           @api.resource(:compute_resources).call(:index, {:per_page => 999999})['results'].each do |compute_resource|
             compute_resource = @api.resource(:compute_resources).call(:show, {'id' => compute_resource['id']})
 
             name = compute_resource['name']
-            count = 1
             organizations = export_column(compute_resource, 'organizations', 'name')
             locations = export_column(compute_resource, 'locations', 'name')
             description = compute_resource['description']
             provider = compute_resource['provider']
             url = compute_resource['url']
-            csv << [name, count, organizations, locations, description, provider, url]
+            csv << [name, organizations, locations, description, provider, url]
           end
         end
       end
@@ -55,7 +43,7 @@ module HammerCLICsv
       end
 
       def create_compute_resources_from_csv(line)
-        line[COUNT].to_i.times do |number|
+        count(line[COUNT]).times do |number|
           name = namify(line[NAME], number)
           if !@existing.include? name
             print "Creating compute resource '#{name}'..." if option_verbose?

data/lib/hammer_cli_csv/containers.rb

@@ -0,0 +1,65 @@
+module HammerCLICsv
+  class CsvCommand
+    class ContainersCommand < BaseCommand
+      command_name 'containers'
+      desc         'import or export containers'
+
+      REGISTRY = 'Registry'
+      REPOSITORY = 'Repository:Tag'
+      COMPUTERESOURCE = 'Compute Resource'
+      ATTACH = 'Attach I/O'
+      ENTRYPOINT = 'Entry Point'
+      COMMAND = 'Command'
+
+      def export
+        CSV.open(option_file || '/dev/stdout', 'wb') do |csv|
+          csv << [NAME, REGISTRY, REPOSITORY, COMPUTERESOURCE, ATTACH, ENTRYPOINT, COMMAND]
+          @api.resource(:containers).call(:index, {'per_page' => 999999})['results'].each do |container|
+            csv << [container['name'],
+                    container['registry_name'],
+                    "#{container['repository_name']}:#{container['tag']}",
+                    container['compute_resource_name'],
+                    export_attach_types(container),
+                    container['entrypoint'],
+                    container['command']]
+          end
+        end
+      end
+
+      def import
+        @existing = {}
+
+        thread_import do |line|
+          create_containers_from_csv(line)
+        end
+      end
+
+      def create_containers_from_csv(line)
+        # TODO: containers cannot be updated (no api)
+        # count(line[COUNT]).times do |number|
+        #   name = namify(line[NAME], number)
+        #   params =  { 'id' => foreman_container(:name => name),
+        #               'container' => {
+        #                 'name' => name,
+        #                 'command' => line[COMMAND]
+        #               }
+        #             }
+        #   print "Updating container '#{name}'..." if option_verbose?
+        #   @api.resource(:containers).call(:update, params)
+        # end
+        # print "done\n" if option_verbose?
+      end
+
+      private
+
+      def export_attach_types(container)
+        types = []
+        types << 'tty' if container['tty']
+        types << 'stdin' if container['attach_stdin']
+        types << 'stdout' if container['attach_stdout']
+        types << 'stderr' if container['attach_stderr']
+        types.join(',')
+      end
+    end
+  end
+end

data/lib/hammer_cli_csv/content_hosts.rb

@@ -1,14 +1,3 @@
-# Copyright 2013-2014 Red Hat, Inc.
-#
-# This software is licensed to you under the GNU General Public
-# License as published by the Free Software Foundation; either version
-# 2 of the License (GPLv2) or (at your option) any later version.
-# There is NO WARRANTY for this software, express or implied,
-# including the implied warranties of MERCHANTABILITY,
-# NON-INFRINGEMENT, or FITNESS FOR A PARTICULAR PURPOSE. You should
-# have received a copy of GPLv2 along with this software; if not, see
-# http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt.
-
 module HammerCLICsv
   class CsvCommand
     class ContentHostsCommand < BaseCommand
@@ -17,8 +6,6 @@ module HammerCLICsv
       command_name 'content-hosts'
       desc         'import or export content hosts'
 
-      option %w(--organization), 'ORGANIZATION', 'Only process organization matching this name'
-
       ORGANIZATION = 'Organization'
       ENVIRONMENT = 'Environment'
       CONTENTVIEW = 'Content View'
@@ -35,8 +22,8 @@ module HammerCLICsv
       SUBSCRIPTIONS = 'Subscriptions'
 
       def export
-        CSV.open(option_csv_file || '/dev/stdout', 'wb', {:force_quotes => false}) do |csv|
-          csv << [NAME, COUNT, ORGANIZATION, ENVIRONMENT, CONTENTVIEW, HOSTCOLLECTIONS, VIRTUAL, HOST,
+        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => false}) do |csv|
+          csv << [NAME, ORGANIZATION, ENVIRONMENT, CONTENTVIEW, HOSTCOLLECTIONS, VIRTUAL, HOST,
                   OPERATINGSYSTEM, ARCHITECTURE, SOCKETS, RAM, CORES, SLA, PRODUCTS, SUBSCRIPTIONS]
           if @server_status['release'] == 'Headpin'
             export_sam csv
@@ -63,7 +50,6 @@ module HammerCLICsv
           host_subscriptions = @headpin.get("systems/#{host_id}/subscriptions")['entitlements']
 
           name = host['name']
-          count = 1
           organization_name = host['owner']['displayName']
           environment = host['environment']['name']
           contentview = host['content_view']['name']
@@ -95,7 +81,7 @@ module HammerCLICsv
           end
           subscriptions.delete!("\n")
 
-          csv << [name, count, organization_name, environment, contentview, hostcollections, virtual, hypervisor,
+          csv << [name, organization_name, environment, contentview, hostcollections, virtual, hypervisor,
                   operatingsystem, architecture, sockets, ram, cores, sla, products, subscriptions]
         end
       end
@@ -114,7 +100,6 @@ module HammerCLICsv
             })
 
             name = host['name']
-            count = 1
             organization_name = organization['name']
             environment = host['environment']['label']
             contentview = host['content_view']['name']
@@ -148,7 +133,7 @@ module HammerCLICsv
               end
             end
             subscriptions.delete!("\n")
-            csv << [name, count, organization_name, environment, contentview, hostcollections, virtual, hypervisor_host,
+            csv << [name, organization_name, environment, contentview, hostcollections, virtual, hypervisor_host,
                     operatingsystem, architecture, sockets, ram, cores, sla, products, subscriptions]
           end
         end
@@ -164,7 +149,7 @@ module HammerCLICsv
       end
 
       def import_remotely
-        params = {'content' => ::File.new(::File.expand_path(option_csv_file), 'rb')}
+        params = {'content' => ::File.new(::File.expand_path(option_file), 'rb')}
         headers = {:content_type => 'multipart/form-data', :multipart => true}
         task_progress(@api.resource(:csv).call(:import_content_hosts, params, headers))
       end
@@ -185,8 +170,8 @@ module HammerCLICsv
                 'guest_ids' => guest_ids
             })
           end
+          puts _('done') if option_verbose?
         end
-        puts _('done') if option_verbose?
       end
 
       def create_content_hosts_from_csv(line)
@@ -204,7 +189,7 @@ module HammerCLICsv
           (total / 20 + 2).to_i.times do |page|
             @api.resource(:systems).call(:index, {
                 'organization_id' => foreman_organization(:name => line[ORGANIZATION]),
-                'page' => page,
+                'page' => page + 1,
                 'per_page' => 20
             })['results'].each do |host|
               @existing[line[ORGANIZATION]][host['name']] = host['uuid'] if host
@@ -212,7 +197,7 @@ module HammerCLICsv
           end
         end
 
-        line[COUNT].to_i.times do |number|
+        count(line[COUNT]).times do |number|
           name = namify(line[NAME], number)
 
           if !@existing[line[ORGANIZATION]].include? name
@@ -314,10 +299,9 @@ module HammerCLICsv
       end
 
       def update_subscriptions(host_id, line)
-        existing_subscriptions = @api.resource(:subscriptions).call(:index, {
-            'organization_id' => foreman_organization(:name => line[ORGANIZATION]),
+        existing_subscriptions = @api.resource(:systems).call(:subscriptions, {
             'per_page' => 999999,
-            'system_id' => host_id
+            'id' => host_id
         })['results']
         if existing_subscriptions.length > 0
           @api.resource(:subscriptions).call(:destroy, {
@@ -326,6 +310,18 @@ module HammerCLICsv
           })
         end
 
+        # existing_subscriptions = @api.resource(:subscriptions).call(:index, {
+        #     'organization_id' => foreman_organization(:name => line[ORGANIZATION]),
+        #     'per_page' => 999999,
+        #     'system_id' => host_id
+        # })['results']
+        # if existing_subscriptions.length > 0
+        #   @api.resource(:subscriptions).call(:destroy, {
+        #     'system_id' => host_id,
+        #     'id' => existing_subscriptions[0]['id']
+        #   })
+        # end
+
         return if line[SUBSCRIPTIONS].nil? || line[SUBSCRIPTIONS].empty?
 
         subscriptions = CSV.parse_line(line[SUBSCRIPTIONS], {:skip_blanks => true}).collect do |details|

data/lib/hammer_cli_csv/content_view_filters.rb

@@ -1,14 +1,3 @@
-# Copyright 2013-2014 Red Hat, Inc.
-#
-# This software is licensed to you under the GNU General Public
-# License as published by the Free Software Foundation; either version
-# 2 of the License (GPLv2) or (at your option) any later version.
-# There is NO WARRANTY for this software, express or implied,
-# including the implied warranties of MERCHANTABILITY,
-# NON-INFRINGEMENT, or FITNESS FOR A PARTICULAR PURPOSE. You should
-# have received a copy of GPLv2 along with this software; if not, see
-# http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt.
-
 # NOTE:
 #   rpm -qa --queryformat "%{NAME}|=|%{VERSION}-%{RELEASE},"
 
@@ -18,8 +7,6 @@ module HammerCLICsv
       command_name 'content-view-filters'
       desc         'import or export content-view-filters'
 
-      option %w(--organization), 'ORGANIZATION', 'Only process organization matching this name'
-
       CONTENTVIEW = 'Content View'
       ORGANIZATION = 'Organization'
       DESCRIPTION = 'Description'
@@ -28,14 +15,13 @@ module HammerCLICsv
       RULES = 'Rules'
 
       def export
-        CSV.open(option_csv_file || '/dev/stdout', 'wb', {:force_quotes => false}) do |csv|
-          csv << [NAME, COUNT, CONTENTVIEW, ORGANIZATION, TYPE, DESCRIPTION, REPOSITORIES, RULES]
+        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => false}) do |csv|
+          csv << [NAME, CONTENTVIEW, ORGANIZATION, TYPE, DESCRIPTION, REPOSITORIES, RULES]
           @api.resource(:organizations).call(:index, {
               :per_page => 999999
           })['results'].each do |organization|
             next if option_organization && organization['name'] != option_organization
 
-            composite_contentviews = []
             @api.resource(:content_views).call(:index, {
                 'per_page' => 999999,
                 'organization_id' => organization['id'],
@@ -44,34 +30,23 @@ module HammerCLICsv
               @api.resource(:content_view_filters).call(:index, {
                   'content_view_id' => contentview['id']
               })['results'].collect do |filter|
-                filter_type = "#{filter['inclusion'] == true ? 'Include' : 'Exclude'} #{filter['type']}"
+                filter_type = "#{filter['inclusion'] == true ? 'Include' : 'Exclude'} #{export_filter_type(filter['type'])}"
 
+                rules = nil
                 case filter['type']
                 when /rpm/
-                  rules = CSV.generate do |column|
-                    column << filter['rules'].collect do |rule|
-                      if rule['version']
-                        "#{rule['name']}|=|#{rule['version']}"
-                      elsif rule['min_version'] && rule['max_version']
-                        "#{rule['name']}|-|#{rule['min_version']},#{rule['max_version']}"
-                      elsif rule['min_version']
-                        "#{rule['name']}|>|#{rule['min_version']}"
-                      elsif rule['max_version']
-                        "#{rule['name']}|<|#{rule['max_version']}"
-                      end
-                    end
-                  end
-                  rules.delete!("\n")
+                  rules = export_rpm_rules(filter)
                 when /erratum/
+                  rules = export_erratum_rules(filter)
                 when /package_group/
+                  rules = export_package_group_rules(filter)
                 else
                   raise "Unknown filter rule type '#{filter['type']}'"
                 end
-                #puts "#{filter['type']} -> #{rule}"
 
-                name = contentview['name']
+                name = filter['name']
                 repositories = export_column(filter, 'repositories', 'name')
-                csv << [name, 1, contentview['name'], organization['name'], filter_type, filter['description'],
+                csv << [name, contentview['name'], organization['name'], filter_type, filter['description'],
                         repositories, rules]
               end
             end
@@ -87,6 +62,7 @@ module HammerCLICsv
         end
       end
 
+      # rubocop:disable CyclomaticComplexity
       def create_filters_from_csv(line)
         return if option_organization && line[ORGANIZATION] != option_organization
 
@@ -105,40 +81,39 @@ module HammerCLICsv
           katello_repository(line[ORGANIZATION], :name => repository)
         end
 
-        line[COUNT].to_i.times do |number|
-          name = namify(line[NAME], number)
+        count(line[COUNT]).times do |number|
+          filter_name = namify(line[NAME], number)
 
-          filter_id = @existing_filters[line[ORGANIZATION]][line[CONTENTVIEW]][name]
+          filter_id = @existing_filters[line[ORGANIZATION]][line[CONTENTVIEW]][filter_name]
+          filter_type = import_filter_type(line[TYPE])
           if !filter_id
-            print "Creating filter '#{name}' for content view filter '#{line[CONTENTVIEW]}'..." if option_verbose?
+            print "Creating filter '#{filter_name}' for content view filter '#{line[CONTENTVIEW]}'..." if option_verbose?
             filter_id = @api.resource(:content_view_filters).call(:create, {
                 'content_view_id' => katello_contentview(line[ORGANIZATION], :name => line[CONTENTVIEW]),
-                'name' => name,
+                'name' => filter_name,
                 'description' => line[DESCRIPTION],
-                'type' => filter_type(line[TYPE]),
+                'type' => filter_type,
                 'inclusion' => filter_inclusion?(line[TYPE]),
                 'repository_ids' => repository_ids
             })['id']
-            @existing_filters[line[ORGANIZATION]][name] = filter_id
+            @existing_filters[line[ORGANIZATION]][filter_name] = filter_id
           else
-            print "Updating filter '#{name}' for content view filter '#{line[CONTENTVIEW]}'..." if option_verbose?
+            print "Updating filter '#{filter_name}' for content view filter '#{line[CONTENTVIEW]}'..." if option_verbose?
             @api.resource(:content_view_filters).call(:update, {
                 'id' => filter_id,
                 'description' => line[DESCRIPTION],
-                'type' => filter_type(line[TYPE]),
+                'type' => filter_type,
                 'inclusion' => filter_inclusion?(line[TYPE]),
                 'repository_ids' => repository_ids
             })
           end
 
-          @existing_rules ||= {}
-          @existing_rules[line[ORGANIZATION]] ||= {}
-          @existing_rules[line[ORGANIZATION]][line[CONTENTVIEW]] ||= {}
+          existing_rules = {}
           @api.resource(:content_view_filter_rules).call(:index, {
               'per_page' => 999999,
               'content_view_filter_id' => filter_id
           })['results'].each do |rule|
-            @existing_rules[line[ORGANIZATION]][line[CONTENTVIEW]][rule['name']] = rule
+            existing_rules[rule['name']] = rule
           end
 
           collect_column(line[RULES]) do |rule|
@@ -147,29 +122,29 @@ module HammerCLICsv
               'content_view_filter_id' => filter_id,
               'name' => name
             }
-            if type == '='
-              params['type'] = 'equal',
-                               params['version'] = version
+            if type == 'all'
+              # empty
+            elsif type == '='
+              params['version'] = version
             elsif type == '<'
-              params['type'] = 'less',
-                               params['max_version'] = version
+              params['max_version'] = version
             elsif type == '>'
-              params['type'] = 'greater',
-                               params['min_version'] = version
+              params['min_version'] = version
             elsif type == '-'
-              params['type'] = 'range',
-                               min_version, max_version = version.split(',')
+              min_version, max_version = version.split(',')
               params['min_version'] = min_version
               params['max_version'] = max_version
+            elsif filter_type == 'package_group'
+              params['uuid'] = name # TODO: this is not right
             else
               raise "Unknown type '#{type}' from '#{line[RULES]}'"
             end
 
-            rule = @existing_rules[line[ORGANIZATION]][line[CONTENTVIEW]][name]
+            rule = existing_rules[name]
             if !rule
               print "." if option_verbose?
               rule = @api.resource(:content_view_filter_rules).call(:create, params)
-              @existing_rules[line[ORGANIZATION]][line[CONTENTVIEW]][rule['name']] = rule
+              existing_rules[rule['name']] = rule
             else
               print "." if option_verbose?
               params['id'] = rule['id']
@@ -186,9 +161,23 @@ module HammerCLICsv
 
       private
 
-      def filter_type(type)
-        if type.split[1] == 'RPM'
+      def import_filter_type(type)
+        case type.split[1..-1].join(' ')
+        when /packages/i
           'rpm'
+        when /package groups/i
+          'package_group'
+        else
+          'unknown'
+        end
+      end
+
+      def export_filter_type(type)
+        case type.split[1]
+        when /rpm/i
+          'Packages'
+        when /package_group/i
+          'Package Groups'
         else
           'unknown'
         end
@@ -201,6 +190,47 @@ module HammerCLICsv
           false
         end
       end
+
+      def export_rpm_rules(filter)
+        rules = CSV.generate do |column|
+          column << filter['rules'].collect do |rule|
+            if rule['version']
+              "#{rule['name']}|=|#{rule['version']}"
+            elsif rule['min_version'] && rule['max_version']
+              "#{rule['name']}|-|#{rule['min_version']},#{rule['max_version']}"
+            elsif rule['min_version']
+              "#{rule['name']}|>|#{rule['min_version']}"
+            elsif rule['max_version']
+              "#{rule['name']}|<|#{rule['max_version']}"
+            else
+              "#{rule['name']}|all"
+            end
+          end
+        end
+        rules.delete!("\n")
+      end
+
+      def export_erratum_rules(filter)
+        rules = CSV.generate do |column|
+          rule = filter['rules'][0]
+          conditions = []
+          conditions << "start = #{DateTime.parse(rule['start_date']).strftime('%F')}" if rule['start_date']
+          conditions << "end = #{DateTime.parse(rule['end_date']).strftime('%F')}" if rule['end_date']
+          conditions << "types = #{rule['types'].join(',')}" if rule['types']
+          conditions << "errata = #{rule['errata_id']}" if rule['errata_id']
+          column << conditions
+        end
+        rules.delete!("\n")
+      end
+
+      def export_package_group_rules(filter)
+        rules = CSV.generate do |column|
+          column << filter['rules'].collect do |rule|
+            rule['name']
+          end
+        end
+        rules.delete!("\n")
+      end
     end
   end
 end