simple-ynab 0.0.3 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9a881eee0ab99fe8a8d82d526a29f16793682fc8
-  data.tar.gz: 02b3e8d3895b1a3e27d0ae71225a2dd44e80a9ca
+  metadata.gz: 01d194396c8cdff4df84b40021ac98a64a572f69
+  data.tar.gz: 06ea74ea4e20eccd2924b295f866f8c695505fc0
 SHA512:
-  metadata.gz: 720f9333ce2434522d154742548f0eb652eb7e1f41b2fe2786fc9161eb3562c16a2b49c6ad16ca594caf14a0a97a6626f706fbab451cc96f3ebcf5c6a809087c
-  data.tar.gz: 2cdc73c61b92a69068de418ba4ac53c8b4ff839241f55590706aa2a62b7c18ab5288e3602bc9e8944bea96218081f2d1f4709abdbad3134d7dade85014f4ee78
+  metadata.gz: 754c93c1147dcea2fa632c8eedb534094a555ce9b00bbbfd2d820c4f102a2a84cc12d87f56ba879c0c5119a80f3ec2dcdb6f5d4e2e724838fd1e004e84e34ba4
+  data.tar.gz: 9fca07efdd7b2db2737093f4236ba47b5940e54fefeb19e7ecad00fad059e00ae8e8be36550dfda0c61056fda17bbad35539b8f535fe236594fb4a86d0aa6447

data/bin/simple-ynab

@@ -1,49 +1,135 @@
 #!/usr/bin/env ruby
 
+# Converts the CSV generated by Simple Bank into a CSV understandable by YNAB
 require 'csv'
+require 'slop'
+require 'colorize'
+require 'Date'
 
+opts = Slop.parse(:banner => 'simple-ynab', :help => true ) do
+    on 'd', 'days=',    'Strips all rows before x days ago, e.g -d 30 ', :argument => true
+    on 'y', 'ynab_date', 'Converts date to MM/DD/YYYY. SEE README.md FOR WARNINGS!!!', :argument => false
+    on 'p', 'pending', 'Include pending transactions', :argument => false
+    on 'v', 'verbose', 'Print debug information',                    :argument => false
+    on 'x', 'very_verbose', 'Prints an uncomfortable amount of debugging information to terminal, does not replace -v',  :argument => false
+end
 
-@simple_keywords = ["Date", "Description", "Category", "Memo", "Amount"]
-@ynab_keywords        = ["Date", "Payee", "Category", "Memo", "Amount"]
-
-@newcsv = []
-puts ""
-
+# Which column headers in the Simple CSV match up with the columns headers needed by YNAB
+simple_keywords = ["Date", "Description", "Category", "Memo", "Amount"]
+ynab_keywords   = ["Date", "Payee", "Category", "Memo", "Amount"]
 
-# Get the csv file from args or prompt user
+##### Get the csv file from args
 ARGV.each do |argument|
-	@csvfile = ARGV[0]
+  @csvfile = ARGV[0]
 end
 
+Kernel.trap('INT') { Kernel.exit } #Exit cleanly with ctrl+c
+
 if ARGV.empty?
-	print "No file detected, enter file name: "
-	@csvfile = gets
+  print "No file detected, enter file name: "
+  @csvfile = gets
 end
 
+
 # Import csv
-csv = CSV.read(@csvfile, headers: true) #http://bit.ly/1mSlqfA
-# @headers = CSV.open('foo.csv','r', :headers => true).read.headers
+if opts.verbose? then puts "Importing CSV".magenta end
+simplecsv = CSV.read(@csvfile) #http://bit.ly/1mSlqfA
+if opts.verbose? then puts "Headers are: #{simplecsv[0]}".blue end
 
-# Rename the Simple csv headers to match the ynab csv headers
-indicies = @simple_keywords.map{ |column| csv.headers.index(column)}
 
