vm-tool 1.0.4__tar.gz → 1.0.6__tar.gz

{vm_tool-1.0.4/vm_tool.egg-info → vm_tool-1.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: vm_tool
-Version: 1.0.4
+Version: 1.0.6
 Summary: A Comprehensive Tool for Setting Up Virtual Machines Using Ansible.
 License: MIT
 Description-Content-Type: text/markdown

{vm_tool-1.0.4 → vm_tool-1.0.6}/setup.py

@@ -7,7 +7,7 @@ with open(os.path.join(os.path.dirname(__file__), 'Readme.md'), encoding='utf-8'
 
 setup(
     name='vm_tool',
-    version='1.0.4',  # This will be updated by bump2version
+    version='1.0.6',  # This will be updated by bump2version
     packages=find_packages(),
     description='A Comprehensive Tool for Setting Up Virtual Machines Using Ansible.',
     long_description=long_description,

vm_tool-1.0.6/vm_tool/ssh.py

@@ -0,0 +1,117 @@
+import os
+import paramiko
+
+class SSHSetup:
+    """
+    A class to set up SSH configuration and keys for a VM.
+
+    Attributes:
+        hostname (str): The hostname of the VM.
+        username (str): The username for SSH login.
+        password (str): The password for SSH login.
+        email (str): The email for SSH key generation.
+        private_key_path (str): The path to the private SSH key.
+        client (paramiko.SSHClient): The SSH client.
+    """
+
+    def __init__(self, hostname, username, password, email):
+        """
+        The constructor for SSHSetup class.
+
+        Parameters:
+            hostname (str): The hostname of the VM.
+            username (str): The username for SSH login.
+            password (str): The password for SSH login.
+            email (str): The email for SSH key generation.
+        """
+        self.hostname = hostname
+        self.username = username
+        self.password = password
+        self.email = email
+        self.private_key_path = os.path.expanduser('~/.ssh/id_rsa')
+        self.client = paramiko.SSHClient()
+        self.client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+
+    def generate_ssh_key(self, email):
+        """
+        Generates an SSH key pair.
+
+        Parameters:
+            email (str): The email for SSH key generation.
+        """
+        print(f"Generating SSH key for email: {email}")
+        os.system(f'ssh-keygen -t rsa -b 4096 -C "{email}"')
+
+    def read_or_generate_public_key(self, email):
+        """
+        Reads the public SSH key or generates a new one if it doesn't exist.
+
+        Parameters:
+            email (str): The email for SSH key generation.
+
+        Returns:
+            str: The public SSH key.
+        """
+        public_key_path = f'{self.private_key_path}.pub'
+        if not os.path.exists(public_key_path):
+            self.generate_ssh_key(email)
+        with open(public_key_path, 'r') as file:
+            print(f"Reading public key from {public_key_path}")
+            return file.read()
+
+    def configure_vm(self, vm_ip, vm_password, public_key):
+        """
+        Configures the VM by adding the public SSH key to the authorized keys.
+
+        Parameters:
+            vm_ip (str): The IP address of the VM.
+            vm_password (str): The password for the VM.
+            public_key (str): The public SSH key.
+        """
+        print(f"Configuring VM at {vm_ip} with provided public key.")
+        self.client.connect(vm_ip, username=self.username, password=vm_password)
+        self.client.exec_command(f'echo "{public_key}" >> ~/.ssh/authorized_keys')
+        self.client.close()
+
+    def update_ssh_config(self):
+        """
+        Updates the local SSH config file with the VM details.
+        """
+        config = f"""
+Host {self.hostname}
+  HostName {self.hostname}
+  User {self.username}
+  StrictHostKeyChecking no
+  IdentityFile {self.private_key_path}
+  ForwardAgent yes
+"""
+        print(f"Updating SSH config for host {self.hostname}.")
+        with open(f'{os.path.expanduser("~")}/.ssh/config', 'a') as file:
+            file.write(config)
+
+    def establish_connection(self):
+        """
+        Establishes an SSH connection to the VM.
+        """
+        print(f"Establishing SSH connection to {self.hostname}.")
+        self.client.connect(self.hostname, username=self.username, key_filename=self.private_key_path)
+
+    def close_connection(self):
+        """
+        Closes the SSH connection.
+        """
+        if self.client:
+            print("Closing SSH connection.")
+            self.client.close()
+
+    def setup(self):
+        """
+        Sets up the SSH configuration and keys for the VM.
+        """
+        print("Starting SSH setup.")
+        public_key = self.read_or_generate_public_key(self.email)
+        self.configure_vm(self.hostname, self.password, public_key)
+        self.update_ssh_config()
+        self.establish_connection()
+        self.close_connection()
+        print("SSH setup completed.")

{vm_tool-1.0.4 → vm_tool-1.0.6/vm_tool.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: vm_tool
-Version: 1.0.4
+Version: 1.0.6
 Summary: A Comprehensive Tool for Setting Up Virtual Machines Using Ansible.
 License: MIT
 Description-Content-Type: text/markdown

vm_tool-1.0.4/vm_tool/ssh.py

@@ -1,53 +0,0 @@
-import os
-import paramiko
-
-class SSHSetup:
-    def __init__(self, hostname, username, password, email):
-        self.hostname = hostname
-        self.username = username
-        self.password = password
-        self.email = email
-        self.private_key_path = os.path.expanduser('~/.ssh/id_rsa')
-        self.client = paramiko.SSHClient()
-        self.client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
-
-    def _generate_ssh_key(self, email):
-        os.system(f'ssh-keygen -t rsa -b 4096 -C "{email}"')
-
-    def _read_or_generate_public_key(self, email):
-        public_key_path = f'{self.private_key_path}.pub'
-        if not os.path.exists(public_key_path):
-            self._generate_ssh_key(email)
-        with open(public_key_path, 'r') as file:
-            return file.read()
-
-    def _configure_vm(self, vm_ip, vm_password, public_key):
-        self.client.connect(vm_ip, username=self.username, password=vm_password)
-        self.client.exec_command(f'echo "{public_key}" >> ~/.ssh/authorized_keys')
-        self.client.close()
-
-    def _update_ssh_config(self):
-        config = f"""
-Host {self.hostname}
-  HostName {self.hostname}
-  User {self.username}
-  StrictHostKeyChecking no
-  IdentityFile {self.private_key_path}
-  ForwardAgent yes
-"""
-        with open(f'{os.path.expanduser("~")}/.ssh/config', 'a') as file:
-            file.write(config)
-
-    def _establish_connection(self):
-        self.client.connect(self.hostname, username=self.username, key_filename=self.private_key_path)
-
-    def _close_connection(self):
-        if self.client:
-            self.client.close()
-
-    def setup(self):
-        public_key = self._read_or_generate_public_key(self.email)
-        self._configure_vm(self.hostname, self.password, public_key)
-        self._update_ssh_config()
-        self._establish_connection()
-        self._close_connection()