X-Git-Url: https://plomlompom.com/repos/berlin_corona.txt?a=blobdiff_plain;f=ledger.py;h=b4d9fd99191af8cdf3a45d2686654b4b371eb3fe;hb=5f3b9e7eb997cacea086decc3dcd04d6619b9121;hp=6e0a482bffa4b66a57202d84027c1f718dfca933;hpb=fc4f07e359dc5e3a7817c8d100a4142c1f1360c4;p=misc diff --git a/ledger.py b/ledger.py index 6e0a482..b4d9fd9 100755 --- a/ledger.py +++ b/ledger.py @@ -1,21 +1,13 @@ from http.server import BaseHTTPRequestHandler, HTTPServer -import sys import os import html import jinja2 import decimal +from datetime import datetime, timedelta from urllib.parse import parse_qs, urlparse -hostName = "localhost" -serverPort = 8082 +from plomlib import PlomDB, PlomException, run_server, run_server, PlomServer - -class HandledException(Exception): - pass - - -def handled_error_exit(msg): - print(f"ERROR: {msg}") - sys.exit(1) +server_port = 8082 def apply_booking_to_account_balances(account_sums, account, currency, amount): @@ -72,7 +64,6 @@ def bookings_to_account_tree(bookings): def parse_lines(lines, validate_bookings=True): - import datetime inside_booking = False date_string, description = None, None booking_lines = [] @@ -80,6 +71,7 @@ def parse_lines(lines, validate_bookings=True): 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 @@ -89,7 +81,7 @@ def parse_lines(lines, validate_bookings=True): 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") + raise PlomException(f"{prefix} booking ends to early") booking = Booking(date_string, description, booking_lines, start_line, validate_bookings) bookings += [booking] # expect new booking to follow so re-zeroall booking data @@ -113,20 +105,23 @@ def parse_lines(lines, validate_bookings=True): toks = non_comment.split(maxsplit=1) date_string = toks[0] try: - datetime.datetime.strptime(date_string, '%Y-%m-%d') + datetime.strptime(date_string, '%Y-%m-%d') except ValueError: - raise HandledException(f"{prefix} bad date string: {date_string}") + raise PlomException(f"{prefix} bad date string: {date_string}") + if last_date > date_string: + raise PlomException(f"{prefix} out-of-order-date") + last_date = date_string try: description = toks[1] except IndexError: - raise HandledException(f"{prefix} bad description: {description}") + raise PlomException(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 if len(toks) > 3: - raise HandledException(f"{prefix} too many booking line tokens: {toks}") + raise PlomException(f"{prefix} too many booking line tokens: {toks}") amount, currency = None, None account_name = toks[0] if account_name[0] == '[' and account_name[-1] == ']': @@ -142,10 +137,10 @@ def parse_lines(lines, validate_bookings=True): try: amount = decimal.Decimal(toks[2]) except decimal.InvalidOperation: - raise HandledException(f"{prefix} no decimal number in: {toks[1:]}") + raise PlomException(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}") + raise PlomException(f"{prefix} currency starts with int, dot, or minus: {currency}") elif len(toks) == 2: value = toks[1] inside_amount = False @@ -169,26 +164,25 @@ def parse_lines(lines, validate_bookings=True): if inside_amount: if c not in decimal_chars: if len(currency) > 0: - raise HandledException(f"{prefix} amount has non-decimal chars: {value}") + raise PlomException(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}") + raise PlomException(f"{prefix} amount has non-start '-': {value}") if c == '.': if dots_counted > 1: - raise HandledException(f"{prefix} amount has multiple dots: {value}") + raise PlomException(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) + raise PlomException(f"{prefix} currency missing: {value}") + 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") + raise PlomException(f"{prefix} last booking unfinished") return bookings, comments @@ -214,7 +208,7 @@ class Booking: _, 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}") + raise PlomException(f"{prefix} relates more than one empty value of same currency {currency}") empty_values += 1 continue if currency not in sums: @@ -223,14 +217,14 @@ class Booking: if empty_values == 0: for k, v in sums.items(): if v != 0: - raise HandledException(f"{prefix} does not add up to zero / {k} {v}") + raise PlomException(f"{prefix} does not add up to zero / {k} {v}") 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") + raise PlomException(f"{prefix} has empty value that cannot be filled") def parse_booking_lines_to_account_changes(self): account_changes = {} @@ -256,50 +250,68 @@ class Booking: -class Database: +class LedgerDB(PlomDB): def __init__(self): - db_name = "_ledger" - self.db_file = db_name + ".json" - self.lock_file = db_name+ ".lock" self.bookings = [] self.comments = [] self.real_lines = [] - if os.path.exists(self.db_file): - with open(self.db_file, "r") as f: - self.real_lines += [l.rstrip() for l in f.readlines()] + super().__init__('_ledger') ret = parse_lines(self.real_lines) self.bookings += ret[0] self.comments += ret[1] + def read_db_file(self, f): + self.real_lines += [l.rstrip() for l in f.readlines()] + def get_lines(self, start, end): return self.real_lines[start:end] - def replace(self, start, end, lines): - 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() - total_lines = self.real_lines[:start] + lines + self.real_lines[end:] + def write_db(self, text, mode='w'): + self.write_text_to_db(text) + + def insert_at_date(self, lines, date): + start_at = len(self.real_lines) + for b in self.bookings: + if b.date_string == date: + start_at = b.start_line + break + elif b.date_string > date: + break + if start_at == len(self.real_lines): + lines = [''] + lines + return self.write_lines_in_total_lines_at(self.real_lines, start_at, lines) + + def update(self, start, end, lines, date): + total_lines = self.real_lines[:start] + self.real_lines[end:] + n_original_lines = end - start + start_at = len(total_lines) + for b in self.bookings: + if b.date_string == date: + if start_at == len(total_lines) or b.start_line == start: + start_at = b.start_line + if b.start_line > start: + start_at -= n_original_lines + elif b.date_string > date: + break + if start_at == len(total_lines): + lines = [''] + lines + return self.write_lines_in_total_lines_at(total_lines, start_at, lines) + + def write_lines_in_total_lines_at(self, total_lines, start_at, lines): + total_lines = total_lines[:start_at] + lines + [''] + total_lines[start_at:] + _, _ = parse_lines(lines) text = '\n'.join(total_lines) - with open(self.db_file, 'w') as f: - f.write(text); - os.remove(self.lock_file) - - def append(self, lines): - 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' + '\n'.join(lines) + '\n\n'); - os.remove(self.lock_file) + self.write_db(text) + return start_at + + def get_nth_for_booking_of_start_line(self, start_line): + nth = 0 + for b in self.bookings: + if b.start_line >= start_line: + break + nth += 1 + return nth def add_taxes(self, lines, finish=False): ret = [] @@ -318,6 +330,7 @@ class Database: buffer_expenses = 0 kk_expenses = 0 est_expenses = 0 + months_passed = -int(finish) for b in self.bookings: if date == b.date_string: break @@ -330,16 +343,18 @@ class Database: kk_expenses += b.account_changes[acc_kk]['â¬'] if acc_est in acc_keys: est_expenses += b.account_changes[acc_est]['â¬'] - if finish and acc_kk_add in acc_keys and acc_kk_minimum in acc_keys: - 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 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_before_anything = - 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}â¬'] + ret += [f' {acc_est} {-last_monthbreak_est}⬠; for old assumption of needed income: {old_needed_income_before_anything}â¬'] _, account_sums = bookings_to_account_tree(bookings) - expenses_so_far = -1 * account_sums[acc_assets]['â¬'] - old_needed_income + expenses_so_far = -1 * account_sums[acc_assets]['â¬'] + old_needed_income_before_anything needed_income_before_kk = expenses_so_far ESt_this_month = 0 left_over = needed_income_before_kk - ESt_this_month @@ -349,7 +364,6 @@ class Database: E1 = decimal.Decimal(15999) E2 = decimal.Decimal(62809) E3 = decimal.Decimal(277825) - months_passed = int(date[5:7]) - 1 while True: zvE = buffer_expenses - kk_expenses + (12 - months_passed) * needed_income_before_kk if finish: @@ -388,24 +402,37 @@ class Database: 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_so_far = account_sums[acc_kk_add]['â¬'] if acc_kk_add in account_sums.keys() else 0 + kk_add = needed_income_before_kk / (1 - kk_factor) - needed_income_before_kk - kk_minimum_tax + hit_kk_minimum_income_limit = False + if kk_add_so_far + kk_add < 0: + hit_kk_minimum_income_limit = True + kk_add_uncorrect = kk_add + kk_add = -(kk_add + kk_add_so_far) 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}â¬)'] + ret += [f' {acc_kk_add} {-last_monthbreak_kk_add}⬠; for old assumption of needed income'] 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}â¬)'] + if hit_kk_minimum_income_limit: + ret += [f' {acc_kk_add} {kk_add}⬠; {needed_income_before_kk:.2f}⬠/ (1 - {kk_factor:.3f}) - {needed_income_before_kk:.2f}⬠- {kk_minimum_tax}⬠= {kk_add_uncorrect:.2f}⬠would reduce current {acc_kk_dd} ({kk_add_so_far:.2f}â¬) below 0'] + else: + ret += [f' {acc_kk_add} {kk_add}⬠; {needed_income_before_kk:.2f}⬠/ (1 - {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 + final_minus = expenses_so_far - old_needed_income_before_anything + diff ret += [f' {acc_assets} {-diff} â¬'] ret += [f' {acc_assets} {final_minus} â¬'] - ret += [f' {acc_buffer} {-final_minus} â¬'] + year_needed = buffer_expenses + final_minus + (12 - months_passed - 1) * final_minus + if finish: + ret += [f' {acc_buffer} {-final_minus} â¬'] + else: + 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): +class LedgerServer(PlomServer): header = """
-ledger1 -ledger2 +ledger balance add free add structured{{date}} {{desc}} {{head_comment|e}}
+
{{date}} {{desc}} {{head_comment|e}}
{content}" def ledger_as_html(self, db): - elements_to_write = [] - account_sums = {} ## - nth_of_same_date = 0 - same_date = '' - for booking in db.bookings: - 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 - 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 ledger2_as_html(self, db): - single_c_tmpl = jinja2.Template('{{c|e}}