Add --export-format {csv,json} to portfolio command

pytr/main.py

@@ -190,8 +190,13 @@ def formatter(prog):
         help=info,
         description=info,
     )
-    parser_portfolio.add_argument("-o", "--output", help="Output path of CSV file", type=Path)
-
+    parser_portfolio.add_argument("-o", "--output", help="Output path of the file", type=Path)
+    parser_portfolio.add_argument(
+        "--export-format",
+        choices=("json", "csv"),
+        default="csv",
+        help="The output file format.",
+    )
     # details
     info = "Get details for an ISIN"
     parser_details = parser_cmd.add_parser(
@@ -367,7 +372,10 @@ def main():
         )
         p.get()
         if args.output is not None:
-            p.portfolio_to_csv(args.output)
+            if args.export_format == "csv":
+                p.portfolio_to_csv(args.output)
+            elif args.export_format == "json":
+                p.portfolio_to_json(args.output)
     elif args.command == "export_transactions":
         events = [Event.from_dict(item) for item in json.load(args.input)]
         TransactionExporter(

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"