import decimal
from datetime import datetime, timedelta
from urllib.parse import parse_qs, urlparse
-from plomlib import PlomDB, PlomException, run_server, run_server, PlomServer
+from plomlib import PlomDB, PlomException, run_server, PlomHandler
server_port = 8082
db_path = '/home/plom/org/ledger2023.dat'
.full_line_comment { display: block; white-space: nowrap; width: 0; }
</style>
<body>
-<a href="/">ledger</a>
-<a href="/balance">balance</a>
-<a href="/add_free">add free</a>
-<a href="/add_structured">add structured</a>
+<a href="{{prefix}}/ledger">ledger</a>
+<a href="{{prefix}}/balance">balance</a>
+<a href="{{prefix}}/add_free">add free</a>
+<a href="{{prefix}}/add_structured">add structured</a>
<hr />
"""
booking_html = """
-<p id="{{nth}}"><a href="#{{nth}}">{{date}}</a> {{desc}} <span class="comment">{{head_comment|e}}</span><br />
-<span class="meta">[edit: <a href="/add_structured?start={{start}}&end={{end}}">structured</a>
-/ <a href="/add_free?start={{start}}&end={{end}}">free</a>
-| copy:<a href="/copy_structured?start={{start}}&end={{end}}">structured</a>
-/ <a href="/copy_free?start={{start}}&end={{end}}">free</a>
-| move {% if move_up %}<a href="/move_up?start={{start}}&end={{end}}">up</a>{% else %}up{% endif %}/{% if move_down %}<a href="/move_down?start={{start}}&end={{end}}">down</a>{% else %}down{% endif %}
-| <a href="/balance?stop={{nth+1}}">balance after</a>
+<p id="{{nth}}"><a href="{{prefix}}#{{nth}}">{{date}}</a> {{desc}} <span class="comment">{{head_comment|e}}</span><br />
+<span class="meta">[edit: <a href="{{prefix}}/add_structured?start={{start}}&end={{end}}">structured</a>
+/ <a href="{{prefix}}/add_free?start={{start}}&end={{end}}">free</a>
+| copy:<a href="{{prefix}}/copy_structured?start={{start}}&end={{end}}">structured</a>
+/ <a href="{{prefix}}/copy_free?start={{start}}&end={{end}}">free</a>
+| 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 %}
+| <a href="{{prefix}}/balance?stop={{nth+1}}">balance after</a>
]</span>
<table>
{% for l in booking_lines %}
<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" />
+<input name="replace_from" />
+<input type="submit" name="replace" value="-> replace ->" />
+<input name="replace_to" />
+<input type="submit" name="add_mirror" value="add mirror" />
<br />
<input name="date" value="{{date|e}}" size=9 />
<input name="description" value="{{desc|e}}" list="descriptions" />
class LedgerDB(PlomDB):
- def __init__(self):
+ def __init__(self, prefix):
+ self.prefix = prefix
self.bookings = []
self.comments = []
self.real_lines = []
return self.real_lines[start:end]
def write_db(self, text, mode='w'):
+ if text[-1] != '\n':
+ text += '\n'
self.write_text_to_db(text)
def insert_at_date(self, lines, date):
- start_at = len(self.real_lines)
- for b in self.bookings:
- if b.date_string == date:
- start_at = b.start_line
- break
- elif b.date_string > date:
- break
- if start_at == len(self.real_lines):
- lines = [''] + lines
+ print("DEBUG insert_at_date")
+ start_at = 0
+ if len(self.bookings) > 0:
+ if date >= self.bookings[-1].date_string:
+ start_at = len(self.real_lines)
+ lines = [''] + lines
+ else:
+ for b in self.bookings:
+ if b.date_string == date:
+ start_at = b.start_line
+ elif b.date_string > date:
+ start_at = b.start_line
+ break
return self.write_lines_in_total_lines_at(self.real_lines, start_at, lines)
def update(self, start, end, lines, date):
+ print("DEBUG update")
total_lines = self.real_lines[:start] + self.real_lines[end:]
n_original_lines = end - start
start_at = len(total_lines)
ret += [f' {acc_buffer} {-final_minus} € ; assume as to earn in year: {acc_buffer} + {12 - months_passed - 1} * this = {year_needed}']
return ret
+ def add_mirror(self, lines):
+ ret = []
+ bookings, _ = parse_lines(lines)
+ booking = bookings[0]
+ for line in booking.lines[1:]:
+ ret += [f'? {-line[1]} {line[2]}']
+ return ret
-class LedgerServer(PlomServer):
-
- def pre_init(self):
- self.html_head += [html_head]
-
- def do_POST(self):
- try:
- parsed_url = urlparse(self.path)
- length = int(self.headers['content-length'])
- postvars = parse_qs(self.rfile.read(length).decode(), keep_blank_values=1)
- start = int(postvars['start'][0])
- end = int(postvars['end'][0])
- db = LedgerDB()
- add_empty_line = None
- lines = []
- # get inputs
- if '/add_structured' == parsed_url.path and not 'revert' in postvars.keys():
- lines, add_empty_line = self.booking_lines_from_postvars(postvars, db)
- elif '/add_free' == parsed_url.path and not 'revert' in postvars.keys():
- lines = postvars['booking'][0].splitlines()
- # validate where appropriate
- if ('save' in postvars.keys()) or ('check' in postvars.keys()):
- _, _ = parse_lines(lines)
- # if saving, process where to and where to redirect after
- if 'save' in postvars.keys():
- last_date = str(datetime.now())[:10]
- if len(db.bookings) > 0:
- last_date = db.bookings[-1].date_string
- target_date = last_date[:]
- first_line_tokens = lines[0].split() if len(lines) > 0 else ''
- first_token = first_line_tokens[0] if len(first_line_tokens) > 0 else ''
- try:
- datetime.strptime(first_token, '%Y-%m-%d')
- target_date = first_token
- except ValueError:
- pass
- if start == end == 0:
- start = db.insert_at_date(lines, target_date)
- nth = db.get_nth_for_booking_of_start_line(start)
- else:
- new_start = db.update(start, end, lines, target_date)
- nth = db.get_nth_for_booking_of_start_line(new_start)
- if new_start > start:
- nth -= 1
- self.redirect( f'/#{nth}')
- # otherwise just re-build editing form
- else:
- if '/add_structured' == parsed_url.path:
- edit_content = self.add_structured(db, start, end, temp_lines=lines, add_empty_line=add_empty_line)
- else:
- edit_content = self.add_free(db, start, end)
- self.send_HTML(edit_content)
- except PlomException as e:
- self.fail_400(e)
-
- def do_GET(self):
- try:
- parsed_url = urlparse(self.path)
- params = parse_qs(parsed_url.query)
- start = int(params.get('start', ['0'])[0])
- end = int(params.get('end', ['0'])[0])
- db = LedgerDB()
- if parsed_url.path == '/balance':
- stop = params.get('stop', [None])[0]
- page = self.balance_as_html(db, stop)
- elif parsed_url.path == '/add_free':
- page = self.add_free(db, start, end)
- elif parsed_url.path == '/add_structured':
- page = self.add_structured(db, start, end)
- elif parsed_url.path == '/copy_free':
- page = self.add_free(db, start, end, copy=True)
- elif parsed_url.path == '/copy_structured':
- page = self.add_structured(db, start, end, copy=True)
- elif parsed_url.path == '/move_up':
- nth = self.move_up(db, start, end)
- self.redirect(f'/#{nth}')
- return
- elif parsed_url.path == '/move_down':
- nth = self.move_down(db, start, end)
- self.redirect(f'/#{nth}')
- return
- else:
- page = self.ledger_as_html(db)
- self.send_HTML(page)
- except PlomException as e:
- self.fail_400(e)
-
- def booking_lines_from_postvars(self, postvars, db):
- add_empty_line = None
- date = postvars['date'][0]
- description = postvars['description'][0]
- start_comment = postvars['line_0_comment'][0]
- start_line = f'{date} {description}'
- if start_comment.rstrip() != '':
- start_line += f' ; {start_comment}'
- lines = [start_line]
- 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
- elif f'line_{i}_delete_after' in postvars.keys():
- break
- elif 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]
- comment = postvars[f'line_{i}_comment'][0]
- i += 1
- new_main = f' {account} {amount}'
- if '' == new_main.rstrip() == comment.rstrip(): # don't write empty lines, ignore currency if nothing else set
- continue
- if len(amount.rstrip()) > 0:
- new_main += f' {currency}'
- j += 1
- new_line = new_main
- if comment.rstrip() != '':
- new_line += f' ; {comment}'
- lines += [new_line]
- if 'add_sink' in postvars.keys():
- temp_lines = lines.copy() + ['_']
- try:
- temp_bookings, _ = parse_lines(temp_lines)
- for currency in temp_bookings[0].sink:
- amount = temp_bookings[0].sink[currency]
- lines += [f'Assets {amount:.2f} {currency}']
- except PlomException:
- pass
- if 'add_taxes' in postvars.keys():
- lines += db.add_taxes(lines, finish=False)
- elif 'add_taxes2' in postvars.keys():
- lines += db.add_taxes(lines, finish=True)
- return lines, add_empty_line
-
- def balance_as_html(self, db, until=None):
- bookings = db.bookings[:until if until is None else int(until)]
- lines = []
- account_tree, account_sums = bookings_to_account_tree(bookings)
- def print_subtree(lines, indent, node, subtree, path):
- line = f"{indent}{node}"
- n_tabs = 5 - (len(line) // 8)
- line += n_tabs * "\t"
- if "€" in account_sums[path + node].keys():
- amount = account_sums[path + node]["€"]
- line += f"{amount:9.2f} €\t"
- else:
- 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):
+ def ledger_as_html(self):
booking_tmpl = jinja2.Template(booking_html)
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):
- move_up = nth > 0 and db.bookings[nth - 1].date_string == booking.date_string
- move_down = nth < len(db.bookings) - 1 and db.bookings[nth + 1].date_string == booking.date_string
+ for nth, booking in enumerate(self.bookings):
+ move_up = nth > 0 and self.bookings[nth - 1].date_string == booking.date_string
+ move_down = nth < len(self.bookings) - 1 and self.bookings[nth + 1].date_string == booking.date_string
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 != ''] ##
+ elements_to_write += [single_c_tmpl.render(c=c) for c in self.comments[last_i:i] if c != ''] ##
for booking_line in booking.lines[1:]:
i += 1 ##
- comment = db.comments[i] ##
+ comment = self.comments[i] ##
if booking_line == '':
booking_lines += [{'acc': None, 'money': None, 'comment': comment}] ##
continue
money = f'{booking_line[1]} {booking_line[2]}'
booking_lines += [{'acc': booking_line[0], 'money':money, 'comment':comment}] ##
elements_to_write += [booking_tmpl.render(
+ prefix=self.prefix,
nth=nth,
start=booking.start_line,
end=booking_end,
date=booking.date_string,
desc=booking.description,
- head_comment=db.comments[booking.start_line],
+ head_comment=self.comments[booking.start_line],
move_up=move_up,
move_down=move_down,
booking_lines = booking_lines)]
- elements_to_write += [single_c_tmpl.render(c=c) for c in db.comments[last_i:] if c != ''] #
+ elements_to_write += [single_c_tmpl.render(c=c) for c in self.comments[last_i:] if c != ''] #
return '\n'.join(elements_to_write)
- def add_free(self, db, start=0, end=0, copy=False):
+ def balance_as_html(self, until=None):
+ bookings = self.bookings[:until if until is None else int(until)]
+ lines = []
+ account_tree, account_sums = bookings_to_account_tree(bookings)
+ def print_subtree(lines, indent, node, subtree, path):
+ line = f"{indent}{node}"
+ n_tabs = 5 - (len(line) // 8)
+ line += n_tabs * "\t"
+ if "€" in account_sums[path + node].keys():
+ amount = account_sums[path + node]["€"]
+ line += f"{amount:9.2f} €\t"
+ else:
+ 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 add_free(self, start=0, end=0, copy=False):
tmpl = jinja2.Template(add_form_header + add_free_html + add_form_footer)
- lines = db.get_lines(start, end)
+ lines = self.get_lines(start, end)
if copy:
start = end = 0
- return tmpl.render(action='add_free', start=start, end=end, lines=lines)
+ return tmpl.render(action=self.prefix + '/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):
+ def add_structured(self, start=0, end=0, copy=False, temp_lines=[], add_empty_line=None):
tmpl = jinja2.Template(add_form_header + add_structured_html + add_form_footer)
- lines = temp_lines if len(''.join(temp_lines)) > 0 else db.get_lines(start, end)
+ lines = temp_lines if len(''.join(temp_lines)) > 0 else self.get_lines(start, end)
bookings, comments = parse_lines(lines, validate_bookings=False)
if len(bookings) > 1:
raise PlomException('can only structurally edit single Booking')
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'
+ action = self.prefix + '/add_structured'
datalist_sets = {'descriptions': set(), 'accounts': set(), 'currencies': set()}
- for b in db.bookings:
+ for b in self.bookings:
datalist_sets['descriptions'].add(b.description)
for account, moneys in b.account_changes.items():
datalist_sets['accounts'].add(account)
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
else:
booking = bookings[0]
desc = booking.description
date = today if copy else booking.date_string
head_comment=comments[0]
- last_line = len(comments)
for i in range(1, len(comments)):
account = amount = currency = ''
if i < len(booking.lines) and booking.lines[i] != '':
'curr': currency if currency else '€',
'comment': comments[i],
'comm_cols': len(comments[i])}]
+ for i in range(len(comments), len(comments) + 8):
+ booking_lines += [{'i': i, 'acc': '', 'amt': '', 'curr': '€', 'comment': ''}]
content += tmpl.render(
action=action,
date=date,
end=end)
return content
- def move_up(self, db, start, end):
+ def move_up(self, start, end):
prev_booking = None
- for redir_nth, b in enumerate(db.bookings):
+ for redir_nth, b in enumerate(self.bookings):
if b.start_line >= start:
break
prev_booking = b
start_at = prev_booking.start_line
- self.make_move(db, start, end, start_at)
+ self.make_move(start, end, start_at)
return redir_nth - 1
- def move_down(self, db, start, end):
+ def move_down(self, start, end):
next_booking = None
- for redir_nth, b in enumerate(db.bookings):
+ for redir_nth, b in enumerate(self.bookings):
if b.start_line > start:
next_booking = b
break
start_at = next_booking.start_line + len(next_booking.lines) - (end - start) + 1
- self.make_move(db, start, end, start_at)
+ self.make_move(start, end, start_at-1)
return redir_nth
- def make_move(self, db, start, end, start_at):
- lines = db.get_lines(start, end)
- total_lines = db.real_lines[:start] + db.real_lines[end:]
- db.write_lines_in_total_lines_at(total_lines, start_at, lines)
+ def make_move(self, start, end, start_at):
+ lines = self.get_lines(start, end)
+ 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
+ self.write_lines_in_total_lines_at(total_lines, start_at, lines)
+
+ def booking_lines_from_postvars(self, postvars):
+ add_empty_line = None
+ date = postvars['date'][0]
+ description = postvars['description'][0]
+ start_comment = postvars['line_0_comment'][0]
+ start_line = f'{date} {description}'
+ if start_comment.rstrip() != '':
+ start_line += f' ; {start_comment}'
+ lines = [start_line]
+ 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
+ elif f'line_{i}_delete_after' in postvars.keys():
+ break
+ elif 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]
+ comment = postvars[f'line_{i}_comment'][0]
+ i += 1
+ new_main = f' {account} {amount}'
+ if '' == new_main.rstrip() == comment.rstrip(): # don't write empty lines, ignore currency if nothing else set
+ continue
+ if len(amount.rstrip()) > 0:
+ new_main += f' {currency}'
+ j += 1
+ new_line = new_main
+ if comment.rstrip() != '':
+ new_line += f' ; {comment}'
+ lines += [new_line]
+ if 'add_sink' in postvars.keys():
+ temp_lines = lines.copy() + ['_']
+ try:
+ temp_bookings, _ = parse_lines(temp_lines)
+ for currency in temp_bookings[0].sink:
+ amount = temp_bookings[0].sink[currency]
+ lines += [f'Assets {amount:.2f} {currency}']
+ except PlomException:
+ pass
+ elif 'add_taxes' in postvars.keys():
+ lines += self.add_taxes(lines, finish=False)
+ elif 'add_taxes2' in postvars.keys():
+ lines += self.add_taxes(lines, finish=True)
+ elif 'replace' in postvars.keys():
+ for i, line in enumerate(lines):
+ lines[i] = line.replace(postvars['replace_from'][0], postvars['replace_to'][0])
+ elif 'add_mirror' in postvars.keys():
+ lines += self.add_mirror(lines)
+ return lines, add_empty_line
+
+
+
+class LedgerHandler(PlomHandler):
+
+ def app_init(self, handler):
+ default_path = '/ledger'
+ handler.add_route('GET', default_path, self.forward_gets)
+ handler.add_route('POST', default_path, self.forward_posts)
+ return 'ledger', default_path
+
+ def do_POST(self):
+ self.try_do(self.forward_posts)
+
+ def forward_posts(self):
+ prefix = self.apps['ledger'] if hasattr(self, 'apps') else ''
+ parsed_url = urlparse(self.path)
+ length = int(self.headers['content-length'])
+ postvars = parse_qs(self.rfile.read(length).decode(), keep_blank_values=1)
+ start = int(postvars['start'][0])
+ end = int(postvars['end'][0])
+ db = LedgerDB(prefix)
+ add_empty_line = None
+ lines = []
+ # get inputs
+ if prefix + '/add_structured' == parsed_url.path and not 'revert' in postvars.keys():
+ lines, add_empty_line = db.booking_lines_from_postvars(postvars)
+ elif prefix + '/add_free' == parsed_url.path and not 'revert' in postvars.keys():
+ lines = postvars['booking'][0].splitlines()
+ # validate where appropriate
+ if ('save' in postvars.keys()) or ('check' in postvars.keys()):
+ _, _ = parse_lines(lines)
+ # if saving, process where to and where to redirect after
+ if 'save' in postvars.keys():
+ last_date = str(datetime.now())[:10]
+ if len(db.bookings) > 0:
+ last_date = db.bookings[-1].date_string
+ target_date = last_date[:]
+ first_line_tokens = lines[0].split() if len(lines) > 0 else ''
+ first_token = first_line_tokens[0] if len(first_line_tokens) > 0 else ''
+ try:
+ datetime.strptime(first_token, '%Y-%m-%d')
+ target_date = first_token
+ except ValueError:
+ pass
+ if start == end == 0:
+ start = db.insert_at_date(lines, target_date)
+ nth = db.get_nth_for_booking_of_start_line(start)
+ else:
+ new_start = db.update(start, end, lines, target_date)
+ nth = db.get_nth_for_booking_of_start_line(new_start)
+ if new_start > start:
+ nth -= 1
+ self.redirect(prefix + f'/#{nth}')
+ # otherwise just re-build editing form
+ else:
+ if prefix + '/add_structured' == parsed_url.path:
+ edit_content = db.add_structured(start, end, temp_lines=lines, add_empty_line=add_empty_line)
+ else:
+ edit_content = db.add_free(start, end)
+ header = jinja2.Template(html_head).render(prefix=prefix)
+ self.send_HTML(header + edit_content)
+
+ def do_GET(self):
+ self.try_do(self.forward_gets)
+
+ def forward_gets(self):
+ prefix = self.apps['ledger'] if hasattr(self, 'apps') else ''
+ parsed_url = urlparse(self.path)
+ params = parse_qs(parsed_url.query)
+ start = int(params.get('start', ['0'])[0])
+ end = int(params.get('end', ['0'])[0])
+ db = LedgerDB(prefix=prefix)
+ if parsed_url.path == prefix + '/balance':
+ stop = params.get('stop', [None])[0]
+ page = db.balance_as_html(stop)
+ elif parsed_url.path == prefix + '/add_free':
+ page = db.add_free(start, end)
+ elif parsed_url.path == prefix + '/add_structured':
+ page = db.add_structured(start, end)
+ elif parsed_url.path == prefix + '/copy_free':
+ page = db.add_free(start, end, copy=True)
+ elif parsed_url.path == prefix + '/copy_structured':
+ page = db.add_structured(start, end, copy=True)
+ elif parsed_url.path == prefix + '/move_up':
+ nth = db.move_up(start, end)
+ self.redirect(prefix + f'/#{nth}')
+ return
+ elif parsed_url.path == prefix + '/move_down':
+ nth = db.move_down(start, end)
+ self.redirect(prefix + f'/#{nth}')
+ return
+ else:
+ page = db.ledger_as_html()
+ header = jinja2.Template(html_head).render(prefix=prefix)
+ self.send_HTML(header + page)
+
if __name__ == "__main__":
- run_server(server_port, LedgerServer)
+ run_server(server_port, LedgerHandler)
projects / misc / blobdiff