clan-lib: Remove override args from parse_deployment_address, instead use the Remote.override method where necessary
This commit is contained in:
@@ -62,8 +62,7 @@ class DeployInfo:
|
|||||||
remote = Remote.from_deployment_address(
|
remote = Remote.from_deployment_address(
|
||||||
machine_name="clan-installer",
|
machine_name="clan-installer",
|
||||||
address=addr,
|
address=addr,
|
||||||
password=password,
|
).override(host_key_check=host_key_check, password=password)
|
||||||
).override(host_key_check=host_key_check)
|
|
||||||
addrs.append(remote)
|
addrs.append(remote)
|
||||||
else:
|
else:
|
||||||
msg = f"Invalid address format: {addr}"
|
msg = f"Invalid address format: {addr}"
|
||||||
@@ -72,9 +71,7 @@ class DeployInfo:
|
|||||||
remote = Remote.from_deployment_address(
|
remote = Remote.from_deployment_address(
|
||||||
machine_name="clan-installer",
|
machine_name="clan-installer",
|
||||||
address=tor_addr,
|
address=tor_addr,
|
||||||
password=password,
|
).override(host_key_check=host_key_check, tor_socks=True, password=password)
|
||||||
tor_socks=True,
|
|
||||||
).override(host_key_check=host_key_check)
|
|
||||||
addrs.append(remote)
|
addrs.append(remote)
|
||||||
|
|
||||||
return DeployInfo(addrs=addrs)
|
return DeployInfo(addrs=addrs)
|
||||||
|
|||||||
@@ -1,7 +1,6 @@
|
|||||||
import re
|
import re
|
||||||
import urllib.parse
|
import urllib.parse
|
||||||
from pathlib import Path
|
from typing import TYPE_CHECKING
|
||||||
from typing import TYPE_CHECKING, Any
|
|
||||||
|
|
||||||
from clan_lib.errors import ClanError
|
from clan_lib.errors import ClanError
|
||||||
|
|
||||||
@@ -13,12 +12,14 @@ def parse_deployment_address(
|
|||||||
*,
|
*,
|
||||||
machine_name: str,
|
machine_name: str,
|
||||||
address: str,
|
address: str,
|
||||||
forward_agent: bool = True,
|
|
||||||
meta: dict[str, Any] | None = None,
|
|
||||||
private_key: Path | None = None,
|
|
||||||
password: str | None = None,
|
|
||||||
tor_socks: bool = False,
|
|
||||||
) -> "Remote":
|
) -> "Remote":
|
||||||
|
"""
|
||||||
|
Parses an SSH URI into a Remote object.
|
||||||
|
The address can be in the form of:
|
||||||
|
- `ssh://[user@]hostname[:port]?option=value&option2=value2`
|
||||||
|
- `[user@]hostname[:port]`
|
||||||
|
The specification can be found here: https://www.ietf.org/archive/id/draft-salowey-secsh-uri-00.html
|
||||||
|
"""
|
||||||
if address.startswith("ssh://"):
|
if address.startswith("ssh://"):
|
||||||
# Strip the `ssh://` prefix if it exists
|
# Strip the `ssh://` prefix if it exists
|
||||||
address = address[len("ssh://") :]
|
address = address[len("ssh://") :]
|
||||||
@@ -68,10 +69,6 @@ def parse_deployment_address(
|
|||||||
address=hostname,
|
address=hostname,
|
||||||
user=user,
|
user=user,
|
||||||
port=port,
|
port=port,
|
||||||
private_key=private_key,
|
|
||||||
password=password,
|
|
||||||
command_prefix=machine_name,
|
command_prefix=machine_name,
|
||||||
forward_agent=forward_agent,
|
|
||||||
ssh_options=options,
|
ssh_options=options,
|
||||||
tor_socks=tor_socks,
|
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -61,6 +61,8 @@ class Remote:
|
|||||||
*,
|
*,
|
||||||
host_key_check: HostKeyCheck | None = None,
|
host_key_check: HostKeyCheck | None = None,
|
||||||
private_key: Path | None = None,
|
private_key: Path | None = None,
|
||||||
|
password: str | None = None,
|
||||||
|
tor_socks: bool | None = None,
|
||||||
) -> "Remote":
|
) -> "Remote":
|
||||||
"""
|
"""
|
||||||
Returns a new Remote instance with the same data but with a different host_key_check.
|
Returns a new Remote instance with the same data but with a different host_key_check.
|
||||||
@@ -71,14 +73,14 @@ class Remote:
|
|||||||
command_prefix=self.command_prefix,
|
command_prefix=self.command_prefix,
|
||||||
port=self.port,
|
port=self.port,
|
||||||
private_key=private_key if private_key is not None else self.private_key,
|
private_key=private_key if private_key is not None else self.private_key,
|
||||||
password=self.password,
|
password=password if password is not None else self.password,
|
||||||
forward_agent=self.forward_agent,
|
forward_agent=self.forward_agent,
|
||||||
host_key_check=host_key_check
|
host_key_check=host_key_check
|
||||||
if host_key_check is not None
|
if host_key_check is not None
|
||||||
else self.host_key_check,
|
else self.host_key_check,
|
||||||
verbose_ssh=self.verbose_ssh,
|
verbose_ssh=self.verbose_ssh,
|
||||||
ssh_options=self.ssh_options,
|
ssh_options=self.ssh_options,
|
||||||
tor_socks=self.tor_socks,
|
tor_socks=tor_socks if tor_socks is not None else self.tor_socks,
|
||||||
_control_path_dir=self._control_path_dir,
|
_control_path_dir=self._control_path_dir,
|
||||||
_askpass_path=self._askpass_path,
|
_askpass_path=self._askpass_path,
|
||||||
)
|
)
|
||||||
@@ -93,23 +95,12 @@ class Remote:
|
|||||||
*,
|
*,
|
||||||
machine_name: str,
|
machine_name: str,
|
||||||
address: str,
|
address: str,
|
||||||
forward_agent: bool = True,
|
|
||||||
private_key: Path | None = None,
|
|
||||||
password: str | None = None,
|
|
||||||
tor_socks: bool = False,
|
|
||||||
) -> "Remote":
|
) -> "Remote":
|
||||||
"""
|
"""
|
||||||
Parse a deployment address and return a Host object.
|
Parse a deployment address and return a Host object.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
return parse_deployment_address(
|
return parse_deployment_address(machine_name=machine_name, address=address)
|
||||||
machine_name=machine_name,
|
|
||||||
address=address,
|
|
||||||
forward_agent=forward_agent,
|
|
||||||
private_key=private_key,
|
|
||||||
password=password,
|
|
||||||
tor_socks=tor_socks,
|
|
||||||
)
|
|
||||||
|
|
||||||
def run_local(
|
def run_local(
|
||||||
self,
|
self,
|
||||||
|
|||||||
Reference in New Issue
Block a user