From 52d124c8792cf3721e2966615131e0dcad1435e9 Mon Sep 17 00:00:00 2001
From: Zakhar Izmaylov <ptdev@kedruss.ru>
Date: Sat, 11 Jan 2025 18:43:14 +0300
Subject: [PATCH] Add timeout

---
 handlers/keys/subscriptions.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/handlers/keys/subscriptions.py b/handlers/keys/subscriptions.py
index 38ae40cc..a8be7c15 100644
--- a/handlers/keys/subscriptions.py
+++ b/handlers/keys/subscriptions.py
@@ -13,7 +13,8 @@ from logger import logger
 async def fetch_url_content(url, tg_id):
     try:
         logger.info(f"Получение URL: {url} для tg_id: {tg_id}")
-        async with aiohttp.ClientSession() as session:
+        timeout = aiohttp.ClientTimeout(total=5)
+        async with aiohttp.ClientSession(timeout=timeout) as session:
             async with session.get(url, ssl=False) as response:
                 if response.status == 200:
                     content = await response.text()
@@ -24,6 +25,9 @@ async def fetch_url_content(url, tg_id):
                         f"Не удалось получить {url} для tg_id: {tg_id}, статус: {response.status}"
                     )
                     return []
+    except asyncio.TimeoutError:
+        logger.error(f"Таймаут при получении {url} для tg_id: {tg_id}")
+        return []
     except Exception as e:
         logger.error(f"Ошибка при получении {url} для tg_id: {tg_id}: {e}")
         return []