def set_metadata(self, x: Directive, lineno: Optional[int] = None):
meta = dict({}, **(x.meta or {}))
meta['filename'] = self.filename
if lineno is None:
lineno = self.new_lineno + 1
meta['lineno'] = lineno
return x._replace(meta=meta)
def _partially_book_entry(orig_entry: Directive,
booked_entry: Directive) -> Directive:
"""Computes a partially-booked entry.
Given a pre-booking entry and a post-booking entry, returns a modified copy
of the pre-booking entry that includes missing units.
"""
if not isinstance(orig_entry, Transaction): return orig_entry
assert isinstance(booked_entry, Transaction)
booked_postings_by_meta = dict() # type: Dict[int, List[Posting]]
for posting in booked_entry.postings:
meta = posting.meta
if meta is None: continue
booked_postings_by_meta.setdefault(id(meta), []).append(posting)
partially_booked_postings = [] # type: List[Posting]
empty_list = [] # type: List[Posting]
for posting in orig_entry.postings:
booked_matches = booked_postings_by_meta.get(
id(posting.meta), empty_list)
if len(booked_matches) != 1:
partially_booked_postings.append(posting)
continue
booked_match = booked_matches[0]
if posting.units is MISSING and booked_match.units is not MISSING:
posting = posting._replace(units=booked_match.units)
partially_booked_postings.append(posting)
return orig_entry._replace(postings=partially_booked_postings)
def _format_entry(entry: Directive, currency_column: int) -> str:
"""Wrapper that strips unnecessary whitespace from format_entry."""
meta = {
key: entry.meta[key] for key in entry.meta if not key.startswith("_")
}
entry = entry._replace(meta=meta)
string = align(format_entry(entry), currency_column)
string = string.replace("<class 'beancount.core.number.MISSING'>", "")
return "\n".join((line.rstrip() for line in string.split("\n")))
def categorize_entry(catmap: Dict[Account, Cat],
entry: data.Directive) -> Tuple[Cat]:
"""Assigns metadata to each posting."""
postings = []
for posting in entry.postings:
category = catmap[posting.account]
if category is None:
category = Cat.OTHER if posting.cost is None else Cat.OTHERASSET
meta = posting.meta.copy() if posting.meta else {}
meta["category"] = category
postings.append(posting._replace(meta=meta))
return entry._replace(postings=postings)
def merge_narration(entry: Directive, comment_narrations: Dict[str, Dict[int, str]]) -> Directive:
if not isinstance(entry, Transaction):
return entry
narrations = []
for posting in entry.postings:
if posting.meta:
narration = posting.meta.get('narration')
comment_narration = comment_narrations.get(posting.meta.get('filename')).get(posting.meta.get('lineno'))
if narration is None and comment_narration:
posting.meta['narration'] = comment_narration
narration = comment_narration
if narration:
narrations.append(narration.strip())
if entry.narration:
return entry
return entry._replace(narration=' | '.join(narrations))