import libvirt

from .exceptions import VMError, VMNotFound


class VirtualMachineBase:

    def __init__(self, session: 'LibvirtSession', name: str):
        self.domname = name
        self.session = session.session  # virConnect object
        self.config = session.config  # ConfigLoader object
        self.domain = self._get_domain(name)

    def _get_domain(self, name: str) -> libvirt.virDomain:
        """Get virDomain object by name to manipulate with domain."""
        try:
            domain = self.session.lookupByName(name)
            if domain is not None:
                return domain
            raise VMNotFound(name)
        except libvirt.libvirtError as err:
            raise VMError(err) from err