added functionality to save output to the RPi

run.py

@@ -3,6 +3,7 @@
 import os
 from dotenv import load_dotenv
 import re
+from datetime import datetime
 
 def establishConnection(ssh_server, ssh_port, user, pasd):
     ssh_client = paramiko.client.SSHClient()
@@ -28,7 +29,9 @@ def isCommandInstalled(base_cmd, ssh_client):
 
 def runCommand(base_cmd, arguments):
     try:
-        full_command = f"{base_cmd} {arguments}"
+        makedir = f"mkdir -p $HOME/offloaded_scans/ && "
+        out_redirect = f"| tee $HOME/offloaded_scans/{datetime.now().strftime(f'%d-%m-%Y_%H-%M-%S_{base_cmd}')}.txt"
+        full_command = f"{makedir} {base_cmd} {arguments} {out_redirect}"
         stdin, stdout, stderr = ssh_client.exec_command(full_command)
         output = stdout.read().decode('utf-8')
         print(output)
@@ -44,7 +47,7 @@ def runCommand(base_cmd, arguments):
     if len(sys.argv) != 3 or sys.argv[1] == '-h':
         print("Usage: python3 script.py <base command> <arguments>")
         sys.exit(1)
-    
+
     base_cmd = sys.argv[1]
     arguments = sys.argv[2]
 

todo.md

@@ -9,3 +9,4 @@
 - [x] create venv
 - [x] installation instructions
 - [x] update readme
+- [x] save output to file (as datetime-base_cmd.txt)