use host-specific ports in SSH config utility

roscisz · roscisz · commit e344071e1008 · 2021-06-23T15:21:41.000+02:00
limit SSH connections to the configured hosts
diff --git a/tensorhive/core/ssh.py b/tensorhive/core/ssh.py
@@ -36,10 +36,12 @@ def build_dedicated_config_for(host: Hostname, user: Username) -> Tuple[HostsCon
     valid `config` and `pconfig` parameter to `get_client()` function.
     """
     assert host and user, 'Arguments must not be None!'
+    assert host in SSH.AVAILABLE_NODES
     hosts_config = {
         host: {
             'user': user,
-            'pkey': SSH.KEY_FILE
+            'pkey': SSH.KEY_FILE,
+            'port': SSH.AVAILABLE_NODES[host]['port']
         }
     }
     # Read config extracted from hosts_config.ini (proxy is common for all hosts)

Original file line number	Diff line number	Diff line change
`@@ -36,10 +36,12 @@ def build_dedicated_config_for(host: Hostname, user: Username) -> Tuple[HostsCon`
`36`	`36`	valid `config` and `pconfig` parameter to `get_client()` function.
`37`	`37`	`"""`
`38`	`38`	`assert host and user, 'Arguments must not be None!'`
	`39`	`+ assert host in SSH.AVAILABLE_NODES`
`39`	`40`	`hosts_config = {`
`40`	`41`	`host: {`
`41`	`42`	`'user': user,`
`42`		`- 'pkey': SSH.KEY_FILE`
	`43`	`+ 'pkey': SSH.KEY_FILE,`
	`44`	`+ 'port': SSH.AVAILABLE_NODES[host]['port']`
`43`	`45`	`}`
`44`	`46`	`}`
`45`	`47`	`# Read config extracted from hosts_config.ini (proxy is common for all hosts)`