Update preview link model

koval01 · Dec 13, 2024 · 57c3c8f · 57c3c8f
1 parent 11b35bb
commit 57c3c8f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/app/telegram/models/post.py b/app/telegram/models/post.py
@@ -144,16 +144,16 @@ class PreviewLink(BaseModel):
     Attributes:
         title (Optional[str]): The title of the linked content, if available
         url (HttpUrl): This URL of preview page
-        site_name (str): The name of the website or platform the link is from
+        site_name (Optional[str]): The name of the website or platform the link is from
         description (Optional[ParsedAndRaw]): The description of the linked content,
             containing both parsed text and raw HTML formats
-        thumb (HttpUrl): The URL of the preview thumbnail image
+        thumb (Optional[HttpUrl]): The URL of the preview thumbnail image
     """
     title: Optional[str]
     url: HttpUrl
-    site_name: str
+    site_name: Optional[str]
     description: Optional[ParsedAndRaw]
-    thumb: HttpUrl
+    thumb: Optional[HttpUrl]
 
 
 class ContentPost(BaseModel):

diff --git a/app/telegram/parser/types/post.py b/app/telegram/parser/types/post.py
@@ -84,11 +84,12 @@ def preview_link(buble: LexborNode) -> Union[dict, None]:
                 "html": Utils.get_text_html(description)
             }
 
+        site_name = preview.css_first(".link_preview_site_name")
         title = preview.css_first(".link_preview_title")
         thumb = preview.css_first(".tgme_widget_message_link_preview > .link_preview_right_image")
 
         return {
-            "site_name": preview.css_first(".link_preview_site_name").text(strip=True),
+            "site_name": site_name.text(strip=True) if site_name else None,
             "url": preview.attributes.get("href"),
             "title": title.text(strip=True) if title else None,
             "description": description,