Sha256: a2df7cda1556f9a1d105c8f428c536f969c64a6f6b195cf5621c0754c27ad59f
Contents?: true
Size: 1.92 KB
Versions: 6
Compression:
Stored size: 1.92 KB
Contents
# frozen_string_literal: true
module MasterDataTool
module Dump
class Executor
Error = Struct.new(:table, :exception)
DEFAULT_IGNORE_TABLES = %w[ar_internal_metadata schema_migrations master_data_statuses]
DEFAULT_IGNORE_COLUMNS = %w[created_at updated_at]
def initialize(ignore_empty_table: true, ignore_tables: [], ignore_column_names: [], verbose: false)
@ignore_empty_table = ignore_empty_table
@ignore_tables = DEFAULT_IGNORE_TABLES + Array(MasterDataTool.config.dump_ignore_tables) + ignore_tables
@ignore_column_names = DEFAULT_IGNORE_COLUMNS + Array(MasterDataTool.config.dump_ignore_columns) + ignore_column_names
end
def execute
[].tap do |errors|
ApplicationRecord.connection.tables.each do |table|
if @ignore_tables.include?(table)
print_message "[ignore] #{table}"
next
end
dump_to_csv(table)
rescue => e
errors << Error.new(table, e)
end
end
end
private
def print_message(message)
return unless @verbose
puts message
end
def dump_to_csv(table)
model_klass = Rails.const_get(table.classify)
if ignore?(model_klass)
print_message "[ignore] #{table}"
return
end
csv_path = Pathname.new(MasterDataTool.config.master_data_dir).join("#{table}.csv")
CSV.open(csv_path, 'w', force_quotes: true) do |csv|
headers = model_klass.column_names - @ignore_column_names
csv << headers
model_klass.all.find_each do |record|
items = []
headers.each do |name|
items << record[name]
end
csv << items
end
end
end
def ignore?(model_klass)
return false unless @ignore_empty_table
model_klass.count < 1
end
end
end
end
Version data entries
6 entries across 6 versions & 1 rubygems