ruby heroku_db.rake

Posted 2021-05-16

require 'csv'

namespace :csv do
  # Call like this:
  # `rake csv:dump[[e-mail]]`
  task :dump, [:email] => [:environment] do |t, args|
    db_url = Bundler.with_clean_env do
      `heroku config:get DATABASE_URL`
    end
    md = db_url.match(/^postgres:\/\/([^:]+):([^@]+)@([^:]+):(\d+)\/(.+)$/)

    production_config = {}
    production_config['adapter']  = 'postgresql'
    production_config['encoding'] = 'unicode'
    production_config['pool']     = 7
    production_config['username'] = md[1]
    production_config['password'] = md[2]
    production_config['host']     = md[3]
    production_config['port']     = md[4]
    production_config['database'] = md[5]
    ActiveRecord::Base.establish_connection(production_config)
    # Do something with models, etc.
    user = User.find_by_email(args[:email])
    
    # Export jobs
    CSV.open('jobs.csv', 'wb') do |csv|
      csv << ['id', 'name', 'attention', 'created_at', 'completed_at', 'referrer', 'archived_reason']
      user.account.jobs.each do |job|
        csv << [
          job.id,
          job.name,
          job.attention,
          job.created_at,
          job.completed_at,
          job.referrer,
          job.archived_reason
        ]
      end
    end
  end
end