Fix some typing issues

gvm/transforms.py

@@ -8,7 +8,7 @@
 from lxml import etree
 
 from .errors import GvmError, GvmResponseError, GvmServerError
-from .xml import create_parser
+from .xml import Element, create_parser
 
 
 class EtreeTransform:
@@ -19,18 +19,18 @@ class EtreeTransform:
     def __init__(self):
         self._parser = create_parser()
 
-    def _convert_response(self, response: str) -> etree.Element:
+    def _convert_response(self, response: str) -> Element:
         return etree.XML(response, parser=self._parser)
 
-    def __call__(self, response: str) -> etree.Element:
+    def __call__(self, response: str) -> Element:
         return self._convert_response(response)
 
 
-def check_command_status(root: etree.Element):
+def check_command_status(root: Element):
     status = root.get("status")
 
     if status is None:
-        raise GvmServerError("No status in response.", root)
+        raise GvmServerError("No status in response.", str(root))
 
     if status[0] == "4":
         raise GvmResponseError(status=status, message=root.get("status_text"))
@@ -46,7 +46,7 @@ class CheckCommandTransform(EtreeTransform):
     response was an error response
     """
 
-    def __call__(self, response: str) -> str:
+    def __call__(self, response: str) -> str:  # type: ignore[override]
         root = self._convert_response(response)
 
         check_command_status(root)
@@ -60,7 +60,7 @@ class EtreeCheckCommandTransform(EtreeTransform):
     response was an error response
     """
 
-    def __call__(self, response: str) -> etree.Element:
+    def __call__(self, response: str) -> Element:
         root = self._convert_response(response)
 
         check_command_status(root)

gvm/utils.py

@@ -30,8 +30,8 @@ class TypesDict(dict):
     """For dot.notation access to dictionary attributes"""
 
     __getattr__ = dict.get
-    __setattr__ = dict.__setitem__
-    __delattr__ = dict.__delitem__
+    __setattr__ = dict.__setitem__  # type: ignore[assignment]
+    __delattr__ = dict.__delitem__  # type: ignore[assignment]
 
 
 def deprecation(message: str):