Merge pull request #11 from timkelty/force-strings-yaml-compatibility

Force symbols from deploy.rb into strings to be YAML compatible
sgruhier · Apr 12, 2013 · 5186d89 · 5186d89
2 parents 229c2a6 + 7ec2eae
commit 5186d89
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/capistrano-db-tasks/database.rb b/lib/capistrano-db-tasks/database.rb
@@ -61,7 +61,7 @@ def initialize(cap_instance)
       @cap.run("cat #{@cap.current_path}/config/database.yml") do |c, s, d|
         @config += d
       end
-      @config = YAML.load(@config)[@cap.rails_env]
+      @config = YAML.load(@config)[@cap.rails_env.to_s]
     end
 
     def dump
@@ -86,7 +86,7 @@ def load(file, cleanup)
   class Local < Base
     def initialize(cap_instance)
       super(cap_instance)
-      @config = YAML.load_file(File.join('config', 'database.yml'))[@cap.local_rails_env]
+      @config = YAML.load_file(File.join('config', 'database.yml'))[@cap.local_rails_env.to_s]
       puts "local #{@config}"
     end