hammer_cli_csv 1.0.1 → 1.0.2

data/lib/hammer_cli_csv/settings.rb

@@ -0,0 +1,49 @@
+module HammerCLICsv
+  class CsvCommand
+    class SettingsCommand < BaseCommand
+      command_name 'settings'
+      desc         'import or export settings'
+
+      VALUE = 'Value'
+
+      def export
+        CSV.open(option_file || '/dev/stdout', 'wb') do |csv|
+          csv << [NAME, VALUE]
+          @api.resource(:settings).call(:index, {'per_page' => 999999})['results'].each do |setting|
+            csv << [setting['name'], setting['value']]
+          end
+        end
+      end
+
+      def import
+        @existing = {}
+
+        thread_import do |line|
+          create_settings_from_csv(line)
+        end
+      end
+
+      def create_settings_from_csv(line)
+        count(line[COUNT]).times do |number|
+          name = namify(line[NAME], number)
+          params =  { 'id' => get_setting_id(name),
+                      'setting' => {
+                        'value' => line[VALUE]
+                      }
+                    }
+          print "Updating setting '#{name}'..." if option_verbose?
+          @api.resource(:settings).call(:update, params)
+        end
+        print "done\n" if option_verbose?
+      end
+
+      private
+
+      def get_setting_id(name)
+        results = @api.resource(:settings).call(:index, { :search => "name=\"#{name}\"" })['results']
+        raise "Setting '#{name}' not found" if !results || results.empty?
+        results[0]['id']
+      end
+    end
+  end
+end

data/lib/hammer_cli_csv/smart_proxies.rb

@@ -1,18 +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'
-
 module HammerCLICsv
   class CsvCommand
     class SmartProxiesCommand < BaseCommand
@@ -25,16 +10,15 @@ module HammerCLICsv
       LIFECYCLE_ENVIRONMENTS = 'Lifecycle Environments'
 
       def export
