X-Git-Url: https://plomlompom.com/repos/%7B%7Bdb.prefix%7D%7D/task?a=blobdiff_plain;f=ledger.py;h=d8b504a74398ce18315a4430b345ac7a1eaf417a;hb=11df024d5f24e8d7916f57c907975ebe27db617f;hp=334318ff301eb186efa6684809c9ebac1807d177;hpb=c3b5860ae1ca76c2a1dc3b61d3c1242d581424f0;p=misc diff --git a/ledger.py b/ledger.py index 334318f..d8b504a 100755 --- a/ledger.py +++ b/ledger.py @@ -1,316 +1,615 @@ -from http.server import BaseHTTPRequestHandler, HTTPServer -import sys -hostName = "localhost" -serverPort = 8082 +import os +import decimal +from datetime import datetime, timedelta +from urllib.parse import parse_qs, urlparse +from jinja2 import Environment as JinjaEnv, FileSystemLoader as JinjaFSLoader +from plomlib import PlomDB, PlomException, run_server, PlomHandler +from os.path import split as path_split +db_path = '/home/plom/org/ledger2024.dat' +server_port = 8082 +j2env = JinjaEnv(loader=JinjaFSLoader('ledger_templates')) -class HandledException(Exception): - pass +class EditableException(PlomException): + def __init__(self, booking_index, *args, **kwargs): + self.booking_index = booking_index + super().__init__(*args, **kwargs) + +class LedgerTextLine: -def handled_error_exit(msg): - print(f"ERROR: {msg}") - sys.exit(1) + def __init__(self, text_line): + self.text_line = text_line + self.comment = '' + split_by_comment = text_line.rstrip().split(sep=';', maxsplit=1) + self.non_comment = split_by_comment[0].rstrip() + self.empty = len(split_by_comment) == 1 and len(self.non_comment) == 0 + if self.empty: + return + if len(split_by_comment) == 2: + self.comment = split_by_comment[1].lstrip() -def apply_booking_to_account_balances(account_sums, account, currency, amount): - if not account in account_sums: - account_sums[account] = {currency: amount} - elif not currency in account_sums[account].keys(): - account_sums[account][currency] = amount - else: - account_sums[account][currency] += amount +class Wealth: + def __init__(self): + self.money_dict = {} -class Booking: + def __iadd__(self, moneys): + money_dict = moneys + if type(moneys) == Wealth: + moneys = moneys.money_dict + for currency, amount in moneys.items(): + if not currency in self.money_dict.keys(): + self.money_dict[currency] = 0 + self.money_dict[currency] += amount + return self - def __init__(self, date_string, description, booking_lines, start_line): - self.date_string = date_string - self.description = description - self.lines = booking_lines - self.start_line = start_line - self.validate_booking_lines() - self.account_changes = self.parse_booking_lines_to_account_changes() - - def validate_booking_lines(self): - prefix = f"booking at line {self.start_line}" - sums = {} - empty_values = 0 - for line in self.lines: - if line == '': - continue - _, amount, currency = line - if amount is None: - if empty_values > 0: - raise HandledException(f"{prefix} relates more than one empty value of same currency {currency}") - empty_values += 1 + @property + def sink_empty(self): + return len(self.as_sink) == 0 + + @property + def as_sink(self): + sink = {} + for currency, amount in self.money_dict.items(): + if 0 == amount: continue - if currency not in sums: - sums[currency] = 0 - sums[currency] += amount - if empty_values == 0: - for k, v in sums.items(): - if v != 0: - raise HandledException(f"{prefix} does not sum up to zero") + sink[currency] = -amount + return sink + + +class Account: + + def __init__(self, own_name, parent): + self.own_name = own_name + self.parent = parent + if self.parent: + self.parent.children += [self] + self.own_moneys = Wealth() + self.children = [] + + def add_wealth(self, moneys): + self.own_moneys += moneys + + @property + def full_name(self): + if self.parent and len(self.parent.own_name) > 0: + return f'{self.parent.full_name}:{self.own_name}' else: - sinkable = False - for k, v in sums.items(): - if v != 0: - sinkable = True - if not sinkable: - raise HandledException(f"{prefix} has empty value that cannot be filled") - - def parse_booking_lines_to_account_changes(self): - account_changes = {} - debt = {} - sink_account = None - for line in self.lines: - if line == '': - continue - account, amount, currency = line - if amount is None: - sink_account = account + return self.own_name + + @property + def full_moneys(self): + full_moneys = Wealth() + full_moneys += self.own_moneys + for child in self.children: + full_moneys += child.full_moneys + return full_moneys + + +def parse_lines_to_bookings(lines, ignore_editable_exceptions=False): + lines = [LedgerTextLine(line) for line in lines] + lines += [LedgerTextLine('')] # to simulate ending of last booking + bookings = [] + inside_booking = False + booking_lines = [] + booking_start_i = 0 + last_date = '' + for i, line in enumerate(lines): + intro = f'file line {i}' + if line.empty: + if inside_booking: + booking = Booking(lines=booking_lines, starts_at=booking_start_i) + if last_date > booking.date and not ignore_editable_exceptions: + raise EditableException(len(bookings), f'{intro}: out-of-order date (follows {last_date})') + last_date = booking.date + bookings += [booking] + inside_booking =False + booking_lines = [] + else: + if not inside_booking: + booking_start_i = i + inside_booking = True + booking_lines += [line] + if inside_booking: + raise PlomException(f'{intro}: last booking unfinished') + return bookings + + +class TransferLine: + + def __init__(self, line=None, account='', amount=None, currency='', comment='', validate=True): + self.account = account + self.amount = amount + self.currency = currency + self.comment = comment + if line: + if line.empty: + raise PlomException('line empty') + self.comment = line.comment + toks = line.non_comment.split() + if (len(toks) not in {1, 3}): + if validate: + raise PlomException(f'number of non-comment tokens not 1 or 3') + elif len(toks) == 2: + toks += [''] + else: + toks = 3*[''] + self.account = toks[0] + if len(toks) != 1: + try: + self.amount = decimal.Decimal(toks[1]) + except decimal.InvalidOperation: + if validate: + raise PlomException(f'invalid token for Decimal: {toks[1]}') + else: + self.comment = f'unparsed: {toks[1]}; {self.comment}' + self.currency = toks[2] + + @property + def amount_fmt(self): + if self.amount is None: + return '' + elif self.amount.as_tuple().exponent < -2: + return f'{self.amount:.1f}â¦' + else: + return f'{self.amount:.2f}' + + @property + def for_writing(self): + line = f' {self.account}' + if self.amount is not None: + line += f' {self.amount} {self.currency}' + if self.comment != '': + line += f' ; {self.comment}' + return line + + @property + def comment_cols(self): + return max(20, len(self.comment)) + + +class Booking: + + def __init__(self, lines=None, starts_at='?', date='', description='', top_comment='', transfer_lines=None, validate=True): + self.validate = validate + self.starts_at = starts_at + self.intro = f'booking starting at line {self.starts_at}' + self.clean() + if lines: + self.lines = lines + self.parse_lines() + else: + self.date = date + self.description = description + self.top_comment = top_comment + self.validate_head() + self.transfer_lines = transfer_lines if transfer_lines else [] + if self.validate and len(self.transfer_lines) < 2: + raise PlomException(f'{self.intro}: too few transfer lines') + self.calculate_account_changes() + self.lines = [LedgerTextLine(l) for l in self.for_writing] + + @classmethod + def from_postvars(cls, postvars, starts_at='?', validate=True): + date = postvars['date'][0] + description = postvars['description'][0] + top_comment = postvars['top_comment'][0] + transfer_lines = [] + for i, account in enumerate(postvars['account']): + if len(account) == 0: continue - apply_booking_to_account_balances(account_changes, account, currency, amount) - if currency not in debt: - debt[currency] = amount + amount = None + if len(postvars['amount'][i]) > 0: + amount = decimal.Decimal(postvars['amount'][i]) + transfer_lines += [TransferLine(None, account, amount, postvars['currency'][i], postvars['comment'][i])] + return cls(None, starts_at, date, description, top_comment, transfer_lines, validate=validate) + + def clean(self): + self.transfer_lines = [] + self.account_changes = {} + + def parse_lines(self): + if len(self.lines) < 3 and self.validate: + raise PlomException(f'{self.intro}: ends with less than 3 lines:' + str(self.lines)) + top_line = self.lines[0] + if top_line.empty and self.validate: + raise PlomException('{self.intro}: headline empty') + self.top_comment = top_line.comment + toks = top_line.non_comment.split(maxsplit=1) + if len(toks) < 2: + if self.validate: + raise PlomException(f'{self.intro}: headline missing elements: {non_comment}') + elif 0 == len(toks): + toks = 2*[''] + else: + toks += [''] + self.date = toks[0] + self.description = toks[1] + self.validate_head() + for i, line in enumerate(self.lines[1:]): + try: + self.transfer_lines += [TransferLine(line, validate=self.validate)] + except PlomException as e: + raise PlomException(f'{self.intro}, transfer line {i}: {e}') + self.calculate_account_changes() + + def calculate_account_changes(self): + sink_account = None + money_changes = Wealth() + for i, transfer_line in enumerate(self.transfer_lines): + intro = f'{self.intro}, transfer line {i}' + if transfer_line.amount is None: + if sink_account is not None and self.validate: + raise PlomException(f'{intro}: second sink found (only one allowed)') + sink_account = transfer_line.account else: - debt[currency] += amount - if sink_account: - for currency, amount in debt.items(): - apply_booking_to_account_balances(account_changes, sink_account, currency, -amount) - return account_changes + if not transfer_line.account in self.account_changes.keys(): + self.account_changes[transfer_line.account] = Wealth() + money = {transfer_line.currency: transfer_line.amount} + self.account_changes[transfer_line.account] += money + money_changes += money + if sink_account is None and (not money_changes.sink_empty) and self.validate: + raise PlomException(f'{intro}: does not balance (undeclared non-empty sink)') + if sink_account is not None: + if not sink_account in self.account_changes.keys(): + self.account_changes[sink_account] = Wealth() + self.account_changes[sink_account] += money_changes.as_sink + + @property + def for_writing(self): + lines = [f'{self.date} {self.description}'] + if self.top_comment is not None and self.top_comment.rstrip() != '': + lines[0] += f' ; {self.top_comment}' + for line in self.transfer_lines: + lines += [line.for_writing] + return lines + + @property + def comment_cols(self): + return max(20, len(self.top_comment)) + + def validate_head(self): + if not self.validate: + return + if len(self.date) == 0: + raise PlomException(f'{self.intro}: missing date') + if len(self.description) == 0: + raise PlomException(f'{self.intro}: missing description') + try: + datetime.strptime(self.date, '%Y-%m-%d') + except ValueError: + raise PlomException(f'{self.intro}: bad headline date format: {self.date}') + + def fill_sink(self): + replacement_lines = [] + for i, line in enumerate(self.transfer_lines): + if line.amount is None: + for currency, amount in self.account_changes[line.account].money_dict.items(): + replacement_lines += [TransferLine(None, f'{line.account}', amount, currency).for_writing] + break + lines = self.lines[:i+1] + [LedgerTextLine(l) for l in replacement_lines] + self.lines[i+2:] + self.clean() + self.lines = lines + self.parse_lines() + def mirror(self): + new_transfer_lines = [] + for transfer_line in self.transfer_lines: + uncommented_source = LedgerTextLine(transfer_line.for_writing).non_comment + comment = f're: {uncommented_source.lstrip()}' + new_account = '?' + new_transfer_lines += [TransferLine(None, new_account, -transfer_line.amount, transfer_line.currency, comment)] + for transfer_line in new_transfer_lines: + self.lines += [LedgerTextLine(transfer_line.for_writing)] + self.clean() + self.parse_lines() + def replace(self, replace_from, replace_to): + lines = [] + for l in self.for_writing: + lines += [l.replace(replace_from, replace_to)] + self.lines = [LedgerTextLine(l) for l in lines] + self.clean() + self.parse_lines() -class Database: + def add_taxes(self, db): + acc_kk_add = 'Reserves:KrankenkassenBeitragsWachstum' + acc_kk_minimum = 'Reserves:Monthly:KrankenkassenDefaultBeitrag' + acc_kk = 'Expenses:KrankenKasse' + acc_ESt = 'Reserves:EinkommensSteuer' + acc_assets = 'Assets' + acc_neuanfangspuffer_expenses = 'Reserves:NeuAnfangsPuffer:Ausgaben' + months_passed = datetime.strptime(self.date, '%Y-%m-%d').month - 1 + past_kk_expenses = 0 + past_kk_add = 0 + past_neuanfangspuffer_expenses = 0 + for b in db.bookings: + if b.date == self.date: + break + if acc_neuanfangspuffer_expenses in b.account_changes.keys(): + past_neuanfangspuffer_expenses -= b.account_changes[acc_neuanfangspuffer_expenses].money_dict['â¬'] + if acc_kk_add in b.account_changes.keys(): + past_kk_add += b.account_changes[acc_kk_add].money_dict['â¬'] + if acc_kk_minimum in b.account_changes.keys(): + past_kk_expenses += b.account_changes[acc_kk_minimum].money_dict['â¬'] - def __init__(self, load_from_file=True): - import os - db_name = "_ledger" - self.db_file = db_name + ".json" - self.lock_file = db_name+ ".lock" + needed_netto = -self.account_changes['Assets'].money_dict['â¬'] + past_taxed_needs_before_kk = past_neuanfangspuffer_expenses - past_kk_expenses + ESt_this_month = 0 + E0 = decimal.Decimal(11604) + E1 = decimal.Decimal(17006) + E2 = decimal.Decimal(66761) + E3 = decimal.Decimal(277826) + taxed_income_before_kk = needed_netto + too_low = 0 + too_high = 2 * needed_netto + while True: + estimate_for_remaining_year = (12 - months_passed) * taxed_income_before_kk + zvE = past_taxed_needs_before_kk + estimate_for_remaining_year + if zvE < E0: + ESt_year = decimal.Decimal(0) + elif zvE < E1: + y = (zvE - E0)/10000 + ESt_year = (decimal.Decimal(922.98) * y + 1400) * y + elif zvE < E2: + y = (zvE - E1)/10000 + ESt_year = (decimal.Decimal(181.19) * y + 2397) * y + decimal.Decimal(1025.38) + elif zvE < E3: + ESt_year = decimal.Decimal(0.42) * zvE - 10602.13 + else: + ESt_year = decimal.Decimal(0.45) * zvE - 18936.88 + ESt_this_month = ESt_year / 12 + taxed_income_minus_ESt = taxed_income_before_kk - ESt_this_month + if abs(taxed_income_minus_ESt - needed_netto) < 0.001: + break + elif taxed_income_minus_ESt < needed_netto: + too_low = taxed_income_before_kk + elif taxed_income_minus_ESt > needed_netto: + too_high = taxed_income_before_kk + taxed_income_before_kk = too_low + (too_high - too_low)/2 + ESt_this_month = ESt_this_month.quantize(decimal.Decimal('0.00')) + comment = f'estimated zvE: {past_taxed_needs_before_kk}⬠+ {estimate_for_remaining_year:.2f}⬠= {zvE:.2f}⬠â year ESt: {ESt_year:.2f} â needed taxed income before Krankenkasse: {taxed_income_before_kk:.2f}â¬' + self.transfer_lines += [TransferLine(None, acc_ESt, ESt_this_month, 'â¬', comment)] + + kk_factor = decimal.Decimal(1.197) + kk_minimum_income = 1178.33 + kk_minimum_tax = decimal.Decimal(232.13).quantize(decimal.Decimal('0.00')) + if self.date < '2024-02-01': + kk_minimum_income = 1131.67 + kk_minimum_tax = decimal.Decimal(222.94).quantize(decimal.Decimal('0.00')) + comment = f'assumed minimum income of {kk_minimum_income:.2f}⬠* {kk_factor:.3f}' + self.transfer_lines += [TransferLine(None, acc_kk_minimum, kk_minimum_tax, 'â¬', comment)] + kk_add = taxed_income_before_kk * kk_factor - taxed_income_before_kk - kk_minimum_tax + kk_add = decimal.Decimal(kk_add).quantize(decimal.Decimal('0.00')) + if past_kk_add + kk_add < 0: # *if* kk_add would actually kill all earlier kk_add ⦠+ kk_add = - past_kk_add # ⦠shrink it so it won't push the kk_add total below 0 + comment = f'local negative as large as possible without moving {acc_kk_add} below zero' + else: + comment = f'({taxed_income_before_kk:.2f}⬠* {kk_factor:.3f}) - {taxed_income_before_kk:.2f} - {kk_minimum_tax}' + self.transfer_lines += [TransferLine(None, acc_kk_add, kk_add, 'â¬', comment)] + + diff_through_taxes_and_kk = ESt_this_month + kk_minimum_tax + kk_add + comment = f'{ESt_this_month} + {kk_minimum_tax} + {kk_add}' + self.transfer_lines += [TransferLine(None, acc_assets, -diff_through_taxes_and_kk, 'â¬', comment)] + final_loss = diff_through_taxes_and_kk + needed_netto + comment = f'{needed_netto} + {diff_through_taxes_and_kk}' + self.transfer_lines += [TransferLine(None, acc_assets, final_loss, 'â¬', comment)] + self.transfer_lines += [TransferLine(None, acc_neuanfangspuffer_expenses, -final_loss, 'â¬')] + + +class LedgerDB(PlomDB): + + def __init__(self, prefix, ignore_editable_exceptions=False): + self.prefix = prefix self.bookings = [] self.comments = [] - if load_from_file and os.path.exists(self.db_file): - with open(self.db_file, "r") as f: - self.parse_lines(f.readlines()) - - def parse_lines(self, lines): - import datetime - import decimal - inside_booking = False - date_string, description = None, None - booking_lines = [] - start_line = 0 - for i, line in enumerate(lines): - prefix = f"line {i}" - # we start with the case of an utterly empty line - self.comments += [""] - stripped_line = line.rstrip() - if stripped_line == '': - if inside_booking: - # assume we finished a booking, finalize, and commit to DB - if len(booking_lines) < 2: - raise HandledException(f"{prefix} booking ends to early") - booking = Booking(date_string, description, booking_lines, start_line) - self.bookings += [booking] - # expect new booking to follow so re-zeroall booking data - inside_booking = False - date_string, description = None, None - booking_lines = [] - continue - # if non-empty line, first get comment if any, and commit to DB - split_by_comment = stripped_line.split(sep=";", maxsplit=1) - if len(split_by_comment) == 2: - self.comments[i] = split_by_comment[1] - # if pre-comment empty: if inside_booking, this must be a comment-only line so we keep it for later ledger-output to capture those comments; otherwise, no more to process for this line - non_comment = split_by_comment[0].rstrip() - if non_comment.rstrip() == '': - if inside_booking: - booking_lines += [''] - continue - # if we're starting a booking, parse by first-line pattern - if not inside_booking: - start_line = i - toks = non_comment.split(maxsplit=1) - date_string = toks[0] - try: - datetime.datetime.strptime(date_string, '%Y-%m-%d') - except ValueError: - raise HandledException(f"{prefix} bad date string: {date_string}") - try: - description = toks[1] - except IndexError: - raise HandledException(f"{prefix} bad description: {description}") - inside_booking = True - continue - # otherwise, read as transfer data - toks = non_comment.split() # ignore specification's allowance of single spaces in names - if len(toks) > 3: - raise HandledException(f"{prefix} too many booking line tokens: {toks}") - amount, currency = None, None - account_name = toks[0] - if account_name[0] == '[' and account_name[-1] == ']': - # ignore specification's differentiation of "virtual" accounts - account_name = account_name[1:-1] - decimal_chars = ".-0123456789" - if len(toks) == 3: - i_currency = 1 - try: - amount = decimal.Decimal(toks[1]) - i_currency = 2 - except decimal.InvalidOperation: - try: - amount = decimal.Decimal(toks[2]) - except decimal.InvalidOperation: - raise HandledException(f"{prefix} no decimal number in: {toks[1:]}") - currency = toks[i_currency] - if currency[0] in decimal_chars: - raise HandledException(f"{prefix} currency starts with int, dot, or minus: {currency}") - elif len(toks) == 2: - value = toks[1] - inside_amount = False - inside_currency = False - amount_string = "" - currency = "" - dots_counted = 0 - for i, c in enumerate(value): - if i == 0: - if c in decimal_chars: - inside_amount = True - else: - inside_currency = True - if inside_currency: - if c in decimal_chars and len(amount_string) == 0: - inside_currency = False - inside_amount = True - else: - currency += c - continue - if inside_amount: - if c not in decimal_chars: - if len(currency) > 0: - raise HandledException(f"{prefix} amount has non-decimal chars: {value}") - inside_currency = True - inside_amount = False - currency += c - continue - if c == '-' and len(amount_string) > 1: - raise HandledException(f"{prefix} amount has non-start '-': {value}") - if c == '.': - if dots_counted > 1: - raise HandledException(f"{prefix} amount has multiple dots: {value}") - dots_counted += 1 - amount_string += c - if len(amount_string) == 0: - raise HandledException(f"{prefix} amount missing: {value}") - if len(currency) == 0: - raise HandledException(f"{prefix} currency missing: {value}") - amount = decimal.Decimal(amount_string) - booking_lines += [(account_name, amount, currency)] + self.text_lines = [] + super().__init__(db_path) + self.bookings = parse_lines_to_bookings(self.text_lines, ignore_editable_exceptions) + + def read_db_file(self, f): + self.text_lines += f.readlines() + + def insert_booking_at_date(self, booking): + place_at = 0 + if len(self.bookings) > 0: + for i, iterated_booking in enumerate(self.bookings): + if booking.date < iterated_booking.date: + break + elif booking.date == iterated_booking.date: + place_at = i + break + else: + place_at = i + 1 + self.bookings.insert(place_at, booking) def ledger_as_html(self): + for index, booking in enumerate(self.bookings): + booking.can_up = index > 0 and self.bookings[index - 1].date == booking.date + booking.can_down = index < len(self.bookings) - 1 and self.bookings[index + 1].date == booking.date + return j2env.get_template('ledger.html').render(bookings=self.bookings) + + 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 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] + date_today = str(datetime.now())[:10] + if copy: + content.date = date_today + index = -1 + elif -1 == index and (content is None or [] == content): + content = Booking(date=date_today, validate=False) + 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=-1 == index) + + 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 comment in self.comments: - lines += [f"; {comment}" if comment != '' else ''] - for booking in self.bookings: - i = booking.start_line - suffix = f" {lines[i]}" if len(lines[i]) > 0 else "" - lines[i] = f"{booking.date_string} {booking.description}{suffix}" - for booking_line in booking.lines: - i += 1 - if booking_line == '': - continue - suffix = f" {lines[i]}" if len(lines[i]) > 0 else "" - value = f" {booking_line[1]} {booking_line[2]}" if booking_line[1] else "" - lines[i] = f"{booking_line[0]}{value}{suffix}" - content = "\n".join(lines) - return f"
{content}" - - def balance_as_html(self): - account_sums = {} - for booking in self.bookings: - for account, changes in booking.account_changes.items(): - for currency, amount in changes.items(): - apply_booking_to_account_balances(account_sums, account, currency, amount) - account_tree = {} - def collect_branches(account_name, path): - node = account_tree - path_copy = path[:] - while len(path_copy) > 0: - step = path_copy.pop(0) - node = node[step] - toks = account_name.split(":", maxsplit=1) - parent = toks[0] - if parent in node.keys(): - child = node[parent] - else: - child = {} - node[parent] = child - if len(toks) == 2: - k, v = collect_branches(toks[1], path + [parent]) - if k not in child.keys(): - child[k] = v + for i, booking in enumerate(self.bookings): + if i > 0: + lines += [''] + lines += booking.for_writing + self.write_text_to_db('\n'.join(lines) + '\n') + + +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 + + def do_POST(self): + self.try_do(self.forward_posts) + + def forward_posts(self): + prefix = self.apps['ledger'] if hasattr(self, 'apps') else '' + length = int(self.headers['content-length']) + postvars = parse_qs(self.rfile.read(length).decode(), keep_blank_values=1) + db = LedgerDB(prefix, ignore_editable_exceptions=True) + index = 0 + parsed_url = urlparse(self.path) + site = path_split(parsed_url.path)[1] + for string in {'update', 'add', 'check', 'mirror', 'fill_sink', 'textarea', 'table', 'move_up', 'move_down', 'add_taxes', 'replace'}: + if string in postvars.keys(): + submit_button = string + break + if 'ledger' == site and submit_button in {'move_up', 'move_down'}: + mover = getattr(db, submit_button) + index = mover(int(postvars[submit_button][0])) + elif 'edit' == site: + index = int(postvars['index'][0]) + edit_mode = postvars['edit_mode'][0] + validate = submit_button in {'update', 'add', 'copy', 'check'} + starts_at = '?' if index == -1 else db.bookings[index].starts_at + delete = False + if 'textarea' == edit_mode: + lines = [LedgerTextLine(line) for line in postvars['booking'][0].rstrip().split('\n')] + if len(lines) == 1 and lines[0].text_line == 'delete': + delete = True else: - child[k].update(v) - return parent, child - for account_name in sorted(account_sums.keys()): - k, v = collect_branches(account_name, []) - if k not in account_tree.keys(): - account_tree[k] = v + booking = Booking(lines, starts_at, validate=validate) else: - account_tree[k].update(v) - def collect_totals(parent_path, tree_node): - for k, v in tree_node.items(): - child_path = parent_path + ":" + k - for currency, amount in collect_totals(child_path, v).items(): - apply_booking_to_account_balances(account_sums, parent_path, currency, amount) - return account_sums[parent_path] - for account_name in account_tree.keys(): - account_sums[account_name] = collect_totals(account_name, account_tree[account_name]) - lines = [] - 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"
{content}" - - -class MyServer(BaseHTTPRequestHandler): + booking = Booking.from_postvars(postvars, starts_at, validate) + if submit_button in {'update', 'add'}: + if submit_button == 'update': + if 'textarea' == edit_mode and delete: + del db.bookings[index] + # if not creating new Booking, and date unchanged, keep it in place + elif booking.date == db.bookings[index].date: + db.bookings[index] = booking + else: + del db.bookings[index] + db.insert_booking_at_date(booking) + else: + db.insert_booking_at_date(booking) + else: # non-DB-writing calls + error_msg = None + if 'check' == submit_button: + error_msg = 'All looks fine!' + elif submit_button in {'mirror', 'fill_sink', 'add_taxes'}: + if 'add_taxes' == submit_button: + booking.add_taxes(db) + else: + getattr(booking, submit_button)() + elif 'replace' == submit_button: + booking.replace(postvars['replace_from'][0], postvars['replace_to'][0]) + elif submit_button in {'textarea', 'table'}: + edit_mode = submit_button + page = db.edit(index, booking, error_msg=error_msg, edit_mode=edit_mode) + self.send_HTML(page) + return + db.write_db() + index = index if index >= 0 else len(db.bookings) - 1 + self.redirect('ledger#{index}') def do_GET(self): - self.send_response(200) - self.send_header("Content-type", "text/html") - self.end_headers() - db = Database() - # page = db.ledger_as_html() - page = db.balance_as_html() - self.wfile.write(bytes(page, "utf-8")) - - -db = Database() -if __name__ == "__main__": - webServer = HTTPServer((hostName, serverPort), MyServer) - print(f"Server started http://{hostName}:{serverPort}") - try: - webServer.serve_forever() - except KeyboardInterrupt: - pass - webServer.server_close() - print("Server stopped.") + self.try_do(self.forward_gets) + + def forward_gets(self): + prefix = self.apps['ledger'] if hasattr(self, 'apps') else '' + try: + db = LedgerDB(prefix=prefix) + except EditableException as e: + # We catch the EditableException for further editing, and then + # re-run the DB initiation without it blocking DB creation. + db = LedgerDB(prefix=prefix, ignore_editable_exceptions=True) + page = db.edit(index=e.booking_index, error_msg=f'ERROR: {e}') + self.send_HTML(page) + return + parsed_url = urlparse(self.path) + site = path_split(parsed_url.path)[1] + params = parse_qs(parsed_url.query) + if site == 'balance': + stop = params.get('until_after', [None])[0] + page = db.balance_as_html(stop) + elif site == 'edit': + index = params.get('i', [-1])[0] + copy = params.get('copy', [0])[0] + page = db.edit(int(index), copy=bool(copy)) + else: + page = db.ledger_as_html() + self.send_HTML(page) + + + +if __name__ == "__main__": + run_server(server_port, LedgerHandler)