X-Git-Url: https://plomlompom.com/repos/new_day?a=blobdiff_plain;f=ledger.py;h=c3f0ab9c22d028d92b1c2741e7c411a4d9efa499;hb=2deeead813fbdf0fec4f94d238a43fd93514f94f;hp=334318ff301eb186efa6684809c9ebac1807d177;hpb=c3b5860ae1ca76c2a1dc3b61d3c1242d581424f0;p=misc diff --git a/ledger.py b/ledger.py index 334318f..c3f0ab9 100755 --- a/ledger.py +++ b/ledger.py @@ -1,5 +1,6 @@ from http.server import BaseHTTPRequestHandler, HTTPServer import sys +import os hostName = "localhost" serverPort = 8082 @@ -15,19 +16,138 @@ def handled_error_exit(msg): def apply_booking_to_account_balances(account_sums, account, currency, amount): if not account in account_sums: - account_sums[account] = {currency: amount} + account_sums[account] = {currency: amount} elif not currency in account_sums[account].keys(): account_sums[account][currency] = amount else: account_sums[account][currency] += amount +def parse_lines(lines): + import datetime + import decimal + inside_booking = False + date_string, description = None, None + booking_lines = [] + start_line = 0 + bookings = [] + comments = [] + for i, line in enumerate(lines): + prefix = f"line {i}" + # we start with the case of an utterly empty line + comments += [""] + stripped_line = line.rstrip() + if stripped_line == '': + if inside_booking: + # assume we finished a booking, finalize, and commit to DB + if len(booking_lines) < 2: + raise HandledException(f"{prefix} booking ends to early") + booking = Booking(date_string, description, booking_lines, start_line) + bookings += [booking] + # expect new booking to follow so re-zeroall booking data + inside_booking = False + date_string, description = None, None + booking_lines = [] + continue + # if non-empty line, first get comment if any, and commit to DB + split_by_comment = stripped_line.split(sep=";", maxsplit=1) + if len(split_by_comment) == 2: + comments[i] = split_by_comment[1] + # 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 + non_comment = split_by_comment[0].rstrip() + if non_comment.rstrip() == '': + if inside_booking: + booking_lines += [''] + continue + # if we're starting a booking, parse by first-line pattern + if not inside_booking: + start_line = i + toks = non_comment.split(maxsplit=1) + date_string = toks[0] + try: + datetime.datetime.strptime(date_string, '%Y-%m-%d') + except ValueError: + raise HandledException(f"{prefix} bad date string: {date_string}") + try: + description = toks[1] + except IndexError: + raise HandledException(f"{prefix} bad description: {description}") + inside_booking = True + continue + # otherwise, read as transfer data + toks = non_comment.split() # ignore specification's allowance of single spaces in names + if len(toks) > 3: + raise HandledException(f"{prefix} too many booking line tokens: {toks}") + amount, currency = None, None + account_name = toks[0] + if account_name[0] == '[' and account_name[-1] == ']': + # ignore specification's differentiation of "virtual" accounts + account_name = account_name[1:-1] + decimal_chars = ".-0123456789" + if len(toks) == 3: + i_currency = 1 + try: + amount = decimal.Decimal(toks[1]) + i_currency = 2 + except decimal.InvalidOperation: + try: + amount = decimal.Decimal(toks[2]) + except decimal.InvalidOperation: + raise HandledException(f"{prefix} no decimal number in: {toks[1:]}") + currency = toks[i_currency] + if currency[0] in decimal_chars: + raise HandledException(f"{prefix} currency starts with int, dot, or minus: {currency}") + elif len(toks) == 2: + value = toks[1] + inside_amount = False + inside_currency = False + amount_string = "" + currency = "" + dots_counted = 0 + for i, c in enumerate(value): + if i == 0: + if c in decimal_chars: + inside_amount = True + else: + inside_currency = True + if inside_currency: + if c in decimal_chars and len(amount_string) == 0: + inside_currency = False + inside_amount = True + else: + currency += c + continue + if inside_amount: + if c not in decimal_chars: + if len(currency) > 0: + raise HandledException(f"{prefix} amount has non-decimal chars: {value}") + inside_currency = True + inside_amount = False + currency += c + continue + if c == '-' and len(amount_string) > 1: + raise HandledException(f"{prefix} amount has non-start '-': {value}") + if c == '.': + if dots_counted > 1: + raise HandledException(f"{prefix} amount has multiple dots: {value}") + dots_counted += 1 + amount_string += c + if len(amount_string) == 0: + raise HandledException(f"{prefix} amount missing: {value}") + if len(currency) == 0: + raise HandledException(f"{prefix} currency missing: {value}") + amount = decimal.Decimal(amount_string) + booking_lines += [(account_name, amount, currency)] + if inside_booking: + raise HandledException(f"{prefix} last booking unfinished") + return bookings, comments + class Booking: def __init__(self, date_string, description, booking_lines, start_line): self.date_string = date_string - self.description = description - self.lines = booking_lines + self.description = description + self.lines = booking_lines self.start_line = start_line self.validate_booking_lines() self.account_changes = self.parse_booking_lines_to_account_changes() @@ -37,7 +157,7 @@ class Booking: sums = {} empty_values = 0 for line in self.lines: - if line == '': + if line == '': continue _, amount, currency = line if amount is None: @@ -56,7 +176,7 @@ class Booking: sinkable = False for k, v in sums.items(): if v != 0: - sinkable = True + sinkable = True if not sinkable: raise HandledException(f"{prefix} has empty value that cannot be filled") @@ -65,7 +185,7 @@ class Booking: debt = {} sink_account = None for line in self.lines: - if line == '': + if line == '': continue account, amount, currency = line if amount is None: @@ -86,7 +206,6 @@ class Booking: class Database: def __init__(self, load_from_file=True): - import os db_name = "_ledger" self.db_file = db_name + ".json" self.lock_file = db_name+ ".lock" @@ -94,167 +213,87 @@ class Database: self.comments = [] if load_from_file and os.path.exists(self.db_file): with open(self.db_file, "r") as f: - self.parse_lines(f.readlines()) + ret = parse_lines(f.readlines()) + self.bookings += ret[0] + self.comments += ret[1] - def parse_lines(self, lines): - import datetime - import decimal - inside_booking = False - date_string, description = None, None - booking_lines = [] - start_line = 0 - for i, line in enumerate(lines): - prefix = f"line {i}" - # we start with the case of an utterly empty line - self.comments += [""] - stripped_line = line.rstrip() - if stripped_line == '': - if inside_booking: - # assume we finished a booking, finalize, and commit to DB - if len(booking_lines) < 2: - raise HandledException(f"{prefix} booking ends to early") - booking = Booking(date_string, description, booking_lines, start_line) - self.bookings += [booking] - # expect new booking to follow so re-zeroall booking data - inside_booking = False - date_string, description = None, None - booking_lines = [] - continue - # if non-empty line, first get comment if any, and commit to DB - split_by_comment = stripped_line.split(sep=";", maxsplit=1) - if len(split_by_comment) == 2: - self.comments[i] = split_by_comment[1] - # 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 - non_comment = split_by_comment[0].rstrip() - if non_comment.rstrip() == '': - if inside_booking: - booking_lines += [''] - continue - # if we're starting a booking, parse by first-line pattern - if not inside_booking: - start_line = i - toks = non_comment.split(maxsplit=1) - date_string = toks[0] - try: - datetime.datetime.strptime(date_string, '%Y-%m-%d') - except ValueError: - raise HandledException(f"{prefix} bad date string: {date_string}") - try: - description = toks[1] - except IndexError: - raise HandledException(f"{prefix} bad description: {description}") - inside_booking = True - continue - # otherwise, read as transfer data - toks = non_comment.split() # ignore specification's allowance of single spaces in names - if len(toks) > 3: - raise HandledException(f"{prefix} too many booking line tokens: {toks}") - amount, currency = None, None - account_name = toks[0] - if account_name[0] == '[' and account_name[-1] == ']': - # ignore specification's differentiation of "virtual" accounts - account_name = account_name[1:-1] - decimal_chars = ".-0123456789" - if len(toks) == 3: - i_currency = 1 - try: - amount = decimal.Decimal(toks[1]) - i_currency = 2 - except decimal.InvalidOperation: - try: - amount = decimal.Decimal(toks[2]) - except decimal.InvalidOperation: - raise HandledException(f"{prefix} no decimal number in: {toks[1:]}") - currency = toks[i_currency] - if currency[0] in decimal_chars: - raise HandledException(f"{prefix} currency starts with int, dot, or minus: {currency}") - elif len(toks) == 2: - value = toks[1] - inside_amount = False - inside_currency = False - amount_string = "" - currency = "" - dots_counted = 0 - for i, c in enumerate(value): - if i == 0: - if c in decimal_chars: - inside_amount = True - else: - inside_currency = True - if inside_currency: - if c in decimal_chars and len(amount_string) == 0: - inside_currency = False - inside_amount = True - else: - currency += c - continue - if inside_amount: - if c not in decimal_chars: - if len(currency) > 0: - raise HandledException(f"{prefix} amount has non-decimal chars: {value}") - inside_currency = True - inside_amount = False - currency += c - continue - if c == '-' and len(amount_string) > 1: - raise HandledException(f"{prefix} amount has non-start '-': {value}") - if c == '.': - if dots_counted > 1: - raise HandledException(f"{prefix} amount has multiple dots: {value}") - dots_counted += 1 - amount_string += c - if len(amount_string) == 0: - raise HandledException(f"{prefix} amount missing: {value}") - if len(currency) == 0: - raise HandledException(f"{prefix} currency missing: {value}") - amount = decimal.Decimal(amount_string) - booking_lines += [(account_name, amount, currency)] + def append(self, lines): + import shutil + if os.path.exists(self.lock_file): + raise HandledException('Sorry, lock file!') + if os.path.exists(self.db_file): + shutil.copy(self.db_file, self.db_file + ".bak") + f = open(self.lock_file, 'w+') + f.close() + with open(self.db_file, 'a') as f: + f.write('\n' + '\n'.join(lines) + '\n'); + os.remove(self.lock_file) - def ledger_as_html(self): - lines = [] - for comment in self.comments: - lines += [f"; {comment}" if comment != '' else ''] - for booking in self.bookings: - i = booking.start_line - suffix = f" {lines[i]}" if len(lines[i]) > 0 else "" - lines[i] = f"{booking.date_string} {booking.description}{suffix}" - for booking_line in booking.lines: - i += 1 - if booking_line == '': - continue - suffix = f" {lines[i]}" if len(lines[i]) > 0 else "" - value = f" {booking_line[1]} {booking_line[2]}" if booking_line[1] else "" - lines[i] = f"{booking_line[0]}{value}{suffix}" - content = "\n".join(lines) - return f"
{content}" - def balance_as_html(self): +class MyServer(BaseHTTPRequestHandler): + header = 'ledger balance add
{content}" - + return f"
{content}" -class MyServer(BaseHTTPRequestHandler): + def ledger_as_html(self, db): + lines = [] + for comment in db.comments: + lines += [f"; {comment}" if comment != '' else ''] + for booking in db.bookings: + i = booking.start_line + suffix = f" {lines[i]}" if len(lines[i]) > 0 else "" + lines[i] = f"{booking.date_string} {booking.description}{suffix}" + for booking_line in booking.lines: + i += 1 + if booking_line == '': + continue + suffix = f" {lines[i]}" if len(lines[i]) > 0 else "" + value = f" {booking_line[1]} {booking_line[2]}" if booking_line[1] else "" + lines[i] = f"{booking_line[0]}{value}{suffix}" + content = "\n".join(lines) + return f"
{content}" - def do_GET(self): - self.send_response(200) - self.send_header("Content-type", "text/html") - self.end_headers() - db = Database() - # page = db.ledger_as_html() - page = db.balance_as_html() - self.wfile.write(bytes(page, "utf-8")) + def add(self): + return '' db = Database() -if __name__ == "__main__": +if __name__ == "__main__": webServer = HTTPServer((hostName, serverPort), MyServer) print(f"Server started http://{hostName}:{serverPort}") try: