首页 > 解决方案 > 在 Rails 中将 2 个表导出为 CSV [如何选择特定列?]

问题描述

我一直在尝试在 Rails 中将数据库导出为 CSV。经过多次试验和错误,我终于让它工作了,现在我的 CSV 包含完整的 2 个表。

但是,我似乎无法从每个表中仅提取某些列 - 小时日志名称、小时日志描述、小时日志小时、小时日志 created_at 和项目名称(在所附照片中标记为红色)。

我在网上找到的所有解决方案都是用于提取整个表格。

这是我的模型(用于小时日志):

class Hourlog < ApplicationRecord

    belongs_to :user
    belongs_to :project
    accepts_nested_attributes_for :project
    validates :name, presence: true
    validates :description, presence: true
    validates :hours, presence: true
    validates :date, presence: true
    validates :project, presence: true

      def self.to_csv
        CSV.generate do |csv|
          csv << column_names
          all.each do |hourlog|
            csv << (hourlog.attributes.values_at(*column_names) + hourlog.project.attributes.values )
          end
        end
      end
end

这是我的控制器(用于小时日志):

def index 
    @user = User.all
    @filter = Filter.first
    @project = Project.all
    [... a bunch of code that defines @hourlogs...] 

    respond_to do |format|
      format.html
      format.csv { render text: @hourlogs.to_csv }
    end
end 

这是视图:

<%= link_to "CSV", hourlogs_path(format: "csv") %>

这是小时日志的架构:

create_table "hourlogs", force: :cascade do |t|
    t.string   "name"
    t.string   "project"
    t.text     "description"
    t.integer  "hours"
    t.datetime "created_at",     null: false
    t.datetime "updated_at",     null: false
    t.integer  "user_id"
    t.datetime "date"
    t.integer  "project_id"
    t.datetime "date_from"
    t.datetime "date_to"
    t.string   "project_filter"
    t.index ["project_id"], name: "index_hourlogs_on_project_id", using: :btree
    t.index ["user_id"], name: "index_hourlogs_on_user_id", using: :btree
  end

这是项目的架构:

create_table "projects", force: :cascade do |t|
    t.string   "name"
    t.text     "description"
    t.string   "plan"
    t.integer  "planhours"
    t.integer  "thours"
    t.datetime "created_at",                    null: false
    t.datetime "updated_at",                    null: false
    t.datetime "plandate"
    t.datetime "starthourtrack"
    t.integer  "duration"
    t.text     "notes"
    t.boolean  "status",         default: true
  end

任何帮助将不胜感激,我现在​​已经挣扎了 10 个小时,似乎无法得到它。我尝试创建一个单独的 index.csv.erb 文件,概述标题名称,选择 hourlog.name,而不是 hourlog.attributes.values_at(*column_names),但我似乎无法获得正确的语法/逻辑。谢谢!

[编辑:解决方案] 看起来我只需要一些括号 - 我添加了特定的列名来限制显示的列数,然后我写出了我需要的每一列(hourlog.name、hourlog.description 等) - 我没有'没有意识到我需要把那些放在[]中,在“csv <<”之后,它成功了。

def self.to_csv
        CSV.generate do |csv|
          column_names = %w(Project Name Description Hours Date)
          csv << column_names
          all.each do |hourlog|
            csv << [hourlog.project.name, hourlog.name, hourlog.description, hourlog.hours, hourlog.created_at]
          end
        end
      end

我的 CSV 导出现在的样子 - 我已将我希望在 CSV 文件中包含的唯一列标记为红色

标签: ruby-on-railscsvdatatablesexport-to-csv

解决方案


看起来我只需要一些括号 - 我添加了特定的列名来限制显示的列数,然后我写出了我需要的每一列(hourlog.name、hourlog.description 等) - 我没有意识到我需要把它们放在 [] 中,在 "csv <<" 之后就可以了。

def self.to_csv
        CSV.generate do |csv|
          column_names = %w(Project Name Description Hours Date)
          csv << column_names
          all.each do |hourlog|
            csv << [hourlog.project.name, hourlog.name, hourlog.description, hourlog.hours, hourlog.created_at]
          end
        end
end

推荐阅读