NPO-Accounting/conservancy_beancount
6
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

reconcile: Update helper to show N/A when no results, similar to Perl.

Browse source 
Also clarified that a full account name should be passed. This aligns with the
Perl behaviour.
This commit is contained in:
Ben Sturmfels 2022-01-28 10:08:01 +11:00
parent 970fea86fd
commit dbe8d02b78
Signed by: bsturmfels
GPG key ID: 023C05E2C9C068F0
1 changed files with 19 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
conservancy_beancount/reconcile/helper.py
View file

@ -33,18 +33,26 @@ def end_of_month(date):
def format_record_for_grep(row, homedir):
"""Return a line in a grep-style.
This is so the line can be fed into Emacs grep-mode for quickly jumping to
the relevant lines in the books.
"""
file = row[0].replace(homedir, '~')
return [f'{file}:{row[1]}:'] + row[2:]
def max_column_widths(rows):
"""Return the max width for each column in a table of data."""
maxes = [0] * len(rows[0])
for row in rows:
for i, val in enumerate(row):
length = len(str(val))
maxes[i] = max(maxes[i], length)
return maxes
if not rows:
return []
else:
maxes = [0] * len(rows[0])
for row in rows:
for i, val in enumerate(row):
length = len(str(val))
maxes[i] = max(maxes[i], length)
return maxes
def tabulate(rows, headers=None):
@ -68,10 +76,11 @@ def tabulate(rows, headers=None):
print('', file=output)
return output.getvalue().strip()
# Parse all the arguments
parser = argparse.ArgumentParser(description='Reconciliation helper')
parser.add_argument('--beancount-file', required=True)
parser.add_argument('--account', help='Account regexp', 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.')
@ -156,7 +165,9 @@ print(f"START RECONCILIATION FOR {account} ENDING {lastDateInPeriod} (previous e
entries, _, options = loader.load_file(beancount_file)
for desc, query in QUERIES.items():
rtypes, rrows = run_query(entries, options, query, numberify=True)
if desc.startswith('04'):
if not rrows:
print(f'{desc:<55} N/A')
elif desc.startswith('04'):
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]