fix: live check_connectivity behaviour

acgnhiki · Jun 21, 2024 · 1206d6e · 1206d6e
1 parent 772eb5e
commit 1206d6e
Showing 1 changed file with 6 additions and 4 deletions.
diff --git 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(