From 63a167f57662db75d6a2610fe450af21831e30e0 Mon Sep 17 00:00:00 2001 From: bral <54455457+bralbral@users.noreply.github.com> Date: Fri, 22 Sep 2023 15:35:50 +0400 Subject: [PATCH] wrapper.py -> Skip processing None in transform_html --- src/sulguk/wrapper.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/sulguk/wrapper.py b/src/sulguk/wrapper.py index 99f1323..442d362 100644 --- 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()