fix: Avoid exception on empty response

Lookyloo · Dec 15, 2024 · b325b39 · b325b39
1 parent 0850089
commit b325b39
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/har2tree/har2tree.py b/har2tree/har2tree.py
@@ -640,10 +640,10 @@ def make_tree(self) -> URLNode:
         if self.har.downloaded_filename and self.har.downloaded_file:
             downloaded_file = (self.har.downloaded_filename if self.har.downloaded_filename else '',
                                self.har.downloaded_file if self.har.downloaded_file else None)
-
-        self.rendered_node.add_rendered_features(list(self.all_url_requests.keys()),
-                                                 rendered_html=self.har.html_content if self.har.html_content else None,
-                                                 downloaded_file=downloaded_file)
+        if not self.rendered_node.empty_response:
+            self.rendered_node.add_rendered_features(list(self.all_url_requests.keys()),
+                                                     rendered_html=self.har.html_content if self.har.html_content else None,
+                                                     downloaded_file=downloaded_file)
 
         # Initialize the hostname tree root
         self.hostname_tree.add_url(self.url_tree)

diff --git a/har2tree/nodes.py b/har2tree/nodes.py
@@ -340,6 +340,7 @@ def load_har_entry(self, har_entry: MutableMapping[str, Any], all_requests: list
         if not self.response['content'].get('text') or self.response['content']['text'] == '':
             # If the content of the response is empty, skip.
             self.add_feature('empty_response', True)
+            self.add_feature('mimetype', 'inode/x-empty')
         else:
             self.add_feature('empty_response', False)
             if self.response['content'].get('encoding') == 'base64':