mirror of
https://github.com/kernelkit/infix.git
synced 2026-07-30 12:33:02 +02:00
Mattias Walström
35a909edf0
This is now possible for SSH/RESTCONF/NETCONF, most useful with SSH and adding/modify users.
57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
import subprocess
|
|
|
|
from dataclasses import dataclass
|
|
from . import env
|
|
|
|
|
|
@dataclass
|
|
class Location:
|
|
host: str
|
|
username: str
|
|
password: str
|
|
port: int = 22
|
|
|
|
|
|
class Device(object):
|
|
def __init__(self, name: str, location: Location):
|
|
self.name = name
|
|
self.location = location
|
|
def __str__(self):
|
|
nm = f"{self.name}"
|
|
if env.ENV.ltop:
|
|
nm += f"({env.ENV.ltop.xlate(self.name)})"
|
|
return nm + " [SSH]"
|
|
|
|
def _mangle_subprocess_args(self, args, kwargs):
|
|
if not args:
|
|
return None
|
|
|
|
args = list(args)
|
|
if type(args[0]) is str:
|
|
if kwargs.get("shell"):
|
|
args[0] = ["/bin/sh", "-c", args[0]]
|
|
kwargs["shell"] = False
|
|
else:
|
|
args[0] = [args[0]]
|
|
|
|
args[0] = ["ssh",
|
|
"-oStrictHostKeyChecking no",
|
|
"-oUserKnownHostsFile /dev/null",
|
|
"-oLogLevel QUIET",
|
|
f"-l{self.location.username}",
|
|
self.location.host] + args[0]
|
|
|
|
if self.location.password:
|
|
args[0] = ["sshpass", f"-p{self.location.password}"] + args[0]
|
|
|
|
return args, kwargs
|
|
|
|
def run(self, *args, **kwargs):
|
|
args, kwargs = self._mangle_subprocess_args(args, kwargs)
|
|
return subprocess.run(*args, **kwargs)
|
|
|
|
def runsh(self, script, *args, **kwargs):
|
|
return self.run("/bin/sh", text=True, input=script,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.STDOUT, *args, **kwargs)
|