From e7ccd70e564fd55b017acca56492703409cd0236 Mon Sep 17 00:00:00 2001
From: YoursFunny <admin@yoursfunny.top>
Date: Mon, 24 Jun 2024 02:17:26 +0800
Subject: [PATCH] fix deal with invalid url

---
 main.py           | 2 ++
 utils/telegram.py | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/main.py b/main.py
index d9b03a0..5dc02ab 100644
--- a/main.py
+++ b/main.py
@@ -41,6 +41,8 @@ async def inline_query(update: Update, context: ContextTypes.DEFAULT_TYPE) -> No
         return
     logger.info(f"Query: {query}")
     async with Telegram(query) as tweet:
+        if not tweet:
+            return
         await update.inline_query.answer(tweet.inline_query_result())
 
 
diff --git a/utils/telegram.py b/utils/telegram.py
index 583f013..d2b4b37 100644
--- a/utils/telegram.py
+++ b/utils/telegram.py
@@ -17,6 +17,8 @@ class Telegram:
         elif PIXIV_REFRESH_TOKEN and pixiv_url.match(self._url):
             async with TelegramPixiv(self._url) as pixiv:
                 return pixiv
+        else:
+            return None
 
     async def __aexit__(self, exc_type, exc_val, exc_tb):
         pass