1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
#!/usr/bin/env python3
from importers import acme
from importers import csvbank
from importers import ofx
from importers import utrade
from beancount.core import data
import beangulp
importers = [
utrade.Importer("USD",
"Assets:US:UTrade",
"Assets:US:UTrade:Cash",
"Income:US:UTrade:{}:Dividend",
"Income:US:UTrade:{}:Gains",
"Expenses:Financial:Fees",
"Assets:US:BofA:Checking"),
ofx.Importer("379700001111222",
"Liabilities:US:CreditCard",
"bofa"),
acme.Importer("Assets:US:ACMEBank"),
csvbank.Importer("Assets:US:CSVBank", "USD"),
]
def clean_up_descriptions(extracted_entries):
"""Example filter function; clean up cruft from narrations.
Args:
extracted_entries: A list of directives.
Returns:
A new list of directives with possibly modified payees and narration
fields.
"""
clean_entries = []
for entry in extracted_entries:
if isinstance(entry, data.Transaction):
if entry.narration and " / " in entry.narration:
left_part, _ = entry.narration.split(" / ")
entry = entry._replace(narration=left_part)
if entry.payee and " / " in entry.payee:
left_part, _ = entry.payee.split(" / ")
entry = entry._replace(payee=left_part)
clean_entries.append(entry)
return clean_entries
def process_extracted_entries(extracted_entries_list, ledger_entries):
"""Example filter function; clean up cruft from narrations.
Args:
extracted_entries_list: A list of (filename, entries) pairs, where
'entries' are the directives extract from 'filename'.
ledger_entries: If provided, a list of directives from the existing
ledger of the user. This is non-None if the user provided their
ledger file as an option.
Returns:
A possibly different version of extracted_entries_list, a list of
(filename, entries), to be printed.
"""
return [(filename, clean_up_descriptions(entries), account, importer)
for filename, entries, account, importer in extracted_entries_list]
hooks = [process_extracted_entries]
if __name__ == '__main__':
ingest = beangulp.Ingest(importers, hooks)
ingest()
|
