X-Git-Url: https://plomlompom.com/repos/%7B%7Bdb.prefix%7D%7D/task?a=blobdiff_plain;f=ledger.py;h=d8b504a74398ce18315a4430b345ac7a1eaf417a;hb=11df024d5f24e8d7916f57c907975ebe27db617f;hp=aa295d28203d7279137d9b143b5c236e98c92e31;hpb=8822ee7e3069193162469438bb54ee4629e1ae19;p=misc
diff --git a/ledger.py b/ledger.py
index aa295d2..d8b504a 100755
--- a/ledger.py
+++ b/ledger.py
@@ -4,6 +4,7 @@ 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
@@ -15,92 +16,18 @@ class EditableException(PlomException):
super().__init__(*args, **kwargs)
-# html_head = """
-#
-#
-# ledger
-# ledger2
-# balance
-# balance2
-# add free
-# add structured
-# add
-#
-# """
-# booking_html = """
-# {{date}} {{desc}}
-# [edit: structured
-# / free
-# | copy:structured
-# / free
-# | move {% if move_up %}up{% else %}up{% endif %}/{% if move_down %}down{% else %}down{% endif %}
-# | balance after
-# ]
-#
-# {% for l in booking_lines %}
-# {% if l.acc %}
-# {{l.acc|e}} | {{l.money|e}} |
-# {% else %}
-# |
-# {% endif %}
-# {% endfor %}
-#
-# """
-# add_form_header = """
-# """
-# add_free_html = """
-#
-# """
-# add_structured_html = """
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-# {% for line in booking_lines %}
-#
-#
-#
-#
-#
-#
-#
-#
-# {% endfor %}
-# {% for name, items in datalist_sets.items() %}
-#
-# {% endfor %}
-# """
+class LedgerTextLine:
+
+ 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()
class Wealth:
@@ -108,17 +35,22 @@ class Wealth:
def __init__(self):
self.money_dict = {}
- def add_money_dict(self, moneys):
+ 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 add_wealth(self, moneys):
- self.add_money_dict(moneys.money_dict)
+ @property
+ def sink_empty(self):
+ return len(self.as_sink) == 0
@property
- def as_sink_dict(self):
+ def as_sink(self):
sink = {}
for currency, amount in self.money_dict.items():
if 0 == amount:
@@ -138,7 +70,7 @@ class Account:
self.children = []
def add_wealth(self, moneys):
- self.own_moneys.add_wealth(moneys)
+ self.own_moneys += moneys
@property
def full_name(self):
@@ -150,67 +82,15 @@ class Account:
@property
def full_moneys(self):
full_moneys = Wealth()
- full_moneys.add_wealth(self.own_moneys)
+ full_moneys += self.own_moneys
for child in self.children:
- full_moneys.add_wealth(child.full_moneys)
+ full_moneys += child.full_moneys
return full_moneys
-# 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
-#
-#
-# 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_lines2(lines, ignore_editable_exceptions=False):
- lines = lines.copy() + [''] # to simulate ending of last booking
+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 = []
@@ -218,8 +98,7 @@ def parse_lines2(lines, ignore_editable_exceptions=False):
last_date = ''
for i, line in enumerate(lines):
intro = f'file line {i}'
- stripped_line = line.rstrip()
- if stripped_line == '':
+ 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:
@@ -238,129 +117,6 @@ def parse_lines2(lines, ignore_editable_exceptions=False):
return bookings
-# def parse_lines(lines, validate_bookings=True):
-# 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
-# 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 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
-# 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:
-# 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() == '':
-# 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.strptime(date_string, '%Y-%m-%d')
-# except ValueError:
-# 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 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 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] == ']':
-# # 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 PlomException(f"{prefix} no decimal number in: {toks[1:]}")
-# currency = toks[i_currency]
-# if currency[0] in decimal_chars:
-# raise PlomException(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 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 PlomException(f"{prefix} amount has non-start '-': {value}")
-# if c == '.':
-# if dots_counted > 1:
-# raise PlomException(f"{prefix} amount has multiple dots: {value}")
-# dots_counted += 1
-# amount_string += c
-# if len(currency) == 0:
-# 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 PlomException(f"{prefix} last booking unfinished")
-# return bookings, comments
-
-
class TransferLine:
def __init__(self, line=None, account='', amount=None, currency='', comment='', validate=True):
@@ -369,13 +125,10 @@ class TransferLine:
self.currency = currency
self.comment = comment
if line:
- split_by_comment = line.rstrip().split(sep=';', maxsplit=1)
- if len(split_by_comment) == 0:
+ if line.empty:
raise PlomException('line empty')
- elif len(split_by_comment) == 2:
- self.comment = split_by_comment[1].lstrip()
- non_comment = split_by_comment[0].rstrip()
- toks = non_comment.split()
+ 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')
@@ -391,7 +144,7 @@ class TransferLine:
if validate:
raise PlomException(f'invalid token for Decimal: {toks[1]}')
else:
- self.amount = toks[1]
+ self.comment = f'unparsed: {toks[1]}; {self.comment}'
self.currency = toks[2]
@property
@@ -403,6 +156,7 @@ class TransferLine:
else:
return f'{self.amount:.2f}'
+ @property
def for_writing(self):
line = f' {self.account}'
if self.amount is not None:
@@ -424,18 +178,18 @@ class Booking:
self.intro = f'booking starting at line {self.starts_at}'
self.clean()
if lines:
- self.real_lines = 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 len(self.transfer_lines) < 2 and self.validate:
+ 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.real_lines = self.for_writing()
+ self.lines = [LedgerTextLine(l) for l in self.for_writing]
@classmethod
def from_postvars(cls, postvars, starts_at='?', validate=True):
@@ -457,16 +211,13 @@ class Booking:
self.account_changes = {}
def parse_lines(self):
- self.top_comment = ''
- if len(self.real_lines) < 3 and self.validate:
- raise PlomException(f'{self.intro}: ends with less than 3 lines:' + str(self.real_lines))
- split_by_comment = self.real_lines[0].rstrip().split(sep=";", maxsplit=1)
- if len(split_by_comment) == 0 and self.validate:
+ 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')
- elif len(split_by_comment) == 2 and self.validate:
- self.top_comment = split_by_comment[1].lstrip()
- non_comment = split_by_comment[0].rstrip()
- toks = non_comment.split()
+ 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}')
@@ -477,7 +228,7 @@ class Booking:
self.date = toks[0]
self.description = toks[1]
self.validate_head()
- for i, line in enumerate(self.real_lines[1:]):
+ for i, line in enumerate(self.lines[1:]):
try:
self.transfer_lines += [TransferLine(line, validate=self.validate)]
except PlomException as e:
@@ -497,14 +248,27 @@ class Booking:
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].add_money_dict(money)
- money_changes.add_money_dict(money)
- if sink_account is None and len(money_changes.as_sink_dict) > 0 and self.validate:
- raise PlomException(f'{intro}: does not balance (unused non-empty sink)')
+ 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].add_money_dict(money_changes.as_sink_dict)
+ 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:
@@ -523,172 +287,116 @@ class Booking:
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()]
+ replacement_lines += [TransferLine(None, f'{line.account}', amount, currency).for_writing]
break
- lines = self.real_lines[:i+1] + replacement_lines + self.real_lines[i+2:]
+ lines = self.lines[:i+1] + [LedgerTextLine(l) for l in replacement_lines] + self.lines[i+2:]
self.clean()
- self.real_lines = lines
+ self.lines = lines
self.parse_lines()
- def add_mirror(self):
+ def mirror(self):
new_transfer_lines = []
- for line in self.transfer_lines:
- new_transfer_lines += [TransferLine(None, f'({line.account})', line.amount, line.currency, line.comment)]
+ 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.real_lines += [transfer_line.for_writing()]
+ self.lines += [LedgerTextLine(transfer_line.for_writing)]
self.clean()
self.parse_lines()
- def add_taxes(self):
+ 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()
+
+ def add_taxes(self, db):
acc_kk_add = 'Reserves:KrankenkassenBeitragsWachstum'
- acc_kk_minimum = 'Reserves:Month:KrankenkassenDefaultBeitrag'
+ acc_kk_minimum = 'Reserves:Monthly:KrankenkassenDefaultBeitrag'
acc_kk = 'Expenses:KrankenKasse'
- acc_est = 'Reserves:EinkommensSteuer'
+ acc_ESt = 'Reserves:EinkommensSteuer'
acc_assets = 'Assets'
- acc_buffer = 'Reserves:NeuAnfangsPuffer:Ausgaben'
- buffer_expenses = 0 # FIXME: hardcoded for now
- kk_expenses = 0 # FIXME: hardcoded for now
- est_expenses = 0 # FIXME: hardcoded for now
- months_passed = 0 # FIXME: hardcoded for now
- last_monthbreak_assets = 0 # FIXME: hardcoded for now
- last_monthbreak_est = 0 # FIXME: hardcoded for now
- last_monthbreak_kk_minimum = 0 # FIXME: hardcoded for now
- last_monthbreak_kk_add = 0 # FIXME: hardcoded for now
- expenses_so_far = 0 # FIXME: hardcoded for now
- needed_netto = self.account_changes['Assets'].money_dict['â¬']
+ 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['â¬']
+
+ needed_netto = -self.account_changes['Assets'].money_dict['â¬']
+ past_taxed_needs_before_kk = past_neuanfangspuffer_expenses - past_kk_expenses
ESt_this_month = 0
- left_over = needed_netto - ESt_this_month
+ 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
- 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_netto
- # if finish:
- # zvE += last_monthbreak_assets + last_monthbreak_kk_add + last_monthbreak_kk_minimum
+ 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 = decimal.Decimal(0)
+ ESt_year = decimal.Decimal(0)
elif zvE < E1:
y = (zvE - E0)/10000
- ESt = (decimal.Decimal(979.18) * y + 1400) * y
+ ESt_year = (decimal.Decimal(922.98) * y + 1400) * y
elif zvE < E2:
y = (zvE - E1)/10000
- ESt = (decimal.Decimal(192.59) * y + 2397) * y + decimal.Decimal(966.53)
+ ESt_year = (decimal.Decimal(181.19) * y + 2397) * y + decimal.Decimal(1025.38)
elif zvE < E3:
- ESt = decimal.Decimal(0.42) * (zvE - decimal.Decimal(62809)) + decimal.Decimal(16405.54)
+ ESt_year = decimal.Decimal(0.42) * zvE - 10602.13
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_netto - ESt_this_month
- if abs(left_over - expenses_so_far) < 0.001:
+ 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 left_over < expenses_so_far:
- too_low = needed_netto
- elif left_over > expenses_so_far:
- too_high = needed_netto
- needed_netto = too_low + (too_high - too_low)/2
+ 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'))
- self.transfer_lines += [TransferLine(None, acc_est, ESt_this_month, 'â¬')]
- kk_minimum_income = 1131.67
- kk_factor = decimal.Decimal(0.197)
- kk_minimum_tax = decimal.Decimal(222.94).quantize(decimal.Decimal('0.00'))
- kk_add_so_far = account_sums[acc_kk_add]['â¬'] if acc_kk_add in account_sums.keys() else 0
- kk_add = needed_netto / (1 - kk_factor) - needed_netto - 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)
+ 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'))
- self.transfer_lines += [TransferLine(None, acc_kk_minimum, kk_minimum_tx, 'â¬')]
- self.transfer_lines += [TransferLine(None, acc_kk_add, kk_add, 'â¬')]
- 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_before_anything + diff
- diff += kk_minimum_tax
- final_minus = expenses_so_far - old_needed_income_before_anything + diff
- self.transfer_lines += [TransferLine(None, acc_assets, -diff, 'â¬')]
- self.transfer_lines += [TransferLine(None, acc_assets, final_minus, 'â¬')]
- self.transfer_lines += [TransferLine(None, acc_buffer, -final_minus, 'â¬')]
-
- 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))
-
-
-# class Booking:
-#
-# 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
-# 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[1:]:
-# if line == '':
-# continue
-# _, amount, currency = line
-# if amount is None:
-# if empty_values > 0:
-# raise PlomException(f"{prefix} relates more than one empty value of same currency {currency}")
-# empty_values += 1
-# 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 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 PlomException(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[1:]:
-# if line == '':
-# continue
-# account, amount, currency = line
-# if amount is None:
-# sink_account = account
-# continue
-# apply_booking_to_account_balances(account_changes, account, currency, amount)
-# if currency not in debt:
-# debt[currency] = amount
-# else:
-# debt[currency] += amount
-# 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
+ 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):
@@ -697,70 +405,12 @@ class LedgerDB(PlomDB):
self.prefix = prefix
self.bookings = []
self.comments = []
- self.real_lines = []
+ self.text_lines = []
super().__init__(db_path)
- self.bookings = parse_lines2(self.real_lines, ignore_editable_exceptions)
+ self.bookings = parse_lines_to_bookings(self.text_lines, ignore_editable_exceptions)
def read_db_file(self, f):
- self.real_lines += [l.rstrip() for l in f.readlines()] # TODO is this necessary? (parser already removes lines?)
-
- # def get_lines(self, start, end):
- # return self.real_lines[start:end]
-
- # def write_db(self, text, mode='w'):
- # if text[-1] != '\n':
- # text += '\n'
- # self.write_text_to_db(text)
-
- # def insert_at_date(self, lines, date):
- # start_at = 0
- # if len(self.bookings) > 0:
- # if date >= self.bookings[-1].date_string:
- # start_at = len(self.real_lines)
- # lines = [''] + lines
- # else:
- # for b in self.bookings:
- # if b.date_string == date:
- # start_at = b.start_line
- # elif b.date_string > date:
- # start_at = b.start_line
- # break
- # lines += [''] # DEBUG is new
- # return self.write_lines_in_total_lines_at(self.real_lines, start_at, lines)
-
- # def update(self, start, end, lines, date):
- # remaining_lines = self.real_lines[:start] + self.real_lines[end:]
- # n_original_lines = end - start
- # start_at = len(remaining_lines)
- # for b in self.bookings:
- # if b.date_string == date:
- # if start_at == len(remaining_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
- # # print("DEBUG update start_at", start_at, "len(remaining_lines)", len(remaining_lines), "len(self.real_lines)", len(self.real_lines), "end", end)
- # if start_at != 0 and end != len(self.real_lines) and start_at == len(remaining_lines):
- # # Add empty predecessor line if appending.
- # lines = [''] + lines
- # return self.write_lines_in_total_lines_at(remaining_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:]
- # total_lines = total_lines[:start_at] + lines + total_lines[start_at:]
- # _, _ = parse_lines(lines)
- # text = '\n'.join(total_lines)
- # 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
+ self.text_lines += f.readlines()
def insert_booking_at_date(self, booking):
place_at = 0
@@ -773,179 +423,16 @@ class LedgerDB(PlomDB):
break
else:
place_at = i + 1
- self.bookings = self.bookings[:place_at] + [booking] + self.bookings[place_at:]
-
- 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_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_before_anything}â¬']
- _, account_sums = bookings_to_account_tree(bookings)
- 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
- 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_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}⬠; 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}']
- 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_before_anything + diff
- ret += [f' {acc_assets} {-diff} â¬']
- ret += [f' {acc_assets} {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
-
- # def add_mirror(self, lines):
- # ret = []
- # bookings, _ = parse_lines(lines)
- # booking = bookings[0]
- # for line in booking.lines[1:]:
- # ret += [f'? {-line[1]} {line[2]}']
- # return ret
+ self.bookings.insert(place_at, booking)
def ledger_as_html(self):
- for nth, booking in enumerate(self.bookings):
- booking.can_up = nth > 0 and self.bookings[nth - 1].date == booking.date
- booking.can_down = nth < len(self.bookings) - 1 and self.bookings[nth + 1].date == booking.date
+ 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 ledger_as_html(self):
- # booking_tmpl = jinja2.Template(booking_html)
- # single_c_tmpl = jinja2.Template('
') ##
- # elements_to_write = []
- # last_i = i = 0 ##
- # for nth, booking in enumerate(self.bookings):
- # move_up = nth > 0 and self.bookings[nth - 1].date_string == booking.date_string
- # move_down = nth < len(self.bookings) - 1 and self.bookings[nth + 1].date_string == booking.date_string
- # booking_end = last_i = booking.start_line + len(booking.lines)
- # booking_lines = []
- # i = booking.start_line ##
- # elements_to_write += [single_c_tmpl.render(c=c) for c in self.comments[last_i:i] if c != ''] ##
- # for booking_line in booking.lines[1:]:
- # i += 1 ##
- # comment = self.comments[i] ##
- # if booking_line == '':
- # booking_lines += [{'acc': None, 'money': None, 'comment': comment}] ##
- # continue
- # account = booking_line[0]
- # money = ''
- # if booking_line[1] is not None:
- # money = f'{booking_line[1]} {booking_line[2]}'
- # booking_lines += [{'acc': booking_line[0], 'money':money, 'comment':comment}] ##
- # elements_to_write += [booking_tmpl.render(
- # prefix=self.prefix,
- # nth=nth,
- # start=booking.start_line,
- # end=booking_end,
- # date=booking.date_string,
- # desc=booking.description,
- # head_comment=self.comments[booking.start_line],
- # move_up=move_up,
- # move_down=move_down,
- # booking_lines = booking_lines)]
- # elements_to_write += [single_c_tmpl.render(c=c) for c in self.comments[last_i:] if c != ''] #
- # return '\n'.join(elements_to_write)
-
- def balance_as_html(self, until=None):
- bookings = self.bookings[:(until if until is None else int(until)+1)]
+ 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:
@@ -973,222 +460,54 @@ class LedgerDB(PlomDB):
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"{content}
"
-
- def edit(self, index, sent=None, error_msg=None, edit_mode='table'):
+ 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()
+ 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)
-
- # 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, nth):
- return self.move(nth, -1)
-
- def move_down(self, nth):
- return self.move(nth, +1)
-
- def move(self, nth, direction):
- to_move = self.bookings[nth]
- swap_nth = nth+1*(direction)
- to_swap = self.bookings[swap_nth]
- self.bookings[nth] = to_swap
- self.bookings[nth+1*(direction)] = to_move
- return swap_nth
+ 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 i, booking in enumerate(self.bookings):
if i > 0:
lines += ['']
- lines += booking.for_writing()
+ 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.real_lines[end+1:]
- # lines = [''] + lines
- # start_at += 1
- # else:
- # total_lines = self.real_lines[:start-1] + self.real_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):
@@ -1208,104 +527,59 @@ class LedgerHandler(PlomHandler):
db = LedgerDB(prefix, ignore_editable_exceptions=True)
index = 0
parsed_url = urlparse(self.path)
- for string in {'save', 'copy', 'check', 'mirror', 'fill sink', 'as textarea', 'as table', 'move up', 'move down', 'add taxes'}:
+ 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 prefix + '/ledger' == parsed_url.path:
- if submit_button == 'move up':
- index = db.move_up(int(postvars['move up'][0]))
- elif submit_button == 'move down':
- index = db.move_down(int(postvars['move down'][0]))
- elif prefix + '/edit' == parsed_url.path:
+ 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])
- starts_at = '?' if index == -1 else db.bookings[index].starts_at
edit_mode = postvars['edit_mode'][0]
- validate = submit_button in {'save', 'copy', 'check'}
+ 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 = postvars['booking'][0].rstrip().split('\n')
- booking = Booking(lines, starts_at, validate=validate)
+ 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:
+ booking = Booking(lines, starts_at, validate=validate)
else:
booking = Booking.from_postvars(postvars, starts_at, validate)
- if submit_button in {'save', 'copy'}:
- if index != -1 and submit_button != 'copy':
- if booking.date == db.bookings[index].date:
- db.bookings[index] = booking
- booking_is_placed = True
- else:
- db.bookings = db.bookings[:index] + db.bookings[index+1:]
- db.insert_booking_at_date(booking)
+ 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:
+ else: # non-DB-writing calls
error_msg = None
if 'check' == submit_button:
error_msg = 'All looks fine!'
- elif 'mirror' == submit_button:
- booking.add_mirror()
- elif 'fill sink' == submit_button:
- booking.fill_sink()
- elif 'add taxes' == submit_button:
- booking.add_taxes()
- elif submit_button in {'as textarea', 'as table'}:
- edit_mode = submit_button[len('as '):]
+ 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(prefix + f'/ledger#{index}')
-
- # def forward_posts(self):
- # prefix = self.apps['ledger'] if hasattr(self, 'apps') else ''
- # parsed_url = urlparse(self.path)
- # length = int(self.headers['content-length'])
- # postvars = parse_qs(self.rfile.read(length).decode(), keep_blank_values=1)
- # start = int(postvars['start'][0])
- # end = int(postvars['end'][0])
- # print("DEBUG start, end", start, end)
- # db = LedgerDB(prefix)
- # add_empty_line = None
- # lines = []
- # # get inputs
- # if prefix + '/add_structured' == parsed_url.path and not 'revert' in postvars.keys():
- # lines, add_empty_line = db.booking_lines_from_postvars(postvars)
- # elif prefix + '/add_free' == parsed_url.path and not 'revert' in postvars.keys():
- # lines = postvars['booking'][0].splitlines()
- # # validate where appropriate
- # if ('save' in postvars.keys()) or ('check' in postvars.keys()):
- # _, _ = parse_lines(lines)
- # # if saving, process where to and where to redirect after
- # if 'save' in postvars.keys():
- # last_date = str(datetime.now())[:10]
- # if len(db.bookings) > 0:
- # last_date = db.bookings[-1].date_string
- # target_date = last_date[:]
- # first_line_tokens = lines[0].split() if len(lines) > 0 else ''
- # first_token = first_line_tokens[0] if len(first_line_tokens) > 0 else ''
- # try:
- # datetime.strptime(first_token, '%Y-%m-%d')
- # target_date = first_token
- # except ValueError:
- # pass
- # if start == end == 0:
- # start = db.insert_at_date(lines, target_date)
- # nth = db.get_nth_for_booking_of_start_line(start)
- # else:
- # new_start = db.update(start, end, lines, target_date)
- # print("DEBUG save", new_start, start, end, lines)
- # nth = db.get_nth_for_booking_of_start_line(new_start)
- # if new_start > start:
- # nth -= 1
- # self.redirect(prefix + f'/#{nth}')
- # # otherwise just re-build editing form
- # else:
- # if prefix + '/add_structured' == parsed_url.path:
- # edit_content = db.add_structured(start, end, temp_lines=lines, add_empty_line=add_empty_line)
- # else:
- # edit_content = db.add_free(start, end)
- # header = jinja2.Template(html_head).render(prefix=prefix)
- # self.send_HTML(header + edit_content)
+ self.redirect('ledger#{index}')
def do_GET(self):
self.try_do(self.forward_gets)
@@ -1315,18 +589,22 @@ class LedgerHandler(PlomHandler):
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 parsed_url.path == prefix + '/balance':
- stop = params.get('stop', [None])[0]
+ if site == 'balance':
+ stop = params.get('until_after', [None])[0]
page = db.balance_as_html(stop)
- elif parsed_url.path == prefix + '/edit':
+ elif site == 'edit':
index = params.get('i', [-1])[0]
- page = db.edit(int(index))
+ copy = params.get('copy', [0])[0]
+ page = db.edit(int(index), copy=bool(copy))
else:
page = db.ledger_as_html()
self.send_HTML(page)