lib/multiverse.rb in multiverse-0.2.0 vs lib/multiverse.rb in multiverse-0.2.1
- old
+ new
@@ -9,22 +9,16 @@
def db
@db ||= ENV["DB"].presence
end
def db_dir
- db_dir = db ? "db/#{db}" : "db"
- abort "Unknown DB: #{db}" if db && !Dir.exist?(db_dir)
+ db_dir = "#{Rails.application.config.paths["db"].first}/#{db}"
+ abort "Unknown DB: #{db}" unless Dir.exist?(db_dir)
db_dir
end
def parent_class_name
- if db
- "#{db.camelize}Record"
- elsif ActiveRecord::VERSION::MAJOR >= 5
- "ApplicationRecord"
- else
- "ActiveRecord::Base"
- end
+ "#{db.camelize}Record"
end
def migrate_path
"#{db_dir}/migrate"
end