Class: BankScrap::Exporter::Csv

Inherits:
Object
  • Object
show all
Defined in:
lib/bankscrap/exporters/csv.rb

Overview

Csv exporter

Instance Method Summary collapse

Constructor Details

#initialize(account) ⇒ Csv

Returns a new instance of Csv.



8
9
10
 
# File 'lib/bankscrap/exporters/csv.rb', line 8

def initialize()
  @account = 
end

Instance Method Details

#check_array_class(data, tclass) ⇒ Object 



32
33
34
 
# File 'lib/bankscrap/exporters/csv.rb', line 32

def check_array_class(data, tclass)
  data.all? { |x| x.is_a? tclass }
end

#get_filename(data, output) ⇒ Object 



12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
 
# File 'lib/bankscrap/exporters/csv.rb', line 12

def get_filename(data, output)
  if output
    if output == '-'
      '/dev/stdout'
    else
      output
    end
  else
    if check_array_class(data, Bankscrap::Transaction)
      'transactions.json'
    elsif check_array_class(data, Bankscrap::Account)
      'accounts.json'
    elsif check_array_class(data, Bankscrap::Loan)
      'loans.json'
    elsif check_array_class(data, Bankscrap::Card)
      'cards.json'
    end
  end
end

#write_to_file(data, output) ⇒ Object 



36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
 
# File 'lib/bankscrap/exporters/csv.rb', line 36

def write_to_file(data, output)
  output = get_filename(data, output)
  if check_array_class(data, Bankscrap::Transaction)
    CSV.open(output, 'wb') do |csv|
      csv << %w[Date Description Amount]
      data.each { |line| csv << line.to_a }
    end
    STDERR.puts "Transactions for: #{@account.description} (#{@account.iban}) exported to #{output}".cyan
  elsif check_array_class(data, Bankscrap::Account)
    CSV.open(output, 'wb') do |csv|
      csv << %w[Id Iban Name Description Bank Balance]
      data.each { |line| csv << line.to_a }
    end
    STDERR.puts "Accounts exported to #{output}".cyan
  elsif check_array_class(data, Bankscrap::Loan)
    CSV.open(output, 'wb') do |csv|
      csv << %w[Id Name Description Amount]
      data.each { |line| csv << line.to_a }
    end
    STDERR.puts "Accounts exported to #{output}".cyan
  elsif check_array_class(data, Bankscrap::Card)
    CSV.open(output, 'wb') do |csv|
      csv << %w[Id Name Description Pan Amount Avaliable Is_credit]
      data.each { |line| csv << line.to_a }
    end
    STDERR.puts "Accounts exported to #{output}".cyan
  end
end