scripts/q7s-cp.py

@@ -49,6 +49,13 @@ def handle_args():
         action="store_true",
         help="Copy from Q7S to host instead. Always copies to current directory.",
     )
+    parser.add_argument(
+        "-f",
+        "--flatsat",
+        default=False,
+        action="store_true",
+        help="Copy to flatsat instead"
+    )
     # Positional argument(s)
     parser.add_argument(
         "source", help="Source files to copy or target files to copy back to host"
@@ -61,18 +68,21 @@ def build_cmd(args):
     cmd = "scp "
     if args.recursive:
         cmd += "-r "
-    target = args.target
-    if args.invert and target == "":
-        target = "."
-    elif target == "":
-        target = f"/tmp"
-    if args.invert:
-        cmd += f"-P {args.port} root@localhost:{args.source} {target}"
+    if args.flatsat:
+        cmd += f"{args.source} eive@flatsat.eive.absatvirt.lw:{target}"
     else:
-        cmd += f"-P {args.port} {args.source} root@localhost:{target}"
-    if args.target:
-        cmd += args.target
-    return cmd
+        target = args.target
+        if args.invert and target == "":
+            target = "."
+        elif target == "":
+            target = f"/tmp"
+        if args.invert:
+            cmd += f"-P {args.port} root@localhost:{args.source} {target}"
+        else:
+            cmd += f"-P {args.port} {args.source} root@localhost:{target}"
+        if args.target:
+            cmd += args.target
+        return cmd
 
 
 if __name__ == "__main__":