hammer_cli_csv 2.0.0 → 2.1.0

data/lib/hammer_cli_csv/reports.rb

@@ -13,17 +13,13 @@ module HammerCLICsv
       PENDING = 'Pending'
       METRICS = 'Metrics'
 
-      def export
-        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => false}) do |csv|
-          csv << [NAME]
-          @api.resource(:reports).call(:index, {
-              'per_page' => 999999
-          })['results'].each do |report|
-            csv << [report['host_name'], report['metrics'].to_json]
-          end
+      def export(csv)
+        csv << [NAME]
+        @api.resource(:reports).call(:index, {
+            'per_page' => 999999
+        })['results'].each do |report|
+          csv << [report['host_name'], report['metrics'].to_json]
         end
-
-        HammerCLI::EX_OK
       end
 
       def import

data/lib/hammer_cli_csv/roles.rb

@@ -10,25 +10,21 @@ module HammerCLICsv
       ORGANIZATIONS = 'Organizations'
       LOCATIONS = 'Locations'
 
-      def export
-        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => false}) do |csv|
-          csv << [NAME, RESOURCE, SEARCH, PERMISSIONS, ORGANIZATIONS, LOCATIONS]
-          @api.resource(:roles).call(:index, {'per_page' => 999999})['results'].each do |role|
-            @api.resource(:filters).call(:index, {
-                'per_page' => 999999,
-                'search' => "role=\"#{role['name']}\""
-            })['results'].each do |filter|
-              filter = @api.resource(:filters).call(:show, 'id' => filter['id'])
+      def export(csv)
+        csv << [NAME, RESOURCE, SEARCH, PERMISSIONS, ORGANIZATIONS, LOCATIONS]
+        @api.resource(:roles).call(:index, {'per_page' => 999999})['results'].each do |role|
+          @api.resource(:filters).call(:index, {
+              'per_page' => 999999,
+              'search' => "role=\"#{role['name']}\""
+          })['results'].each do |filter|
+            filter = @api.resource(:filters).call(:show, 'id' => filter['id'])
 
-              permissions = export_column(filter, 'permissions', 'name')
-              organizations = export_column(filter, 'organizations', 'name')
-              locations = export_column(filter, 'locations', 'name')
-              csv << [role['name'], filter['resource_type'], filter['search'] || '', permissions, organizations, locations]
-            end
+            permissions = export_column(filter, 'permissions', 'name')
+            organizations = export_column(filter, 'organizations', 'name')
+            locations = export_column(filter, 'locations', 'name')
+            csv << [role['name'], filter['resource_type'], filter['search'] || '', permissions, organizations, locations]
           end
         end
-
-        HammerCLI::EX_OK
       end
 
       def import

data/lib/hammer_cli_csv/settings.rb

@@ -6,12 +6,14 @@ module HammerCLICsv
 
       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
+      def self.supported?
+        true
+      end
+
+      def export(csv)
+        csv << [NAME, VALUE]
+        @api.resource(:settings).call(:index, {'per_page' => 999999})['results'].each do |setting|
+          csv << [setting['name'], setting['value']]
         end
       end
 

data/lib/hammer_cli_csv/smart_proxies.rb

@@ -9,17 +9,15 @@ module HammerCLICsv
       URL = 'URL'
       LIFECYCLE_ENVIRONMENTS = 'Lifecycle Environments'
 
