Plugins » Katello » Hammer CLI CSV

module HammerCLICsv

  class CsvCommand

    class OrganizationsCommand < BaseCommand

      command_name 'organizations'

      desc         'import or export organizations'

      LABEL = 'Label'

      DESCRIPTION = 'Description'

      def export(csv)

        csv << [NAME, LABEL, DESCRIPTION]

        @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

      def import

        @existing = {}

        @api.resource(:organizations).call(:index, {:per_page => 999999})['results'].each do |organization|

          @existing[organization['name']] = organization['id'] if organization

        end

        thread_import do |line|

          create_organizations_from_csv(line)

        end

      end

      def create_organizations_from_csv(line)

        count(line[COUNT]).times do |number|

          name = namify(line[NAME], number)

          return if option_organization && name != option_organization

          label = namify(line[LABEL], number)

          organization_id = @existing[name]

          if organization_id.nil?

            print "Creating organization '#{name}'... " if option_verbose?

            @api.resource(:organizations).call(:create, {

                'name' => name,

                'organization' => {

                    'name' => name,

                    'label' => label,

                    'description' => line[DESCRIPTION]

                }

            })

          else

            print "Updating organization '#{name}'... " if option_verbose?

            organization = @api.resource(:organizations).call(:show, {'id' => organization_id})

            @api.resource(:organizations).call(:update, {

                'id' => organization_id,

                'organization' => {

                    'id' => organization_id,

                    'description' => line[DESCRIPTION]

                }

            })

          end

          print "done\n" if option_verbose?

        end

      end

    end

  end

end