Ruby - CSV ファイルを HTML table タグに変換！

[ プログラミング ] [ HTML, Ruby ]

２年くらい前には、CSV データを HTML の table タグに変換する GUI アプリを Microsoft Visual Basic で作成したことを記事にしました。

最近はプライベートで Windows 系 OS に触れることがないので、同様のツール（ただし、CUI ベース）を Ruby で作成しました。

単純に CSV ファイルを読み込んで、 HTML に変換しているだけですが、以下で作成した Ruby スクリプトを紹介します。

0. 前提条件

  • Linux Mint 14 (64bit) での作業を想定。
  • Ruby 2.0.0-p247 で作業・動作確認。
  • CSV ファイルのヘッダ行と明細行の区別はしない。

1. CSV ファイル作成

変換に使用する CSV ファイルを用意しておく。
今回は以下のような CSV ファイル（ファイル名："test.csv"）を使用する。

test.csv
1
2
3
4
5
6
7
8
9
10
11

ヘッダ１,ヘッダ２,ヘッダ３,ヘッダ４
test1,a,1,あああああ
test2,bb,22,いいいい
test3,ccc,333,ううう
test4,dddd,4444,ええ
test5,eeeee,55555,お
test6,f,666666,かかかかか
test7,gg,7777777,きききき
test8,hhh,88888888,くくく
test9,iiii,999999999,けけ
test10,jjjjj,0000000000,こ

2. Ruby スクリプト作成

例として、以下のように Ruby スクリプトを作成してみた。
第１引数に CSV ファイル名を指定、第２引数に align(left: l or L, center: c or C, right: r or R)（省略可） を指定するようにしている。
その他については、特に難しいことは行なっていないので、スクリプトを見れば理解できると思う。
あと、csv ライブラリを使用しなくても、普通にテキストファイルとして読み込んで split(",") を使用すれば同様に実装可能。

csv2table.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113

#= CSV -> table 変換
# ：CSV ファイルを table タグに変換する
#   第１引数：変換対象の CSV ファイル名
#   第２引数：align を c,l,l,r,r,l のように指定（省略可、デフォルトは "l"）
#---------------------------------------------------------------------------------
#++
require 'csv'

# [CLASS] 引数
class Arg
  def initialize
    @aligns = Array.new
  end

  # 引数取得
  def get_args
    begin
      # CSV ファイル名取得
      @csv_file = ARGV[0]
      # CSV ファイル存在チェック
      unless File.exist?(@csv_file)
        puts "[ERROR] Can't found #{@csv_file}"
        exit
      end

      # align 情報取得
      return unless ARGV[1]
      ARGV[1].split(",").collect do |a|
        if ["l", "L"].include?(a[0])
          @aligns << "l"
        elsif ["c", "C"].include?(a[0])
          @aligns << "c"
        elsif ["r", "R"].include?(a[0])
          @aligns << "r"
        else
          @aligns << "l"
        end
      end
    rescue => e
      STDERR.puts "[ERROR][#{self.class.name}.get_args] #{e}"
      exit 1
    end
  end

  # CSV ファイル名返却
  def get_csv_file
    return @csv_file
  end

  # align 情報返却
  def get_aligns
    return @aligns
  end
end

# [CLASS] CSV -> table 変換
class Csv2Table
  def initialize(csv_file, aligns)
    @csv_file = csv_file
    @aligns   = aligns
    @html_file = File.basename(csv_file, File.extname(csv_file)) + ".html"
  end

  # CSV -> table 変換
  def convert
    begin
      # CSV ファイル読み込み
      csv = CSV.open(@csv_file, "r")

      # HTML ファイル書き込み
      html = File.open(@html_file, "w")
      html.puts "<table>"
      csv.each do |row|
        html.puts "  <tr>"
        row.each_with_index do |col, i|
          align = @aligns[i]
          html.print "    <td"
          case align
          when "l"
            html.print " align=\"left\">"
          when "c"
            html.print " align=\"center\">"
          when "r"
            html.print " align=\"right\">"
          else
            html.print ">"
          end
          html.puts "#{col}</td>"
        end
        html.puts "  </tr>"
      end
      html.puts "</table>"
    rescue => e
      STDERR.puts "[ERROR][#{self.class.name}.convert] #{e}"
      exit 1
    end
  end
end

begin
  # 引数取得
  obj_args = Arg.new
  obj_args.get_args
  csv_file = obj_args.get_csv_file
  aligns   = obj_args.get_aligns

  # CSV -> table 変換
  obj_main = Csv2Table.new(csv_file, aligns)
  obj_main.convert
rescue => e
  STDERR.puts "[EXCEPTION] #{e}"
  exit 1
end

4. Ruby スクリプト実行

以下のようにして、作成した Ruby スクリプトを実行してみる。

1

$ ruby csv2table.rb test.csv r,c,r,l

5. 結果確認

CSV ファイルと同じディレクトリに、以下のような内容の HTML ファイル（CSV ファイル名が “test.csv” なら “test.html” というファイル名の HTML ファイル）が作成される。

test.html
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

<table>
  <tr>
    <td align="right">ヘッダ１</td>
    <td align="center">ヘッダ２</td>
    <td align="right">ヘッダ３</td>
    <td align="left">ヘッダ４</td>
  </tr>
  <tr>
    <td align="right">test1</td>
    <td align="center">a</td>
    <td align="right">1</td>
    <td align="left">あああああ</td>
  </tr>
  <tr>
    <td align="right">test2</td>
    <td align="center">bb</td>
    <td align="right">22</td>
    <td align="left">いいいい</td>
  </tr>
  <tr>
    <td align="right">test3</td>
    <td align="center">ccc</td>
    <td align="right">333</td>
    <td align="left">ううう</td>
  </tr>
  <tr>
    <td align="right">test4</td>
    <td align="center">dddd</td>
    <td align="right">4444</td>
    <td align="left">ええ</td>
  </tr>
  <tr>
    <td align="right">test5</td>
    <td align="center">eeeee</td>
    <td align="right">55555</td>
    <td align="left">お</td>
  </tr>
  <tr>
    <td align="right">test6</td>
    <td align="center">f</td>
    <td align="right">666666</td>
    <td align="left">かかかかか</td>
  </tr>
  <tr>
    <td align="right">test7</td>
    <td align="center">gg</td>
    <td align="right">7777777</td>
    <td align="left">きききき</td>
  </tr>
  <tr>
    <td align="right">test8</td>
    <td align="center">hhh</td>
    <td align="right">88888888</td>
    <td align="left">くくく</td>
  </tr>
  <tr>
    <td align="right">test9</td>
    <td align="center">iiii</td>
    <td align="right">999999999</td>
    <td align="left">けけ</td>
  </tr>
  <tr>
    <td align="right">test10</td>
    <td align="center">jjjjj</td>
    <td align="right">0000000000</td>
    <td align="left">こ</td>
  </tr>
</table>

当然、ブラウザで確認すると左寄せ・中央揃え・右寄せで表示される。

CSV2TABLE

どうしても table タグを自分で手入力しないといけない場合には、ちょっとした table タグでも意外と面倒に感じることがありますが、これで大分手間が省けます。

以上。