diff options
Diffstat (limited to 'src/journal_lib/parse/parsers/p_ledger.py')
| -rw-r--r-- | src/journal_lib/parse/parsers/p_ledger.py | 112 |
1 files changed, 62 insertions, 50 deletions
diff --git a/src/journal_lib/parse/parsers/p_ledger.py b/src/journal_lib/parse/parsers/p_ledger.py index ad01a79..6b5d573 100644 --- a/src/journal_lib/parse/parsers/p_ledger.py +++ b/src/journal_lib/parse/parsers/p_ledger.py @@ -2,13 +2,20 @@ from datetime import datetime from journal_lib.parse.parsewrapper import ParseWrapper from journal_lib.parse.lexers.l_ledger import JournalLexer -from journal_lib.dataclasses import Journal, JournalEntry, JournalEntryTransaction, JournalAccountDef, JournalCommodityDef +from journal_lib.dataclasses import ( + Journal, + JournalEntry, + JournalEntryTransaction, + JournalAccountDef, + JournalCommodityDef, +) + class JournalParser(ParseWrapper): tokens = JournalLexer.tokens def p_journal(self, p): - '''journal : elements ''' + """journal : elements""" p[0] = Journal.from_elements(p[1]) if self.debug: @@ -16,41 +23,41 @@ class JournalParser(ParseWrapper): print(repr(x)) def p_elements(self, p): - '''elements : elements element - | element''' + """elements : elements element + | element""" if len(p) == 3: p[0] = p[1] + [p[2]] else: p[0] = [p[1]] def p_element_entry(self, p): - '''element : DATE effective_date status TEXT transactions''' + """element : DATE effective_date status TEXT transactions""" p[0] = JournalEntry( - date = p[1], - effective_date = p[2], - cleared = p[3]['cleared'], - pending = p[3]['pending'], - title = p[4], - transactions = p[5]['transactions'], - comments = p[5]['comments'] + date=p[1], + effective_date=p[2], + cleared=p[3]["cleared"], + pending=p[3]["pending"], + title=p[4], + transactions=p[5]["transactions"], + comments=p[5]["comments"], ) def p_element_account(self, p): - '''element : KW_ACCOUNT TEXT COMMENT - | KW_ACCOUNT TEXT''' + """element : KW_ACCOUNT TEXT COMMENT + | KW_ACCOUNT TEXT""" p[0] = JournalAccountDef(account=p[2], comment=p[3] if len(p) > 3 else None) def p_element_commodity(self, p): - '''element : KW_COMMODITY TEXT commodity_attributes ''' + """element : KW_COMMODITY TEXT commodity_attributes""" p[0] = JournalCommodityDef(commodity=p[2], **p[3]) def p_commodity_attributes(self, p): - '''commodity_attributes : COMMODITY_FORMAT TEXT commodity_attributes + """commodity_attributes : COMMODITY_FORMAT TEXT commodity_attributes | COMMODITY_NOTE TEXT commodity_attributes | COMMODITY_NOMARKET commodity_attributes | COMMODITY_DEFAULT commodity_attributes | empty - ''' + """ p[0] = {} if p[1] == "format": p[0]["format"] = p[2] @@ -64,66 +71,72 @@ class JournalParser(ParseWrapper): p[0].update(p[3]) def p_effective_date(self, p): - '''effective_date : ENTRY_EFFECTIVE_DATE_SEPARATOR DATE - | empty - ''' + """effective_date : ENTRY_EFFECTIVE_DATE_SEPARATOR DATE + | empty""" p[0] = p[2] if p[1] else None def p_status(self, p): - '''status : ENTRY_STATUS - | empty - ''' - p[0] = {'cleared': p[1] == "*", 'pending': p[1] == "!"} if p[1] else {'cleared': False, 'pending': False} + """status : ENTRY_STATUS + | empty""" + p[0] = ( + {"cleared": p[1] == "*", "pending": p[1] == "!"} + if p[1] + else {"cleared": False, "pending": False} + ) def p_empty(self, p): - '''empty :''' + """empty :""" pass def p_transactions(self, p): - '''transactions : transactions transaction''' - p[0] = {"comments": p[1]['comments'], - "transactions": p[1]['transactions'] + [p[2]]} + """transactions : transactions transaction""" + p[0] = { + "comments": p[1]["comments"], + "transactions": p[1]["transactions"] + [p[2]], + } def p_comments(self, p): - '''transactions : transactions COMMENT''' - p[0] = {"comments": p[1]['comments'] + [p[2]], - "transactions": p[1]['transactions']} + """transactions : transactions COMMENT""" + p[0] = { + "comments": p[1]["comments"] + [p[2]], + "transactions": p[1]["transactions"], + } def p_transactions_single(self, p): - '''transactions : transaction ''' + """transactions : transaction""" p[0] = {"comments": [], "transactions": [p[1]]} def p_transactions_comment_single(self, p): - '''transactions : COMMENT ''' + """transactions : COMMENT""" p[0] = {"comments": [p[1]], "transactions": []} def p_amount_prefixed(self, p): - '''amount : CURRENCY AMOUNT''' - p[0] = {'currency': p[1], 'amount': p[2]} + """amount : COMMODITY AMOUNT""" + p[0] = {"currency": p[1], "amount": p[2]} def p_amount_suffixed(self, p): - '''amount : AMOUNT CURRENCY''' - p[0] = {'currency': p[2], 'amount': p[1]} + """amount : AMOUNT COMMODITY""" + p[0] = {"currency": p[2], "amount": p[1]} def p_transaction_with_amount(self, p): - '''transaction : TEXT amount INLINE_COMMENT - | TEXT amount''' + """transaction : TEXT amount INLINE_COMMENT + | TEXT amount""" p[0] = JournalEntryTransaction( - account = p[1], - currency = p[2]['currency'], - amount = p[2]['amount'], - comment = p[3] if len(p) > 3 else None + account=p[1], + currency=p[2]["currency"], + amount=p[2]["amount"], + comment=p[3] if len(p) > 3 else None, ) def p_transaction_without_amount(self, p): - '''transaction : TEXT INLINE_COMMENT - | TEXT''' + """transaction : TEXT INLINE_COMMENT + | TEXT""" p[0] = JournalEntryTransaction( - account = p[1], - currency = None, - amount = None, - comment = p[2] if len(p) > 2 else None + account=p[1], + currency=None, + amount=None, + comment=p[2] if len(p) > 2 else None, ) def p_error(self, p): @@ -131,4 +144,3 @@ class JournalParser(ParseWrapper): self._hl_token(p) else: print("Syntax error at EOF") - |