From df13fc64930a7d05caf4ded471a27c4ab39faf32 Mon Sep 17 00:00:00 2001
From: Christian Heller
Date: Thu, 26 Oct 2023 04:52:06 +0200
Subject: [PATCH] Improve ledger.py.
---
ledger.py | 83 ++++++++++++++++++++++++++++++++++---------------------
1 file changed, 52 insertions(+), 31 deletions(-)
diff --git a/ledger.py b/ledger.py
index 60bbf64..80ca0f6 100755
--- a/ledger.py
+++ b/ledger.py
@@ -27,7 +27,7 @@ def apply_booking_to_account_balances(account_sums, account, currency, amount):
def add_taxes(lines):
import decimal
- bookings, _ = parse_lines(lines.copy())
+ bookings, _ = parse_lines(lines)
_, account_sums = bookings_to_account_tree(bookings)
expenses_so_far = -1 * account_sums['Assets']['â¬']
needed_income_before_krankenkasse = expenses_so_far
@@ -132,7 +132,7 @@ def parse_lines(lines):
start_line = 0
bookings = []
comments = []
- lines += [''] # to ensure a booking-ending last line
+ lines = lines.copy() + [''] # to ensure a booking-ending last line
for i, line in enumerate(lines):
prefix = f"line {i}"
# we start with the case of an utterly empty line
@@ -376,13 +376,22 @@ input[type=number] { text-align: right; font-family: monospace; }
length = int(self.headers['content-length'])
postvars = parse_qs(self.rfile.read(length).decode(), keep_blank_values=1)
parsed_url = urlparse(self.path)
- if '/add_structured' == parsed_url.path:
+ lines = []
+ add_empty_line = None
+ if '/add_structured' == parsed_url.path and not 'revert' in postvars.keys():
date = postvars['date'][0]
description = postvars['description'][0]
start_comment = postvars['line_0_comment'][0]
lines = [f'{date} {description} ; {start_comment}']
- i = 1
+ 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
+ if 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]
@@ -391,30 +400,38 @@ input[type=number] { text-align: right; font-family: monospace; }
new_main = f'{account} {amount} {currency}'
if '' == new_main.rstrip() == comment.rstrip(): # don't write empty lines
continue
+ j += 1
new_line = new_main
if comment.rstrip() != '':
new_line += f' ; {comment}'
lines += [new_line]
- if 'add_income_tax' in postvars.keys():
+ if 'add_taxes' in postvars.keys():
lines += add_taxes(lines)
elif '/add_free' == parsed_url.path:
lines = postvars['booking'][0].splitlines()
+ lines += [''] # to ensure Booking-ending last line
start = int(postvars['start'][0])
end = int(postvars['end'][0])
try:
_, _ = parse_lines(lines)
- if start == end == 0:
- db.append(lines)
+ if 'save' in postvars.keys():
+ if start == end == 0:
+ db.append(lines)
+ else:
+ db.replace(start, end, lines)
+ self.send_response(301)
+ redir_url = '/'
+ self.send_header('Location', redir_url)
+ self.end_headers()
else:
- db.replace(start, end, lines)
- self.send_response(301)
- redir_url = '/'
- if 'add_income_tax' in postvars.keys():
- redir_url = f'/add_structured?start={start}&end={start+len(lines)-1}'
- self.send_header('Location', redir_url)
- self.end_headers()
+ page = self.header + self.add_structured(db, start, end, temp_lines=lines, add_empty_line=add_empty_line) + self.footer
+ self.send_response(200)
+ self.send_header("Content-type", "text/html")
+ self.end_headers()
+ self.wfile.write(bytes(page, "utf-8"))
except HandledException as e:
self.send_response(400)
+ self.send_header("Content-type", "text/html")
self.end_headers()
page = f'{self.header}ERROR: {e}{self.footer}'
self.wfile.write(bytes(page, "utf-8"))
@@ -429,7 +446,7 @@ input[type=number] { text-align: right; font-family: monospace; }
params = parse_qs(parsed_url.query)
start = int(params.get('start', ['0'])[0])
end = int(params.get('end', ['0'])[0])
- bonus_lines = int(params.get('bonus_lines', ['0'])[0])
+ # bonus_lines = int(params.get('bonus_lines', ['0'])[0])
if parsed_url.path == '/balance':
page += self.balance_as_html(db)
elif parsed_url.path == '/add_free':
@@ -476,7 +493,7 @@ input[type=number] { text-align: right; font-family: monospace; }
for booking in db.bookings:
i = booking.start_line
booking_end = booking.start_line + len(booking.lines)
- lines[i] = f"""{booking.date_string} {booking.description} {lines[i]}
+ lines[i] = f"""{booking.date_string} {booking.description} {lines[i]}
[edit: structured
/ free
| copy:structured
@@ -493,7 +510,7 @@ input[type=number] { text-align: right; font-family: monospace; }
money = f'{booking_line[1]} {booking_line[2]}'
account = booking_line[0]
lines[i] = f'
{booking_line[0]} | {money} | {comment} |
'
- lines[i] = lines[i] + "\n" + ''
+ lines[i] = lines[i] + "\n" + '
'
return '\n'.join(lines)
def header_add_form(self, action):
@@ -505,10 +522,10 @@ input[type=number] { text-align: right; font-family: monospace; }
return f"""
-
+
"""
- def textarea(self, name, lines, min_rows=1, min_cols=80):
+ def textarea(self, name, lines, min_rows=1, min_cols=20):
safe_content = html.escape(''.join(lines))
n_rows = max(min_rows, len(lines))
n_cols = min_cols
@@ -518,15 +535,18 @@ input[type=number] { text-align: right; font-family: monospace; }
def add_free(self, db, start=0, end=0, copy=False):
lines = db.get_lines(start, end)
- return f'{self.header_add_form("add_free")}{self.textarea("booking",lines,10)}{self.footer_add_form(start, end, copy)}'
+ return f'{self.header_add_form("add_free")}{self.textarea("booking",lines,10,80)}{self.footer_add_form(start, end, copy)}'
- def add_structured(self, db, start=0, end=0, bonus_lines=10, copy=False):
+ def add_structured(self, db, start=0, end=0, copy=False, temp_lines=[], add_empty_line=None):
import datetime
- lines = db.get_lines(start, end)
+ lines = temp_lines if len(''.join(temp_lines)) > 0 else db.get_lines(start, end)
bookings, comments = parse_lines(lines)
if len(bookings) > 1:
raise HandledException('can only edit single Booking')
- last_line = 0
+ 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:]
def inpu(name, val="", datalist="", input_type='text', size=-1):
val = val if val is not None else ""
safe_val = html.escape(str(val))
@@ -534,26 +554,27 @@ input[type=number] { text-align: right; font-family: monospace; }
number_step = '' if input_type != 'number' else ' step=0.01'
size_string = '' if size < 0 else f' size={size}'
return f''
- input_lines = inpu('add_income_tax', 'add income tax', '', 'submit') + '
'
+ input_lines = inpu('check', 'check', '', 'submit') +\
+ inpu('revert', 'revert', '', 'submit') +\
+ inpu('add_taxes', 'add taxes', '', 'submit') +\
+ '
'
today = str(datetime.datetime.now())[:10]
if len(bookings) == 0:
- input_lines += f'{inpu("date", today, size=9)} {inpu("description", "", "descriptions")} ; {inpu("line_0_comment")}
'
- last_line = 1
+ input_lines += f'{inpu("date", today, size=9)} {inpu("description", "", "descriptions")} ; {inpu("line_0_comment")} {inpu(f"line_0_add", "[+]", "", "submit")}
'
+ for i in range(1, 3):
+ input_lines += f'{inpu(f"line_{i}_account", "", "accounts", size=40)} {inpu(f"line_{i}_amount", "", "", "number", size=10)} {inpu(f"line_{i}_currency", "", "currencies", size=3)} ; {self.textarea(f"line_{i}_comment", "")} {inpu(f"line_{i}_delete", "[x]", "", "submit")} {inpu(f"line_{i}_add", "[+]", "", "submit")}
'
else:
booking = bookings[0]
last_line = len(comments)
date_string = today if copy else booking.date_string
- input_lines += f'{inpu("date", date_string, size=9)} {inpu("description", booking.description, "descriptions")} ; {self.textarea("line_0_comment", [comments[0]])}
'
+ input_lines += f'{inpu("date", date_string, size=9)} {inpu("description", booking.description, "descriptions")} ; {self.textarea("line_0_comment", [comments[0]])} {inpu(f"line_0_add", "[+]", "", "submit")}
'
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]
- input_lines += f'{inpu(f"line_{i}_account", account, "accounts", size=40)} {inpu(f"line_{i}_amount", amount, "", "number", size=10)} {inpu(f"line_{i}_currency", currency, "currencies", size=3)} ; {self.textarea(f"line_{i}_comment", [comments[i]])}
'
- for j in range(bonus_lines):
- i = j + last_line
- input_lines += f'{inpu(f"line_{i}_account", "", "accounts", size=40)} {inpu(f"line_{i}_amount", "", "", "number", size=10)} {inpu(f"line_{i}_currency", "", "currencies", size=3)} ; {self.textarea(f"line_{i}_comment", [""])}
'
+ input_lines += f'{inpu(f"line_{i}_account", account, "accounts", size=40)} {inpu(f"line_{i}_amount", amount, "", "number", size=10)} {inpu(f"line_{i}_currency", currency, "currencies", size=3)} ; {self.textarea(f"line_{i}_comment", [comments[i]])} {inpu(f"line_{i}_delete", "[x]", "", "submit")} {inpu(f"line_{i}_add", "[+]", "", "submit")}
'
datalist_sets = {'descriptions': set(), 'accounts': set(), 'currencies': set()}
for b in db.bookings:
datalist_sets['descriptions'].add(b.description)
--
2.30.2