add option to query account details including personal data

pytr/accountdetails.py

@@ -0,0 +1,16 @@
+import json
+
+class Accountdetails:
+    def __init__(self, tr):
+        self.tr = tr
+        self.data = None
+
+    def get(self):
+        self.data = self.tr.get_account_details()
+        print(f"{self.data}")
+        return self.data
+
+    def data_to_file(self, output_path):
+        with open(output_path, "w", encoding="utf-8") as f:
+            json.dump(self.data, f)
+

pytr/api.py

@@ -262,6 +262,13 @@ def resume_websession(self):
                 return True
         return False
 
+    def get_account_details(self):
+        r = self._websession.get(
+            f"{self._host}/api/v2/auth/account",
+        )
+        j = r.json()
+        return j
+
     def _web_request(self, url_path, payload=None, method="GET"):
         if self._web_session_token_expires_at < time.time():
             r = self._websession.get(f"{self._host}/api/v1/auth/web/session")

pytr/main.py

@@ -17,6 +17,7 @@
 from pytr.portfolio import Portfolio
 from pytr.alarms import Alarms
 from pytr.details import Details
+from pytr.accountdetails import Accountdetails
 
 
 def get_main_parser():
@@ -200,6 +201,18 @@ def formatter(prog):
         help='Two letter language code or "auto" for system language',
         default="auto",
     )
+    # account details
+    info = "Show account details"
+    parser_accountdetails = parser_cmd.add_parser(
+        "accountdetails",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+        parents=[parser_login_args],
+        help=info,
+        description=info,
+    )
+    parser_accountdetails.add_argument(
+        "-j", "--jsonoutput", help="Output path of JSON file", metavar="OUTPUT", type=Path
+    )
 
     info = "Print shell tab completion"
     parser_completion = parser_cmd.add_parser(
@@ -287,6 +300,13 @@ def main():
         installed_version = version("pytr")
         print(installed_version)
         check_version(installed_version)
+    elif args.command == "accountdetails":
+        ad = Accountdetails(
+            login(phone_no=args.phone_no, pin=args.pin, web=not args.applogin),
+        )
+        ad.get()
+        if args.jsonoutput is not None:
+            ad.data_to_file(args.jsonoutput)
     else:
         parser.print_help()