-      def export
-        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']
-            organizations = export_column(smart_proxy, 'organizations', 'name')
-            locations = export_column(smart_proxy, 'locations', 'name')
-            url = smart_proxy['url']
-            csv << [name, organizations, locations, url]
-          end
+      def export(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']
+          organizations = export_column(smart_proxy, 'organizations', 'name')
+          locations = export_column(smart_proxy, 'locations', 'name')
+          url = smart_proxy['url']
+          csv << [name, organizations, locations, url]
         end
       end
 
@@ -63,8 +61,6 @@ module HammerCLICsv
 
           print "done\n" if option_verbose?
         end
-      rescue RuntimeError => e
-        raise "#{e}\n       #{line}"
       end
     end
   end

data/lib/hammer_cli_csv/splice.rb

@@ -130,8 +130,6 @@ module HammerCLICsv
         update_host_collections(host_id, line)
 
         puts _('done') if option_verbose?
-      rescue RuntimeError => e
-        raise "#{e}\n       #{line}"
       end
 
       private
@@ -266,7 +264,7 @@ module HammerCLICsv
         subscriptions = CSV.parse_line(line[SUBSCRIPTIONS], {:skip_blanks => true}).collect do |details|
           (amount, sku, name) = details.split('|')
           {
-            :id => katello_subscription(line[ORGANIZATION], :name => name),
+            :id => get_subscription(line[ORGANIZATION], :name => name),
             :quantity => (amount.nil? || amount.empty? || amount == 'Automatic') ? 0 : amount.to_i
           }
         end

data/lib/hammer_cli_csv/subnets.rb

@@ -19,33 +19,31 @@ module HammerCLICsv
       DNS_SECONDARY = 'DNS Secondary'
       VLAN_ID = 'VLAN ID'
 
-      def export
-        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']})
+      def export(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']
-            organizations = export_column(subnet, 'organizations', 'name')
-            locations = export_column(subnet, 'locations', 'name')
-            network = subnet['network']
-            network_mask = subnet['mask']
-            network_from = subnet['from']
-            network_to = subnet['to']
-            domains = export_column(subnet, 'domains', 'name')
-            gateway = subnet['gateway']
-            dhcp_proxy = (subnet['dhcp'] && subnet['dhcp'].key?('name')) ? subnet['dhcp']['name'] : ''
-            tftp_proxy = (subnet['tftp'] && subnet['tftp'].key?('name')) ? subnet['tftp']['name'] : ''
-            dns_proxy = (subnet['dns'] && subnet['dns'].key?('name')) ? subnet['dns']['name'] : ''
-            dns_primary = subnet['dns_primary']
-            dns_secondary = subnet['dns_secondary']
-            vlan_id = subnet['vlanid']
-            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
+          name = subnet['name']
+          organizations = export_column(subnet, 'organizations', 'name')
+          locations = export_column(subnet, 'locations', 'name')
+          network = subnet['network']
+          network_mask = subnet['mask']
+          network_from = subnet['from']
+          network_to = subnet['to']
+          domains = export_column(subnet, 'domains', 'name')
+          gateway = subnet['gateway']
+          dhcp_proxy = (subnet['dhcp'] && subnet['dhcp'].key?('name')) ? subnet['dhcp']['name'] : ''
+          tftp_proxy = (subnet['tftp'] && subnet['tftp'].key?('name')) ? subnet['tftp']['name'] : ''
+          dns_proxy = (subnet['dns'] && subnet['dns'].key?('name')) ? subnet['dns']['name'] : ''
+          dns_primary = subnet['dns_primary']
+          dns_secondary = subnet['dns_secondary']
+          vlan_id = subnet['vlanid']
+          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
       end
 
@@ -100,8 +98,6 @@ module HammerCLICsv
 
           print "done\n" if option_verbose?
         end
-      rescue RuntimeError => e
-        raise "#{e}\n       #{line}"
       end
     end
   end

data/lib/hammer_cli_csv/subscriptions.rb

@@ -4,110 +4,60 @@ module HammerCLICsv
       command_name 'subscriptions'
       desc         'import or export subscriptions'
 
+      def self.supported?
+        true
+      end
+
       ORGANIZATION = 'Organization'
       MANIFEST = 'Manifest File'
-      CONTENT_SET = 'Content Set'
-      ARCH = 'Arch'
-      RELEASE = 'Release'
+      SUBSCRIPTION = 'Subscription Name'
+      QUANTITY = 'Quantity'
+      SKU = 'Product SKU'
+      CONTRACT = 'Contract Number'
+      ACCOUNT = 'Account Number'
 
-      def export
-        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, {
-                'per_page' => 999999,
-                'organization_id' => organization['id'],
-                'enabled' => true
-            })['results'].each do |product|
-              if product['redhat']
-                name = product['name']
-                @api.resource(:repository_sets).call(:index, {
-                    'per_page' => 999999,
-                    'organization_id' => organization['id'],
-                    'product_id' => product['id']
-                })['results'].each do |repository_set|
-                  content_set = repository_set['name']
-                  repository_set['repositories'].each do |repository|
-                    name_split = repository['name'].split(' ')
-                    arch = name_split[-2]
-                    release = name_split[-1]
-                    csv << [name, organization['name'], nil, content_set, arch, release]
-                  end
-                end
-              end
-            end
-          end
+      def export(csv)
+        csv << [NAME, ORGANIZATION, MANIFEST, SUBSCRIPTION, QUANTITY, SKU, CONTRACT, ACCOUNT]
+        @api.resource(:organizations).call(:index, {:per_page => 999999})['results'].each do |organization|
+          next if option_organization && organization['name'] != option_organization
+          organization = @api.resource(:organizations).call(:show, {'id' => organization['id']})
+          export_manifest(csv, organization)
+          export_subscriptions(csv, organization)
         end
       end
 
-      def import
-        thread_import do |line|
-          if line[MANIFEST] && !line[MANIFEST].empty?
-            import_manifest_from_csv(line)
-          else
-            enable_products_from_csv(line)
-          end
-        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
+      def export_manifest(csv, organization)
+        info = organization['owner_details']['upstreamConsumer']
+        return if info.nil?
 
-        results = @api.resource(:products).call(:index, {
-            'per_page' => 999999,
-            'organization_id' => foreman_organization(:name => line[ORGANIZATION]),
-            'name' => line[NAME]
-        })['results']
-        raise "No match for product '#{line[NAME]}'" if results.length == 0
-        raise "Multiple matches for product '#{line[NAME]}'" if results.length != 1
-        product = @api.resource(:products).call(:show, {'id' => results[0]['id']})
+        csv << ["# Manifest Name", organization['name'], info['name']]
+        csv << ["# Manifest URL", organization['name'], "https://#{info['webUrl']}#{info['uuid']}"]
+      end
 
-        results = @api.resource(:repository_sets).call(:index, {
+      def export_subscriptions(csv, organization)
+        @api.resource(:subscriptions).call(:index, {
             'per_page' => 999999,
-            'organization_id' => foreman_organization(:name => line[ORGANIZATION]),
-            'product_id' => product['id'],
-            'name' => line[CONTENT_SET]
-        })['results']
-        raise "No match for content set '#{line[CONTENT_SET]}'" if results.length == 0
-        raise "Multiple matches for content set '#{line[CONTENT_SET]}'" if results.length != 1
-        repository_set = results[0]
-
-        repository = repository_set['repositories'].find do |repo|
-          if line[RELEASE].nil? || line[RELEASE].empty?
-            repo['name'].end_with?("#{line[ARCH]}")
-          else
-            repo['name'].end_with?("#{line[ARCH]} #{line[RELEASE]}")
-          end
+            'organization_id' => organization['id']
+        })['results'].each do |subscription|
+          next if subscription['product_id'].to_i != 0  # Red Hat subs do not have number SKU
+          name = subscription['host'].nil? ? "# Subscription" : "# Guest Subscription for Host '#{subscription['host']['name']}'"
+          quantity = subscription['quantity'] < 0 ? "Unlimited" : subscription['quantity']
+          csv << [name, organization['name'], nil, subscription['name'],
+                  quantity, subscription['product_id'], subscription['contract_number'],
+                  subscription['account_number'],
+                  subscription['start_date'], subscription['end_date']]
         end
+      end
 
-        if repository.nil?
-          print "Enabling repository #{line[CONTENT_SET]} #{line[ARCH]} #{line[RELEASE]}..." if option_verbose?
-          product_content = product['product_content'].find do |content|
-            content['content']['name'] == line[CONTENT_SET]
+      def import
+        thread_import do |line|
+          if line[MANIFEST] && !line[MANIFEST].empty?
+            import_manifest(line)
           end
-          raise "No match for content set '#{line[CONTENT_SET]}'" if !product_content
-
-          params = {
-              'id' => product_content['content']['id'],
-              'product_id' => product['id'],
-              '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?
         end
       end
 
-      def import_manifest_from_csv(line)
+      def import_manifest(line)
         return if option_organization && line[ORGANIZATION] != option_organization
         args = %W{
           --server #{ @server } --username #{ @username } --password #{ @password }
@@ -116,8 +66,6 @@ module HammerCLICsv
         }
         hammer.run(args)
 
-      rescue RuntimeError => e
-        raise "#{e}\n       #{line}"
       end
     end
   end

data/lib/hammer_cli_csv/sync_plans.rb

@@ -11,32 +11,30 @@ module HammerCLICsv
       INTERVAL = 'Interval'
       PRODUCTS = 'Products'
 
-      def export
-        CSV.open(option_file || '/dev/stdout', 'wb', {:force_quotes => false}) do |csv|
-          csv << [NAME, ORGANIZATION, DESCRIPTION, ENABLED, STARTDATE, INTERVAL, PRODUCTS]
+      def export(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
+        @api.resource(:organizations).call(:index, {:per_page => 999999})['results'].each do |organization|
+          next if option_organization && organization['name'] != option_organization
 
-            @api.resource(:sync_plans).call(:index, {
-                 'per_page' => 999999,
-                 'organization_id' => foreman_organization(:name => organization['name'])
-            })['results'].each do |sync_plan|
-              name = sync_plan['name']
-              organization_name = organization['name']
-              description = sync_plan['description']
-              enabled = sync_plan['enabled'] ? 'Yes' : 'No'
-              start_date = sync_plan['sync_date']
-              interval = sync_plan['interval']
-              products = CSV.generate do |column|
-                column << sync_plan['products'].collect do |product|
-                  product['name']
-                end
+          @api.resource(:sync_plans).call(:index, {
+               'per_page' => 999999,
+               'organization_id' => foreman_organization(:name => organization['name'])
+          })['results'].each do |sync_plan|
+            name = sync_plan['name']
+            organization_name = organization['name']
+            description = sync_plan['description']
+            enabled = sync_plan['enabled'] ? 'Yes' : 'No'
+            start_date = sync_plan['sync_date']
+            interval = sync_plan['interval']
+            products = CSV.generate do |column|
+              column << sync_plan['products'].collect do |product|
+                product['name']
               end
-              products.delete!("\n")
-              csv << [name, organization_name, description, enabled, start_date, interval,
-                      products]
             end
+            products.delete!("\n")
+            csv << [name, organization_name, description, enabled, start_date, interval,
+                    products]
           end
         end
       end
@@ -89,8 +87,6 @@ module HammerCLICsv
           puts "done" if option_verbose?
         end
 
-      rescue RuntimeError => e
-        raise "#{e}\n       #{line}"
       end
 
       private

data/lib/hammer_cli_csv/users.rb

@@ -12,39 +12,37 @@ module HammerCLICsv
       ADMIN = 'Administrator'
       ROLES = 'Roles'
 
-      def export
-        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|
-                column << user['organizations'].collect do |organization|
-                  organization['name']
-                end
+      def export(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|
+              column << user['organizations'].collect do |organization|
+                organization['name']
               end
-              organizations.delete!("\n")
             end
-            if user['locations']
-              locations = CSV.generate do |column|
-                column << user['locations'].collect do |location|
-                  location['name']
-                end
+            organizations.delete!("\n")
+          end
+          if user['locations']
+            locations = CSV.generate do |column|
+              column << user['locations'].collect do |location|
+                location['name']
               end
-              locations.delete!("\n")
             end
-            if user['roles']
-              roles = CSV.generate do |column|
-                column << user['roles'].collect do |role|
-                  role['name']
-                end
+            locations.delete!("\n")
+          end
+          if user['roles']
+            roles = CSV.generate do |column|
+              column << user['roles'].collect do |role|
+                role['name']
               end
-              roles.delete!("\n")
-            end
-            admin = user['admin'] ? 'Yes' : 'No'
-            if user['login'] != 'admin' && !user['login'].start_with?('hidden-')
-              csv << [user['login'], user['firstname'], user['lastname'], user['mail'],
-                      organizations, locations, admin, roles]
             end
+            roles.delete!("\n")
+          end
+          admin = user['admin'] ? 'Yes' : 'No'
+          if user['login'] != 'admin' && !user['login'].start_with?('hidden-')
+            csv << [user['login'], user['firstname'], user['lastname'], user['mail'],
+                    organizations, locations, admin, roles]
           end
         end
       end
@@ -81,8 +79,6 @@ module HammerCLICsv
           end
           print "done\n" if option_verbose?
         end
-      rescue RuntimeError => e
-        raise "#{e}\n       #{line}"
       end
 
       def create_user(line, name, roles, organizations, locations)

data/lib/hammer_cli_csv/utils/subscriptions.rb

@@ -0,0 +1,130 @@
+module HammerCLICsv
+  module Utils
+    module Subscriptions
+      SUBSCRIPTIONS = 'Subscriptions'
+      SUBS_NAME = 'Subscription Name'
+      SUBS_TYPE = 'Subscription Type'
+      SUBS_QUANTITY = 'Subscription Quantity'
+      SUBS_SKU = 'Subscription SKU'
+      SUBS_CONTRACT = 'Subscription Contract'
+      SUBS_ACCOUNT = 'Subscription Account'
+      SUBS_START = 'Subscription Start'
+      SUBS_END = 'Subscription End'
+
+      def get_all_subscriptions(organization)
+        @api.resource(:subscriptions).call(:index, {
+            :per_page => 999999,
+            'organization_id' => foreman_organization(:name => organization)
+        })['results']
+      end
+
+      def get_subscription(organization, options = {})
+        @subscriptions ||= {}
+        @subscriptions[organization] ||= {}
+
+        if options[:name]
+          return nil if options[:name].nil? || options[:name].empty?
+          options[:id] = @subscriptions[organization][options[:name]]
+          if !options[:id]
+            results = @api.resource(:subscriptions).call(:index, {
+                :per_page => 999999,
+                'organization_id' => foreman_organization(:name => organization),
+                'search' => "name = \"#{options[:name]}\""
+            })
+            raise "No subscriptions match '#{options[:name]}'" if results['subtotal'] == 0
+            raise "Too many subscriptions match '#{options[:name]}'" if results['subtotal'] > 1
+            subscription = results['results'][0]
+            @subscriptions[organization][options[:name]] = subscription['id']
+            options[:id] = @subscriptions[organization][options[:name]]
+            raise "Subscription '#{options[:name]}' not found" if !options[:id]
+          end
+          result = options[:id]
+        else
+          return nil if options[:id].nil?
+          options[:name] = @subscriptions.key(options[:id])
+          if !options[:name]
+            subscription = @api.resource(:subscriptions).call(:show, {'id' => options[:id]})
+            raise "Subscription '#{options[:name]}' not found" if !subscription || subscription.empty?
+            options[:name] = subscription['name']
+            @subscriptions[options[:name]] = options[:id]
+          end
+          result = options[:name]
+        end
+
+        result
+      end
+
+      def matches_by_sku_and_name(matches, line, subscriptions)
+        if line[SUBS_SKU]
+          matches = subscriptions.select do |subscription|
+            line[SUBS_SKU] == subscription['product_id']
+          end
+          raise _("No subscriptions match SKU '%{sku}'") % {:sku => line[SUBS_SKU]} if matches.empty?
+        elsif line[SUBS_NAME]
+          matches = subscriptions.select do |subscription|
+            line[SUBS_NAME] == subscription['name']
+          end
+          raise _("No subscriptions match name '%{name}'") % {:name => line[SUBS_NAME]} if matches.empty?
+        end
+        matches
+      end
+
+      def matches_by_type(matches, line)
+        if line[SUBS_TYPE] == 'Red Hat' || line[SUBS_TYPE] == 'Custom'
+          matches = matches.select do |subscription|
+            subscription['type'] == 'NORMAL'
+          end
+        elsif line[SUBS_TYPE] == 'Red Hat Guest'
+          matches = matches.select do |subscription|
+            subscription['type'] == 'STACK_DERIVED'
+          end
+        elsif line[SUBS_TYPE] == 'Red Hat Temporary'
+          matches = matches.select do |subscription|
+            subscription['type'] == 'UNMAPPED_GUEST'
+          end
+        end
+        raise _("No subscriptions match type '%{type}'") % {:type => line[SUBS_TYPE]} if matches.empty?
+        matches
+      end
+
+      def matches_by_account(matches, line)
+        if matches.length > 1 && line[SUBS_ACCOUNT]
+          refined = matches.select do |subscription|
+            line[SUBS_ACCOUNT] == subscription['account_number']
+          end
+          matches = refined unless refined.empty?
+        end
+        matches
+      end
+
+      def matches_by_contract(matches, line)
+        if matches.length > 1 && line[SUBS_CONTRACT]
+          refined = matches.select do |subscription|
+            line[SUBS_CONTRACT] == subscription['contract_number']
+          end
+          matches = refined unless refined.empty?
+        end
+        matches
+      end
+
+      def matches_by_quantity(matches, line)
+        if line[SUBS_QUANTITY] && line[SUBS_QUANTITY] != 'Automatic'
+          refined = matches.select do |subscription|
+            subscription['available'] == -1 || line[SUBS_QUANTITY].to_i <= subscription['available']
+          end
+          raise _("No '%{name}' subscription with quantity %{quantity} or more available") %
+            {:name => matches[0]['name'], :quantity => line[SUBS_QUANTITY]} if refined.empty?
+          matches = refined
+        end
+        matches
+      end
+
+      # Subscription amount, SKU, name, contract number, and account number separated by '|'
+      # or simply the subscription name.
+      def split_subscription_details(details)
+        details = details.split('|')
+        details.length == 1 ? ['Automatic', nil, details[0], nil, nil] : details
+      end
+    end
+  end
+end

data/lib/hammer_cli_csv/version.rb

@@ -1,5 +1,5 @@
 module HammerCLICsv
   def self.version
-    @version ||= Gem::Version.new('2.0.0')
+    @version ||= Gem::Version.new('2.1.0')
   end
 end