module ActiveRecord::Tasks::DatabaseTasks

def each_current_configuration(environment)

def each_current_configuration(environment)
  environments = [environment]
  # add test environment only if no RAILS_ENV was specified.
  environments << 'test' if environment == 'development' && ENV['RAILS_ENV'].nil?
  configurations = ActiveRecord::Base.configurations.values_at(*environments)
  configurations.compact.each do |configuration|
    yield configuration unless configuration['database'].blank?
  end
end