-        CSV.open(option_csv_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
-          csv << [NAME, COUNT, ORGANIZATIONS, LOCATIONS, URL, LIFECYCLE_ENVIRONMENTS]
+        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
+          csv << [NAME, ORGANIZATIONS, LOCATIONS, URL, LIFECYCLE_ENVIRONMENTS]
           @api.resource(:smart_proxies).call(:index, {:per_page => 999999})['results'].each do |smart_proxy|
             smart_proxy = @api.resource(:smart_proxies).call(:show, {'id' => smart_proxy['id']})
             name = smart_proxy['name']
-            count = 1
             organizations = export_column(smart_proxy, 'organizations', 'name')
             locations = export_column(smart_proxy, 'locations', 'name')
             url = smart_proxy['url']
-            csv << [name, count, organizations, locations, url]
+            csv << [name, organizations, locations, url]
           end
         end
       end
@@ -51,9 +35,10 @@ module HammerCLICsv
       end
 
       def create_smart_proxies_from_csv(line)
-        line[COUNT].to_i.times do |number|
+        count(line[COUNT]).times do |number|
           name = namify(line[NAME], number)
-          if !@existing.include? line[URL]
+          id = @existing[line[URL]]
+          if id.nil?
             print "Creating smart proxy '#{name}'..." if option_verbose?
             id = @api.resource(:smart_proxies).call(:create, {
                 'smart_proxy' => {
@@ -63,13 +48,13 @@ module HammerCLICsv
             })['id']
           else
             print "Updating smart proxy '#{name}'..." if option_verbose?
-            id = @api.resource(:smart_proxies).call(:update, {
-                'id' => @existing[name],
+            @api.resource(:smart_proxies).call(:update, {
+                'id' => id,
                 'smart_proxy' => {
                     'name' => name,
                     'url' => line[URL]
                 }
-            })['smart_proxy']['id']
+            })
           end
 
           # Update associated resources

data/lib/hammer_cli_csv/splice.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 'openssl'
 require 'date'
 
@@ -18,7 +7,6 @@ module HammerCLICsv
       command_name 'splice'
       desc         'import Satellite-5 splice data'
 
-      option %w(--organization), 'ORGANIZATION', 'Only process organization matching this name'
       option %w(--dir), 'DIR',
           'Directory of Splice exported CSV files (default pwd)'
       option %w(--mapping-dir), 'DIR',

data/lib/hammer_cli_csv/subnets.rb

@@ -1,18 +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'
-
 module HammerCLICsv
   class CsvCommand
     class SubnetsCommand < BaseCommand
@@ -35,15 +20,14 @@ module HammerCLICsv
       VLAN_ID = 'VLAN ID'
 
       def export
-        CSV.open(option_csv_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
-          csv << [NAME, COUNT, ORGANIZATIONS, LOCATIONS, NETWORK, NETWORK_MASK,
+        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
+          csv << [NAME, ORGANIZATIONS, LOCATIONS, NETWORK, NETWORK_MASK,
                   NETWORK_FROM, NETWORK_TO, DOMAINS, GATEWAY, DHCP_PROXY, TFTP_PROXY, DNS_PROXY,
                   DNS_PRIMARY, DNS_SECONDARY, VLAN_ID]
           @api.resource(:subnets).call(:index, {:per_page => 999999})['results'].each do |subnet|
             subnet = @api.resource(:subnets).call(:show, {'id' => subnet['id']})
 
             name = subnet['name']
-            count = 1
             organizations = export_column(subnet, 'organizations', 'name')
             locations = export_column(subnet, 'locations', 'name')
             network = subnet['network']
@@ -58,7 +42,7 @@ module HammerCLICsv
             dns_primary = subnet['dns_primary']
             dns_secondary = subnet['dns_secondary']
             vlan_id = subnet['vlanid']
-            csv << [name, count, organizations, locations, network, network_mask,
+            csv << [name, organizations, locations, network, network_mask,
                     network_from, network_to, domains, gateway, dhcp_proxy, tftp_proxy, dns_proxy,
                     dns_primary, dns_secondary, vlan_id]
           end
@@ -81,13 +65,18 @@ module HammerCLICsv
           foreman_domain(:name => domain)
         end
 
-        line[COUNT].to_i.times do |number|
+        count(line[COUNT]).times do |number|
           name = namify(line[NAME], number)
           if !@existing.include? name
             print "Creating subnet '#{name}'..." if option_verbose?
             id = @api.resource(:subnets).call(:create, {
                 'subnet' => {
-                    'name' => name
+                  'name' => name,
+                  'network' => line[NETWORK],
+                  'mask' => line[NETWORK_MASK],
+                  #'from' => line[NETWORK_FROM],
+                  #'to' => line[NETWORK_TO],
+                  #'domain_ids' => line[DOMAINS]
                 }
             })['id']
           else

data/lib/hammer_cli_csv/subscriptions.rb

@@ -1,23 +1,9 @@
-# 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 SubscriptionsCommand < BaseCommand
       command_name 'subscriptions'
       desc         'import or export subscriptions'
 
-      option %w(--organization), 'ORGANIZATION', 'Only process organization matching this name'
-
       ORGANIZATION = 'Organization'
       MANIFEST = 'Manifest File'
       CONTENT_SET = 'Content Set'
@@ -25,8 +11,8 @@ module HammerCLICsv
       RELEASE = 'Release'
 
       def export
-        CSV.open(option_csv_file || '/dev/stdout', 'wb', {:force_quotes => false}) do |csv|
-          csv << [NAME, COUNT, ORGANIZATION, MANIFEST, CONTENT_SET, ARCH, RELEASE]
+        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => false}) do |csv|
+          csv << [NAME, ORGANIZATION, MANIFEST, CONTENT_SET, ARCH, RELEASE]
           @api.resource(:organizations).call(:index, {:per_page => 999999})['results'].each do |organization|
             next if option_organization && organization['name'] != option_organization
             @api.resource(:products).call(:index, {
@@ -34,7 +20,7 @@ module HammerCLICsv
                 'organization_id' => organization['id'],
                 'enabled' => true
             })['results'].each do |product|
-              if product['provider']['name'] == 'Red Hat'
+              if product['redhat']
                 name = product['name']
                 @api.resource(:repository_sets).call(:index, {
                     'per_page' => 999999,
@@ -46,7 +32,7 @@ module HammerCLICsv
                     name_split = repository['name'].split(' ')
                     arch = name_split[-2]
                     release = name_split[-1]
-                    csv << [name, 1, organization['name'], nil, content_set, arch, release]
+                    csv << [name, organization['name'], nil, content_set, arch, release]
                   end
                 end
               end
@@ -65,7 +51,12 @@ module HammerCLICsv
         end
       end
 
+      # FIXME: TODO remove this rubocop
+      # rubocop:disable CyclomaticComplexity
       def enable_products_from_csv(line)
+        organization = line[ORGANIZATION] || option_organization
+        raise "Organization is required in either input CSV or by option --organization" if organization.nil? || organization.empty?
+        line[ORGANIZATION] = organization
         return if option_organization && line[ORGANIZATION] != option_organization
 
         results = @api.resource(:products).call(:index, {
@@ -75,7 +66,7 @@ module HammerCLICsv
         })['results']
         raise "No match for product '#{line[NAME]}'" if results.length == 0
         raise "Multiple matches for product '#{line[NAME]}'" if results.length != 1
-        product = results[0]
+        product = @api.resource(:products).call(:show, {'id' => results[0]['id']})
 
         results = @api.resource(:repository_sets).call(:index, {
             'per_page' => 999999,
@@ -88,7 +79,11 @@ module HammerCLICsv
         repository_set = results[0]
 
         repository = repository_set['repositories'].find do |repo|
-          repo['name'].end_with?("#{line[ARCH]} #{line[RELEASE]}")
+          if line[RELEASE].nil? || line[RELEASE].empty?
+            repo['name'].end_with?("#{line[ARCH]}")
+          else
+            repo['name'].end_with?("#{line[ARCH]} #{line[RELEASE]}")
+          end
         end
 
         if repository.nil?
@@ -98,12 +93,14 @@ module HammerCLICsv
           end
           raise "No match for content set '#{line[CONTENT_SET]}'" if !product_content
 
-          @api.resource(:repository_sets).call(:enable, {
+          params = {
               'id' => product_content['content']['id'],
               'product_id' => product['id'],
-              'basearch' => line[ARCH],
-              'releasever' => line[RELEASE]
-          })
+              'basearch' => line[ARCH]
+          }
+          params['releasever'] = line[RELEASE] unless line[RELEASE].nil? || line[RELEASE].empty?
+
+          @api.resource(:repository_sets).call(:enable, params)
           puts 'done' if option_verbose?
         else
           puts "Repository #{repository['name']} already enabled" if option_verbose?

data/lib/hammer_cli_csv/sync_plans.rb

@@ -1,23 +1,9 @@
-# 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 SyncPlansCommand < BaseCommand
       command_name 'sync-plans'
       desc         'import or export repository sync plans'
 
-      option %w(--organization), 'ORGANIZATION', 'Only process organization matching this name'
-
       ORGANIZATION = 'Organization'
       DESCRIPTION = 'Description'
       ENABLED = 'Enabled'
@@ -26,8 +12,8 @@ module HammerCLICsv
       PRODUCTS = 'Products'
 
       def export
-        CSV.open(option_csv_file || '/dev/stdout', 'wb', {:force_quotes => false}) do |csv|
-          csv << [NAME, COUNT, ORGANIZATION, DESCRIPTION, ENABLED, STARTDATE, INTERVAL, PRODUCTS]
+        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => false}) do |csv|
+          csv << [NAME, ORGANIZATION, DESCRIPTION, ENABLED, STARTDATE, INTERVAL, PRODUCTS]
 
           @api.resource(:organizations).call(:index, {:per_page => 999999})['results'].each do |organization|
             next if option_organization && organization['name'] != option_organization
@@ -37,7 +23,6 @@ module HammerCLICsv
                  'organization_id' => foreman_organization(:name => organization['name'])
             })['results'].each do |sync_plan|
               name = sync_plan['name']
-              count = 1
               organization_name = organization['name']
               description = sync_plan['description']
               enabled = sync_plan['enabled'] ? 'Yes' : 'No'
@@ -49,7 +34,7 @@ module HammerCLICsv
                 end
               end
               products.delete!("\n")
-              csv << [name, count, organization_name, description, enabled, start_date, interval,
+              csv << [name, organization_name, description, enabled, start_date, interval,
                       products]
             end
           end
@@ -77,7 +62,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
             print "Creating sync plan '#{name}'..." if option_verbose?

data/lib/hammer_cli_csv/users.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 UsersCommand < BaseCommand
@@ -24,8 +13,8 @@ module HammerCLICsv
       ROLES = 'Roles'
 
       def export
-        CSV.open(option_csv_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
-          csv << [NAME, COUNT, FIRSTNAME, LASTNAME, EMAIL, ORGANIZATIONS, LOCATIONS, ADMIN, ROLES]
+        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => true}) do |csv|
+          csv << [NAME, FIRSTNAME, LASTNAME, EMAIL, ORGANIZATIONS, LOCATIONS, ADMIN, ROLES]
           @api.resource(:users).call(:index, {:per_page => 999999})['results'].each do |user|
             if user['organizations']
               organizations = CSV.generate do |column|
@@ -53,7 +42,7 @@ module HammerCLICsv
             end
             admin = user['admin'] ? 'Yes' : 'No'
             if user['login'] != 'admin' && !user['login'].start_with?('hidden-')
-              csv << [user['login'], 1, user['firstname'], user['lastname'], user['mail'],
+              csv << [user['login'], user['firstname'], user['lastname'], user['mail'],
                       organizations, locations, admin, roles]
             end
           end
@@ -72,7 +61,7 @@ module HammerCLICsv
       end
 
       def create_users_from_csv(line)
-        line[COUNT].to_i.times do |number|
+        count(line[COUNT]).times do |number|
           name = namify(line[NAME], number)
 
           roles = collect_column(line[ROLES]) do |role|

data/lib/hammer_cli_csv/version.rb

@@ -1,16 +1,5 @@
-# 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
   def self.version
-    @version ||= Gem::Version.new('1.0.1')
+    @version ||= Gem::Version.new('1.0.2')
   end
 end

data/test/content_hosts_test.rb

@@ -17,7 +17,7 @@ describe 'content-hosts' do
       file.rewind
 
       stdout,stderr = capture {
-        hammer.run(%W{csv content-hosts --verbose --csv-file #{file.path}})
+        hammer.run(%W{csv content-hosts --verbose --file #{file.path}})
       }
       stderr.must_equal ''
       stdout[0..-2].must_equal "Creating content host '#{hostname}'...done\nUpdating hypervisor and guest associations...done"

data/test/content_views_test.rb

@@ -0,0 +1,52 @@
+require File.join(File.dirname(__FILE__), 'csv_test_helper')
+
+require 'stringio'
+require 'tempfile'
+
+describe 'content-views' do
+  extend CommandTestHelper
+
+  before :each do
+    HammerCLI::Settings.load_from_file 'test/config.yml'
+  end
+
+  context "import" do
+    it "hammer csv content-views --verbose --file does-not-exist" do
+      stdout,stderr = capture {
+        hammer.run(%W{csv content-views --verbose --file does-not-exist})
+      }
+      stdout.must_equal ''
+      stderr[0..-2].must_equal('Error: No such file or directory - does-not-exist')
+    end
+
+    it "hammer csv content-views --verbose --file tempfile" do
+      contentview = "contentview#{rand(10000)}"
+      file = Tempfile.new('content_views_test')
+      file.write("Name,Count,Organization,Description,Composite,Repositories,Lifecycle Environments\n")
+      file.write("#{contentview},1,Mega Corporation,Katello - The Sysadmin's Fortress,No,Default_Organization_View,Library\n")
+      file.rewind
+
+      stdout,stderr = capture {
+        hammer.run(%W{csv content-views --verbose --file #{file.path}})
+      }
+      stdout[0..-2].must_equal "Creating content view '#{contentview}'...done"
+
+      file.unlink
+    end
+
+    it "hammer csv content-views --verbose --file tempfile (no Count column)" do
+      contentview = "contentview#{rand(10000)}"
+      file = Tempfile.new('content_views_test')
+      file.write("Name,Organization,Description,Composite,Repositories,Lifecycle Environments\n")
+      file.write("#{contentview},Mega Corporation,Katello - The Sysadmin's Fortress,No,Default_Organization_View,Library\n")
+      file.rewind
+
+      stdout,stderr = capture {
+        hammer.run(%W{csv content-views --verbose --file #{file.path}})
+      }
+      stdout[0..-2].must_equal "Creating content view '#{contentview}'...done"
+
+      file.unlink
+    end
+  end
+end