34 lines
920 B
Python
34 lines
920 B
Python
from pathlib import Path
|
|
from contextlib import AbstractContextManager
|
|
|
|
import libvirt
|
|
|
|
from .config import ConfigLoader
|
|
|
|
|
|
class LibvirtSessionError(Exception):
|
|
"""Something went wrong while connecting to libvirt."""
|
|
|
|
|
|
class LibvirtSession(AbstractContextManager):
|
|
def __init__(self, config: Path | None = None):
|
|
self.config = ConfigLoader(config)
|
|
self.session = self._connect(self.config['libvirt']['uri'])
|
|
|
|
def __enter__(self):
|
|
return self
|
|
|
|
def __exit__(self, exception_type, exception_value, exception_traceback):
|
|
self.close()
|
|
|
|
def _connect(self, connection_uri: str):
|
|
try:
|
|
return libvirt.open(connection_uri)
|
|
except libvirt.libvirtError as err:
|
|
raise LibvirtSessionError(
|
|
f'Failed to open connection to the hypervisor: {err}'
|
|
) from err
|
|
|
|
def close(self) -> None:
|
|
self.session.close()
|