Sha256: 4c0e017294d7a6de7d3af380aba672b49012e24b0c43c78fecb381db9af7860e
Contents?: true
Size: 1.74 KB
Versions: 2
Compression:
Stored size: 1.74 KB
Contents
# frozen_string_literal: true require 'json' class ActiveReport::Record < ActiveReport::Base attr_accessor :datum, :model, :only, :except, :headers, :options def initialize(datum, model: nil, only: nil, except: nil, headers: nil, options: {}) @datum = datum @model = model @only = munge(only) @except = munge(except) @headers = headers @options = csv_options.merge(options) end def self.export(datum, only: nil, except: nil, headers: nil, options: {}) klass = new(datum, only: only, except: except, headers: headers, options: options) klass.export end def self.import(datum, only: nil, except: nil, headers: nil, options: {}, model: nil) klass = new(datum, only: only, except: except, headers: headers, options: options, model: model) klass.import end def export @datum = if @datum.is_a?(ActiveRecord::Relation) JSON.parse(@datum.to_json).flatten else merge(@datum.attributes) end @only.map!(&:to_s) @except.map!(&:to_s) CSV.generate(@options) do |csv| csv << (@headers || filter_humanize_keys(@datum)) @datum.lazy.each { |data| csv << filter_values(data) } end end def import if @model.nil? || (@model.superclass != ActiveRecord::Base) raise ArgumentError, 'Model must be an ActiveRecord::Base object.' end @datum = ActiveReport::Hash.import(@datum, headers: @headers, options: @options) @datum = munge(@datum) @datum.lazy.each do |data| params = {} data.lazy.each do |key, value| key = key.to_s.downcase.gsub(/ |-/, '_').to_sym params[key] = value end filter(params) params.delete(:id) @model.create(params) end end end
Version data entries
2 entries across 2 versions & 1 rubygems
Version | Path |
---|---|
active_report-5.0.1 | lib/active_report/record.rb |
active_report-5.0.0 | lib/active_report/record.rb |