-# Create a new csv file with the columns in the ynab order
-@newcsv = csv.map { |row| row.values_at(*indicies) }
+# Get the 'simple' header indicies
+simple_headers = simple_keywords.map{ |column| simplecsv[0].index(column)}
+if opts.verbose? then puts "Simple_headers indexes are #{simple_headers}".blue end
 
 
-filename  = File.basename(@csvfile,".*")
-pathname = File.join( File.dirname(@csvfile), "#{filename}.simple.csv" )
+# Create a new array that omits the CSV rows when the value in the 'Pending' column = true
+if opts.verbose? then puts "Removing pending transactions".magenta end
+if opts.pending? then puts "User chose to include pending transactions".blue end
+no_pending_array = Array.new()
+pending = simplecsv[0].find_index('Pending') # Column number that contains 'Pending'
+simplecsv.each do |row|
+  # If -p (Pending) option is not provided, Only push rows with 'false' in the pending column. 
+  # Make sure header (row 0) gets included too. 
+  if opts.pending? or row[pending].to_s.casecmp('false') == 0 or row[pending].to_s.casecmp('Pending') == 0 
+    no_pending_array.push(row)
+  else
+    if opts.verbose? then puts "Removing pending transaction #{row}".yellow end
+  end
+end
+
+
+# Create a new array that only includes the YNAB columns
+ynab_array = Array.new()
+no_pending_array.each do |row| #Itterate over every row in new 'pending free' array
+    newrow = Array.new() #Create a temporary array
+    simple_headers.each do | desired_index | #Itterate over each of the columsn we want e.g. [0, 7, 9, 16, 3]
+      newrow.push(row[desired_index]) #Add the new value to the temporary array
+    end
+   ynab_array.push(newrow) #Add each row to the new YNAB array
+end
 
 
-# Save headers to file
-puts "Saving new file to #{pathname}"
-CSV.open(pathname, 'w') do |the_csv|
-	the_csv << @ynab_keywords
+# Replace the headers with the YNAB headers
+if opts.verbose?
+  puts "Replacing csv headers with ynab headers".magenta
+  puts "Simple column headers #{ynab_array[0]}".blue
 end
+ynab_array[0] = ynab_keywords
+if opts.verbose? then puts "YNAB column headers are #{ynab_array[0]}".blue end
+
+
+# Only include dates within the specified range
+ynab_array_less_dates = Array.new()
+ynab_array_less_dates[0] = ynab_array[0] # Copy the header to new array
+days_ago = Date.today() - opts[:days].to_i
+date_column = ynab_array[0].find_index('Date') #Find the index of colum that contains 'Date' data (usually [0])
+ynab_array.each do | row |
+  if opts.days? # User specified number of days to copy from CSV
+    if row[date_column] =~ /\d/ and Date.parse(row[date_column]) >= days_ago  #Only apply to rows that contain dates, skip rows without nummbers (e.g. header row)
+      if opts.very_verbose? then puts "Transaction was less than #{opts[:days].to_i} days ago: #{row}".light_green end
+      ynab_array_less_dates.push(row) # Only push rows less than x number of days old
+    else
+      if opts.very_verbose? then puts "Transaction was more than #{opts[:days].to_i} days ago: #{row}".yellow end
+    end
+  else
+    ynab_array_less_dates.push(row) # Push all rows
+  end
+end
+
+
+# Change the dates from YYYY-MM-DD to MM-DD-YYYY, saves 1 click when importing in YNAB
+# Has side affect of sometimes causing YNAB to fail to match previously imported transactions
+if opts.ynab_date?
+  puts "WARNING: YNAB may fail to match transactions previously imported with different date formats".red
+  puts "Converting dates to MM/DD/YYYY".red
+  if opts.verbose? then puts "Changing simple dates to ynab dates".magenta end
+  if ynab_array_less_dates[0].any?{|s| s.casecmp('date')==0}
+    date_column = ynab_array_less_dates[0].find_index('Date') #Find the index of colum that contains 'Date' data (usually [0])
+  else
+    puts "Unable to find word 'Date' in csv header. #{ynab_array_less_dates[0]}".red
+    exit
+  end
+  if opts.verbose? then puts "The date column in ynab_array_less_dates is #{date_column}: #{ynab_array_less_dates[0][date_column]}".blue end
+    ynab_array_less_dates.each do |row|
+      if row[date_column] =~ /\d/ #Only apply to rows that contain dates, skip rows without nummbers (e.g. header row)
+        row[date_column] = Date.parse(row[date_column]).strftime(format='%m/%d/%Y').to_s
+      else
+        if opts.verbose? then puts "Not changing date syntax on header row #{row}".blue end
+      end
+  end
+end
+
+
+# Save to file
+if opts.verbose? then puts "Saving to CSV file".magenta end
+
+filename = File.basename(@csvfile,".*")
+pathname = File.join( File.dirname(@csvfile), "#{filename}.simple.csv" )
+puts ""
+puts "Saving to new file: #{pathname}".green
 
-# Append the new csv to the file
-CSV.open(pathname, 'a+') do |the_csv|
-	@newcsv.each do |row|
-		the_csv << row
-	end
+CSV.open(pathname,'w') do |csv|
+  ynab_array_less_dates.each do | row |
+    csv << row
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: simple-ynab
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors:
 - Spencer Owen
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-09-19 00:00:00.000000000 Z
+date: 2014-11-10 00:00:00.000000000 Z
 dependencies: []
 description: Converts simple bank csv to ynab csv
 email: owenspencer@gmail.com
@@ -18,7 +18,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - bin/simple-ynab
-homepage: 
+homepage: https://github.com/spuder/simple-ynab
 licenses:
 - MIT
 metadata: {}
@@ -43,3 +43,4 @@ signing_key:
 specification_version: 4
 summary: Converts csv from simple format to ynab format
 test_files: []
+has_rdoc: