mirror of
ssh://git.janware.com/janware/proj/jw-pkg
synced 2026-04-24 09:13:37 +02:00
lib.ExecContext.__init__(): Add parameter uri
Take a positional uri argument to the constructor of ExecContext, forcing SSHClient to follow suit. The latter was instantiated with a hostname as only argument up to now, which still works as a special case of an uri. Signed-off-by: Jan Lindemann <jan@janware.com>
This commit is contained in:
parent
db9bf1826d
commit
888c1e7f16
3 changed files with 20 additions and 10 deletions
|
|
@ -11,7 +11,8 @@ class Result(NamedTuple):
|
||||||
|
|
||||||
class ExecContext(abc.ABC):
|
class ExecContext(abc.ABC):
|
||||||
|
|
||||||
def __init__(self, interactive: bool=True, verbose_default=False):
|
def __init__(self, uri: str, interactive: bool=True, verbose_default=False):
|
||||||
|
self.__uri = uri
|
||||||
self.__interactive = interactive
|
self.__interactive = interactive
|
||||||
self.__verbose_default = verbose_default
|
self.__verbose_default = verbose_default
|
||||||
assert verbose_default is not None
|
assert verbose_default is not None
|
||||||
|
|
@ -21,6 +22,10 @@ class ExecContext(abc.ABC):
|
||||||
return verbose
|
return verbose
|
||||||
return self.__verbose_default
|
return self.__verbose_default
|
||||||
|
|
||||||
|
@property
|
||||||
|
def uri(self) -> str:
|
||||||
|
return self.__uri
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def interactive(self) -> bool:
|
def interactive(self) -> bool:
|
||||||
return self.__interactive
|
return self.__interactive
|
||||||
|
|
|
||||||
|
|
@ -7,14 +7,16 @@ import os, abc, shlex, sys
|
||||||
from .util import run_cmd
|
from .util import run_cmd
|
||||||
from .log import *
|
from .log import *
|
||||||
from .ExecContext import ExecContext, Result
|
from .ExecContext import ExecContext, Result
|
||||||
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
class SSHClient(ExecContext):
|
class SSHClient(ExecContext):
|
||||||
|
|
||||||
def __init__(self, hostname: str) -> None:
|
def __init__(self, uri: str, *args, **kwargs) -> None:
|
||||||
super().__init__(interactive=False, verbose_default=False)
|
super().__init__(uri=uri, *args, **kwargs)
|
||||||
self.___ssh = None
|
parsed = urlparse(uri)
|
||||||
self.__hostname = hostname
|
self.__hostname = parsed.hostname
|
||||||
self.__password: str|None = None
|
self.__password: parsed.password
|
||||||
|
self.__username = parsed.username
|
||||||
|
|
||||||
@abc.abstractmethod
|
@abc.abstractmethod
|
||||||
async def _run_cmd(self, cmd: list[str]) -> Result:
|
async def _run_cmd(self, cmd: list[str]) -> Result:
|
||||||
|
|
@ -101,8 +103,8 @@ class SSHClient(ExecContext):
|
||||||
|
|
||||||
class SSHClientInternal(SSHClient): # export
|
class SSHClientInternal(SSHClient): # export
|
||||||
|
|
||||||
def __init__(self, hostname: str) -> None:
|
def __init__(self, *args, **kwargs) -> None:
|
||||||
super().__init__(hostname=hostname)
|
super().__init__(*args, **kwargs)
|
||||||
self.__timeout: float|None = None # Untested
|
self.__timeout: float|None = None # Untested
|
||||||
self.___ssh: Any|None = None
|
self.___ssh: Any|None = None
|
||||||
|
|
||||||
|
|
@ -136,10 +138,10 @@ class SSHClientInternal(SSHClient): # export
|
||||||
|
|
||||||
class SSHClientCmd(SSHClient): # export
|
class SSHClientCmd(SSHClient): # export
|
||||||
|
|
||||||
def __init__(self, hostname: str) -> None:
|
def __init__(self, *args, **kwargs) -> None:
|
||||||
self.__askpass: str|None = None
|
self.__askpass: str|None = None
|
||||||
self.__askpass_orig: dict[str, str|None] = dict()
|
self.__askpass_orig: dict[str, str|None] = dict()
|
||||||
super().__init__(hostname=hostname)
|
super().__init__(*args, **kwargs)
|
||||||
|
|
||||||
def __del__(self):
|
def __del__(self):
|
||||||
for key, val in self.__askpass_orig.items():
|
for key, val in self.__askpass_orig.items():
|
||||||
|
|
|
||||||
|
|
@ -11,6 +11,9 @@ from ..util import pretty_cmd
|
||||||
|
|
||||||
class Local(Base):
|
class Local(Base):
|
||||||
|
|
||||||
|
def __init__(self, uri='local', *args, **kwargs) -> None:
|
||||||
|
super().__init__(uri, *args, **kwargs)
|
||||||
|
|
||||||
async def _run(
|
async def _run(
|
||||||
self,
|
self,
|
||||||
args: list[str],
|
args: list[str],
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue