flickrの各画像サイズのurl

最初helperにしようかと思ったけどmodelにメソッド追加した

class Photo < ActiveRecord::Base
  def source_url_size(size = :medium)
    if size == :square
      return source_url.sub(/\.jpg/, '_s.jpg')
    elsif size == :thumb
      return source_url.sub(/\.jpg/, '_t.jpg')
    elsif size == :small
      return source_url.sub(/\.jpg/, '_m.jpg')
    elsif size == :large
      return source_url.sub(/\.jpg/, '_b.jpg')
    end
    return source_url
  end
end

viewから呼ぶ

<%=h photo.source_url_size(:thumb) %>