From b5ee3d38a3bb4c220d017c64e58399bbaa4c9346 Mon Sep 17 00:00:00 2001 From: xssdoctor Date: Sat, 20 Apr 2024 11:56:01 -0400 Subject: [PATCH] added session log to view your sessions --- installer/client/cli/fabric.py | 6 ++++++ installer/client/cli/helper.py | 6 ++++++ 2 files changed, 12 insertions(+) diff --git a/installer/client/cli/fabric.py b/installer/client/cli/fabric.py index e0b4fe2..fe81397 100755 --- a/installer/client/cli/fabric.py +++ b/installer/client/cli/fabric.py @@ -31,6 +31,7 @@ def main(): parser.add_argument('--session', '-S', help="Continue your previous conversation. Default is your previous conversation", nargs="?", const="default") parser.add_argument('--clearsession', help="deletes indicated session. Use 'all' to delete all dessions") + parser.add_argument('--sessionlog', help="View the log of a session") parser.add_argument( "--gui", help="Use the GUI (Node and npm need to be installed)", action="store_true") parser.add_argument( @@ -133,6 +134,11 @@ def main(): else: print(f"Session {args.clearsession} cleared") sys.exit() + if args.sessionlog: + from .helper import Session + session = Session() + print(session.session_log(args.sessionlog)) + sys.exit() standalone = Standalone(args, args.pattern) if args.list: try: diff --git a/installer/client/cli/helper.py b/installer/client/cli/helper.py index 9fefef2..da4815d 100644 --- a/installer/client/cli/helper.py +++ b/installer/client/cli/helper.py @@ -49,3 +49,9 @@ class Session: os.remove(os.path.join(self.sessions_folder, file)) else: os.remove(os.path.join(self.sessions_folder, session)) + def session_log(self, session): + file = os.path.join(self.sessions_folder, session) + if not os.path.exists(file): + return None + with open(file, "r") as f: + return f.read()