scripts/q7s-cp.py

@@ -68,20 +68,28 @@ def build_cmd(args):
     cmd = "scp "
     if args.recursive:
         cmd += "-r "
+    address = ""
+    port_args = ""
+    target = args.target
     if args.flatsat:
-        cmd += f"{args.source} eive@flatsat.eive.absatvirt.lw:{target}"
+        address = "eive@flatsat.eive.absatvirt.lw"
+    else:
+        address = "root@localhost"
+        port_args=f"-P {args.port}"
+    if args.invert:
+        if target == "":
+            target = "."
         else:
             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
+        if target == "":
+            target = f"/tmp"
+        else:
+            target = args.target
+    if args.invert:
+        cmd += f"{port_args} {address}:{args.source} {target}"
+    else:
+        cmd += f"{port_args} {args.source} {address}:{target}"
     return cmd