hammer_cli_csv 2.0.0 → 2.1.0

data/lib/hammer_cli_csv/import.rb

@@ -6,21 +6,32 @@ module HammerCLICsv
       command_name 'import'
       desc         'import by directory'
 
+      def self.supported?
+        true
+      end
+
       option %w(-v --verbose), :flag, _('be verbose')
       option %w(--threads), 'THREAD_COUNT', _('Number of threads to hammer with'),
              :default => 1, :hidden => true
       option '--dir', 'DIRECTORY', _('directory to import from')
       option %w(--organization), 'ORGANIZATION', _('Only process organization matching this name')
-      option %w(--prefix), 'PREFIX', _('Prefix for all name columns')
+      option %w(--prefix), 'PREFIX', _('Prefix for all name columns'),
+             :hidden => true
 
-      RESOURCES = %w( settings organizations locations puppet_environments operating_systems
-                      domains architectures partition_tables lifecycle_environments host_collections
-                      provisioning_templates
-                      subscriptions products content_views content_view_filters activation_keys
-                      hosts content_hosts smart_proxies compute_resources reports roles users )
+      RESOURCES = %w(
+        settings organizations locations puppet_environments operating_systems
+        domains architectures partition_tables lifecycle_environments host_collections
+        provisioning_templates
+        subscriptions products content_views content_view_filters activation_keys
+        hosts content_hosts smart_proxies compute_resources reports roles users
+      )
+      SUPPORTED_RESOURCES = %w(
+        settings
+      )
       RESOURCES.each do |resource|
         dashed = resource.gsub('_', '-')
-        option "--#{dashed}", 'FILE', "csv file for #{dashed}"
+        option "--#{dashed}", 'FILE', "csv file for #{dashed}",
+               :hidden => !SUPPORTED_RESOURCES.include?(resource)
       end
 
       def execute
@@ -67,15 +78,18 @@ module HammerCLICsv
 
       def hammer_resource(resource)
         return if !self.send("option_#{resource}") && !option_dir
-        options_file = "#{option_dir}/#{resource.gsub('_', '-')}.csv" || self.send("option_#{resource}")
+        options_file = option_dir ? "#{option_dir}/#{resource.gsub('_', '-')}.csv" :  self.send("option_#{resource}")
         unless options_file_exists? options_file
           if option_dir
-            puts _("Skipping #{resource} because '#{options_file}' does not exist") if option_verbose?
+            return unless SUPPORTED_RESOURCES.include?(resource)
+            puts _("Skipping %{resource} because '%{options_file}' does not exist") %
+              {:resource => resource, :options_file => options_file} if option_verbose?
             return
           end
           raise "File for #{resource} '#{options_file}' does not exist"
         end
