wrapper.py -> Skip processing None in transform_html

Tishka17 · Sep 26, 2023 · 63a167f · 63a167f
1 parent 4b8d7d0
commit 63a167f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/sulguk/wrapper.py b/src/sulguk/wrapper.py
@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import List
+from typing import List, Optional
 
 from sulguk.render.state import MessageEntity, State
 from .transformer import Transformer
@@ -11,7 +11,11 @@ class RenderResult:
     entities: List[MessageEntity]
 
 
-def transform_html(raw_html: str) -> RenderResult:
+def transform_html(raw_html: Optional[str]) -> RenderResult:
+
+    if raw_html is None:
+        return RenderResult(text="", entities=[])
+
     transformer = Transformer()
     transformer.feed(raw_html)
     state = State()