From 1206d6e80fae545681e04c11ecdb88b88074cefd Mon Sep 17 00:00:00 2001
From: imkero <kerorek@outlook.com>
Date: Tue, 23 Apr 2024 01:08:11 +0800
Subject: [PATCH] fix: live check_connectivity behaviour

---
 src/blrec/bili/live.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/blrec/bili/live.py b/src/blrec/bili/live.py
index a143d4df..ad91d32b 100644
--- a/src/blrec/bili/live.py
+++ b/src/blrec/bili/live.py
@@ -175,11 +175,13 @@ def is_living(self) -> bool:
 
     async def check_connectivity(self) -> bool:
         try:
-            await self._session.head('https://live.bilibili.com/', timeout=3)
-        except (aiohttp.ClientConnectionError, asyncio.TimeoutError):
-            return False
-        else:
+            await self._session.head('https://live.bilibili.com/', timeout=3, headers={
+                'User-Agent': self._user_agent,
+            })
             return True
+        except Exception as e:
+            self._logger.warning(f'Check connectivity failed: {repr(e)}')
+            return False
 
     async def update_info(self, raise_exception: bool = False) -> bool:
         return all(