- 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"<pre>{content}</pre>"
-
- def ledger_as_html(self, db):
- single_c_tmpl = jinja2.Template('<span class="comment">{{c|e}}</span><br />') ##
- elements_to_write = []
- last_i = i = 0 ##
- for nth, booking in enumerate(db.bookings):
- 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 db.comments[last_i:i] if c != ''] ##
- for booking_line in booking.lines[1:]:
- i += 1 ##
- comment = db.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 += [self.booking_tmpl.render(
- nth=nth,
- 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)]
- elements_to_write += [single_c_tmpl.render(c=c) for c in db.comments[last_i:] if c != ''] #
- return '\n'.join(elements_to_write)
-
- def add_free(self, db, start=0, end=0, copy=False):
- tmpl = jinja2.Template("""
-<form method="POST" action="{{action|e}}">
-<textarea name="booking" rows=10 cols=80>
-{% for line in lines %}{{ line }}
-{% endfor %}
-</textarea>
-""" + self.add_form_footer)
- lines = db.get_lines(start, end)
- if copy:
- start = end = 0
- return tmpl.render(action='add_free', 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("""
-<form method="POST" action="{{action|e}}">
-<input type="submit" name="check" value="check" />
-<input type="submit" name="revert" value="revert" />
-<input type="submit" name="add_taxes" value="add taxes" />
-<input type="submit" name="add_taxes2" value="add taxes2" />
-<input type="submit" name="add_sink" value="add sink" />
-<br />
-<input name="date" value="{{date|e}}" size=9 />
-<input name="description" value="{{desc|e}}" list="descriptions" />
-<textarea name="line_0_comment" rows=1 cols=20>{{head_comment|e}}</textarea>
-<input type="submit" name="line_0_add" value="[+]" />
-<br />
-{% for line in booking_lines %}
-<input name="line_{{line.i}}_account" value="{{line.acc|e}}" size=40 list="accounts" />
-<input type="number" name="line_{{line.i}}_amount" step=0.01 value="{{line.amt}}" size=10 />
-<input name="line_{{line.i}}_currency" value="{{line.curr|e}}" size=3 list="currencies" />
-<input type="submit" name="line_{{line.i}}_delete" value="[x]" />
-<input type="submit" name="line_{{line.i}}_delete_after" value="[XX]" />
-<input type="submit" name="line_{{line.i}}_add" value="[+]" />
-<textarea name="line_{{line.i}}_comment" rows=1 cols={% if line.comm_cols %}{{line.comm_cols}}{% else %}20{% endif %}>{{line.comment|e}}</textarea>
-<br />
-{% endfor %}
-{% for name, items in datalist_sets.items() %}
-<datalist id="{{name}}">
-{% for item in items %}
- <option value="{{item|e}}">{{item|e}}</option>
-{% endfor %}
-</datalist>
-{% endfor %}
-""" + self.add_form_footer)
- lines = temp_lines if len(''.join(temp_lines)) > 0 else db.get_lines(start, end)
- bookings, comments = parse_lines(lines, validate_bookings=False)
- if len(bookings) > 1:
- raise HandledException('can only 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 = '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:
- for i in range(1, 3):
- booking_lines += [{'i': i, 'acc': '', 'amt': '', 'curr': '€', 'comment': ''}]
- date=today
+ booking = Booking.from_postvars(postvars, starts_at, validate)
+ if submit_button in {'update', 'add'}:
+ if submit_button == 'update':
+ if 'textarea' == edit_mode and 'delete' == ''.join([l.text_line for l in lines]).strip():
+ 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: # non-DB-writing calls
+ error_msg = None
+ if 'check' == submit_button:
+ error_msg = 'All looks fine!'
+ 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 do_GET(self):
+ self.try_do(self.forward_gets)
+
+ def forward_gets(self):
+ prefix = self.apps['ledger'] if hasattr(self, 'apps') else ''
+ 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)
+ params = parse_qs(parsed_url.query)
+ if parsed_url.path == f'{prefix}/balance':
+ stop = params.get('until_after', [None])[0]
+ page = db.balance_as_html(stop)
+ elif parsed_url.path == f'{prefix}/edit':
+ index = params.get('i', [-1])[0]
+ copy = params.get('copy', [0])[0]
+ page = db.edit(int(index), copy=bool(copy))