diff options
Diffstat (limited to 'src/virtual/ogOperations.py')
| -rw-r--r-- | src/virtual/ogOperations.py | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/src/virtual/ogOperations.py b/src/virtual/ogOperations.py index 0819a76..3cfe83a 100644 --- a/src/virtual/ogOperations.py +++ b/src/virtual/ogOperations.py @@ -23,6 +23,7 @@ import math import sys import enum import time +from src.log import OgError class OgVM: DEFAULT_CPU = 'host' @@ -97,7 +98,7 @@ class OgQMP: self.sock.connect((self.ip, self.port)) except socket.error as err: if err.errno == errno.ECONNREFUSED: - raise RuntimeError('Cannot connect to QEMU') + raise OgError('Cannot connect to QEMU') elif err.errno == errno.EINPROGRESS: pass @@ -114,11 +115,11 @@ class OgQMP: pass if 'QMP' not in out: - raise RuntimeError('Cannot handshake QEMU') + raise OgError('Cannot handshake QEMU') out = self.talk(str({"execute": "qmp_capabilities"})) if 'return' not in out: - raise RuntimeError('Cannot handshake QEMU') + raise OgError('Cannot handshake QEMU') def disconnect(self): try: @@ -143,9 +144,9 @@ class OgQMP: try: self.sock.send(bytes(data, 'utf-8')) except Exception as e: - raise RuntimeError('Cannot talk to QEMU') from e + raise OgError('Cannot talk to QEMU') from e else: - raise RuntimeError('Timeout when talking to QEMU') + raise OgError('Timeout when talking to QEMU') return self.recv(timeout=timeout) @@ -158,9 +159,9 @@ class OgQMP: out = self.sock.recv(4096).decode('utf-8') out = json.loads(out) except socket.error as err: - raise RuntimeError('Cannot talk to QEMU') from err + raise OgError('Cannot talk to QEMU') from err else: - raise RuntimeError('Timeout when talking to QEMU') + raise OgError('Timeout when talking to QEMU') return out class OgVirtualOperations: |