Skip to content

Instantly share code, notes, and snippets.

@flyerhzm
Forked from jcasimir/exporter.rb
Last active August 29, 2015 14:14
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save flyerhzm/1fbc8a694b03593280d2 to your computer and use it in GitHub Desktop.
Save flyerhzm/1fbc8a694b03593280d2 to your computer and use it in GitHub Desktop.
require 'csv'
module Exporter
DEFAULT_EXPORT_TABLES = [ Invoice, InvoiceItem, Item, Merchant, Transaction, User ]
DESTINATION_FOLDER = "tmp/"
def self.included(klass)
klass.extend ClassLevelMethods
end
def self.export_tables_to_csv(tables = DEFAULT_EXPORT_TABLES)
tables.each &:export_table_to_csv
end
def data
self.class.column_names.map { |column| send(column) }
end
module ClassLevelMethods
def export_table_to_csv
CSV.open(filename_for_class, "w") do |output_file|
output_file << column_names
data.each{ |row| output_file << row }
end
end
def filename_for_class
[DESTINATION_FOLDER, to_s.pluralize.underscore, '.csv'].join
end
def data
all.map(&:data)
end
end
end
class ActiveRecord::Base
include Exporter
end
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment