X-Git-Url: https://plomlompom.com/repos/%7B%7Bprefix%7D%7D/balance?a=blobdiff_plain;f=ledger.py;h=b93b202e462cb47566bbc1fcf2aee598eec72775;hb=169d6b20b9f3d63508180971426db3bc195a3581;hp=c3f0ab9c22d028d92b1c2741e7c411a4d9efa499;hpb=2deeead813fbdf0fec4f94d238a43fd93514f94f;p=misc diff --git a/ledger.py b/ledger.py index c3f0ab9..b93b202 100755 --- a/ledger.py +++ b/ledger.py @@ -1,6 +1,11 @@ from http.server import BaseHTTPRequestHandler, HTTPServer import sys import os +import html +import jinja2 +import decimal +import datetime +from urllib.parse import parse_qs, urlparse hostName = "localhost" serverPort = 8082 @@ -23,15 +28,60 @@ def apply_booking_to_account_balances(account_sums, account, currency, amount): account_sums[account][currency] += amount -def parse_lines(lines): +def bookings_to_account_tree(bookings): + account_sums = {} + for booking in 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 + 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 + 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]) + return account_tree, account_sums + + +def parse_lines(lines, validate_bookings=True): import datetime - import decimal inside_booking = False date_string, description = None, None booking_lines = [] start_line = 0 bookings = [] comments = [] + lines = lines.copy() + [''] # to ensure a booking-ending last line + last_date = '' for i, line in enumerate(lines): prefix = f"line {i}" # we start with the case of an utterly empty line @@ -42,7 +92,7 @@ def parse_lines(lines): # 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) + booking = Booking(date_string, description, booking_lines, start_line, validate_bookings) bookings += [booking] # expect new booking to follow so re-zeroall booking data inside_booking = False @@ -52,7 +102,7 @@ def parse_lines(lines): # 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: - comments[i] = split_by_comment[1] + comments[i] = split_by_comment[1].lstrip() # 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() == '': @@ -68,11 +118,15 @@ def parse_lines(lines): datetime.datetime.strptime(date_string, '%Y-%m-%d') except ValueError: raise HandledException(f"{prefix} bad date string: {date_string}") + if last_date > date_string: + raise HandledException(f"{prefix} out-of-order-date") + last_date = date_string try: description = toks[1] except IndexError: raise HandledException(f"{prefix} bad description: {description}") inside_booking = True + booking_lines += [non_comment] continue # otherwise, read as transfer data toks = non_comment.split() # ignore specification's allowance of single spaces in names @@ -132,31 +186,33 @@ def parse_lines(lines): 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) + if len(amount_string) > 0: + amount = decimal.Decimal(amount_string) booking_lines += [(account_name, amount, currency)] if inside_booking: raise HandledException(f"{prefix} last booking unfinished") return bookings, comments + class Booking: - def __init__(self, date_string, description, booking_lines, start_line): + def __init__(self, date_string, description, booking_lines, start_line, process=True): 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() + if process: + self.validate_booking_lines() + self.sink = {} + 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: + for line in self.lines[1:]: if line == '': continue _, amount, currency = line @@ -171,7 +227,7 @@ class Booking: if empty_values == 0: for k, v in sums.items(): if v != 0: - raise HandledException(f"{prefix} does not sum up to zero") + raise HandledException(f"{prefix} does not add up to zero / {k} {v}") else: sinkable = False for k, v in sums.items(): @@ -184,7 +240,7 @@ class Booking: account_changes = {} debt = {} sink_account = None - for line in self.lines: + for line in self.lines[1:]: if line == '': continue account, amount, currency = line @@ -199,57 +255,324 @@ class Booking: if sink_account: for currency, amount in debt.items(): apply_booking_to_account_balances(account_changes, sink_account, currency, -amount) + self.sink[currency] = -amount return account_changes class Database: - def __init__(self, load_from_file=True): + def __init__(self): db_name = "_ledger" self.db_file = db_name + ".json" self.lock_file = db_name+ ".lock" self.bookings = [] self.comments = [] - if load_from_file and os.path.exists(self.db_file): + self.real_lines = [] + if os.path.exists(self.db_file): with open(self.db_file, "r") as f: - ret = parse_lines(f.readlines()) - self.bookings += ret[0] - self.comments += ret[1] + self.real_lines += [l.rstrip() for l in f.readlines()] + ret = parse_lines(self.real_lines) + self.bookings += ret[0] + self.comments += ret[1] - def append(self, lines): + def get_lines(self, start, end): + return self.real_lines[start:end] + + def write_db(self, text, mode='w'): import shutil if os.path.exists(self.lock_file): raise HandledException('Sorry, lock file!') - if os.path.exists(self.db_file): - shutil.copy(self.db_file, self.db_file + ".bak") f = open(self.lock_file, 'w+') f.close() - with open(self.db_file, 'a') as f: - f.write('\n' + '\n'.join(lines) + '\n'); + + # always back up most recent to .bak + bakpath = f'{self.db_file}.bak' + shutil.copy(self.db_file, bakpath) + + # collect modification times of numbered .bak files + bak_prefix = f'{bakpath}.' + backup_dates = [] + i = 0 + bak_as = f'{bak_prefix}{i}' + while os.path.exists(bak_as): + mod_time = os.path.getmtime(bak_as) + backup_dates += [str(datetime.datetime.fromtimestamp(mod_time))] + i += 1 + bak_as = f'{bak_prefix}{i}' + + # collect what numbered .bak files to save + to_save = [] + datetime_len = 19 + now = str(datetime.datetime.now())[:datetime_len] + while datetime_len > 2: + for i, date in reversed(list(enumerate(backup_dates))): + if date[:datetime_len] == now: + if i not in to_save: + to_save += [i] + break + datetime_len -= 1 + now = now[:datetime_len] + + # remove redundant backup files + j = 0 + for i in reversed(to_save): + if i != j: + source = f'{bak_prefix}{i}' + target = f'{bak_prefix}{j}' + shutil.move(source, target) + else: + print("keeping", i) + j += 1 + for i in range(j, len(backup_dates)): + print("removing", i) + try: + os.remove(f'{bak_prefix}{i}') + except FileNotFoundError: + pass + + # + shutil.copy(self.db_file, f'{bak_prefix}{j}') + with open(self.db_file, mode) as f: + f.write(text); os.remove(self.lock_file) + def replace(self, start, end, lines): + total_lines = self.real_lines[:start] + lines + self.real_lines[end:] + text = '\n'.join(total_lines) + self.write_db(text) + + def append(self, lines): + text = '\n\n' + '\n'.join(lines) + '\n\n' + self.write_db(text, 'a') + + def add_taxes(self, lines, finish=False): + ret = [] + bookings, _ = parse_lines(lines) + date = bookings[0].date_string + acc_kk_add = 'Reserves:KrankenkassenBeitragsWachstum' + acc_kk_minimum = 'Reserves:Month:KrankenkassenDefaultBeitrag' + acc_kk = 'Expenses:KrankenKasse' + acc_est = 'Reserves:Einkommenssteuer' + acc_assets = 'Assets' + acc_buffer = 'Reserves:NeuAnfangsPuffer:Ausgaben' + last_monthbreak_assets = 0 + last_monthbreak_est = 0 + last_monthbreak_kk_minimum = 0 + last_monthbreak_kk_add = 0 + buffer_expenses = 0 + kk_expenses = 0 + est_expenses = 0 + months_passed = -int(finish) + for b in self.bookings: + if date == b.date_string: + break + acc_keys = b.account_changes.keys() + if acc_buffer in acc_keys: + buffer_expenses -= b.account_changes[acc_buffer]['â¬'] + if acc_kk_add in acc_keys: + kk_expenses += b.account_changes[acc_kk_add]['â¬'] + if acc_kk in acc_keys: + kk_expenses += b.account_changes[acc_kk]['â¬'] + if acc_est in acc_keys: + est_expenses += b.account_changes[acc_est]['â¬'] + if acc_kk_add in acc_keys and acc_kk_minimum in acc_keys: + months_passed += 1 + if finish: + last_monthbreak_kk_add = b.account_changes[acc_kk_add]['â¬'] + last_monthbreak_est = b.account_changes[acc_est]['â¬'] + last_monthbreak_kk_minimum = b.account_changes[acc_kk_minimum]['â¬'] + last_monthbreak_assets = b.account_changes[acc_buffer]['â¬'] + old_needed_income = last_monthbreak_assets + last_monthbreak_kk_add + last_monthbreak_kk_minimum + last_monthbreak_est + if finish: + ret += [f' {acc_est} {-last_monthbreak_est}⬠; for old assumption of needed income: {old_needed_income}â¬'] + _, account_sums = bookings_to_account_tree(bookings) + expenses_so_far = -1 * account_sums[acc_assets]['â¬'] - old_needed_income + needed_income_before_kk = expenses_so_far + ESt_this_month = 0 + left_over = needed_income_before_kk - ESt_this_month + too_low = 0 + too_high = 2 * needed_income_before_kk + E0 = decimal.Decimal(10908) + E1 = decimal.Decimal(15999) + E2 = decimal.Decimal(62809) + E3 = decimal.Decimal(277825) + while True: + zvE = buffer_expenses - kk_expenses + (12 - months_passed) * needed_income_before_kk + if finish: + zvE += last_monthbreak_assets + last_monthbreak_kk_add + last_monthbreak_kk_minimum + if zvE < E0: + ESt = decimal.Decimal(0) + elif zvE < E1: + y = (zvE - E0)/10000 + ESt = (decimal.Decimal(979.18) * y + 1400) * y + elif zvE < E2: + y = (zvE - E1)/10000 + ESt = (decimal.Decimal(192.59) * y + 2397) * y + decimal.Decimal(966.53) + elif zvE < E3: + ESt = decimal.Decimal(0.42) * (zvE - decimal.Decimal(62809)) + decimal.Decimal(16405.54) + else: + ESt = decimal.Decimal(0.45) * (zvE - decimal.Decimal(277825)) + decimal.Decimal(106713.52) + ESt_this_month = (ESt + last_monthbreak_est - est_expenses) / (12 - months_passed) + left_over = needed_income_before_kk - ESt_this_month + if abs(left_over - expenses_so_far) < 0.001: + break + elif left_over < expenses_so_far: + too_low = needed_income_before_kk + elif left_over > expenses_so_far: + too_high = needed_income_before_kk + needed_income_before_kk = too_low + (too_high - too_low)/2 + ESt_this_month = ESt_this_month.quantize(decimal.Decimal('0.00')) + ret += [f' {acc_est} {ESt_this_month}⬠; expenses so far: {expenses_so_far:.2f}â¬; zvE: {zvE:.2f}â¬; ESt total: {ESt:.2f}â¬; needed before Krankenkasse: {needed_income_before_kk:.2f}â¬'] + kk_minimum_income = 1131.67 + if date < '2023-02-01': + kk_minimum_income = decimal.Decimal(1096.67) + kk_factor = decimal.Decimal(0.189) + kk_minimum_tax = decimal.Decimal(207.27).quantize(decimal.Decimal('0.00')) + elif date < '2023-08-01': + kk_factor = decimal.Decimal(0.191) + kk_minimum_tax = decimal.Decimal(216.15).quantize(decimal.Decimal('0.00')) + else: + kk_factor = decimal.Decimal(0.197) + kk_minimum_tax = decimal.Decimal(222.94).quantize(decimal.Decimal('0.00')) + kk_add = max(0, kk_factor * needed_income_before_kk - kk_minimum_tax) + kk_add = decimal.Decimal(kk_add).quantize(decimal.Decimal('0.00')) + if finish: + ret += [f' {acc_kk_add} {-last_monthbreak_kk_add}⬠; max(0, {kk_factor:.3f} * {-(old_needed_income - last_monthbreak_est):.2f}⬠- {kk_minimum_tax}â¬)'] + else: + ret += [f' {acc_kk_minimum} {kk_minimum_tax}⬠; assumed minimum income {kk_minimum_income:.2f}⬠* {kk_factor:.3f}'] + ret += [f' {acc_kk_add} {kk_add}⬠; max(0, {kk_factor:.3f} * {needed_income_before_kk:.2f}⬠- {kk_minimum_tax}â¬)'] + diff = - last_monthbreak_est + ESt_this_month - last_monthbreak_kk_add + kk_add + if not finish: + diff += kk_minimum_tax + final_minus = expenses_so_far + old_needed_income + diff + ret += [f' {acc_assets} {-diff} â¬'] + ret += [f' {acc_assets} {final_minus} â¬'] + year_needed = buffer_expenses + final_minus + (12 - months_passed - 1) * final_minus + ret += [f' {acc_buffer} {-final_minus} ⬠; assume as to earn in year: {acc_buffer} + {12 - months_passed - 1} * this = {-year_needed}'] + return ret + class MyServer(BaseHTTPRequestHandler): - header = '
ledger balance add{{date}} {{desc}} {{head_comment|e}}
+
+
{{l.acc|e}} | {{l.money|e}} | {{l.comment|e}} |
{{l.comment|e}} |
{content}" - def ledger_as_html(self, db): - lines = [] - for comment in db.comments: - lines += [f"; {comment}" if comment != '' else ''] + def ledger2_as_html(self, db): + elements_to_write = [] + account_sums = {} ## + nth_of_same_date = 0 + same_date = '' for booking in db.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.date_string == same_date: + nth_of_same_date += 1 + else: + same_date = booking.date_string + nth_of_same_date = 0 + booking_end = booking.start_line + len(booking.lines) + booking_lines = [] + for booking_line in booking.lines[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}" + account = booking_line[0] ## + account_toks = account.split(':') ## + path = '' ## + for tok in account_toks: ## + path += tok ## + if not path in account_sums.keys(): ## + account_sums[path] = {} ## + path += ':' ## + moneys = [] ## + money = '' + if booking_line[1] is not None: + moneys += [(booking_line[1], booking_line[2])] ## + money = f'{moneys[0][0]} {moneys[0][1]}' + else: ## + for currency, amount in booking.sink.items(): ## + moneys += {(amount, currency)} ## + money = '[' ## + for m in moneys: ## + money += f'{m[0]} {m[1]} ' ## + money += ']' ## + balance = '' ## + for amount, currency in moneys: ## + path = '' ## + for tok in account_toks: ## + path += tok ## + if not currency in account_sums[path].keys(): ## + account_sums[path][currency] = 0 ## + account_sums[path][currency] += amount ## + path += ':' ## + balance += f'{account_sums[account][currency]} {currency}' ## + booking_lines += [{'acc': booking_line[0], 'money':money, 'comment':balance}] ## + elements_to_write += [self.booking_tmpl.render( + start=booking.start_line, + end=booking_end, + date=booking.date_string, + desc=booking.description, + head_comment=db.comments[booking.start_line], + steps_after_date=nth_of_same_date, + booking_lines = booking_lines)] + return '\n'.join(elements_to_write) - def add(self): - return '' + def ledger_as_html(self, db): + single_c_tmpl = jinja2.Template('{{c|e}}