44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
from pathlib import Path
|
|
|
|
from .parser import StatementParseError, parse_directory, parse_statement
|
|
|
|
|
|
def main() -> int:
|
|
parser = argparse.ArgumentParser(description="Parse bank statements by header signature.")
|
|
parser.add_argument("path", type=Path, help="Statement file or directory")
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
batches = (
|
|
parse_directory(args.path)
|
|
if args.path.is_dir()
|
|
else parse_statement(args.path)
|
|
)
|
|
except (OSError, StatementParseError) as exc:
|
|
parser.error(str(exc))
|
|
|
|
summaries = [
|
|
{
|
|
"file": batch.source_file.name,
|
|
"sheet": batch.sheet_name,
|
|
"bank": batch.bank_name,
|
|
"template": batch.template_id,
|
|
"header_row": batch.header_row,
|
|
"period_start": batch.period_start.isoformat() if batch.period_start else None,
|
|
"period_end": batch.period_end.isoformat() if batch.period_end else None,
|
|
"transactions": len(batch.transactions),
|
|
"warnings": list(batch.warnings),
|
|
}
|
|
for batch in batches
|
|
]
|
|
print(json.dumps(summaries, ensure_ascii=False, indent=2))
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|