Add --json-output to portfolio command

Author: pschmitt

pytr/main.py

@@ -134,6 +134,7 @@ def formatter(prog):
         help=info,
         description=info,
     )
+    parser_portfolio.add_argument("-j", "--json-output", help="Output path of JSON file", metavar="OUTPUT", type=Path)
     parser_portfolio.add_argument("-o", "--output", help="Output path of CSV file", metavar="OUTPUT", type=Path)
     # details
     info = "Get details for an ISIN"
@@ -304,6 +305,8 @@ def main():
         p.get()
         if args.output is not None:
             p.portfolio_to_csv(args.output)
+        if args.json_output is not None:
+            p.portfolio_to_json(args.json_output)
     elif args.command == "export_transactions":
         export_transactions(args.input, args.output, args.lang, args.sort, args.date_isoformat)
     elif args.version:

pytr/portfolio.py

@@ -1,4 +1,5 @@
 import asyncio
+import json
 
 from pytr.utils import preview
 
@@ -84,6 +85,19 @@ def portfolio_to_csv(self, output_path):
 
         print(f"Wrote {len(csv_lines) + 1} lines to {output_path}")
 
+    def portfolio_to_json(self, output_path):
+        # Combine the portfolio positions with the cash portfolio.
+        data = {
+            "positions": sorted(self.portfolio["positions"], key=lambda x: x["netSize"], reverse=True),
+            "cash": self.cash
+        }
+        if output_path is None or str(output_path) == "-":
+            print(json.dumps(data, indent=2))
+        else:
+            with open(output_path, "w", encoding="utf-8") as f:
+                json.dump(data, f, indent=2)
+                print(f"Wrote portfolio and cash data to {output_path}")
+
     def overview(self):
         print(
             "Name                      ISIN            avgCost *   quantity =    buyCost ->   netValue       diff   %-diff"