+ def balance_as_html(self, until_after=None):
+ bookings = self.bookings[:(until_after if until_after is None else int(until_after)+1)]
+ account_trunk = Account('', None)
+ accounts = {account_trunk.full_name: account_trunk}
+ for booking in bookings:
+ for full_account_name, moneys in booking.account_changes.items():
+ toks = full_account_name.split(':')
+ path = []
+ for tok in toks:
+ parent_name = ':'.join(path)
+ path += [tok]
+ account_name = ':'.join(path)
+ if not account_name in accounts.keys():
+ accounts[account_name] = Account(own_name=tok, parent=accounts[parent_name])
+ accounts[full_account_name].add_wealth(moneys)
+ class Node:
+ def __init__(self, indent, name, moneys):
+ self.indent = indent
+ self.name = name
+ self.moneys = moneys.money_dict
+ nodes = []
+ def walk_tree(nodes, indent, account):
+ nodes += [Node(indent, account.own_name, account.full_moneys)]
+ for child in account.children:
+ walk_tree(nodes, indent+1, child)
+ for acc in account_trunk.children:
+ walk_tree(nodes, 0, acc)
+ return j2env.get_template('balance.html').render(nodes=nodes)
+
+ # def balance_as_html(self, until=None):
+ # bookings = self.bookings[:until if until is None else int(until)]
+ # lines = []
+ # account_tree, account_sums = bookings_to_account_tree(bookings)
+ # def print_subtree(lines, indent, node, subtree, path):
+ # line = f"{indent}{node}"
+ # n_tabs = 5 - (len(line) // 8)
+ # line += n_tabs * "\t"
+ # if "€" in account_sums[path + node].keys():
+ # amount = account_sums[path + node]["€"]
+ # line += f"{amount:9.2f} €\t"
+ # else:
+ # line += f"\t\t"
+ # for currency, amount in account_sums[path + node].items():
+ # if currency != '€' and amount != 0:
+ # line += f"{amount:5.2f} {currency}\t"
+ # lines += [line]
+ # indent += " "
+ # for k, v in sorted(subtree.items()):
+ # print_subtree(lines, indent, k, v, path + node + ":")
+ # for k, v in sorted(account_tree.items()):
+ # print_subtree(lines, "", k, v, "")
+ # content = "\n".join(lines)
+ # return f"<pre>{content}</pre>"
+
+ def edit(self, index, sent=None, error_msg=None, edit_mode='table', copy=False):
+ accounts = set()
+ if sent or -1 == index:
+ content = sent if sent else ([] if 'textarea'==edit_mode else None)
+ else:
+ content = self.bookings[index]
+ if copy:
+ content.date = str(datetime.now())[:10]
+ if 'textarea' == edit_mode and content:
+ content = content.for_writing
+ else:
+ for booking in self.bookings:
+ for transfer_line in booking.transfer_lines:
+ accounts.add(transfer_line.account)
+ return j2env.get_template('edit.html').render(content=content, index=index, error_msg=error_msg, edit_mode=edit_mode, accounts=accounts, adding=(copy or -1 == index))
+
+ # def add_free(self, start=0, end=0, copy=False):
+ # tmpl = jinja2.Template(add_form_header + add_free_html + add_form_footer)
+ # lines = self.get_lines(start, end)
+ # if copy:
+ # start = end = 0
+ # return tmpl.render(action=self.prefix + '/add_free', start=start, end=end, lines=lines)
+
+ # def add_structured(self, start=0, end=0, copy=False, temp_lines=[], add_empty_line=None):
+ # tmpl = jinja2.Template(add_form_header + add_structured_html + add_form_footer)
+ # lines = temp_lines if len(''.join(temp_lines)) > 0 else self.get_lines(start, end)
+ # bookings, comments = parse_lines(lines, validate_bookings=False)
+ # if len(bookings) > 1:
+ # raise PlomException('can only structurally edit single Booking')
+ # if add_empty_line is not None:
+ # comments = comments[:add_empty_line+1] + [''] + comments[add_empty_line+1:]
+ # booking = bookings[0]
+ # booking.lines = booking.lines[:add_empty_line+1] + [''] + booking.lines[add_empty_line+1:]
+ # action = self.prefix + '/add_structured'
+ # datalist_sets = {'descriptions': set(), 'accounts': set(), 'currencies': set()}
+ # for b in self.bookings:
+ # datalist_sets['descriptions'].add(b.description)
+ # for account, moneys in b.account_changes.items():
+ # datalist_sets['accounts'].add(account)
+ # for currency in moneys.keys():
+ # datalist_sets['currencies'].add(currency)
+ # content = ''
+ # today = str(datetime.now())[:10]
+ # booking_lines = []
+ # if copy:
+ # start = end = 0
+ # desc = head_comment = ''
+ # if len(bookings) == 0:
+ # date=today
+ # else:
+ # booking = bookings[0]
+ # desc = booking.description
+ # date = today if copy else booking.date_string
+ # head_comment=comments[0]
+ # for i in range(1, len(comments)):
+ # account = amount = currency = ''
+ # if i < len(booking.lines) and booking.lines[i] != '':
+ # account = booking.lines[i][0]
+ # amount = booking.lines[i][1]
+ # currency = booking.lines[i][2]
+ # booking_lines += [{
+ # 'i': i,
+ # 'acc': account,
+ # 'amt': amount,
+ # 'curr': currency if currency else '€',
+ # 'comment': comments[i],
+ # 'comm_cols': len(comments[i])}]
+ # for i in range(len(comments), len(comments) + 8):
+ # booking_lines += [{'i': i, 'acc': '', 'amt': '', 'curr': '€', 'comment': ''}]
+ # content += tmpl.render(
+ # action=action,
+ # date=date,
+ # desc=desc,
+ # head_comment=head_comment,
+ # booking_lines=booking_lines,
+ # datalist_sets=datalist_sets,
+ # start=start,
+ # end=end)
+ # return content
+
+ def move_up(self, index):
+ return self.move(index, -1)
+
+ def move_down(self, index):
+ return self.move(index, +1)
+
+ def move(self, index, direction):
+ to_move = self.bookings[index]
+ swap_index = index + 1*(direction)
+ to_swap = self.bookings[swap_index]
+ self.bookings[index] = to_swap
+ self.bookings[index + 1*(direction)] = to_move
+ return swap_index
+
+ def write_db(self):
+ lines = []
+ for i, booking in enumerate(self.bookings):
+ if i > 0:
+ lines += ['']
+ lines += booking.for_writing
+ self.write_text_to_db('\n'.join(lines) + '\n')
+
+ # def move_up(self, start, end):
+ # prev_booking = None
+ # for redir_nth, b in enumerate(self.bookings):
+ # if b.start_line >= start:
+ # break
+ # prev_booking = b
+ # start_at = prev_booking.start_line
+ # self.make_move(start, end, start_at)
+ # return redir_nth - 1
+
+ # def move_down(self, start, end):
+ # next_booking = None
+ # for redir_nth, b in enumerate(self.bookings):
+ # if b.start_line > start:
+ # next_booking = b
+ # break
+ # # start_at = next_booking.start_line + len(next_booking.lines) - (end - start) + 1
+ # # self.make_move(start, end, start_at-1)
+ # start_at = next_booking.start_line + len(next_booking.lines) - (end - start)
+ # self.make_move(start, end, start_at)
+ # return redir_nth
+
+ # def make_move(self, start, end, start_at):
+ # # FIXME currently broken due to changed self.write_lines_in_total_lines_at, easy fix would be lines += [""] maybe?
+ # lines = self.get_lines(start, end)
+ # if start == 0:
+ # total_lines = self.text_lines[end+1:]
+ # lines = [''] + lines
+ # start_at += 1
+ # else:
+ # total_lines = self.text_lines[:start-1] + self.text_lines[end:] # -1 because we reduce the original position's two empty limit lines to one in-between line
+ # lines += ['']
+ # self.write_lines_in_total_lines_at(total_lines, start_at, lines)
+
+ # def booking_lines_from_postvars(self, postvars):
+ # add_empty_line = None
+ # date = postvars['date'][0]
+ # description = postvars['description'][0]
+ # start_comment = postvars['line_0_comment'][0]
+ # start_line = f'{date} {description}'
+ # if start_comment.rstrip() != '':
+ # start_line += f' ; {start_comment}'
+ # lines = [start_line]
+ # if 'line_0_add' in postvars.keys():
+ # add_empty_line = 0
+ # i = j = 1
+ # while f'line_{i}_comment' in postvars.keys():
+ # if f'line_{i}_delete' in postvars.keys():
+ # i += 1
+ # continue
+ # elif f'line_{i}_delete_after' in postvars.keys():
+ # break
+ # elif f'line_{i}_add' in postvars.keys():
+ # add_empty_line = j
+ # account = postvars[f'line_{i}_account'][0]
+ # amount = postvars[f'line_{i}_amount'][0]
+ # currency = postvars[f'line_{i}_currency'][0]
+ # comment = postvars[f'line_{i}_comment'][0]
+ # i += 1
+ # new_main = f' {account} {amount}'
+ # if '' == new_main.rstrip() == comment.rstrip(): # don't write empty lines, ignore currency if nothing else set
+ # continue
+ # if len(amount.rstrip()) > 0:
+ # new_main += f' {currency}'
+ # j += 1
+ # new_line = new_main
+ # if comment.rstrip() != '':
+ # new_line += f' ; {comment}'
+ # lines += [new_line]
+ # if 'add_sink' in postvars.keys():
+ # temp_lines = lines.copy() + ['_']
+ # try:
+ # temp_bookings, _ = parse_lines(temp_lines)
+ # for currency in temp_bookings[0].sink:
+ # amount = temp_bookings[0].sink[currency]
+ # # lines += [f'Assets {amount:.2f} {currency}']
+ # lines += [f'Assets {amount} {currency}']
+ # except PlomException:
+ # pass
+ # elif 'add_taxes' in postvars.keys():
+ # lines += self.add_taxes(lines, finish=False)
+ # elif 'add_taxes2' in postvars.keys():
+ # lines += self.add_taxes(lines, finish=True)
+ # elif 'replace' in postvars.keys():
+ # for i, line in enumerate(lines):
+ # lines[i] = line.replace(postvars['replace_from'][0], postvars['replace_to'][0])
+ # elif 'add_mirror' in postvars.keys():
+ # lines += self.add_mirror(lines)
+ # return lines, add_empty_line
+
+
+
+class LedgerHandler(PlomHandler):
+
+ def app_init(self, handler):
+ default_path = '/ledger'
+ handler.add_route('GET', default_path, self.forward_gets)
+ handler.add_route('POST', default_path, self.forward_posts)
+ return 'ledger', default_path