RubyでCSVのヘッダーとデータが別々の2ファイルからCSV::Tableをつくる #Ruby

下記のようにデータレコードとそのヘッダが別々のファイルに保存されているような場合に、2ファイルから1つのCSV::Tableをつくりたい。

header.csv

id, name, score

data.csv

1, john, 10
2, paul, 12

結論としては1.のheadersにヘッダーの配列を渡す方法がすっきり確実そうです。

1. オプションのheadersにヘッダーの配列を渡す

@hanachin_ さんから教えていただいた方法。すっきりしていい感じ…！

require 'csv'

headers = CSV.parse_line(File.read("header.csv"))
p CSV.table('data.csv', headers: headers)

CSV.table('data.csv', headers: headers).each do |r|
  p r.to_h
end

# #<CSV::Table mode:col_or_row row_count:3>
# {:id=>1, :name=>" john", :score=>10}
# {:id=>2, :name=>" paul", :score=>12}

2. ヒアドキュメントでがっちゃんこする

@yancya さんに教えていただいた方法。そんな技が…。知らなかったです。

require "csv"

header = File.read("header.txt").chomp
rows = File.read("data.csv").chomp

table = CSV.parse(<<CSV, headers: true)
#{header}
#{rows}
CSV

p table
# => #<CSV::Table mode:col_or_row row_count:3>

3. シェルコマンドとCSV.newの合せ技

さすがにTempfileを使う方法から頭悪すぎる感が滲んでいたので、周りのエンジニアの方にもお知恵をいただいた結果かなりスッキリしてきました。

str = %x(cat header.csv; echo ""; cat data.csv)
p CSV.new(str, headers: true, header_converters: :symbol).read
# => #<CSV::Table mode:col_or_row row_count:3>

ただしこの方法だと、末尾の改行だったり、文字列が"で囲われているかどうかだったりと、色々なことが整った状態でないとうまくいきません。

4. Tempfileを使う

最初に試した方法。

CSV.table('path/to/csv')

本来であれば上記のように1ファイルでサクッとできるのだが、2ファイルに分かれてしまっているので、Tempfileを使ってヘッダをくっつけた仮のCSVファイルを生成し、そこからCSV::Tableをつくってやる。

header = File.read('header.csv')
data = File.read('data.csv')

csv = nil
Tempfile.open do |t|
  t.write header
  t.write "\n"
  t.write data
  csv = CSV.table(t.path)
end

p csv

# => #<CSV::Table mode:col_or_row row_count:3>

Ref

class CSV::Table