Cleanup imports part 2

[nitsi.git] / src / nitsi / virtual_environ.py

#!/usr/bin/python3

import configparser
import libvirt
import logging
import os

from . import machine
from . import network

logger = logging.getLogger("nitsi.virtual_environ")

# Should return all vms and networks in a list
# and should provide the path to the necessary xml files
class virtual_environ():
    def __init__(self, path):
        self.log = logger.getChild(os.path.basename(os.path.abspath(path)))
        try:
            self.path = os.path.abspath(path)
        except BaseException as e:
            self.log.error("Could not get absolute path")

        self.log.debug(self.path)

        self.settings_file = "{}/settings".format(self.path)
        if not os.path.isfile(self.settings_file):
            self.log.error("No such file: {}".format(self.settings_file))

        self.log.debug(self.settings_file)
        self.config = configparser.ConfigParser()
        self.config.read(self.settings_file)
        self.name = self.config["DEFAULT"]["name"]
        self.machines_string = self.config["DEFAULT"]["machines"]
        self.networks_string = self.config["DEFAULT"]["networks"]

        self.machines = []
        for machine in self.machines_string.split(","):
            self.machines.append(machine.strip())

        self.networks = []
        for network in self.networks_string.split(","):
            self.networks.append(network.strip())

        self.log.debug(self.machines)
        self.log.debug(self.networks)

        # Number of characters of the longest machine name
        self._longest_machine_name = 0

        self.uri = self.config["DEFAULT"]["uri"]

        try:
            self.con = libvirt.open(self.uri)
        except BaseException as error:
            self.log.error("Could not connect to: {}".format(self.uri))

        self.log.debug("Connected to: {}".format(self.uri))

    def get_networks(self):
        networks = {}
        for _network in self.networks:
            self.log.debug(_network)
            networks.setdefault(_network, network.network(self.con, os.path.normpath(self.path + "/" + self.config[_network]["xml_file"])))
        return networks

    def get_machines(self):
        machines = {}
        for _machine in self.machines:
            self.log.debug(_machine)
            machines.setdefault(_machine, machine.machine(
                self.con,
                os.path.normpath(self.path + "/" + self.config[_machine]["xml_file"]),
                os.path.normpath(self.path + "/" + self.config[_machine]["snapshot_xml_file"]),
                self.config[_machine]["image"],
                self.config[_machine]["root_uid"],
                self.config[_machine]["username"],
                self.config[_machine]["password"]))

        return machines

    @property
    def machine_names(self):
        return self.machines

    @property
    def network_names(self):
        return self.networks

    @property
    def longest_machine_name(self):
        if self._longest_machine_name:
            return self._longest_machine_name
        else:
            for _machine in self.machines:
                if len(_machine) > self._longest_machine_name:
                    self._longest_machine_name = len(_machine)

            return self._longest_machine_name