add argument for an ssh key

Opentrons · Sep 19, 2023 · d78336e · d78336e
1 parent ae4a7ff
commit d78336e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/push b/push
@@ -122,6 +122,9 @@ def push(host, repo_path=None, build=True, restart=True):
         return -1
 
 def _push_from_argparse(args):
+    if args.key:
+        _SSH_EXTRA_OPTS.append('-i')
+        _SSH_EXTRA_OPTS.append(args.key)
     return push(args.host, os.path.abspath(args.repo_path), not args.no_build, not args.no_restart)
 
 def _arg_parser(parent=None):
@@ -148,6 +151,11 @@ def _arg_parser(parent=None):
         action='store_true',
         help='Skip restarting robot server'
     )
+    parser.add_argument(
+        '--key',
+        type=str,
+        help='Private SSH key to use'
+    )
     return parser
 
 def _main():