Merge pull request #105 from internetee/115336093-column_missing

115336093-added_registrar_code
This commit is contained in:
Timo Võhmar 2016-03-18 16:25:13 +02:00
commit 59ed022d8d

View file

@ -6,6 +6,8 @@ class AccountActivity < ActiveRecord::Base
belongs_to :bank_transaction
belongs_to :invoice
attr_accessor :registrar
CREATE = 'create'
RENEW = 'renew'
ADD_CREDIT = 'add_credit'
@ -22,12 +24,13 @@ class AccountActivity < ActiveRecord::Base
end
def to_csv
attributes = %w(description activity_type created_at sum)
attributes = %w(registrar description activity_type created_at sum)
CSV.generate(headers: true) do |csv|
csv << %w(description activity_type receipt_date sum)
csv << %w(registrar description activity_type receipt_date sum)
all.each do |x| # rubocop:disable Rails/FindEach
x.registrar = Registrar.find(x.account_id).try(:code)
csv << attributes.map { |attr| x.send(attr) }
end
end