===================================================================
RCS file: /cvs/pta/pta_import.pl,v
retrieving revision 1.2
retrieving revision 1.9
diff -u -p -r1.2 -r1.9
--- pta/pta_import.pl	2020/11/01 20:12:08	1.2
+++ pta/pta_import.pl	2020/11/12 13:08:13	1.9
@@ -16,81 +16,170 @@
 
 use warnings;
 use strict;
-use autodie;
 
-# Please change the account numbers below to match the 
-# desired accounts in your accounts.txt.
+use Getopt::Std qw(getopts);
 
-my %accounts = (
-    CHASE_CREDIT_CARD => '1234',
-    CHASE_PAYMENT_ACCOUNT => '5678',
-    AUTO => '1111',
-    BILLS => '2222',
-    FEES => '3333',
-    FOOD => '4444',
-    GAS => '5555',
-    GIFTS => '6666',
-    GROCERIES => '7777',
-    HEALTH => '8888',
-    HOME => '9999',
-    PERSONAL => '9998',
-    PRO => '9997',
-    SHOP => '9996'
-);
+our ($opt_I);
 
-# This is the end of the user-configurable section.
-
 # === SUBROUTINES  =====================================================
 
-sub import_chase_credit_card() {
-	my %categories = (
-	    'Automotive' => 'AUTO',
-  	    'Bills & Utilities' => 'BILLS',
-	    'Fees & Adjustments' => 'FEES',
-	    'Food & Drink' => 'FOOD',
-	    'Gas' => 'GAS',
-	    'Gifts & Donations' => 'GIFTS',
-	    'Groceries' => 'GROCERIES',
-	    'Health & Wellness' => 'HEALTH',
-	    'Home' => 'HOME',
-	    'Personal' => 'PERSONAL',
-	    'Professional Services' => 'PRO',
-	    'Shopping' => 'SHOP'
-	);
+sub import_chase_credit ($$$$) {
+	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;
-		my ($trans_date, $post_date, $description, $category, 
+		s#(\d+)/(\d+)/(\d+)#$3$1$2#g; 
+		my $line = $_;
+		($trans_date, $post_date, $description, $category, 
 		    $type, $amount) = split /,/;
-		
-		$trans_date =~ s#(\d+)/(\d+)/(\d+)#$3$1$2#; 
-		$post_date =~ s#(\d+)/(\d+)/(\d+)#$3$1$2#; 
 		$description =~ s/#//g;
-		my $booking = $categories{$category} || 
-		    'CHASE_PAYMENT_ACCOUNT';
-		my $contra = $accounts{$booking};
 		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 = $accounts{CHASE_CREDIT_CARD};
-			$debit = $contra;
+			$credit = $csv_account;	
+			$debit = $account;
 		} else {
-			$debit = $accounts{CHASE_CREDIT_CARD};
-			$credit = $contra;
+			$debit = $csv_account;	
+			$credit = $account;
 		}
 		print "$post_date $booking $debit $credit $amount " .
 		    "$description\n";
 	}
 }
 
-# === MAIN PROGRAM =====================================================
+sub import_capital_one_credit ($$$$) {
+	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";
+	}
+}
 
-my $chase_credit_card_header = "Transaction Date,Post Date," .
-    "Description,Category,Type,Amount";
-my $firstLine;
+sub import_optum_hsa {
+        my ($bookings_ref, $accounts_ref, $compiled_ref,
+            $csv_account) = @_;
+        my ($date, $description, $amount,
+            $type);
+        chomp(my $header = <>);
+        while (<>) {
+                next if (/^$/);
+                chomp;
+                s#(\d+)-(\d+)-(\d+)#$1$2$3#;
+                my $line = $_;
+                ($date, $description, $amount,
+                    $type) = split /,/;
+                $description =~ s/#//g;
+                $amount =~ s/\$//;
+                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 "$date $booking $debit $credit $amount " .
+                    "$description\n";
+        }
+}
 
-chomp($firstLine = <>);
-if ($firstLine eq $chase_credit_card_header) {
-	import_chase_credit_card;
+sub usage () {
+	printf STDERR "usage: %s -I bankname csvfilename\n", $0;
+	exit 1;
+}
+
+# === MAIN PROGRAM =====================================================
+
+my ($csv_account, $fn, $in, $sub_name, %accounts, %bookings, @compiled);
+getopts 'I:' or usage;
+if ($opt_I) {
+	$sub_name = "import_" . $opt_I;
+	$fn = $sub_name . ".txt";
+	open $in, $fn or die "$fn: $!";
 } else {
-	print "Undefined bank header\n";
+	usage;
 }
+while (<$in>) {
+	chomp;
+	next if /^(?:#|$)/;
+	my $line = $_;
+	if (/^ACCOUNT\s+(\d+)$/) {
+		$csv_account and die "duplicate ACCOUNT line: $1";
+		$csv_account = $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;
+$csv_account or die "no ACCOUNT line in $fn";
+my $sub_ref = \&$sub_name;
+&$sub_ref(\%bookings, \%accounts, \@compiled, $csv_account);