chg: code cleanup.

weblyzard · Jan 8, 2024 · 57120ce · 57120ce
1 parent ffef3f9
commit 57120ce
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/src/inscriptis/annotation/output/html.py b/src/inscriptis/annotation/output/html.py
@@ -68,9 +68,7 @@ def _get_label_colors(labels: List[str]) -> Dict[str, str]:
             A mapping between the available labels and the corresponding color
             from the COLOR_SCHEMA.
         """
-        return {label: color
-                for label, color in zip({a[2] for a in sorted(labels)},
-                                        cycle(COLOR_SCHEMA))}
+        return dict(zip({a[2] for a in sorted(labels)}, cycle(COLOR_SCHEMA)))
 
     def _get_css(self, labels: List[str]) -> str:
         """Compute the CSS to be included into the HTML output.

diff --git a/src/inscriptis/html_engine.py b/src/inscriptis/html_engine.py
@@ -137,7 +137,7 @@ def handle_starttag(self, tag, attrs):
                 tag, DEFAULT_HTML_ELEMENT).__copy__().set_tag(tag)))
         self.tags.append(cur)
 
-        handler = self.start_tag_handler_dict.get(tag, None)
+        handler = self.start_tag_handler_dict.get(tag)
         if handler:
             handler(attrs)
 
@@ -150,7 +150,7 @@ def handle_endtag(self, tag):
         Args:
           tag: the HTML end tag to process.
         """
-        handler = self.end_tag_handler_dict.get(tag, None)
+        handler = self.end_tag_handler_dict.get(tag)
         if handler:
             handler()