Add deserialize_op (xml/lxml)

Author: bitterpanda63

aikido_zen/sources/xml_sources/lxml.py

@@ -2,19 +2,24 @@
     extract_data_from_xml_body,
 )
 from aikido_zen.helpers.get_argument import get_argument
+from aikido_zen.helpers.register_call import register_call
 from aikido_zen.sinks import on_import, after, patch_function
 
 
 @after
 def _fromstring(func, instance, args, kwargs, return_value):
     text = get_argument(args, kwargs, 0, "text")
+    register_call("lxml.etree.fromstring", "deserialize_op")
+
     if text:
         extract_data_from_xml_body(user_input=text, root_element=return_value)
 
 
 @after
 def _fromstringlist(func, instance, args, kwargs, return_value):
     strings = get_argument(args, kwargs, 0, "strings")
+    register_call("lxml.etree.fromstringlist", "deserialize_op")
+
     for text in strings:
         extract_data_from_xml_body(user_input=text, root_element=return_value)
 

aikido_zen/sources/xml_sources/xml.py

@@ -2,18 +2,23 @@
     extract_data_from_xml_body,
 )
 from aikido_zen.helpers.get_argument import get_argument
+from aikido_zen.helpers.register_call import register_call
 from aikido_zen.sinks import on_import, patch_function, after
 
 
 @after
 def _fromstring(func, instance, args, kwargs, return_value):
     text = get_argument(args, kwargs, 0, "text")
+    register_call("xml.etree.ElementTree.fromstring", "deserialize_op")
+
     extract_data_from_xml_body(user_input=text, root_element=return_value)
 
 
 @after
 def _fromstringlist(func, instance, args, kwargs, return_value):
     strings = get_argument(args, kwargs, 0, "sequence")
+    register_call("xml.etree.ElementTree.fromstringlist", "deserialize_op")
+
     for text in strings:
         extract_data_from_xml_body(user_input=text, root_element=return_value)
 

aikido_zen/storage/statistics/operations.py

@@ -1,4 +1,11 @@
-SUPPORTED_KINDS = ["sql_op", "nosql_op", "outgoing_http_op", "fs_op", "exec_op"]
+SUPPORTED_KINDS = [
+    "sql_op",
+    "nosql_op",
+    "outgoing_http_op",
+    "fs_op",
+    "exec_op",
+    "deserialize_op",
+]
 
 
 class Operations(dict):