'line': line,

    }

def format_record(record: Dict) -> str:

    return f"{record['date'].isoformat()}: {record['amount']:12,.2f} {record['payee'][:20]:<20}"

def sort_records(records: List) -> List:

    return sorted(records, key=lambda x: (x['date'], x['amount']))

def records_match(r1: Dict, r2: Dict) -> Tuple[bool, str]:

    """Do these records represent the same transaction?"""

    date_matches_exactly = r1['date'] == r2['date']

    date_matches_loosly = r1['date'] >= r2['date'] - datetime.timedelta(days=3) and r1['date'] <= r2['date'] + datetime.timedelta(days=3)

    amount_matches = r1['amount'] == r2['amount']

    payee_match_quality = fuzz.token_set_ratio(r1['payee'], r2['payee'])

    payee_matches = payee_match_quality > 50

    if date_matches_exactly and amount_matches and payee_matches:

        return True, 'Matched'

    elif date_matches_loosly and amount_matches and payee_matches:

        return True, 'Matched +/- 3 days'

    elif date_matches_exactly and amount_matches:

    elif date_matches_loosly and amount_matches:

    else:

        return False, ''

parser = argparse.ArgumentParser(description='Reconciliation helper')

parser.add_argument('--beancount-file', required=True)

parser.add_argument('--csv-statement', required=True)

parser.add_argument('--account', required=True, help='eg. Liabilities:CreditCard:AMEX')

parser.add_argument('--grep-output-filename')

# parser.add_argument('--report-group-regex')

parser.add_argument('--show-reconciled-matches', action='store_true')

args = parser.parse_args()

# TODO: Should put in a sanity check to make sure the statement you're feeding

# in matches the account you've provided.

if 'AMEX' in args.account:

    standardize_statement_record = standardize_amex_record

else:

    standardize_statement_record = standardize_fr_record

with open(args.csv_statement) as f:

    reader = csv.DictReader(f)

    statement_trans = sort_records([standardize_statement_record(row, reader.line_num) for row in reader])

# the books. If found, the books transaction is marked off so that it can only

# be matched once. Some transactions will be matched, some will be on the

# statement but not the books and some on the books but not the statement.

#

# Doesn't currently find exact matches when there are a bunch of transactions

# for the same amount on the same day. Probably ought to do a first pass through

# the books transactions to find an closely matching payee, then do another pass

# disregarding payee.

for r1 in statement_trans:

    for r2 in books_trans:

        match, note = records_match(r1, r2)

        if match:

            if not r2['statement'] or args.show_reconciled_matches:

                    matches.append([r2['date'], f'{format_record(r1)}  →  {format_record(r2)}  ✓ {note}'])

            if not r2['statement']:

                metadata_to_apply.append((r2['filename'], r2['line'], f'    bank-statement: "{os.path.basename(args.csv_statement)}:{r2["line"]}"\n'))

            books_trans.remove(r2)

            break

    else:

        matches.append([r1['date'], f'{format_record(r1)}  →  {" ":^45}  ✗ Not in books ({os.path.basename(args.csv_statement)}:{r1["line"]})'])

for r2 in books_trans:

    matches.append([r2['date'], f'{" ":^45}  →  {format_record(r2)}  ✗ Not on statement ({os.path.basename(r2["filename"])}:{r2["line"]})'])

print('-' * 155)

print('-' * 155)

for _, output in sorted(matches):

    print(output)

print('-' * 155)

# Write statement metadata back to books

if metadata_to_apply:

    print('Mark matched transactions as reconciled in the books? (y/N) ', end='')

    if input().lower() == 'y':

        files = {}

        # Query results aren't necessarily sequential in a file, so need to sort

        # so that our line number offsets work.

        for filename, line, metadata in sorted(metadata_to_apply):

            if filename not in files:

                with open(filename, 'r') as f:

                    # print(f'Opening {filename}.')

                    files[filename] = [0, f.readlines()]  # Offset and contents

            files[filename][1].insert(line + files[filename][0], metadata)

            files[filename][0] += 1

            # print(f'File {filename} offset {files[filename][0]}')

        for filename in files:

            with open(filename, 'w') as f:

                f.writelines(files[filename][1])

                print(f'Wrote {filename}.')