X-Git-Url: https://plomlompom.com/repos/do_day?a=blobdiff_plain;f=ledger.py;h=2a9a89c3f494cedd2472d34dcd6a251e47740405;hb=01c660c082033cb534453bbd7d597e240c30610b;hp=60bbf640a4d117ca548a407d727375cc60dbad1d;hpb=6061ff7f1948b83dc601f52df1119a57b6e91649;p=misc
diff --git a/ledger.py b/ledger.py
index 60bbf64..2a9a89c 100755
--- a/ledger.py
+++ b/ledger.py
@@ -2,6 +2,7 @@ from http.server import BaseHTTPRequestHandler, HTTPServer
import sys
import os
import html
+import jinja2
from urllib.parse import parse_qs, urlparse
hostName = "localhost"
serverPort = 8082
@@ -27,7 +28,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 +133,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
@@ -244,6 +245,7 @@ def parse_lines(lines):
raise HandledException(f"{prefix} last booking unfinished")
return bookings, comments
+
class Booking:
def __init__(self, date_string, description, booking_lines, start_line):
@@ -316,7 +318,7 @@ class Database:
self.real_lines = []
if os.path.exists(self.db_file):
with open(self.db_file, "r") as f:
- self.real_lines += f.readlines()
+ self.real_lines += [l.rstrip() for l in f.readlines()]
ret = parse_lines(self.real_lines)
self.bookings += ret[0]
self.comments += ret[1]
@@ -332,7 +334,9 @@ class Database:
shutil.copy(self.db_file, self.db_file + ".bak")
f = open(self.lock_file, 'w+')
f.close()
- text = ''.join(self.real_lines[:start]) + '\n'.join(lines) + ''.join(self.real_lines[end:])
+ total_lines = self.real_lines[:start] + lines + self.real_lines[end:]
+ text = '\n'.join(total_lines)
+ # text = '\n'.join(self.real_lines[:start]) + '\n'.join(lines) + '\n'.join(self.real_lines[end:])
with open(self.db_file, 'w') as f:
f.write(text);
os.remove(self.lock_file)
@@ -346,7 +350,7 @@ class Database:
f = open(self.lock_file, 'w+')
f.close()
with open(self.db_file, 'a') as f:
- f.write('\n' + '\n'.join(lines) + '\n');
+ f.write('\n\n' + '\n'.join(lines) + '\n\n');
os.remove(self.lock_file)
@@ -376,13 +380,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 +404,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 +450,6 @@ 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])
if parsed_url.path == '/balance':
page += self.balance_as_html(db)
elif parsed_url.path == '/add_free':
@@ -470,107 +490,156 @@ input[type=number] { text-align: right; font-family: monospace; }
return f"
{content} "
def ledger_as_html(self, db):
- lines = []
- for comment in db.comments:
- lines += [f'' if comment != '' else '']
+ single_c_tmpl = jinja2.Template(' ')
+ booking_tmpl = jinja2.Template("""
+{{date}} {{desc}}
+[edit: structured
+/ free
+| copy:structured
+/ free ]
+
+{% for l in booking_lines %}
+{% if l.acc %}
+{{l.acc}} {{l.money}}
+{% else %}
+
+{% endif %}
+{% endfor %}
+
+""")
+ elements_to_write = []
+ last_i = i = 0
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]}
-[edit: structured
-/ free
-| copy:structured
-/ free ]
-"""
+ elements_to_write += [single_c_tmpl.render(c=c) for c in db.comments[last_i:i] if c != '']
+ booking_end = last_i = booking.start_line + len(booking.lines)
+ booking_lines = []
for booking_line in booking.lines[1:]:
i += 1
+ comment = db.comments[i]
if booking_line == '':
- lines[i] = f' '
+ booking_lines += [{'acc': None, 'money': None, 'comment': comment}]
continue
- comment = f' {lines[i]}' if len(lines[i]) > 0 else ''
money = ''
if booking_line[1]:
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" + '
'
- return '\n'.join(lines)
-
- def header_add_form(self, action):
- return f""""
-
- def textarea(self, name, lines, min_rows=1, min_cols=80):
- safe_content = html.escape(''.join(lines))
- n_rows = max(min_rows, len(lines))
- n_cols = min_cols
- for line in lines:
- n_cols = len(line) if len(line) > n_cols else n_cols
- return f''
+ booking_lines += [{'acc': booking_line[0], 'money':money, 'comment':comment}]
+ elements_to_write += [booking_tmpl.render(
+ start=booking.start_line,
+ end=booking_end,
+ date=booking.date_string,
+ desc=booking.description,
+ head_comment=db.comments[booking.start_line],
+ booking_lines = booking_lines)]
+ return '\n'.join(elements_to_write)
def add_free(self, db, start=0, end=0, copy=False):
+ tmpl = jinja2.Template("""
+
+""")
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)}'
-
- def add_structured(self, db, start=0, end=0, bonus_lines=10, copy=False):
+ if copy:
+ start = end = 0
+ return tmpl.render(start=start, end=end, lines=lines)
+
+ def add_structured(self, db, start=0, end=0, copy=False, temp_lines=[], add_empty_line=None):
+ tmpl = jinja2.Template("""
+
+
+
+
+
+
+
+{{head_comment}}
+
+
+{% for line in booking_lines %}
+
+
+
+{{line.comment}}
+
+
+
+{% endfor %}
+{% for name, items in datalist_sets.items() %}
+
+{% for item in items %}
+ {{item}}
+{% endfor %}
+
+{% endfor %}
+
+
+
+
+""")
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
- def inpu(name, val="", datalist="", input_type='text', size=-1):
- val = val if val is not None else ""
- safe_val = html.escape(str(val))
- datalist_string = '' if datalist == '' else f'list="{datalist}"'
- 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') + ' '
+ 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 = 'add_structured'
+ datalist_sets = {'descriptions': set(), 'accounts': set(), 'currencies': set()}
+ for b in db.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.datetime.now())[:10]
+ booking_lines = []
+ if copy:
+ start = end = 0
+ desc = head_comment = ''
if len(bookings) == 0:
- input_lines += f'{inpu("date", today, size=9)} {inpu("description", "", "descriptions")} ; {inpu("line_0_comment")} '
- last_line = 1
+ for i in range(1, 3):
+ booking_lines += [{'i': i, 'acc': '', 'amt': '', 'curr': '', 'comment': ''}]
+ date=today
else:
booking = bookings[0]
+ desc = booking.description
+ date = today if copy else booking.date_string
+ head_comment=comments[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]])} '
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", [""])} '
- datalist_sets = {'descriptions': set(), 'accounts': set(), 'currencies': set()}
- for b in db.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)
- def build_datalist(name):
- datalist = f'' + "\n"
- for item in datalist_sets[name]:
- safe_item = html.escape(item)
- datalist += f'{safe_item} ' + "\n"
- return f"{datalist} \n"
- datalists = build_datalist('descriptions')
- datalists += build_datalist('accounts')
- datalists += build_datalist('currencies')
- return f'{self.header_add_form("add_structured")}{input_lines}{datalists}{self.footer_add_form(start, end, copy)}'
+ booking_lines += [{
+ 'i': i,
+ 'acc': account,
+ 'amt': amount,
+ 'curr': currency if currency else '',
+ 'comment': comments[i],
+ 'comm_cols': len(comments[i])}]
+ 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
if __name__ == "__main__":