paypalscript.py

#!/usr/bin/env python3
import argparse
import csv
from datetime import datetime


class PayPalConverter:
    """
    Parses Paypal csv files into something GNUCash can read
    """
    data = []
    ignore_datetime = None
    mapping = {'Date': 'Date', 'Time': 'Time', 'Name': 'Name', 'Amount': 'Net', 'Description': 'Type'}
    paypal_header = 'Balance Impact'
    balance_type = 'Credit'

    def __init__(self, in_file, out_file, ignore_datetime):
        self.in_file = in_file
        self.out_file = out_file
        self.ignore_datetime = ignore_datetime

    def _strip_row(self, row):
        return {k: row[v] for k, v in self.mapping.items()}

    def read(self):
        try:
            with open(self.in_file, 'r', encoding='utf-8-sig') as f:
                self.data = [self._strip_row(row) for row in csv.DictReader(f) if row[self.paypal_header] ==
                             self.balance_type]
                self.data = filter(lambda d: self._convert_datetime(d['Date'], d['Time']) > self.ignore_datetime,
                                   self.data)
        except FileNotFoundError:
            exit(f"File { self.in_file } not found")

    def _convert_datetime(self, date, time):
        return datetime.strptime(f'{ date } { time }', "%m/%d/%Y %H:%M:%S")

    def write(self):
        with open(self.out_file, 'w', newline='') as f:
            writer = csv.DictWriter(f, fieldnames=self.mapping.keys())
            writer.writeheader()
            writer.writerows(self.data)


if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Takes a PayPal CSV export file and generates a GnuCash formatted CSV '
                                                 'import file.  Default input file name is Paypal Export.csv.  '
                                                 'Defaults output file name is Output.csv')
    parser.add_argument('-i', '--input', default='Paypal Export.csv', help='Input file name')
    parser.add_argument('-o', '--output', default='Output.csv', help='Output file name')
    parser.add_argument('-d', '--date', required=True, help='Ignore entries including and before this date ('
                                                            'mm/dd/yyyy hh/mm/ss)',
                        type=lambda s: datetime.strptime(s, "%m/%d/%Y %H:%M:%S"))
    args = parser.parse_args()

    print(f'Input File: { args.input }')
    print(f'Output File: { args.output }')
    print(f'Ignore On or Before Date: { args.date }')

    # Parse input file and create output file
    converter = PayPalConverter(in_file=args.input, out_file=args.output, ignore_datetime=args.date)
    converter.read()
    converter.write()