ruby max_value_importer.rb

Posted 2021-05-15

#!/usr/bin/env ruby
# -*- coding: utf-8 -*-
# max_value_importer.rb --file "./abc.xlsx" --sheet "sheet1" --date_range "A2:A100" --value_range "B2:B100" --date_cell "d2" --value_cell "e2"

require 'csv'
require 'optparse'
require 'win32ole'

class WIN32OLE
  def to_a
    [].tap do |array|
      each { |o| array.push o }
    end
  end
end

class MaxValueImporter
  def initialize
    params = optparse
    @file = absolute_path params['file']
    @sheet = params['sheet']
    @date_range = params['date_range']
    @value_range = params['value_range']
    @date_cell = params['date_cell']
    @value_cell = params['value_cell']
    write_excel
  end

  def open_book(file)
    excel = WIN32OLE.new 'Excel.Application'
    excel.visible = false

    begin
      book = excel.Workbooks.Open file
      yield book
    ensure
      book.Close
      excel.Quit
    end
  end

  def write_excel
    open_book @file do |book|
      sheet = book.Worksheets @sheet
      dates = sheet.Range(@date_range)
                   .to_a
                   .each
                   .map(&:value)
                   .compact
      values = sheet.Range(@value_range)
                    .to_a
                    .each
                    .map(&:value)
                    .compact
      max_index = values
                  .each_with_index
                  .map { |value, i| value == values.max ? i : nil }
                  .compact
                  .first
      sheet.Range(@date_cell).Value = dates[max_index]
      sheet.Range(@value_cell).Value = values.max
      book.Save
    end
  end

  def optparse
    params = ARGV.getopts('', 'file:', 'sheet:', 'date_range:', 'value_range:',
                          'date_cell:', 'value_cell:')
    params
  end

  def absolute_path(file)
    fso = WIN32OLE.new 'Scripting.FileSystemObject'
    fso.GetAbsolutePathName file
  end
end

MaxValueImporter.new if $PROGRAM_NAME == __FILE__