Return to pta_import.pl CVS log

Up to [cvsweb.bsd.lv] / pta

support for another credit card;
from Freda Bundchen, minimally tweaked by me to make the diff smaller

#!/usr/bin/perl 

# Copyright (c) 2020 Freda Bundchen

# Permission to use, copy, modify, and distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.

use warnings;
use strict;

# === SUBROUTINES  =====================================================

sub parse_import_file {
	my $fn = $_[0];
	my (@compiled, %bookings, %accounts, $import_type);
	my $csv_account = "";
	if ($fn eq "import_chase_credit.txt") {
		$import_type = "chase_credit";
	} elsif ($fn eq "import_capital_one_credit.txt") { 
		$import_type = "capital_one_credit";
	} else {
		die "Undefined import type for: $fn";
	}
	open my $in, $fn or die "$fn: $!";
	while (<$in>) {
		chomp;
		next if /^(?:#|$)/;
		my $line = $_;
		if (/^ACCOUNT/) {
			$csv_account = $line;
			$csv_account =~ s/^ACCOUNT\s+(\d+)$/$1/;
			next;
		}
		s/^(.*),\s+(\d+)\s+(\S+)// or
		    die "$fn import parse error: $line";
		my ($reg, $account, $booking) = ($1, $2, $3);
		$reg =~ s/(?:^|(?<=,))(?:$|(?=,))/[^,]*/g;
		$reg = qr/$reg/;
		push @compiled, $reg;
		$bookings{$reg} = $booking;
		$accounts{$reg} = $account;
	}
	close $in;
	die "CSV account not defined in $fn" if ($csv_account eq "");
	return (\%bookings, \%accounts, \@compiled, $csv_account, $import_type);
}

sub import_chase_credit_card {
	my ($bookings_ref, $accounts_ref, $compiled_ref,
	    $csv_account) = @_;
	my ($trans_date, $post_date, $description, $category, 
	    $type, $amount);
	chomp(my $header = <>);
	while (<>) {
		next if (/^$/);
		chomp;
		s#(\d+)/(\d+)/(\d+)#$3$1$2#g; 
		my $line = $_;
		($trans_date, $post_date, $description, $category, 
		    $type, $amount) = split /,/;
		$description =~ s/#//g;
		my ($debit, $credit);
		my $booking = "";
		my $account = "";
		foreach my $regex (@$compiled_ref) {
			if ($line =~ /$regex/) {
				$account = %$accounts_ref{$regex};
				$booking = %$bookings_ref{$regex};
				last;
			}
		}
		if ($booking eq "" || $account eq "") {
			die "import parse error: $line";
		}
		if ($amount <= 0) {
			$amount = substr $amount, 1;
			$credit = $csv_account;	
			$debit = $account;
		} else {
			$debit = $csv_account;	
			$credit = $account;
		}
		print "$post_date $booking $debit $credit $amount " .
		    "$description\n";
	}
}

sub import_capital_one_credit_card {
	my ($bookings_ref, $accounts_ref, $compiled_ref,
	    $csv_account) = @_;
	my ($trans_date, $post_date, $card_num, $description,
	    $category, $csv_debit, $csv_credit, $amount);
	chomp(my $header = <>);
	while (<>) {
		next if (/^$/);
		chomp;
		s/(\d+)-(\d+)-(\d+)/$1$2$3/g; 
		my $line = $_;
		($trans_date, $post_date, $card_num, $description,
		    $category, , $csv_debit, $csv_credit) = split /,/;
		$description =~ s/#//g;
		my ($debit, $credit);
		my $booking = "";
		my $account = "";
		foreach my $regex (@$compiled_ref) {
			if ($line =~ /$regex/) {
				$account = %$accounts_ref{$regex};
				$booking = %$bookings_ref{$regex};
				last;
			}
		}
		if ($booking eq "" || $account eq "") {
			die "import parse error: $line";
		}
		if ($csv_debit =~ /^$/) {
			$amount = $csv_credit;
			$credit = $account;
			$debit = $csv_account;
		} else {
			$amount = $csv_debit;
			$credit = $csv_account;
			$debit = $account;
		}
		print "$post_date $booking $debit $credit $amount " .
		    "$description\n";
	}
}
# === MAIN PROGRAM =====================================================

# This begins the user-editable section to select the formatting file. 
my $fn = "import_capital_one_credit.txt";	
# my $fn = "import_chase_credit.txt";	
# This ends the user-editable section to select the formatting file. 

my ($bookings_ref, $accounts_ref, $compiled_ref, $csv_account, $import_type) =
    parse_import_file($fn);
if ($import_type eq "chase_credit") {
	import_chase_credit_card($bookings_ref, $accounts_ref,
	    $compiled_ref, $csv_account);
} elsif ($import_type eq "capital_one_credit") {
	import_capital_one_credit_card($bookings_ref, $accounts_ref,
	$compiled_ref, $csv_account);
} else {
	die "Undefined import type: $import_type";
}