def cmd_chart(config, opts):
chart = get_chart(config, opts)
for account in sorted(fullset(select_accounts(chart, opts))):
if opts['--verbose']:
line = [account.full_name()]
else:
line = ['\t' * (account.depth() - 1) + account.bare_name()]
if account.label:
line.append('[%s]' % (account.label,))
if account.atype and not (account.parent and account.parent.atype == account.atype):
line.append('=%s' % (abo.account.atype_to_tag[account.atype]))
for tag in account.tags:
line.append('=%s' % (tag,))
yield ' '.join(line)
def tree(self, title, accounts, amount_columns, depth=None):
if depth is not None:
accounts = set(a for a in accounts if a.depth() <= depth)
accounts = filter_display_accounts(accounts, self.opts) # TODO refactor filter_display_accounts() as method of Formatter
subaccounts = parentset(accounts)
all_accounts = [None] + list(sorted(accounts | subaccounts))
while all_accounts:
account = all_accounts.pop(0)
is_subaccount = account in subaccounts
if not self.opt_subtotals and self.opt_fullnames and is_subaccount:
continue
if account is None:
if self.opt_bare:
continue
text = 'TOTAL ' + title
elif self.opt_fullnames:
text = str(account)
else:
def has_sibling(account):
if not all_accounts:
return False
for a in all_accounts:
if a.depth() <= account.depth():
break
return a.depth() == account.depth()
graph = ['├─╴' if has_sibling(account) else '└─╴']
for a in account.all_parents():
graph.append('│ ' if has_sibling(a) else ' ')
if self.opt_bare:
graph.pop()
text = ''.join(reversed(graph)) + account.bare_name()
columns = []
for acol in amount_columns:
if (account is None or self.opt_subtotals or not is_subaccount) and account in acol:
columns.append(self.config.format_money(acol[account]))
else:
columns.append('')
yield self.fmt(text + ' ', ((' ' + c if c else '') for c in columns),
label= (account and account.label) or '',
text_fill= '' if is_subaccount else '.',
column_fill= '' if is_subaccount else '.',
strong= self.opt_subtotals and not is_subaccount)
