clan-lib: Rename parse_deployment_address to parse_ssh_uri
This commit is contained in:
@@ -167,7 +167,7 @@ def update_hardware_config_command(args: argparse.Namespace) -> None:
|
|||||||
)
|
)
|
||||||
|
|
||||||
if args.target_host:
|
if args.target_host:
|
||||||
target_host = Remote.from_deployment_address(
|
target_host = Remote.from_ssh_uri(
|
||||||
machine_name=machine.name, address=args.target_host
|
machine_name=machine.name, address=args.target_host
|
||||||
).override(host_key_check=host_key_check)
|
).override(host_key_check=host_key_check)
|
||||||
else:
|
else:
|
||||||
|
|||||||
@@ -185,7 +185,7 @@ def install_command(args: argparse.Namespace) -> None:
|
|||||||
host_key_check = args.host_key_check
|
host_key_check = args.host_key_check
|
||||||
|
|
||||||
if target_host_str is not None:
|
if target_host_str is not None:
|
||||||
target_host = Remote.from_deployment_address(
|
target_host = Remote.from_ssh_uri(
|
||||||
machine_name=machine.name, address=target_host_str
|
machine_name=machine.name, address=target_host_str
|
||||||
).override(host_key_check=host_key_check)
|
).override(host_key_check=host_key_check)
|
||||||
else:
|
else:
|
||||||
|
|||||||
@@ -275,7 +275,7 @@ def update_command(args: argparse.Namespace) -> None:
|
|||||||
with AsyncRuntime() as runtime:
|
with AsyncRuntime() as runtime:
|
||||||
for machine in machines:
|
for machine in machines:
|
||||||
if args.target_host:
|
if args.target_host:
|
||||||
target_host = Remote.from_deployment_address(
|
target_host = Remote.from_ssh_uri(
|
||||||
machine_name=machine.name,
|
machine_name=machine.name,
|
||||||
address=args.target_host,
|
address=args.target_host,
|
||||||
).override(host_key_check=host_key_check)
|
).override(host_key_check=host_key_check)
|
||||||
|
|||||||
@@ -46,7 +46,7 @@ class DeployInfo:
|
|||||||
if not host:
|
if not host:
|
||||||
msg = "Hostname cannot be empty."
|
msg = "Hostname cannot be empty."
|
||||||
raise ClanError(msg)
|
raise ClanError(msg)
|
||||||
remote = Remote.from_deployment_address(
|
remote = Remote.from_ssh_uri(
|
||||||
machine_name="clan-installer", address=host
|
machine_name="clan-installer", address=host
|
||||||
).override(host_key_check=host_key_check)
|
).override(host_key_check=host_key_check)
|
||||||
remotes.append(remote)
|
remotes.append(remote)
|
||||||
@@ -59,7 +59,7 @@ class DeployInfo:
|
|||||||
|
|
||||||
for addr in data.get("addrs", []):
|
for addr in data.get("addrs", []):
|
||||||
if isinstance(addr, str):
|
if isinstance(addr, str):
|
||||||
remote = Remote.from_deployment_address(
|
remote = Remote.from_ssh_uri(
|
||||||
machine_name="clan-installer",
|
machine_name="clan-installer",
|
||||||
address=addr,
|
address=addr,
|
||||||
).override(host_key_check=host_key_check, password=password)
|
).override(host_key_check=host_key_check, password=password)
|
||||||
@@ -68,7 +68,7 @@ class DeployInfo:
|
|||||||
msg = f"Invalid address format: {addr}"
|
msg = f"Invalid address format: {addr}"
|
||||||
raise ClanError(msg)
|
raise ClanError(msg)
|
||||||
if tor_addr := data.get("tor"):
|
if tor_addr := data.get("tor"):
|
||||||
remote = Remote.from_deployment_address(
|
remote = Remote.from_ssh_uri(
|
||||||
machine_name="clan-installer",
|
machine_name="clan-installer",
|
||||||
address=tor_addr,
|
address=tor_addr,
|
||||||
).override(host_key_check=host_key_check, tor_socks=True, password=password)
|
).override(host_key_check=host_key_check, tor_socks=True, password=password)
|
||||||
|
|||||||
@@ -258,8 +258,6 @@ def get_host(
|
|||||||
return None
|
return None
|
||||||
|
|
||||||
return RemoteSource(
|
return RemoteSource(
|
||||||
data=Remote.from_deployment_address(
|
data=Remote.from_ssh_uri(machine_name=machine.name, address=host_str),
|
||||||
machine_name=machine.name, address=host_str
|
|
||||||
),
|
|
||||||
source=source,
|
source=source,
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ if TYPE_CHECKING:
|
|||||||
from clan_lib.ssh.remote import Remote
|
from clan_lib.ssh.remote import Remote
|
||||||
|
|
||||||
|
|
||||||
def parse_deployment_address(
|
def parse_ssh_uri(
|
||||||
*,
|
*,
|
||||||
machine_name: str,
|
machine_name: str,
|
||||||
address: str,
|
address: str,
|
||||||
|
|||||||
@@ -21,7 +21,7 @@ from clan_lib.colors import AnsiColor
|
|||||||
from clan_lib.errors import ClanError # Assuming these are available
|
from clan_lib.errors import ClanError # Assuming these are available
|
||||||
from clan_lib.nix import nix_shell
|
from clan_lib.nix import nix_shell
|
||||||
from clan_lib.ssh.host_key import HostKeyCheck, hostkey_to_ssh_opts
|
from clan_lib.ssh.host_key import HostKeyCheck, hostkey_to_ssh_opts
|
||||||
from clan_lib.ssh.parse import parse_deployment_address
|
from clan_lib.ssh.parse import parse_ssh_uri
|
||||||
from clan_lib.ssh.sudo_askpass_proxy import SudoAskpassProxy
|
from clan_lib.ssh.sudo_askpass_proxy import SudoAskpassProxy
|
||||||
|
|
||||||
cmdlog = logging.getLogger(__name__)
|
cmdlog = logging.getLogger(__name__)
|
||||||
@@ -90,7 +90,7 @@ class Remote:
|
|||||||
return f"{self.user}@{self.address}"
|
return f"{self.user}@{self.address}"
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_deployment_address(
|
def from_ssh_uri(
|
||||||
cls,
|
cls,
|
||||||
*,
|
*,
|
||||||
machine_name: str,
|
machine_name: str,
|
||||||
@@ -100,7 +100,7 @@ class Remote:
|
|||||||
Parse a deployment address and return a Host object.
|
Parse a deployment address and return a Host object.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
return parse_deployment_address(machine_name=machine_name, address=address)
|
return parse_ssh_uri(machine_name=machine_name, address=address)
|
||||||
|
|
||||||
def run_local(
|
def run_local(
|
||||||
self,
|
self,
|
||||||
|
|||||||
@@ -110,7 +110,7 @@ def test_parse_deployment_address(
|
|||||||
|
|
||||||
with maybe_check_exception:
|
with maybe_check_exception:
|
||||||
machine_name = "foo"
|
machine_name = "foo"
|
||||||
result = Remote.from_deployment_address(
|
result = Remote.from_ssh_uri(
|
||||||
machine_name=machine_name,
|
machine_name=machine_name,
|
||||||
address=test_addr,
|
address=test_addr,
|
||||||
).override(host_key_check="strict")
|
).override(host_key_check="strict")
|
||||||
@@ -130,7 +130,7 @@ def test_parse_deployment_address(
|
|||||||
|
|
||||||
def test_parse_ssh_options() -> None:
|
def test_parse_ssh_options() -> None:
|
||||||
addr = "root@example.com:2222?IdentityFile=/path/to/private/key&StrictRemoteKeyChecking=yes"
|
addr = "root@example.com:2222?IdentityFile=/path/to/private/key&StrictRemoteKeyChecking=yes"
|
||||||
host = Remote.from_deployment_address(machine_name="foo", address=addr).override(
|
host = Remote.from_ssh_uri(machine_name="foo", address=addr).override(
|
||||||
host_key_check="strict"
|
host_key_check="strict"
|
||||||
)
|
)
|
||||||
assert host.address == "example.com"
|
assert host.address == "example.com"
|
||||||
|
|||||||
Reference in New Issue
Block a user