Rubyでカレンダー

2012-12-27
#ruby

Rubyでカレンダーを表示するには、Dateモジュールの機能をいろいろ使って実現する。Dateインスタンスのwdayで曜日（0=日曜、6=土曜）を取得、またDate.new(year, month, -1)でその月の最後の日が取得できることを利用する。

require 'date'

# Returns array of weeks of the year/month.
#   wday_start: Start day of week (0=Sun, 1=Mon, ..., 6=Sat)
#
# Example:
#   cal(2013, 1)
#     => [[nil,nil,  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]]
def cal(year, month, wday_start = 0)
  last_day = Date.new(year, month, -1).day
  wday = (Date.new(year, month, 1).wday - wday_start) % 7
  weeks = (wday + last_day + 6) / 7
 
  (0 ... weeks).map do |i|
    day = i * 7 - wday + 1
    if day <= 0  # First week of the month.
      [nil] * wday + (1 .. 7 - wday).to_a
    elsif day + 7 > last_day  # Last week of the month.                         
      (day .. last_day).to_a
    else
      (day .. day + 6).to_a
    end
  end
end


if $0 == __FILE__
=begin
2013/1
 .  .  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
=end
  now = Time.now
  month = (ARGV.shift || now.month).to_i
  year = (ARGV.shift || now.year).to_i

  weeks = cal(year, month)

  puts "#{year}/#{month}"
  weeks.each do |week|
    puts week.map {|d| d ? sprintf('%2d', d) : ' .'}.join(' ')
  end
end

Codepadで動作を見てみる：http://codepad.org/r7uuRL37

(wday + last_day + 6) / 7でその月に何週あるかを割り出して、初めの週と終わりの週に特殊処理をして、それらを配列として返すので、あとはなんなりと。

月曜始まりにしたい場合とかを考慮して、何曜日を先頭にするかを引数として渡せるようにした。

関連記事

新着記事