-        puts _("Importing #{resource} from '#{options_file}'") if option_verbose?
+        puts _("Importing %{resource} from '%{options_file}'") %
+          {:resource => resource, :options_file => options_file} if option_verbose?
 
         args = %W( csv #{resource.gsub('_', '-')} --file #{options_file} )
         args << '-v' if option_verbose?

data/lib/hammer_cli_csv/installation_media.rb

@@ -0,0 +1,74 @@
+module HammerCLICsv
+  class CsvCommand
+    class InstallationMediaCommand < BaseCommand
+      command_name 'installation-media'
+      desc         'import or export media'
+
+      ORGANIZATIONS = 'Organizations'
+      LOCATIONS = 'Locations'
+      PATH = 'Path'
+      OSFAMILY = 'OS Family'
+      OPERATING_SYSTEMS = 'Operating Systems'
+
+      def export(csv)
+        csv << [NAME, ORGANIZATIONS, LOCATIONS, PATH, OSFAMILY, OPERATING_SYSTEMS]
+        @api.resource(:media).call(:index, {:per_page => 999999})['results'].each do |medium|
+          medium = @api.resource(:media).call(:show, :id => medium['id'])
+          name = medium['name']
+          organizations = export_column(medium, 'organizations', 'name')
+          locations = export_column(medium, 'locations', 'name')
+          count = 1
+          path = medium['path']
+          os_family = medium['os_family']
+          operating_systems = export_column(medium, 'operatingsystems', 'title')
+          csv << [name, organizations, locations, path, os_family, operating_systems]
+        end
+      end
+
+      def import
+        @existing = {}
+        @api.resource(:media).call(:index, {:per_page => 999999})['results'].each do |medium|
+          @existing[medium['name']] = medium['id'] if medium
+        end
+
+        thread_import do |line|
+          create_from_csv(line)
+        end
+      end
+
+      def create_from_csv(line)
+        params = {
+          'medium' => {
+            'organization_ids' => collect_column(line[ORGANIZATIONS]) do |organization|
+              foreman_organization(:name => organization)
+            end,
+            'location_ids' => collect_column(line[LOCATIONS]) do |location|
+              foreman_location(:name => location)
+            end,
+            'path' => line[PATH],
+            'os_family' => line[OSFAMILY],
+            'operatingsystem_ids' => collect_column(line[OPERATING_SYSTEMS]) do |os|
+              foreman_operatingsystem(:name => os)
+            end
+          }
+        }
+
+        count(line[COUNT]).times do |number|
+          name = namify(line[NAME], number)
+          params['medium']['name'] = name
+
+          if !@existing.include? name
+            print _("Creating installation medium '%{name}'... ") % {:name => name} if option_verbose?
+            medium = @api.resource(:media).call(:create, params)
+            @existing[name] = medium['id']
+          else
+            print _("Updating installation medium '%{name}'... ") % {:name => name} if option_verbose?
+            params['id'] = @existing[name]
+            medium = @api.resource(:media).call(:update, params)
+          end
+          puts _('done') if option_verbose?
+        end
+      end
+    end
+  end
+end

data/lib/hammer_cli_csv/job_templates.rb

@@ -17,48 +17,46 @@ module HammerCLICsv
       INPUT_TYPE = 'Input:Type'
       INPUT_PARAMETERS = 'Input:Parameters'
 
-      def export
-        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
-          csv << [NAME, ORGANIZATIONS, LOCATIONS, DESCRIPTION, JOB, PROVIDER, SNIPPET, TEMPLATE,
-                  INPUT_NAME, INPUT_DESCRIPTION, INPUT_REQUIRED, INPUT_TYPE, INPUT_PARAMETERS]
-          @api.resource(:job_templates).call(:index, {
-              :per_page => 999999
-          })['results'].each do |template_id|
-            template = @api.resource(:job_templates).call(:show, {:id => template_id['id']})
-            next if template['locked']
-            next unless option_organization.nil? || template['organizations'].detect { |org| org['name'] == option_organization }
-            name = template['name']
-            description = template['description_format']
-            job = template['job_category']
-            snippet = template['snippet'] ? 'Yes' : 'No'
-            provider = template['provider_type']
-            organizations = export_column(template, 'organizations', 'name')
-            locations = export_column(template, 'locations', 'name')
-            csv << [name, organizations, locations, description, job, provider, snippet, template['template']]
+      def export(csv)
+        csv << [NAME, ORGANIZATIONS, LOCATIONS, DESCRIPTION, JOB, PROVIDER, SNIPPET, TEMPLATE,
+                INPUT_NAME, INPUT_DESCRIPTION, INPUT_REQUIRED, INPUT_TYPE, INPUT_PARAMETERS]
+        @api.resource(:job_templates).call(:index, {
+            :per_page => 999999
+        })['results'].each do |template_id|
+          template = @api.resource(:job_templates).call(:show, {:id => template_id['id']})
+          next if template['locked']
+          next unless option_organization.nil? || template['organizations'].detect { |org| org['name'] == option_organization }
+          name = template['name']
+          description = template['description_format']
+          job = template['job_category']
+          snippet = template['snippet'] ? 'Yes' : 'No'
+          provider = template['provider_type']
+          organizations = export_column(template, 'organizations', 'name')
+          locations = export_column(template, 'locations', 'name')
+          csv << [name, organizations, locations, description, job, provider, snippet, template['template']]
 
-            template_columns = [name] + Array.new(7)
-            @api.resource(:template_inputs).call(:index, {
-                :template_id => template['id']
-            })['results'].each  do|input|
-              input_field = nil
-              input_options = nil
-              case input['input_type']
-              when /user/
-                input_name = export_column(input, 'options') do |value|
-                  value
-                end
-              when /fact/
-                input_name = input['fact_name']
-              when /variable/
-                input_name = input['variable_name']
-              when /puppet_parameter/
-                input_name = "#{input['puppet_class_name']}|#{input['puppet_parameter_name']}"
-              else
-                raise _("Unknown job template input type '%{type}'") % {:type => input['input_type']}
+          template_columns = [name] + Array.new(7)
+          @api.resource(:template_inputs).call(:index, {
+              :template_id => template['id']
+          })['results'].each  do|input|
+            input_field = nil
+            input_options = nil
+            case input['input_type']
+            when /user/
+              input_name = export_column(input, 'options') do |value|
+                value
               end
-              required = input['required'] ? 'Yes' : 'No'
-              csv << template_columns + [input['name'], input['description'], required, input['input_type'], input_name]
+            when /fact/
+              input_name = input['fact_name']
+            when /variable/
+              input_name = input['variable_name']
+            when /puppet_parameter/
+              input_name = "#{input['puppet_class_name']}|#{input['puppet_parameter_name']}"
+            else
+              raise _("Unknown job template input type '%{type}'") % {:type => input['input_type']}
             end
+            required = input['required'] ? 'Yes' : 'No'
+            csv << template_columns + [input['name'], input['description'], required, input['input_type'], input_name]
           end
         end
       end
@@ -138,8 +136,6 @@ module HammerCLICsv
           # end
 
         end
-      rescue RuntimeError => e
-        raise "#{e}\n       #{line[NAME]}"
       end
 
       def create_template(line, number)

data/lib/hammer_cli_csv/lifecycle_environments.rb

@@ -8,24 +8,22 @@ module HammerCLICsv
       PRIORENVIRONMENT = 'Prior Environment'
       DESCRIPTION = 'Description'
 
-      def export
-        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
-          csv << [NAME, ORGANIZATION, PRIORENVIRONMENT, DESCRIPTION]
-          @api.resource(:organizations).call(:index, {
-              'per_page' => 999999
-          })['results'].each do |organization|
-            next if option_organization && organization['name'] != option_organization
+      def export(csv)
+        csv << [NAME, ORGANIZATION, PRIORENVIRONMENT, DESCRIPTION]
+        @api.resource(:organizations).call(:index, {
+            'per_page' => 999999
+        })['results'].each do |organization|
+          next if option_organization && organization['name'] != option_organization
 
-            @api.resource(:lifecycle_environments).call(:index, {
-                'per_page' => 999999,
-                'organization_id' => organization['id']
-            })['results'].sort { |a, b| a['created_at'] <=> b['created_at'] }.each do |environment|
-              if environment['name'] != 'Library'
-                name = environment['name']
-                prior = environment['prior']['name']
-                description = environment['description']
-                csv << [name, organization['name'], prior, description]
-              end
+          @api.resource(:lifecycle_environments).call(:index, {
+              'per_page' => 999999,
+              'organization_id' => organization['id']
+          })['results'].sort { |a, b| a['created_at'] <=> b['created_at'] }.each do |environment|
+            if environment['name'] != 'Library'
+              name = environment['name']
+              prior = environment['prior']['name']
+              description = environment['description']
+              csv << [name, organization['name'], prior, description]
             end
           end
         end
@@ -76,8 +74,6 @@ module HammerCLICsv
           end
           print "done\n" if option_verbose?
         end
-      rescue RuntimeError => e
-        raise "#{e}\n       #{line}"
       end
     end
   end

data/lib/hammer_cli_csv/locations.rb

@@ -6,12 +6,10 @@ module HammerCLICsv
 
       PARENT = 'Parent Location'
 
-      def export
-        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
-          csv << [NAME, PARENT]
-          @api.resource(:locations).call(:index, {:per_page => 999999})['results'].each do |location|
-            csv << [location['name'], '']
-          end
+      def export(csv)
+        csv << [NAME, PARENT]
+        @api.resource(:locations).call(:index, {:per_page => 999999})['results'].each do |location|
+          csv << [location['name'], '']
         end
       end
 

data/lib/hammer_cli_csv/operating_systems.rb

@@ -13,26 +13,24 @@ module HammerCLICsv
       PROVISIONING_TEMPLATES = 'Provisioning Templates'
       PARAMETERS = 'Parameters'
 
-      def export
-        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
-          csv << [NAME, DESCRIPTION, FAMILY, PASSWORD_HASH, PARTITION_TABLES, ARCHITECTURES, MEDIA,
-                  PROVISIONING_TEMPLATES, PARAMETERS]
-          @api.resource(:operatingsystems).call(:index, {:per_page => 999999})['results'].each do |operatingsystem_id|
-            operatingsystem = @api.resource(:operatingsystems).call(:show, {:id => operatingsystem_id['id']})
-            name = build_os_name(operatingsystem['name'], operatingsystem['major'], operatingsystem['minor'])
-            description = operatingsystem['description']
-            family = operatingsystem['family']
-            password_hash = operatingsystem['password_hash']
-            partition_tables = export_column(operatingsystem, 'ptables', 'name')
-            architectures = export_column(operatingsystem, 'architectures', 'name')
-            media = export_column(operatingsystem, 'media', 'name')
-            partition_tables = export_column(operatingsystem, 'ptables', 'name')
-            parameters = export_column(operatingsystem, 'parameters') do |parameter|
-              "#{parameter['name']}|#{parameter['value']}"
-            end
-            csv << [name, description, family, password_hash, partition_tables, architectures,
-                    media, partition_tables, parameters]
+      def export(csv)
+        csv << [NAME, DESCRIPTION, FAMILY, PASSWORD_HASH, PARTITION_TABLES, ARCHITECTURES, MEDIA,
+                PROVISIONING_TEMPLATES, PARAMETERS]
+        @api.resource(:operatingsystems).call(:index, {:per_page => 999999})['results'].each do |operatingsystem_id|
+          operatingsystem = @api.resource(:operatingsystems).call(:show, {:id => operatingsystem_id['id']})
+          name = build_os_name(operatingsystem['name'], operatingsystem['major'], operatingsystem['minor'])
+          description = operatingsystem['description']
+          family = operatingsystem['family']
+          password_hash = operatingsystem['password_hash']
+          partition_tables = export_column(operatingsystem, 'ptables', 'name')
+          architectures = export_column(operatingsystem, 'architectures', 'name')
+          media = export_column(operatingsystem, 'media', 'name')
+          partition_tables = export_column(operatingsystem, 'ptables', 'name')
+          parameters = export_column(operatingsystem, 'parameters') do |parameter|
+            "#{parameter['name']}|#{parameter['value']}"
           end
+          csv << [name, description, family, password_hash, partition_tables, architectures,
+                  media, partition_tables, parameters]
         end
       end
 
@@ -82,8 +80,6 @@ module HammerCLICsv
           end
           print "done\n" if option_verbose?
         end
-      rescue RuntimeError => e
-        raise "#{e}\n       #{line}"
       end
     end
   end

data/lib/hammer_cli_csv/organizations.rb

@@ -7,20 +7,18 @@ module HammerCLICsv
       LABEL = 'Label'
       DESCRIPTION = 'Description'
 
-      def export
-        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
-          csv << [NAME, LABEL, DESCRIPTION]
+      def export(csv)
+        csv << [NAME, LABEL, DESCRIPTION]
 
-          if @server_status['release'] == 'Headpin'
-            @headpin.get(:organizations).each do |organization|
-              next if option_organization && organization['name'] != option_organization
-              csv << [organization['name'], organization['label'], organization['description']]
-            end
-          else
-            @api.resource(:organizations).call(:index, {:per_page => 999999})['results'].each do |organization|
-              next if option_organization && organization['name'] != option_organization
-              csv << [organization['name'], organization['label'], organization['description']]
-            end
+        if @server_status['release'] == 'Headpin'
+          @headpin.get(:organizations).each do |organization|
+            next if option_organization && organization['name'] != option_organization
+            csv << [organization['name'], organization['label'], organization['description']]
+          end
+        else
+          @api.resource(:organizations).call(:index, {:per_page => 999999})['results'].each do |organization|
+            next if option_organization && organization['name'] != option_organization
+            csv << [organization['name'], organization['label'], organization['description']]
           end
         end
       end

data/lib/hammer_cli_csv/partition_tables.rb

@@ -10,7 +10,7 @@ module HammerCLICsv
       OPERATINGSYSTEMS = 'Operating Systems'
       LAYOUT = 'Layout'
 
-      def export
+      def export(csv)
         # TODO: partition-tables do not return their organizations or locations
         # http://projects.theforeman.org/issues/11175
         organizations_map = {}
@@ -28,26 +28,24 @@ module HammerCLICsv
           end
         end
 
-        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
-          csv << [NAME, ORGANIZATIONS, LOCATIONS, OSFAMILY, OPERATINGSYSTEMS, LAYOUT]
-          @api.resource(:ptables).call(:index, {:per_page => 999999})['results'].each do |ptable|
-            ptable = @api.resource(:ptables).call(:show, {'id' => ptable['id']})
-            name = ptable['name']
-            osfamily = ptable['os_family']
-            layout = ptable['layout']
-            operatingsystems = export_column(ptable, 'operatingsystems', 'title')
-
-            organizations = CSV.generate do |column|
-              column << organizations_map[name] if organizations_map[name]
-            end
-            organizations.delete!("\n")
-            locations = CSV.generate do |column|
-              column << locations_map[name] if locations_map[name]
-            end
-            locations.delete!("\n")
+        csv << [NAME, ORGANIZATIONS, LOCATIONS, OSFAMILY, OPERATINGSYSTEMS, LAYOUT]
+        @api.resource(:ptables).call(:index, {:per_page => 999999})['results'].each do |ptable|
+          ptable = @api.resource(:ptables).call(:show, {'id' => ptable['id']})
+          name = ptable['name']
+          osfamily = ptable['os_family']
+          layout = ptable['layout']
+          operatingsystems = export_column(ptable, 'operatingsystems', 'title')
 
-            csv << [name, organizations, locations, osfamily, operatingsystems, layout]
+          organizations = CSV.generate do |column|
+            column << organizations_map[name] if organizations_map[name]
           end
+          organizations.delete!("\n")
+          locations = CSV.generate do |column|
+            column << locations_map[name] if locations_map[name]
+          end
+          locations.delete!("\n")
+
+          csv << [name, organizations, locations, osfamily, operatingsystems, layout]
         end
       end
 
@@ -102,8 +100,6 @@ module HammerCLICsv
           end
           print "done\n" if option_verbose?
         end
-      rescue RuntimeError => e
-        raise "#{e}\n       #{line}"
       end
     end
   end