adding new form

2025-07-29 09:39:41 +01:00
parent 2366d0c2a8
commit bccd344aa7
9310 changed files with 1330402 additions and 1 deletions
@@ -0,0 +1,77 @@
+__author__ = "Oleg Butovich"
+__copyright__ = "(c) Oleg Butovich 2013-2017"
+__license__ = "MIT"
+
+# spell-checker:ignore putfo
+
+import logging
+import os
+
+from proxmoxer.backends.command_base import (
+    CommandBaseBackend,
+    CommandBaseSession,
+    shell_join,
+)
+
+logger = logging.getLogger(__name__)
+logger.setLevel(level=logging.WARNING)
+
+try:
+    import paramiko
+except ImportError:
+    import sys
+
+    logger.error("Chosen backend requires 'paramiko' module\n")
+    sys.exit(1)
+
+
+class SshParamikoSession(CommandBaseSession):
+    def __init__(self, host, user, password=None, private_key_file=None, port=22, **kwargs):
+        super().__init__(**kwargs)
+        self.host = host
+        self.user = user
+        self.password = password
+        self.private_key_file = private_key_file
+        self.port = port
+
+        self.ssh_client = self._connect()
+
+    def _connect(self):
+        ssh_client = paramiko.SSHClient()
+        ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+
+        if self.private_key_file:
+            key_filename = os.path.expanduser(self.private_key_file)
+        else:
+            key_filename = None
+
+        ssh_client.connect(
+            self.host,
+            username=self.user,
+            allow_agent=(not self.password),
+            look_for_keys=True,
+            key_filename=key_filename,
+            password=self.password,
+            timeout=self.timeout,
+            port=self.port,
+        )
+
+        return ssh_client
+
+    def _exec(self, cmd):
+        session = self.ssh_client.get_transport().open_session()
+        session.exec_command(shell_join(cmd))
+        stdout = session.makefile("rb", -1).read().decode()
+        stderr = session.makefile_stderr("rb", -1).read().decode()
+        return stdout, stderr
+
+    def upload_file_obj(self, file_obj, remote_path):
+        sftp = self.ssh_client.open_sftp()
+        sftp.putfo(file_obj, remote_path)
+        sftp.close()
+
+
+class Backend(CommandBaseBackend):
+    def __init__(self, *args, **kwargs):
+        self.session = SshParamikoSession(*args, **kwargs)
+        self.target = self.session.host