fio_parser 0.1.0 → 0.2.0

data/lib/fio_parser.rb

@@ -1,17 +1,27 @@
 # encoding: UTF-8
 
-# fio_parser is simple library for parsing monthly account statements from the Fio Bank CZ. Supports CSV and GPC formats. 
+# fio_parser is simple library for parsing monthly account statements from the Fio Bank CZ.
+# Supports CSV and GPC formats.
+# Support for the transaction email.
 #
 # Author: Jaromír "Cervajz" Čevenka
 # http://www.cervajz.com
 #
 # Copyright © 2012 cervajz@cervajz.com
 #
-# Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the “Software”), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+# Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
+# documentation files (the “Software”), to deal in the Software without restriction, including without limitation
+# the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software,
+# and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
 #
-# The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+# The above copyright notice and this permission notice shall be included in all copies or substantial portions of
+# the Software.
 #
-#THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+# THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
+# INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+# IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR
+# THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
 require 'ostruct'
 require 'date'
@@ -52,25 +62,34 @@ end
 
 class FioParser
   include GPCRowParser
-  attr_accessor :transactions
+  attr_reader :transactions
 
   class << self
-    def parse_csv(csv_file_path)
-      return new csv_file_path, true
+    def parse_csv csv_file_path
+      new csv_file_path, :csv
     end
 
-    def parse_gpc(gpc_file_path)
-      return new gpc_file_path, false
+    def parse_gpc gpc_file_path
+      new gpc_file_path, :gpc
+    end
+
+    def parse_email email_body
+      new email_body, :email
     end
   end
 
-  def initialize file_path, csv=true
+  def initialize file_path, type
     @transactions = []
 
-    if csv
-      parse_csv file_path
-    else
-      parse_gpc file_path
+    case type
+      when :csv
+        parse_csv file_path
+      when :gpc
+        parse_gpc file_path
+      when :email
+        parse_email file_path
+      else
+        Exception.raise 'Unknown input type'
     end
   end
 
@@ -129,17 +148,73 @@ class FioParser
     end
   end
 
+  def parse_email email
+    if File.exist?(email)
+      email_body = file_rows email
+    else
+      email_body = string_rows email
+    end
+
+    class << self
+      attr_reader :account
+      attr_reader :total
+      attr_reader :vs
+      attr_reader :message
+      attr_reader :balance
+      attr_reader :type
+      attr_reader :contra_acc
+      attr_reader :contra_acc_bc
+      attr_reader :ss
+      attr_reader :cs
+    end
+
+    income = true
+
+    email_body.each_with_index do |line, index|
+      out = get_string_from_colon line
+
+      case index
+        when 0 # Account nr.
+          income = mail_income? line
+          @account = out
+        when 1 # Total
+          @total = mail_total out, income
+        when 2 # Variable symbol
+          @vs = out
+        when 3 # Messsage
+          @message = out
+        when 4 # Balance
+          @balance = get_total out
+        when 5 # Contra account & payment type
+          @type = mail_transaction_type out
+          if @type == :transfer
+            @contra_acc = out.split('/')[0]
+            @contra_acc_bc = out.split('/')[1]
+          else
+            @contra_acc = ''
+            @contra_acc_bc = ''
+          end
+        when 6 # Specific symbol
+          @ss = out
+        when 7 # Constant symbol
+          @cs = out
+        else
+          #
+      end
+    end
+  end
+
   def parse_csv_row row
     columns = row.split ';'
 
     t = OpenStruct.new
     t.date = Date.strptime columns[1], '%d.%m.%Y'
-    t.total = columns[2].to_s.delete(' ').gsub(',', '.').to_f
-    t.counter_acc = columns[3]
-    t.bank_code = columns[4]
+    t.total = get_total columns[2]
+    t.contra_acc = columns[3]
+    t.contra_acc_bc = columns[4]
     t.vs = columns[5]
     t.type = get_transaction_type columns[6]
-    t.counter_acc_name = columns[7]
+    t.contra_acc_name = columns[7]
     t.message = columns[8]
     t.comment = columns[9]
 
@@ -151,8 +226,8 @@ class FioParser
 
     t.date = gpc_date row
     t.total = gpc_total row
-    t.counter_acc = delete_zeros row[19..34]
-    t.bank_code = gpc_bank_code row
+    t.contra_acc = delete_zeros row[19..34]
+    t.contra_acc_bc = gpc_bank_code row
     t.vs = delete_zeros row[61..70]
     t.cs = gpc_constant_symbol row
     t.ss = gpc_specific_symbol row
@@ -167,11 +242,11 @@ class FioParser
   end
 
   def get_transaction_type str
-    if ['převodem', 'bezhotovostní', 'připsaný úrok'].any? { |w| str.downcase.include?(w) }
+    if ['převodem', 'prevodem', 'bezhotovostní', 'bezhotovostni', 'připsaný úrok'].any? { |w| str.downcase.include?(w) }
       return :transfer
     end
 
-    if ['kartou'].any? { |w| str.include?(w) }
+    if %w(kartou).any? { |w| str.include?(w) }
       return :card
     end
 
@@ -180,9 +255,44 @@ class FioParser
 
   def file_rows file
     result = []
-    e_c = Encoding::Converter.new 'cp1250', 'utf-8'
-    File.open(file).each_line { |line| result.push e_c.convert line }
+    File.open(file).each_line { |line| result.push convert_line line }
     result
   end
-end
 
+  def string_rows str
+    result = []
+    str.each_line { |line| result.push convert_line line }
+    result
+  end
+
+  def convert_line str
+    if str.encoding.to_s.downcase == "utf-8"
+      str
+    else
+      e_c = Encoding::Converter.new 'cp1250', 'utf-8'
+      e_c.convert str
+    end
+  end
+
+  def get_string_from_colon str
+    str.sub(/^.*?:/i, '').strip
+  end
+
+  def get_total str
+    str.delete(' ').gsub(',', '.').to_f
+  end
+
+  def mail_total str, income
+    total = get_total str
+    total = 0 - total unless income
+    total
+  end
+
+  def mail_income? str
+    ['příjem', 'prijem'].any? { |w| str.downcase.include?(w) }
+  end
+
+  def mail_transaction_type str
+    str.downcase.include?('kartou') ? :card : :transfer
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: fio_parser
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -11,8 +11,8 @@ bindir: bin
 cert_chain: []
 date: 2012-10-24 00:00:00.000000000 Z
 dependencies: []
-description: Gem for parsing monthly statements from the Fio Bank CZ. Supports CSV
-  and GPC formats
+description: Gem for parsing monthly statements from the Fio Bank CZ. Supports CSV,
+  GPC and email format.
 email: cervajz@cervajz.com
 executables: []
 extensions: []
@@ -42,5 +42,6 @@ rubyforge_project:
 rubygems_version: 1.8.24
 signing_key: 
 specification_version: 3
-summary: Gem for parsing monthly statements from the Fio Bank CZ
+summary: Gem for parsing monthly statements from the Fio Bank CZ. Parses transaction
+  emails too.
 test_files: []