z-hypervisor/zhypervisor/machine.py

import logging

from zhypervisor.clients.qmachine import QMachine


class MachineSpec(object):
    """
    Represents a machine we may control
    """
    def __init__(self, master, machine_id, spec):
        """
        Initialize options and properties of the machine. More importantly, initialize the self.machine object which
        should be a subclass of zhypervisor.util.Machine.
        """
        logging.info("Initting machine %s", machine_id)
        self.master = master
        self.machine_id = machine_id

        self.options = spec["options"]
        self.properties = spec["properties"]

        # TODO replace if/else with better system
        if self.options["type"] == "q":
            self.machine = QMachine(self)
        else:
            raise Exception("Unknown machine type: {}".format(self.options["type"]))

    def start(self):
        """
        Start this machine (pass-through)
        """
        self.machine.block_respawns = False
        self.machine.start_machine()

    def stop(self):
        """
        Stop this machine
        """
        self.machine.block_respawns = True
        self.machine.stop_machine()

    def save(self):
        """
        Write the machine's config to disk
        """
        self.master.add_machine(self.machine_id, {"options": self.options, "properties": self.properties},
                                write=True)

    def serialize(self):
        """
        Return a serializable form of this machine's specs
        """
        return {"options": self.options,
                "properties": self.properties}