Rails 重用页面的表格做 CSV 导出

lingceng · 2015年12月23日 · 最后由 torvaldsdb 回复于 2016年12月22日 · 3381 次阅读

rails 做 csv 导出挺容易的，只是大部分还是要组装二维数组。我想直接抓取页面的表格做导出，这样页面显示时的一些转换也可以重用了。

代码如下：

def generate_csv_data(template = nil)
  template ||= "#{controller_name}/#{action_name}.html.slim"
  content = render_to_string(template)
  doc =  Nokogiri::HTML(content)

  table =  doc.at_css('table')
  data = table.css('tr').map do |r|
    r.css('td,th').map(&:text).to_csv
  end.join

  # Convert from utf8 to gbk to make it compatible with Windows Office Excel
  # And Mac number can work with GBK too
  data = data.encode('GBK', undef: :replace, replace: "")
end

全文在这里 http://lingceng.github.io/blog/2015/12/23/export-csv-or-excel-by-re-using-existing-html-view/

ericguo #1 2015年12月26日

我现在的做法是写一个纯 csv 导出，然后用d3.csv，画出表格，正好和你方向相反。。。

class CcbReviewsController < ApplicationController
  def index
    @q = CcbReview.includes(:change_owner).search(params[:q])
    @ccb_reviews = @q.result.paginate(page: params[:page], per_page: 10)

    respond_to do |format|
      format.html
      format.json
      format.csv do
        render_csv_header :wip.to_s
        csv_res = CSV.generate do |csv|
          csv << ['CCB Number', 'State', 'Title', 'Description', 'Reason', 'Purpose']
          CcbReview.all.find_each do |ccb_review|
            values = []
            values << ccb_review.ccb_number
            values << ccb_review.state
            values << ccb_review.title
            values << ccb_review.description
            values << ccb_review.reason
            values << ccb_review.purpose
            csv << values
          end
        end
        send_data "\xEF\xBB\xBF" << csv_res
      end
    end
  end

private

  def render_csv_header(filename = nil)
    filename ||= params[:action]
    filename += '.csv'

    if request.env['HTTP_USER_AGENT'] =~ /msie/i
      headers['Pragma'] = 'public'
      headers['Content-type'] = 'text/plain'
      headers['Cache-Control'] = 'no-cache, must-revalidate, post-check=0, pre-check=0'
      headers['Content-Disposition'] = "attachment; filename=\"#{filename}\""
      headers['Expires'] = '0'
    else
      headers['Content-Type'] ||= 'text/csv'
      headers['Content-Disposition'] = "attachment; filename=\"#{filename}\""
    end
  end
end