【Rails】鉄道会社・路線・駅データをseedで取得・インポートする

Railsで鉄道会社・路線・駅のCSVファイルからデータを挿入する方法を紹介します。

駅データ.jpで鉄道会社・路線・駅データをダウンロードし、Railsプロジェクトに格納
鉄道会社・路線・駅のマイグレーション作成
鉄道会社・路線・駅のモデル作成
seed.rbで鉄道会社・路線・駅データを取得し、テーブルに保存

駅データ.jpで鉄道会社・路線・駅データをダウンロードし、Railsプロジェクトに格納

駅データ.jpで鉄道会社・路線・駅データをダウンロードします。ダウンロードするためには会員登録が必要です。

ダウンロードしたCSVファイルをdb/data配下に格納しておきます。

db/data/companies.csv (鉄道会社データ)
db/data/lines.csv (路線データ)
db/data/stations.csv (駅データ)

駅関連データ提供元：駅データ無料ダウンロード『駅データ.jp』

鉄道会社・路線・駅のマイグレーション作成

まずは、鉄道会社・路線・駅のマイグレーションファイルを作成します。

・db/migrate/20200101000001_create_companies.rb

class CreateCompanies < ActiveRecord::Migration[5.2]

def change

create_table :companies, id: false, options: 'ENGINE=InnoDB COLLATE=utf8_general_ci' do |t|

t.integer :company_id, null: false, unique: true

t.string :name

t.string :kana

t.string :formal_name

t.string :abbreviation_name

t.datetime :created_at

t.datetime :updated_at

t.datetime :deleted_at

end

・db/migrate/20200101000002_create_lines.rb

class CreateLines < ActiveRecord::Migration[5.2]

def change

create_table :lines, id: false, options: 'ENGINE=InnoDB COLLATE=utf8_general_ci' do |t|

t.integer :line_id, null: false, unique: true

t.integer :company_id

t.string :name

t.string :kana

t.decimal :lat, :precision => 9, :scale => 6, null: true

t.decimal :lng, :precision => 9, :scale => 6, null: true

t.datetime :created_at

t.datetime :updated_at

t.datetime :deleted_at

end

・db/migrate/20200101000003_create_stations.rb

class CreateStations < ActiveRecord::Migration[5.2]

def change

create_table :stations, id: false, options: 'ENGINE=InnoDB COLLATE=utf8_general_ci' do |t|

t.integer :station_id, null: false, unique: true

t.integer :line_id

t.string :name

t.string :address

t.decimal :lat, :precision => 9, :scale => 6, null: true

t.decimal :lng, :precision => 9, :scale => 6, null: true

t.datetime :created_at

t.datetime :updated_at

t.datetime :deleted_at

end

鉄道会社・路線・駅のモデル作成

鉄道会社・路線・駅のモデル作成を作成します。各々のリレーションもはっておきます。

・app/models/company.rb

class Company < ApplicationRecord

# リレーション

has_many :lines

end

・app/models/line.rb

class Line < ApplicationRecord

# リレーション

belongs_to :company

has_many :stations

end

・app/models/station.rb

class Station < ApplicationRecord

# リレーション

belongs_to :line

end

seed.rbで鉄道会社・路線・駅データを取得し、テーブルに保存

先ほど保存したCSVファイルから必要なカラムのデータを取得していきます。鉄道会社データから駅データを取得していき、完了です。

・db/seeds.rb

require 'csv'

# 鉄道会社CSVファイルパス

COMPANY_PATH = "db/data/companies.csv"

# 鉄道会社カラム指定定数

COMPANY_CSVROW_COMPANY_ID = 0

COMPANY_CSVROW_NAME = 2

COMPANY_CSVROW_KANA = 3

COMPANY_CSVROW_FORMAL_NAME = 4

COMPANY_CSVROW_ABBREVIATION_NAME = 5

# 鉄道会社CSVを読み込みテーブルに保存

CSV.foreach(COMPANY_PATH) do |row|

company_id = row[COMPANY_CSVROW_COMPANY_ID]

name = row[COMPANY_CSVROW_NAME]

kana = row[COMPANY_CSVROW_KANA]

formal_name = row[COMPANY_CSVROW_FORMAL_NAME]

abbreviation_name = row[COMPANY_CSVROW_ABBREVIATION_NAME]

Company.find_or_create_by(

:company_id => company_id,

:name => name, :kana => kana,

:formal_name => formal_name,

:abbreviation_name => abbreviation_name

)

end

# 路線CSVファイルパス

LINE_PATH = "db/data/lines.csv"

# 路線カラム指定定数

LINE_CSVROW_LINE_ID = 0

LINE_CSVROW_COMPANY_ID = 1

LINE_CSVROW_NAME = 2

LINE_CSVROW_KANA = 3

LINE_CSVROW_LNG = 8

LINE_CSVROW_LAT = 9

# 路線CSVを読み込みテーブルに保存

CSV.foreach(LINE_PATH) do |row|

line_id = row[LINE_CSVROW_LINE_ID]

company_id = row[LINE_CSVROW_COMPANY_ID]

name = row[LINE_CSVROW_NAME]

kana = row[LINE_CSVROW_KANA]

lng = row[LINE_CSVROW_LNG]

lat = row[LINE_CSVROW_LAT]

Line.find_or_create_by(

:line_id => line_id,

:company_id => company_id,

:name => name,

:kana => kana,

:lat => lat,

:lng => lng

)

end

# 駅CSVファイルパス

STATION_PATH = "db/data/stations.csv"

# 駅カラム指定定数

STATION_CSVROW_STATION_ID = 0

STATION_CSVROW_NAME = 2

STATION_CSVROW_LINE_ID = 5

STATION_CSVROW_ADDRESS = 8

STATION_CSVROW_LNG = 9

STATION_CSVROW_LAT = 10

# 駅CSVを読み込みテーブルに保存

CSV.foreach(STATION_PATH) do |row|

station_id = row[STATION_CSVROW_STATION_ID]

name = row[STATION_CSVROW_NAME]

line_id = row[STATION_CSVROW_LINE_ID]

address = row[STATION_CSVROW_ADDRESS]

lng = row[STATION_CSVROW_LNG]

lat = row[STATION_CSVROW_LAT]

Station.find_or_create_by(

:station_id => station_id,

:line_id => line_id,

:name => name,

:address => address,

:lat => lat,

:lng => lng

)

end

以上、鉄道会社・路線・駅データをseedで取得・インポートする方法でした。