Changeset - 71bf8137a56f
[Not reviewed]
0 1 0
Ben Sturmfels (bsturmfels) - 20 months ago 2023-01-11 08:36:43
ben@sturm.com.au
reconcile.helper: Add entry_point to avoid traceback
1 file changed with 13 insertions and 3 deletions:
0 comments (0 inline, 0 general)
conservancy_beancount/reconcile/helper.py
Show inline comments
...
 
@@ -93,75 +93,75 @@ def reconciliation_report(account, end_date, bank_account_balance, uncleared, pr
 
    w.writerow([f'  {account}'])
 
    w.writerow([])
 
    w.writerow(['DATE', 'CHECK NUM', 'PAYEE', 'AMOUNT'])
 
    w.writerow([])
 
    w.writerow([end_date, '', 'BANK ACCOUNT BALANCE', bank_account_balance])
 
    w.writerow([])
 
    for trans in uncleared:
 
        w.writerow(trans)
 
    w.writerow([])
 
    w.writerow([end_date, '', 'OUR ACCOUNT BALANCE', our_account_balance])
 
    if prev_uncleared:
 
        w.writerow([])
 
        w.writerow(['Items Still', 'Outstanding On', prev_end_date, 'Appeared By', end_date])
 
        w.writerow([])
 
        for trans in prev_uncleared:
 
            w.writerow(trans)
 
    return output.getvalue()
 

	
 

	
 
def reconciliation_report_path(account, end_date):
 
    *_, account_name = account.rpartition(':')
 
    return f'Financial/Controls/Reports-for-Treasurer/{end_date}_{account_name}_bank-reconciliation.csv'
 

	
 

	
 
def parse_args():
 
def parse_args(argv):
 
    parser = argparse.ArgumentParser(description='Reconciliation helper')
 
    parser.add_argument('--beancount-file', required=True)
 
    parser.add_argument('--account', help='Full account name, e.g. "Liabilities:CreditCard:AMEX"', required=True)
 
    parser.add_argument('--prev-end-date', type=datetime.date.fromisoformat)
 
    parser.add_argument('--cur-end-date', type=datetime.date.fromisoformat)
 
    parser.add_argument('--month', help='YYYY-MM of ending month. Use with --period.')
 
    parser.add_argument('--period', help='Months in the past to consider. Use with --month.', type=int, choices=[1, 3, 12])
 
    parser.add_argument('--statement-match')
 
    parser.add_argument('--cost-function', default='COST')
 
    parser.add_argument('--grep-output-filename')
 
    # parser.add_argument('--report-group-regex')
 
    args = parser.parse_args()
 
    args = parser.parse_args(args=argv[1:])
 
    if args.month or args.period:
 
        if not (args.month and args.period):
 
            parser.error('--month and --period must be used together')
 
    else:
 
        if not (args.cur_end_date and args.prev_end_date):
 
            parser.error(' --prev-end-date and --cur-end-date must be used together')
 
    return args
 

	
 

	
 
def beancount_file_exists(path):
 
    return os.path.isfile(path)
 

	
 

	
 
args = parse_args()
 
def main(args):
 
    if not beancount_file_exists(args.beancount_file):
 
        sys.exit(f'Beancount file does not exist: {args.beancount_file}')
 
    if args.month or args.period:
 
        parsed_date = datetime.datetime.strptime(args.month, '%Y-%m').date()
 
        preDate = end_of_month(parsed_date - relativedelta(months=args.period)).isoformat()
 
        lastDateInPeriod = end_of_month(parsed_date).isoformat()
 
        month = args.month
 
    else:
 
        preDate = args.prev_end_date
 
        lastDateInPeriod = args.cur_end_date.isoformat()
 
        month = args.cur_end_date.strftime('%Y-%m')
 
    grep_output_file: typing.IO
 
    if args.grep_output_filename:
 
        grep_output_file = open(args.grep_output_filename, 'w')
 
    else:
 
        grep_output_file = tempfile.NamedTemporaryFile(prefix='bc-reconcile-grep-output_', mode='w', delete=False)
 
    beancount_file = args.beancount_file
 
    account = args.account
 
    cost_function = args.cost_function
 
    statement_match = args.statement_match if args.statement_match else month
 

	
 
    QUERIES = {
 
        f"00: CLEARED BAL ENDING DAY BEFORE {preDate}":
 
        # $CONLEDGER -V -C -e "$preDate" bal "/$acct/"
...
 
@@ -224,24 +224,34 @@ for desc, query in QUERIES.items():
 
            homedir = os.getenv('HOME', '')
 
            print(f'{desc}\n   See {grep_output_file.name}')
 
            grep_rows = [format_record_for_grep(row, homedir) for row in rrows]
 
            print(tabulate(grep_rows), file=grep_output_file)
 
        elif len(rrows) == 1 and isinstance(rrows[0][0], decimal.Decimal):
 
            result = rrows[0][0]
 
            print(f'{desc:<55} {result:11,.2f}')
 
            if desc.startswith('02'):
 
                all_trans_balance = result
 
            if desc.startswith('05'):
 
                cleared_balance = result
 
        else:
 
            headers = [c[0].capitalize() for c in rtypes]
 
            if desc.startswith('03'):
 
                uncleared_rows = rrows
 
            print(desc)
 
            print(textwrap.indent(tabulate(rrows, headers=headers), '    '))
 

	
 
    uncleared = [(r[0], r[2], r[4] or r[3], r[1]) for r in uncleared_rows]
 
    report_path = os.path.join(os.getenv('CONSERVANCY_REPOSITORY', ''), reconciliation_report_path(account, lastDateInPeriod))
 
    # TODO: Make the directory if it doesn't exist.
 
    with open(report_path, 'w') as f:
 
        f.write(reconciliation_report(account, lastDateInPeriod, cleared_balance, uncleared, '1900-01-01', all_trans_balance, []))
 
    print(f'Wrote reconciliation report: {report_path}.')
 

	
 

	
 
if __name__ == '__main__':
 
    args = parse_args(sys.argv)
 
    main(args)
 

	
 

	
 
def entry_point():
 
    args = parse_args(sys.argv)
 
    main(args)
0 comments (0 inline, 0 general)