Merge pull request #58 from Nubuki-all/yt_thumbnail_fix

Fixes thumbnail generation for YouTube links

neonize/aioze/client.py

@@ -174,6 +174,7 @@
 from .events import Event
 from ..utils.log import log
 from concurrent.futures import ThreadPoolExecutor
+from linkpreview import link_preview as fallback_link_preview
 
 loop = get_event_loop()
 
@@ -471,6 +472,8 @@ async def _generate_link_preview(self, text: str) -> ExtendedTextMessage | None:
         valid_links = list(filter(validate_link, links))
         if valid_links:
             preview = await link_preview(valid_links[0])
+            if not preview:
+                preview = fallback_link_preview(valid_links[0])
             preview_type = (
                 ExtendedTextMessage.PreviewType.VIDEO
                 if re.match(youtube_url_pattern, valid_links[0])

neonize/aioze/preview/compose.py

@@ -1,5 +1,5 @@
 from typing import Literal, Optional, TypeVar, overload
-from httpx import URL
+from httpx import URL, HTTPStatusError
 from linkpreview import Link, LinkPreview
 from linkpreview.exceptions import InvalidMimeTypeError
 from .grabber import LinkGrabber
@@ -19,6 +19,7 @@ async def link_preview(
             content, url = await grabber.get_content(url)
         except InvalidMimeTypeError:
             content = ""
-
+        except HTTPStatusError:
+            return
     link = Link(url, content)
     return LinkPreview(link, parser=parser)