X-Git-Url: https://plomlompom.com/repos/foo.html?a=blobdiff_plain;ds=sidebyside;f=ledger.py;h=27675da2778e2d7cd94c0354fa7c93848b6eef05;hb=255b10fbe4b0c688828d510868017565af50b57f;hp=60bbf640a4d117ca548a407d727375cc60dbad1d;hpb=6061ff7f1948b83dc601f52df1119a57b6e91649;p=misc diff --git a/ledger.py b/ledger.py index 60bbf64..27675da 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,8 @@ 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) with open(self.db_file, 'w') as f: f.write(text); os.remove(self.lock_file) @@ -346,7 +349,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 +379,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,11 +403,12 @@ 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() @@ -403,18 +416,24 @@ input[type=number] { text-align: right; font-family: monospace; } 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 +448,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 +488,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'{comment}' if comment != '' else ''] + single_c_tmpl = jinja2.Template('{{c|e}}
') + booking_tmpl = jinja2.Template(""" +

{{date}} {{desc}} {{head_comment|e}} +[edit: structured +/ free +| copy:structured +/ free] + +{% for l in booking_lines %} +{% if l.acc %} + +{% else %} + +{% endif %} +{% endfor %} +
{{l.acc|e}}{{l.money|e}}{{l.comment|e}}
{{l.comment|e}}

+""") + 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'' - lines[i] = lines[i] + "\n" + '
{lines[i]}
{booking_line[0]}{money}{comment}
' - return '\n'.join(lines) - - def header_add_form(self, action): - return f"
\n" - - def footer_add_form(self, start, end, copy): - if copy: - start = end = 0 - 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(""" +
+ + + +
+ + + + +
+{% for line in booking_lines %} + + + + + + +
+{% endfor %} +{% for name, items in datalist_sets.items() %} + +{% for item in items %} + +{% 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'' + "\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__":