From 6475ff762c257d92c78967d6621bc7fddc5f5b6f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bj=C3=B6rn=20Ricks?= Date: Wed, 6 Mar 2024 15:35:23 +0100 Subject: [PATCH] Add: Allow to convert an XmlCommandElement into bytes This will allow to transfer an XmlCommandElement and therefore a XmlCommand over a socket. --- gvm/xml.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/gvm/xml.py b/gvm/xml.py index e2b9ae2e3..9d2f65da1 100644 --- a/gvm/xml.py +++ b/gvm/xml.py @@ -70,11 +70,17 @@ def append_xml_str(self, xml_text: str) -> None: self._element.append(node) def to_string(self) -> str: - return xmltostring(self._element).decode("utf-8") + return self.to_bytes().decode("utf-8") + + def to_bytes(self) -> bytes: + return xmltostring(self._element) def __str__(self) -> str: return self.to_string() + def __bytes__(self) -> bytes: + return self.to_bytes() + class XmlCommand(XmlCommandElement): def __init__(self, name: str) -> None: