environment from RAILS_ENV

diff --git a/lib/connection.rb b/lib/connection.rb
index 024b36b..80c4525 100644 (file)
--- a/lib/connection.rb
+++ b/lib/connection.rb
@@ -1,5 +1,6 @@
 require 'yaml'
-options = YAML.load(IO.read("config/database.yml"))["development"]
+environment = ENV["RAILS_ENV"] || "development"
+options = YAML.load(IO.read("config/database.yml"))[environment]
 require 'rubygems'
 gem 'activerecord'
 require 'active_record'