1
/
5

railsでseedデータを簡単に取り込む

railsエンジニアの谷口です。
csv形式のseedデータの取り込みをシンプルにした事例を紹介します。

本記事の動作環境は以下の通りです。


ruby 2.1.2p95
rails 3.2.17

db/seed_datas/ に配置したcsvをシードデータとしてデータベースに取り込みます。
例えばこんなcsvをdelete-insertしたい場合

| id | name |
| 1 | ほげ |
| 2 | ふが |

以前は、下記のように各テーブルごとに取り込んでいました。

seeds.rb


require 'csv'

TestTable.delete_all
test_tables_csv = CSV.readlines("db/seed_datas/test_tables.csv")
test_tables_csv.shift # 1行目はフィールド名なのでとばす
test_tables_csv.each do |row|
record = TestTable.new
record.id = row[0]
record.name = row[1]
record.save!
end



テーブルひとつひとつに対して、毎回このように記述するのは面倒ですし、事故も発生しやすいため、下記のようにすっきり記述するためのモジュールを作成しました。

seeds.rb


include SeedModule

SeedModule.import("test_tables", delete_all: true)



モジュールの中身はこんな感じ
(実際にはこの2倍くらいコードがあるんですが外部公表用に削っています。)

seed_module.rb


require 'csv'

module SeedModule

DEFAULT_ENCODING = 'utf-8'
DEFAULT_SEED_PATH = './db/seed_datas/'

@@encoding = DEFAULT_ENCODING
@@seed_path = DEFAULT_SEED_PATH
@@table_name = nil
@@model = nil

def self.import(table_name, before_options = {}, after_options = {}, import_options = {})
@@table_name = table_name
@@model = table_name.classify.constantize
@@encoding = import_options[:encoding] || DEFAULT_ENCODING

exec_options(before_options, :BEFORE) if before_options.present?

import_all

exec_options(after_options, :AFTER) if after_options.present?
end

def self.exec_options(options, timing)
options.each do |option, value|
case option
when :truncate
next unless value # valueがtrueの場合のみ実行する
sql = "TRUNCATE TABLE "+@@table_name+";"
ActiveRecord::Base.connection.execute(sql)
msg = "全てのレコードを削除:"+sql
when :delete_all
next unless value # valueがtrueの場合のみ実行する
@@model.delete_all
msg = "全てのレコードを削除"
when :delete_before
@@model.delete_all(["id <= " + value.to_s])
msg = "ID#{value}以下のレコードを削除"
else
raise "[#{@@table_name}]#{timing} : 存在しないオプション #{option}"
end

puts "[#{@@table_name}]#{timing} : #{msg}"
end
end
private_class_method :exec_options

def self.import_all
rows, fields = retrieve_rows_and_fields_from_csv

rows.each do |row|
next if row[0].blank? # IDの列が空の行は無視する
record = @@model.find_or_initialize_by_id(row[0])
fields.each_with_index do |field, i|
next if field.blank?
record.send(field+"=", row[i])
end
record.save!
end

puts "[#{@@table_name}]Import : #{rows.size} records"
end
private_class_method :import_all

def self.retrieve_rows_and_fields_from_csv
rows = CSV.read(@@seed_path + @@table_name + ".csv", encoding: @@encoding)
# CSVの1行目からフィールド名を取得して取り除く
fields = rows.shift

return [rows, fields]
end
private_class_method :retrieve_rows_and_fields_from_csv

end
株式会社マネーフォワードでは一緒に働く仲間を募集しています
元記事: railsでseedデータを簡単に取り込む
1 いいね!
1 いいね!
同じタグの記事
今週のランキング