home · contact · privacy
Further improvements to todo accounting.
[misc] / ledger.py
1 import os
2 import jinja2
3 import decimal
4 from datetime import datetime, timedelta
5 from urllib.parse import parse_qs, urlparse
6 from plomlib import PlomDB, PlomException, run_server, PlomHandler 
7
8 server_port = 8082
9 db_path = '/home/plom/org/ledger2023.dat'
10
11 html_head = """
12 <style>
13 body { color: #000000; }
14 table { margin-bottom: 2em; }
15 th, td { text-align: left }
16 input[type=number] { text-align: right; font-family: monospace; }
17 .money { font-family: monospace; text-align: right; }
18 .comment { font-style: italic; color: #777777; }
19 .meta { font-size: 0.75em; color: #777777; }
20 .full_line_comment { display: block; white-space: nowrap; width: 0; }
21 </style>
22 <body>
23 <a href="{{prefix}}/ledger">ledger</a>
24 <a href="{{prefix}}/balance">balance</a>
25 <a href="{{prefix}}/add_free">add free</a>
26 <a href="{{prefix}}/add_structured">add structured</a>
27 <hr />
28 """
29 booking_html = """
30 <p id="{{nth}}"><a href="{{prefix}}#{{nth}}">{{date}}</a> {{desc}} <span class="comment">{{head_comment|e}}</span><br />
31 <span class="meta">[edit: <a href="{{prefix}}/add_structured?start={{start}}&end={{end}}">structured</a>
32 / <a href="{{prefix}}/add_free?start={{start}}&end={{end}}">free</a>
33 | copy:<a href="{{prefix}}/copy_structured?start={{start}}&end={{end}}">structured</a>
34 / <a href="{{prefix}}/copy_free?start={{start}}&end={{end}}">free</a>
35 | move {% if move_up %}<a href="{{prefix}}/move_up?start={{start}}&end={{end}}">up</a>{% else %}up{% endif %}/{% if move_down %}<a href="{{prefix}}/move_down?start={{start}}&end={{end}}">down</a>{% else %}down{% endif %}
36 | <a href="{{prefix}}/balance?stop={{nth+1}}">balance after</a>
37 ]</span>
38 <table>
39 {% for l in booking_lines %}
40 {% if l.acc %}
41 <tr><td>{{l.acc|e}}</td><td class="money">{{l.money|e}}</td><td class="comment">{{l.comment|e}}</td></tr>
42 {% else %}
43 <tr><td><div class="comment full_line_comment">{{l.comment|e}}</div></td></tr>
44 {% endif %}
45 {% endfor %}
46 </table></p>
47 """
48 add_form_header = """<form method="POST" action="{{action|e}}">
49 <input type="submit" name="check" value="check" />
50 <input type="submit" name="revert" value="revert" />
51 """
52 add_form_footer = """
53 <input type="hidden" name="start" value={{start}} />
54 <input type="hidden" name="end" value={{end}} />
55 <input type="submit" name="save" value="save!">
56 </form>
57 """
58 add_free_html = """<br />
59 <textarea name="booking" rows=10 cols=80>
60 {% for line in lines %}{{ line }}
61 {% endfor %}
62 </textarea>
63 """
64 add_structured_html = """
65 <input type="submit" name="add_taxes" value="add taxes" />
66 <input type="submit" name="add_taxes2" value="add taxes2" />
67 <input type="submit" name="add_sink" value="add sink" />
68 <input name="replace_from" />
69 <input type="submit" name="replace" value="-> replace ->" />
70 <input name="replace_to" />
71 <input type="submit" name="add_mirror" value="add mirror" />
72 <br />
73 <input name="date" value="{{date|e}}" size=9 />
74 <input name="description" value="{{desc|e}}" list="descriptions" />
75 <textarea name="line_0_comment" rows=1 cols=20>{{head_comment|e}}</textarea>
76 <input type="submit" name="line_0_add" value="[+]" />
77 <br />
78 {% for line in booking_lines %}
79 <input name="line_{{line.i}}_account" value="{{line.acc|e}}" size=40 list="accounts" />
80 <input type="number" name="line_{{line.i}}_amount" step=0.01 value="{{line.amt}}" size=10 />
81 <input name="line_{{line.i}}_currency" value="{{line.curr|e}}" size=3 list="currencies" />
82 <input type="submit" name="line_{{line.i}}_delete" value="[x]" />
83 <input type="submit" name="line_{{line.i}}_delete_after" value="[XX]" />
84 <input type="submit" name="line_{{line.i}}_add" value="[+]" />
85 <textarea name="line_{{line.i}}_comment" rows=1 cols={% if line.comm_cols %}{{line.comm_cols}}{% else %}20{% endif %}>{{line.comment|e}}</textarea>
86 <br />
87 {% endfor %}
88 {% for name, items in datalist_sets.items() %}
89 <datalist id="{{name}}">
90 {% for item in items %}
91   <option value="{{item|e}}">{{item|e}}</option>
92 {% endfor %}
93 </datalist>
94 {% endfor %}
95 """
96
97
98 def apply_booking_to_account_balances(account_sums, account, currency, amount):
99     if not account in account_sums:
100         account_sums[account] = {currency: amount}
101     elif not currency in account_sums[account].keys():
102         account_sums[account][currency] = amount
103     else:
104         account_sums[account][currency] += amount
105
106
107 def bookings_to_account_tree(bookings):
108     account_sums = {}
109     for booking in bookings:
110         for account, changes in booking.account_changes.items():
111             for currency, amount in changes.items():
112                 apply_booking_to_account_balances(account_sums, account, currency, amount)
113     account_tree = {}
114     def collect_branches(account_name, path):
115         node = account_tree
116         path_copy = path[:]
117         while len(path_copy) > 0:
118             step = path_copy.pop(0)
119             node = node[step]
120         toks = account_name.split(":", maxsplit=1)
121         parent = toks[0]
122         if parent in node.keys():
123             child = node[parent]
124         else:
125             child = {}
126             node[parent] = child
127         if len(toks) == 2:
128             k, v = collect_branches(toks[1], path + [parent])
129             if k not in child.keys():
130                 child[k] = v
131             else:
132                 child[k].update(v)
133         return parent, child
134     for account_name in sorted(account_sums.keys()):
135         k, v = collect_branches(account_name, [])
136         if k not in account_tree.keys():
137             account_tree[k] = v
138         else:
139             account_tree[k].update(v)
140     def collect_totals(parent_path, tree_node):
141         for k, v in tree_node.items():
142             child_path = parent_path + ":" + k
143             for currency, amount in collect_totals(child_path, v).items():
144                 apply_booking_to_account_balances(account_sums, parent_path, currency, amount)
145         return account_sums[parent_path]
146     for account_name in account_tree.keys():
147         account_sums[account_name] = collect_totals(account_name, account_tree[account_name])
148     return account_tree, account_sums
149
150
151 def parse_lines(lines, validate_bookings=True):
152     inside_booking = False
153     date_string, description = None, None
154     booking_lines = []
155     start_line = 0
156     bookings = []
157     comments = []
158     lines = lines.copy() + [''] # to ensure a booking-ending last line
159     last_date = ''
160     for i, line in enumerate(lines):
161         prefix = f"line {i}"
162         # we start with the case of an utterly empty line
163         comments += [""]
164         stripped_line = line.rstrip()
165         if stripped_line == '':
166             if inside_booking:
167                 # assume we finished a booking, finalize, and commit to DB
168                 if len(booking_lines) < 2:
169                     raise PlomException(f"{prefix} booking ends to early")
170                 booking = Booking(date_string, description, booking_lines, start_line, validate_bookings)
171                 bookings += [booking]
172             # expect new booking to follow so re-zeroall booking data
173             inside_booking = False
174             date_string, description = None, None
175             booking_lines = []
176             continue
177         # if non-empty line, first get comment if any, and commit to DB
178         split_by_comment = stripped_line.split(sep=";", maxsplit=1)
179         if len(split_by_comment) == 2:
180             comments[i] = split_by_comment[1].lstrip()
181         # 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
182         non_comment = split_by_comment[0].rstrip()
183         if non_comment.rstrip() == '':
184              if inside_booking:
185                  booking_lines += ['']
186              continue
187         # if we're starting a booking, parse by first-line pattern
188         if not inside_booking:
189             start_line = i
190             toks = non_comment.split(maxsplit=1)
191             date_string = toks[0]
192             try:
193                 datetime.strptime(date_string, '%Y-%m-%d')
194             except ValueError:
195                 raise PlomException(f"{prefix} bad date string: {date_string}")
196             if last_date > date_string:
197                 raise PlomException(f"{prefix} out-of-order-date")
198             last_date = date_string
199             try:
200                 description = toks[1]
201             except IndexError:
202                 raise PlomException(f"{prefix} bad description: {description}")
203             inside_booking = True
204             booking_lines += [non_comment]
205             continue
206         # otherwise, read as transfer data
207         toks = non_comment.split()  # ignore specification's allowance of single spaces in names
208         if len(toks) > 3:
209             raise PlomException(f"{prefix} too many booking line tokens: {toks}")
210         amount, currency = None, None
211         account_name = toks[0]
212         if account_name[0] == '[' and account_name[-1] == ']':
213             # ignore specification's differentiation of "virtual" accounts
214             account_name = account_name[1:-1]
215         decimal_chars = ".-0123456789"
216         if len(toks) == 3:
217             i_currency = 1
218             try:
219                 amount = decimal.Decimal(toks[1])
220                 i_currency = 2
221             except decimal.InvalidOperation:
222                 try:
223                     amount = decimal.Decimal(toks[2])
224                 except decimal.InvalidOperation:
225                     raise PlomException(f"{prefix} no decimal number in: {toks[1:]}")
226             currency = toks[i_currency]
227             if currency[0] in decimal_chars:
228                 raise PlomException(f"{prefix} currency starts with int, dot, or minus: {currency}")
229         elif len(toks) == 2:
230             value = toks[1]
231             inside_amount = False
232             inside_currency = False
233             amount_string = ""
234             currency = ""
235             dots_counted = 0
236             for i, c in enumerate(value):
237                 if i == 0:
238                     if c in decimal_chars:
239                         inside_amount = True
240                     else:
241                         inside_currency = True
242                 if inside_currency:
243                     if c in decimal_chars and len(amount_string) == 0:
244                         inside_currency = False
245                         inside_amount = True
246                     else:
247                         currency += c
248                         continue
249                 if inside_amount:
250                     if c not in decimal_chars:
251                         if len(currency) > 0:
252                             raise PlomException(f"{prefix} amount has non-decimal chars: {value}")
253                         inside_currency = True
254                         inside_amount = False
255                         currency += c
256                         continue
257                     if c == '-' and len(amount_string) > 1:
258                         raise PlomException(f"{prefix} amount has non-start '-': {value}")
259                     if c == '.':
260                         if dots_counted > 1:
261                             raise PlomException(f"{prefix} amount has multiple dots: {value}")
262                         dots_counted += 1
263                     amount_string += c
264             if len(currency) == 0:
265                 raise PlomException(f"{prefix} currency missing: {value}")
266             if len(amount_string) > 0:
267                 amount = decimal.Decimal(amount_string)
268         booking_lines += [(account_name, amount, currency)]
269     if inside_booking:
270         raise PlomException(f"{prefix} last booking unfinished")
271     return bookings, comments
272
273
274 class Booking:
275
276     def __init__(self, date_string, description, booking_lines, start_line, process=True):
277         self.date_string = date_string
278         self.description = description
279         self.lines = booking_lines
280         self.start_line = start_line
281         if process:
282             self.validate_booking_lines()
283             self.sink = {}
284             self.account_changes = self.parse_booking_lines_to_account_changes()
285
286     def validate_booking_lines(self):
287         prefix = f"booking at line {self.start_line}"
288         sums = {}
289         empty_values = 0
290         for line in self.lines[1:]:
291             if line == '':
292                 continue
293             _, amount, currency = line
294             if amount is None:
295                 if empty_values > 0:
296                     raise PlomException(f"{prefix} relates more than one empty value of same currency {currency}")
297                 empty_values += 1
298                 continue
299             if currency not in sums:
300                 sums[currency] = 0
301             sums[currency] += amount
302         if empty_values == 0:
303             for k, v in sums.items():
304                 if v != 0:
305                     raise PlomException(f"{prefix} does not add up to zero / {k} {v}")
306         else:
307             sinkable = False
308             for k, v in sums.items():
309                 if v != 0:
310                     sinkable = True
311             if not sinkable:
312                 raise PlomException(f"{prefix} has empty value that cannot be filled")
313
314     def parse_booking_lines_to_account_changes(self):
315         account_changes = {}
316         debt = {}
317         sink_account = None
318         for line in self.lines[1:]:
319             if line == '':
320                 continue
321             account, amount, currency = line
322             if amount is None:
323                 sink_account = account
324                 continue
325             apply_booking_to_account_balances(account_changes, account, currency, amount)
326             if currency not in debt:
327                 debt[currency] = amount
328             else:
329                 debt[currency] += amount
330         if sink_account:
331             for currency, amount in debt.items():
332                 apply_booking_to_account_balances(account_changes, sink_account, currency, -amount)
333                 self.sink[currency] = -amount
334         return account_changes
335
336
337
338 class LedgerDB(PlomDB):
339
340     def __init__(self, prefix):
341         self.prefix = prefix 
342         self.bookings = []
343         self.comments = []
344         self.real_lines = []
345         super().__init__(db_path)
346         ret = parse_lines(self.real_lines)
347         self.bookings += ret[0]
348         self.comments += ret[1]
349
350     def read_db_file(self, f):
351         self.real_lines += [l.rstrip() for l in f.readlines()]
352
353     def get_lines(self, start, end):
354         return self.real_lines[start:end]
355
356     def write_db(self, text, mode='w'):
357         if text[-1] != '\n':
358             text += '\n'
359         self.write_text_to_db(text)
360
361     def insert_at_date(self, lines, date):
362         print("DEBUG insert_at_date")
363         start_at = 0 
364         if len(self.bookings) > 0:
365             if date >= self.bookings[-1].date_string:
366                 start_at = len(self.real_lines)
367                 lines = [''] + lines
368             else:
369                 for b in self.bookings:
370                     if b.date_string == date:
371                         start_at = b.start_line 
372                     elif b.date_string > date:
373                         start_at = b.start_line 
374                         break
375         return self.write_lines_in_total_lines_at(self.real_lines, start_at, lines)
376
377     def update(self, start, end, lines, date):
378         print("DEBUG update")
379         total_lines = self.real_lines[:start] + self.real_lines[end:]
380         n_original_lines = end - start
381         start_at = len(total_lines)
382         for b in self.bookings:
383             if b.date_string == date:
384                 if start_at == len(total_lines) or b.start_line == start:
385                     start_at = b.start_line 
386                     if b.start_line > start:
387                         start_at -= n_original_lines
388             elif b.date_string > date:
389                 break
390         if start_at == len(total_lines):
391             lines = [''] + lines
392         return self.write_lines_in_total_lines_at(total_lines, start_at, lines)
393
394     def write_lines_in_total_lines_at(self, total_lines, start_at, lines):
395         total_lines = total_lines[:start_at] + lines + [''] + total_lines[start_at:]
396         _, _ = parse_lines(lines)
397         text = '\n'.join(total_lines)
398         self.write_db(text)
399         return start_at
400
401     def get_nth_for_booking_of_start_line(self, start_line):
402         nth = 0
403         for b in self.bookings:
404             if b.start_line >= start_line:
405                 break
406             nth += 1
407         return nth
408
409     def add_taxes(self, lines, finish=False):
410         ret = []
411         bookings, _ = parse_lines(lines)
412         date = bookings[0].date_string
413         acc_kk_add = 'Reserves:KrankenkassenBeitragsWachstum'
414         acc_kk_minimum = 'Reserves:Month:KrankenkassenDefaultBeitrag'
415         acc_kk = 'Expenses:KrankenKasse'
416         acc_est = 'Reserves:Einkommenssteuer'
417         acc_assets = 'Assets'
418         acc_buffer = 'Reserves:NeuAnfangsPuffer:Ausgaben'
419         last_monthbreak_assets = 0
420         last_monthbreak_est = 0
421         last_monthbreak_kk_minimum = 0
422         last_monthbreak_kk_add = 0
423         buffer_expenses = 0
424         kk_expenses = 0
425         est_expenses = 0
426         months_passed = -int(finish) 
427         for b in self.bookings:
428             if date == b.date_string:
429                 break
430             acc_keys = b.account_changes.keys()
431             if acc_buffer in acc_keys:
432                 buffer_expenses -= b.account_changes[acc_buffer]['€']
433             if acc_kk_add in acc_keys:
434                 kk_expenses += b.account_changes[acc_kk_add]['€']
435             if acc_kk in acc_keys:
436                 kk_expenses += b.account_changes[acc_kk]['€']
437             if acc_est in acc_keys:
438                 est_expenses += b.account_changes[acc_est]['€']
439             if acc_kk_add in acc_keys and acc_kk_minimum in acc_keys:
440                 months_passed += 1
441                 if finish:
442                     last_monthbreak_kk_add = b.account_changes[acc_kk_add]['€']
443                     last_monthbreak_est = b.account_changes[acc_est]['€']
444                     last_monthbreak_kk_minimum = b.account_changes[acc_kk_minimum]['€']
445                     last_monthbreak_assets = b.account_changes[acc_buffer]['€']
446         old_needed_income_before_anything = - last_monthbreak_assets - last_monthbreak_kk_add - last_monthbreak_kk_minimum - last_monthbreak_est
447         if finish:
448             ret += [f'  {acc_est}  {-last_monthbreak_est}€ ; for old assumption of needed income: {old_needed_income_before_anything}€']
449         _, account_sums = bookings_to_account_tree(bookings)
450         expenses_so_far = -1 * account_sums[acc_assets]['€'] + old_needed_income_before_anything
451         needed_income_before_kk = expenses_so_far
452         ESt_this_month = 0
453         left_over = needed_income_before_kk - ESt_this_month
454         too_low = 0
455         too_high = 2 * needed_income_before_kk
456         E0 = decimal.Decimal(10908)
457         E1 = decimal.Decimal(15999)
458         E2 = decimal.Decimal(62809)
459         E3 = decimal.Decimal(277825)
460         while True:
461             zvE = buffer_expenses - kk_expenses + (12 - months_passed) * needed_income_before_kk
462             if finish:
463                 zvE += last_monthbreak_assets + last_monthbreak_kk_add + last_monthbreak_kk_minimum
464             if zvE < E0:
465                 ESt = decimal.Decimal(0)
466             elif zvE < E1:
467                 y = (zvE - E0)/10000
468                 ESt = (decimal.Decimal(979.18) * y + 1400) * y
469             elif zvE < E2:
470                 y = (zvE - E1)/10000
471                 ESt = (decimal.Decimal(192.59) * y + 2397) * y + decimal.Decimal(966.53)
472             elif zvE < E3:
473                 ESt = decimal.Decimal(0.42) * (zvE - decimal.Decimal(62809))  + decimal.Decimal(16405.54)
474             else:
475                 ESt = decimal.Decimal(0.45) * (zvE - decimal.Decimal(277825)) + decimal.Decimal(106713.52)
476             ESt_this_month = (ESt + last_monthbreak_est - est_expenses) / (12 - months_passed)
477             left_over = needed_income_before_kk - ESt_this_month
478             if abs(left_over - expenses_so_far) < 0.001:
479                 break
480             elif left_over < expenses_so_far:
481                 too_low = needed_income_before_kk
482             elif left_over > expenses_so_far:
483                 too_high = needed_income_before_kk
484             needed_income_before_kk = too_low + (too_high - too_low)/2
485         ESt_this_month = ESt_this_month.quantize(decimal.Decimal('0.00'))
486         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}€']
487         kk_minimum_income = 1131.67
488         if date < '2023-02-01':
489             kk_minimum_income = decimal.Decimal(1096.67)
490             kk_factor = decimal.Decimal(0.189)
491             kk_minimum_tax = decimal.Decimal(207.27).quantize(decimal.Decimal('0.00'))
492         elif date < '2023-08-01':
493             kk_factor = decimal.Decimal(0.191)
494             kk_minimum_tax = decimal.Decimal(216.15).quantize(decimal.Decimal('0.00'))
495         else:
496             kk_factor = decimal.Decimal(0.197)
497             kk_minimum_tax = decimal.Decimal(222.94).quantize(decimal.Decimal('0.00'))
498         kk_add_so_far = account_sums[acc_kk_add]['€'] if acc_kk_add in account_sums.keys() else 0
499         kk_add = needed_income_before_kk / (1 - kk_factor) - needed_income_before_kk - kk_minimum_tax
500         hit_kk_minimum_income_limit = False
501         if kk_add_so_far + kk_add < 0:
502             hit_kk_minimum_income_limit = True 
503             kk_add_uncorrect = kk_add
504             kk_add = -(kk_add + kk_add_so_far) 
505         kk_add = decimal.Decimal(kk_add).quantize(decimal.Decimal('0.00'))
506         if finish:
507             ret += [f'  {acc_kk_add}  {-last_monthbreak_kk_add}€  ; for old assumption of needed income']
508         else:
509             ret += [f'  {acc_kk_minimum}  {kk_minimum_tax}€  ; assumed minimum income {kk_minimum_income:.2f}€ * {kk_factor:.3f}']
510         if hit_kk_minimum_income_limit:
511             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']
512         else:
513             ret += [f'  {acc_kk_add}  {kk_add}€  ; {needed_income_before_kk:.2f}€ / (1 - {kk_factor:.3f}) - {needed_income_before_kk:.2f}€ - {kk_minimum_tax}€']
514         diff = - last_monthbreak_est + ESt_this_month - last_monthbreak_kk_add + kk_add
515         if not finish:
516             diff += kk_minimum_tax
517         final_minus = expenses_so_far - old_needed_income_before_anything + diff
518         ret += [f'  {acc_assets}  {-diff} €']
519         ret += [f'  {acc_assets}  {final_minus} €']
520         year_needed = buffer_expenses + final_minus + (12 - months_passed - 1) * final_minus 
521         if finish:
522             ret += [f'  {acc_buffer}  {-final_minus} €']
523         else:
524             ret += [f'  {acc_buffer}  {-final_minus} € ; assume as to earn in year: {acc_buffer} + {12 - months_passed - 1} * this = {year_needed}']
525         return ret
526
527     def add_mirror(self, lines):
528         ret = []
529         bookings, _ = parse_lines(lines)
530         booking = bookings[0]
531         for line in booking.lines[1:]:
532             ret += [f'?  {-line[1]} {line[2]}']
533         return ret
534
535     def ledger_as_html(self):
536         booking_tmpl = jinja2.Template(booking_html)
537         single_c_tmpl = jinja2.Template('<span class="comment">{{c|e}}</span><br />')  ##
538         elements_to_write = []
539         last_i = i = 0  ##
540         for nth, booking in enumerate(self.bookings):
541             move_up = nth > 0 and self.bookings[nth - 1].date_string == booking.date_string
542             move_down = nth < len(self.bookings) - 1 and self.bookings[nth + 1].date_string == booking.date_string
543             booking_end = last_i = booking.start_line + len(booking.lines)
544             booking_lines = []
545             i = booking.start_line  ##
546             elements_to_write += [single_c_tmpl.render(c=c) for c in self.comments[last_i:i] if c != '']  ##
547             for booking_line in booking.lines[1:]:
548                  i += 1  ##
549                  comment = self.comments[i]  ##
550                  if booking_line == '':
551                      booking_lines += [{'acc': None, 'money': None, 'comment': comment}]  ##
552                      continue
553                  account = booking_line[0]
554                  money = ''
555                  if booking_line[1] is not None:
556                      money = f'{booking_line[1]} {booking_line[2]}'
557                  booking_lines += [{'acc': booking_line[0], 'money':money, 'comment':comment}]  ##
558             elements_to_write += [booking_tmpl.render(
559                 prefix=self.prefix,
560                 nth=nth,
561                 start=booking.start_line,
562                 end=booking_end,
563                 date=booking.date_string,
564                 desc=booking.description,
565                 head_comment=self.comments[booking.start_line],
566                 move_up=move_up,
567                 move_down=move_down,
568                 booking_lines = booking_lines)]
569         elements_to_write += [single_c_tmpl.render(c=c) for c in self.comments[last_i:] if c != '']  #
570         return '\n'.join(elements_to_write)
571
572     def balance_as_html(self, until=None):
573         bookings = self.bookings[:until if until is None else int(until)]
574         lines = []
575         account_tree, account_sums = bookings_to_account_tree(bookings)
576         def print_subtree(lines, indent, node, subtree, path):
577             line = f"{indent}{node}"
578             n_tabs = 5 - (len(line) // 8)
579             line += n_tabs * "\t"
580             if "€" in account_sums[path + node].keys():
581                 amount = account_sums[path + node]["€"]
582                 line += f"{amount:9.2f} €\t"
583             else:
584                 line += f"\t\t"
585             for currency, amount in account_sums[path + node].items():
586                 if currency != '€' and amount > 0:
587                     line += f"{amount:5.2f} {currency}\t"
588             lines += [line]
589             indent += "  "
590             for k, v in sorted(subtree.items()):
591                 print_subtree(lines, indent, k, v, path + node + ":")
592         for k, v in sorted(account_tree.items()):
593             print_subtree(lines, "", k, v, "")
594         content = "\n".join(lines)
595         return f"<pre>{content}</pre>"
596
597     def add_free(self, start=0, end=0, copy=False):
598         tmpl = jinja2.Template(add_form_header + add_free_html + add_form_footer) 
599         lines = self.get_lines(start, end)
600         if copy:
601             start = end = 0
602         return tmpl.render(action=self.prefix + '/add_free', start=start, end=end, lines=lines)
603
604     def add_structured(self, start=0, end=0, copy=False, temp_lines=[], add_empty_line=None):
605         tmpl = jinja2.Template(add_form_header + add_structured_html + add_form_footer) 
606         lines = temp_lines if len(''.join(temp_lines)) > 0 else self.get_lines(start, end)
607         bookings, comments = parse_lines(lines, validate_bookings=False)
608         if len(bookings) > 1:
609             raise PlomException('can only structurally edit single Booking')
610         if add_empty_line is not None:
611             comments = comments[:add_empty_line+1] + [''] + comments[add_empty_line+1:]
612             booking = bookings[0]
613             booking.lines = booking.lines[:add_empty_line+1] + [''] + booking.lines[add_empty_line+1:]
614         action = self.prefix + '/add_structured'
615         datalist_sets = {'descriptions': set(), 'accounts': set(), 'currencies': set()}
616         for b in self.bookings:
617             datalist_sets['descriptions'].add(b.description)
618             for account, moneys in b.account_changes.items():
619                 datalist_sets['accounts'].add(account)
620                 for currency in moneys.keys():
621                     datalist_sets['currencies'].add(currency)
622         content = ''
623         today = str(datetime.now())[:10]
624         booking_lines = []
625         if copy:
626             start = end = 0
627         desc = head_comment = ''
628         if len(bookings) == 0:
629             date=today
630         else:
631             booking = bookings[0]
632             desc = booking.description
633             date = today if copy else booking.date_string
634             head_comment=comments[0]
635             for i in range(1, len(comments)):
636                 account = amount = currency = ''
637                 if i < len(booking.lines) and booking.lines[i] != '':
638                     account = booking.lines[i][0]
639                     amount = booking.lines[i][1]
640                     currency = booking.lines[i][2]
641                 booking_lines += [{
642                         'i': i,
643                         'acc': account,
644                         'amt': amount,
645                         'curr': currency if currency else '€',
646                         'comment': comments[i],
647                         'comm_cols': len(comments[i])}]
648         for i in range(len(comments), len(comments) + 8):
649             booking_lines += [{'i': i, 'acc': '', 'amt': '', 'curr': '€', 'comment': ''}]
650         content += tmpl.render(
651                 action=action,
652                 date=date,
653                 desc=desc,
654                 head_comment=head_comment,
655                 booking_lines=booking_lines,
656                 datalist_sets=datalist_sets,
657                 start=start,
658                 end=end)
659         return content
660
661     def move_up(self, start, end):
662         prev_booking = None
663         for redir_nth, b in enumerate(self.bookings):
664             if b.start_line >= start:
665                 break
666             prev_booking = b
667         start_at = prev_booking.start_line 
668         self.make_move(start, end, start_at)
669         return redir_nth - 1
670
671     def move_down(self, start, end):
672         next_booking = None
673         for redir_nth, b in enumerate(self.bookings):
674             if b.start_line > start:
675                 next_booking = b
676                 break
677         start_at = next_booking.start_line + len(next_booking.lines) - (end - start) + 1 
678         self.make_move(start, end, start_at-1)
679         return redir_nth
680
681     def make_move(self, start, end, start_at):
682         lines = self.get_lines(start, end)
683         total_lines = self.real_lines[:start-1] + self.real_lines[end:]  # +1 because we reduce the original position's two empty border lines to in-between line
684         self.write_lines_in_total_lines_at(total_lines, start_at, lines)
685
686     def booking_lines_from_postvars(self, postvars):
687         add_empty_line = None
688         date = postvars['date'][0]
689         description = postvars['description'][0]
690         start_comment = postvars['line_0_comment'][0]
691         start_line = f'{date} {description}'
692         if start_comment.rstrip() != '':
693             start_line += f' ; {start_comment}' 
694         lines = [start_line]
695         if 'line_0_add' in postvars.keys():
696             add_empty_line = 0
697         i = j = 1
698         while f'line_{i}_comment' in postvars.keys():
699             if f'line_{i}_delete' in postvars.keys():
700                 i += 1
701                 continue
702             elif f'line_{i}_delete_after' in postvars.keys():
703                 break 
704             elif f'line_{i}_add' in postvars.keys():
705                 add_empty_line = j
706             account = postvars[f'line_{i}_account'][0]
707             amount = postvars[f'line_{i}_amount'][0]
708             currency = postvars[f'line_{i}_currency'][0]
709             comment = postvars[f'line_{i}_comment'][0]
710             i += 1
711             new_main = f'  {account}  {amount}'
712             if '' == new_main.rstrip() == comment.rstrip():  # don't write empty lines, ignore currency if nothing else set
713                 continue
714             if len(amount.rstrip()) > 0:
715                 new_main += f' {currency}'
716             j += 1
717             new_line = new_main
718             if comment.rstrip() != '':
719                 new_line += f'  ; {comment}'
720             lines += [new_line]
721         if 'add_sink' in postvars.keys():
722             temp_lines = lines.copy() + ['_']
723             try:
724                 temp_bookings, _ = parse_lines(temp_lines)
725                 for currency in temp_bookings[0].sink:
726                     amount = temp_bookings[0].sink[currency]
727                     lines += [f'Assets  {amount:.2f} {currency}']
728             except PlomException:
729                 pass
730         elif 'add_taxes' in postvars.keys():
731             lines += self.add_taxes(lines, finish=False)
732         elif 'add_taxes2' in postvars.keys():
733             lines += self.add_taxes(lines, finish=True)
734         elif 'replace' in postvars.keys():
735             for i, line in enumerate(lines):
736                 lines[i] = line.replace(postvars['replace_from'][0], postvars['replace_to'][0])
737         elif 'add_mirror' in postvars.keys():
738             lines += self.add_mirror(lines)
739         return lines, add_empty_line
740
741
742
743 class LedgerHandler(PlomHandler):
744     
745     def app_init(self, handler):
746         default_path = '/ledger'
747         handler.add_route('GET', default_path, self.forward_gets) 
748         handler.add_route('POST', default_path, self.forward_posts) 
749         return 'ledger', default_path 
750
751     def do_POST(self):
752         self.try_do(self.forward_posts)
753
754     def forward_posts(self):
755         prefix = self.apps['ledger'] if hasattr(self, 'apps') else '' 
756         parsed_url = urlparse(self.path)
757         length = int(self.headers['content-length'])
758         postvars = parse_qs(self.rfile.read(length).decode(), keep_blank_values=1)
759         start = int(postvars['start'][0])
760         end = int(postvars['end'][0])
761         db = LedgerDB(prefix)
762         add_empty_line = None
763         lines = []
764         # get inputs
765         if prefix + '/add_structured' == parsed_url.path and not 'revert' in postvars.keys():
766             lines, add_empty_line = db.booking_lines_from_postvars(postvars) 
767         elif prefix + '/add_free' == parsed_url.path and not 'revert' in postvars.keys():
768             lines = postvars['booking'][0].splitlines()
769         # validate where appropriate
770         if ('save' in postvars.keys()) or ('check' in postvars.keys()):
771             _, _ = parse_lines(lines)
772         # if saving, process where to and where to redirect after
773         if 'save' in postvars.keys():
774             last_date = str(datetime.now())[:10]
775             if len(db.bookings) > 0:
776                 last_date = db.bookings[-1].date_string
777             target_date = last_date[:] 
778             first_line_tokens = lines[0].split() if len(lines) > 0 else ''
779             first_token = first_line_tokens[0] if len(first_line_tokens) > 0 else ''
780             try:
781                 datetime.strptime(first_token, '%Y-%m-%d')
782                 target_date = first_token
783             except ValueError:
784                  pass
785             if start == end == 0:
786                 start = db.insert_at_date(lines, target_date)
787                 nth = db.get_nth_for_booking_of_start_line(start) 
788             else:
789                 new_start = db.update(start, end, lines, target_date)
790                 nth = db.get_nth_for_booking_of_start_line(new_start)
791                 if new_start > start: 
792                     nth -= 1 
793             self.redirect(prefix + f'/#{nth}')
794         # otherwise just re-build editing form
795         else:
796             if prefix + '/add_structured' == parsed_url.path: 
797                 edit_content = db.add_structured(start, end, temp_lines=lines, add_empty_line=add_empty_line)
798             else:
799                 edit_content = db.add_free(start, end)
800             header = jinja2.Template(html_head).render(prefix=prefix)
801             self.send_HTML(header + edit_content)
802
803     def do_GET(self):
804         self.try_do(self.forward_gets)
805
806     def forward_gets(self):
807         prefix = self.apps['ledger'] if hasattr(self, 'apps') else '' 
808         parsed_url = urlparse(self.path)
809         params = parse_qs(parsed_url.query)
810         start = int(params.get('start', ['0'])[0])
811         end = int(params.get('end', ['0'])[0])
812         db = LedgerDB(prefix=prefix)
813         if parsed_url.path == prefix + '/balance':
814             stop = params.get('stop', [None])[0]
815             page = db.balance_as_html(stop)
816         elif parsed_url.path == prefix + '/add_free':
817             page = db.add_free(start, end)
818         elif parsed_url.path == prefix + '/add_structured':
819             page = db.add_structured(start, end)
820         elif parsed_url.path == prefix + '/copy_free':
821             page = db.add_free(start, end, copy=True)
822         elif parsed_url.path == prefix + '/copy_structured':
823             page = db.add_structured(start, end, copy=True)
824         elif parsed_url.path == prefix + '/move_up':
825             nth = db.move_up(start, end)
826             self.redirect(prefix + f'/#{nth}')
827             return
828         elif parsed_url.path == prefix + '/move_down':
829             nth = db.move_down(start, end)
830             self.redirect(prefix + f'/#{nth}')
831             return
832         else:
833             page = db.ledger_as_html()
834         header = jinja2.Template(html_head).render(prefix=prefix)
835         self.send_HTML(header + page)
836
837
838
839 if __name__ == "__main__":  
840     run_server(server_port, LedgerHandler)