@@ -15,23 +15,22 @@
Sequence,
)
+from .message import Message
+
class Model:
"""
- Abstract data model, representing some QMP object of some kind.
-
- :param raw: The raw object to be validated.
- :raise KeyError: If any required fields are absent.
- :raise TypeError: If any required fields have the wrong type.
+ Abstract data model, representing a QMP object of some kind.
"""
- def __init__(self, raw: Mapping[str, Any]):
- self._raw = raw
+ @property
+ def _raw(self) -> Mapping[str, Any]:
+ raise NotImplementedError
def _check_key(self, key: str) -> None:
if key not in self._raw:
raise KeyError(f"'{self._name}' object requires '{key}' member")
- def _check_value(self, key: str, type_: type, typestr: str) -> None:
+ def _check_type(self, key: str, type_: type, typestr: str) -> None:
assert key in self._raw
if not isinstance(self._raw[key], type_):
raise TypeError(
@@ -40,7 +39,7 @@ def _check_value(self, key: str, type_: type, typestr: str) -> None:
def _check_member(self, key: str, type_: type, typestr: str) -> None:
self._check_key(key)
- self._check_value(key, type_, typestr)
+ self._check_type(key, type_, typestr)
@property
def _name(self) -> str:
@@ -50,7 +49,37 @@ def __repr__(self) -> str:
return f"{self._name}({self._raw!r})"
-class Greeting(Model):
+class MessageModel(Message, Model):
+ """
+ Abstract data model, representing a QMP Message of some sort.
+
+ Adapter class that glues together `Model` and `Message`.
+ """
+ def __init__(self, raw: Mapping[str, object]):
+ super().__init__(raw)
+
+ @property
+ def _raw(self) -> Mapping[str, Any]:
+ return self._object
+
+ def __setitem__(self, key: str, value: object) -> None:
+ # This is not good OO, but this turns off mutability here.
+ raise RuntimeError("Cannot mutate MessageModel")
+
+
+class StandaloneModel(Model):
+ """
+ Abstract data model, representing a (non-Message) QMP object of some sort.
+ """
+ def __init__(self, raw: Mapping[str, object]):
+ self._raw_mapping = raw
+
+ @property
+ def _raw(self) -> Mapping[str, Any]:
+ return self._raw_mapping
+
+
+class Greeting(MessageModel):
"""
Defined in qmp-spec.txt, section 2.2, "Server Greeting".
@@ -58,8 +87,9 @@ class Greeting(Model):
:raise KeyError: If any required fields are absent.
:raise TypeError: If any required fields have the wrong type.
"""
- def __init__(self, raw: Mapping[str, Any]):
+ def __init__(self, raw: Mapping[str, object]):
super().__init__(raw)
+
#: 'QMP' member
self.QMP: QMPGreeting # pylint: disable=invalid-name
@@ -67,7 +97,7 @@ def __init__(self, raw: Mapping[str, Any]):
self.QMP = QMPGreeting(self._raw['QMP'])
-class QMPGreeting(Model):
+class QMPGreeting(StandaloneModel):
"""
Defined in qmp-spec.txt, section 2.2, "Server Greeting".
@@ -75,8 +105,9 @@ class QMPGreeting(Model):
:raise KeyError: If any required fields are absent.
:raise TypeError: If any required fields have the wrong type.
"""
- def __init__(self, raw: Mapping[str, Any]):
+ def __init__(self, raw: Mapping[str, object]):
super().__init__(raw)
+
#: 'version' member
self.version: Mapping[str, object]
#: 'capabilities' member
@@ -89,7 +120,7 @@ def __init__(self, raw: Mapping[str, Any]):
self.capabilities = self._raw['capabilities']
-class ErrorResponse(Model):
+class ErrorResponse(MessageModel):
"""
Defined in qmp-spec.txt, section 2.4.2, "error".
@@ -97,8 +128,9 @@ class ErrorResponse(Model):
:raise KeyError: If any required fields are absent.
:raise TypeError: If any required fields have the wrong type.
"""
- def __init__(self, raw: Mapping[str, Any]):
+ def __init__(self, raw: Mapping[str, object]):
super().__init__(raw)
+
#: 'error' member
self.error: ErrorInfo
#: 'id' member
@@ -111,7 +143,7 @@ def __init__(self, raw: Mapping[str, Any]):
self.id = raw['id']
-class ErrorInfo(Model):
+class ErrorInfo(StandaloneModel):
"""
Defined in qmp-spec.txt, section 2.4.2, "error".
@@ -119,8 +151,9 @@ class ErrorInfo(Model):
:raise KeyError: If any required fields are absent.
:raise TypeError: If any required fields have the wrong type.
"""
- def __init__(self, raw: Mapping[str, Any]):
+ def __init__(self, raw: Mapping[str, object]):
super().__init__(raw)
+
#: 'class' member, with an underscore to avoid conflicts in Python.
self.class_: str
#: 'desc' member
This allows 'Greeting' to be subclass of 'Message'. We need the adapter classes to avoid some typing problems that occur if we try to put too much into the 'Model' class itself; the exact details of why are left as an exercise to the reader. Why bother? This makes 'Greeting' ⊆ 'Message', which is taxonomically true; but the real motivation is to be able to inherit and abuse all of the Mapping dunders so that we can call dict(greeting) or bytes(greeting), for example. Signed-off-by: John Snow <jsnow@redhat.com> --- python/qemu/aqmp/models.py | 67 ++++++++++++++++++++++++++++---------- 1 file changed, 50 insertions(+), 17 deletions(-)