summaryrefslogtreecommitdiffstats
path: root/src/virtual/ogOperations.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/virtual/ogOperations.py')
-rw-r--r--src/virtual/ogOperations.py15
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: