aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorvg <vg@devys.org>2019-02-06 12:03:57 +0100
committervg <vg@devys.org>2019-02-06 12:03:57 +0100
commit87ec9b2f30b6aa39481ea586447078967447c22f (patch)
treef0bc672ddcfddbcdb2c7b7626257a50b222df1b6
parentcc68421645a3cd28dcf48a045845c13aa0a830f0 (diff)
downloadscripts-87ec9b2f30b6aa39481ea586447078967447c22f.tar.gz
scripts-87ec9b2f30b6aa39481ea586447078967447c22f.tar.bz2
scripts-87ec9b2f30b6aa39481ea586447078967447c22f.zip
new script to convert ofx to json format
-rwxr-xr-xscripts/ofx2json115
1 files changed, 115 insertions, 0 deletions
diff --git a/scripts/ofx2json b/scripts/ofx2json
new file mode 100755
index 0000000..08d9fe9
--- /dev/null
+++ b/scripts/ofx2json
@@ -0,0 +1,115 @@
+#!/usr/bin/env python3
+# Copyright 2019 vg@devys.org
+# SPDX-License-Identifier: MIT
+
+progname = 'ofx2json'
+
+# be explicit about __doc__ to allow progname to be defined first
+__doc__ = f'''
+Convert ofx from input file (or stdin if no FILENAME) to json on stdout.
+
+Usage: {progname} [options] [--] [FILENAME...]
+ {progname} -h|--help
+
+Options:
+ -h, --help Display this help message
+ -n option doing n
+ -N ARG option doing N with ARG
+'''
+
+
+### standard modules
+import collections
+import datetime
+import decimal
+import io
+import json
+import os
+import subprocess
+import sys
+### external modules
+import docopt
+import ofxparse
+
+# This script is not compatible below python3.7.2, always abort (not in main
+# since the syntax itself can cause the script to fail later inconveniently).
+assert sys.hexversion >= 0x03070200
+
+
+def map_ofx_transaction(transaction):
+ return collections.OrderedDict(
+ amount=transaction.amount,
+ checknum=transaction.checknum,
+ date=transaction.date,
+ id=transaction.id,
+ mcc=transaction.mcc,
+ memo=transaction.memo,
+ payee=transaction.payee,
+ sic=transaction.sic,
+ type=transaction.type,
+ )
+
+
+def map_ofx_statement(statement):
+ return collections.OrderedDict(
+ available_balance=statement.available_balance,
+ available_balance_date=statement.available_balance_date,
+ balance=statement.balance,
+ balance_date=statement.balance_date,
+ currency=statement.currency,
+ discarded_entries=statement.discarded_entries,
+ end_date=statement.end_date,
+ start_date=statement.start_date,
+ transactions=[map_ofx_transaction(i)
+ for i in statement.transactions],
+ warnings=statement.warnings,
+ )
+
+
+def map_ofx_account(account):
+ return collections.OrderedDict(
+ account_id=account.account_id,
+ account_type=account.account_type,
+ branch_id=account.branch_id,
+ curdef=account.curdef,
+ institution=account.institution,
+ number=account.number,
+ rounting_number=account.routing_number,
+ statement=map_ofx_statement(account.statement),
+ type=account.type,
+ warnings=account.warnings,
+ )
+
+
+def map_ofx(ofx):
+ return collections.OrderedDict(
+ headers=ofx.headers,
+ accounts=[map_ofx_account(i) for i in ofx.accounts],
+ )
+
+
+def parse_ofx(filepaths):
+ if not filepaths:
+ yield ofxparse.OfxParser.parse(io.StringIO(sys.stdin.read()))
+ else:
+ for filepath in filepaths:
+ with open(filepath) as fh:
+ yield ofxparse.OfxParser.parse(fh)
+
+
+def main():
+ 'function called only when script invoked directly on command line'
+ args = docopt.docopt(__doc__)
+
+ def json_default(obj):
+ if isinstance(obj, decimal.Decimal):
+ return str(obj) # no rounding by dumping a string, not float
+ elif hasattr(obj, 'isoformat'):
+ return obj.isoformat()
+ print(f'error for {obj}')
+ raise TypeError
+
+ print(json.dumps(list(map(map_ofx, parse_ofx(args['FILENAME']))),
+ indent=2, default=json_default))
+
+main()