mirror of
https://github.com/xfgryujk/blivechat.git
synced 2024-12-26 21:00:15 +08:00
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
# -*- coding: utf-8 -*-
|
||
import asyncio
|
||
from typing import *
|
||
|
||
import aiohttp
|
||
|
||
# 不带这堆头部有时候也能成功请求，但是带上后成功的概率更高
|
||
BILIBILI_COMMON_HEADERS = {
|
||
'Origin': 'https://www.bilibili.com',
|
||
'Referer': 'https://www.bilibili.com/',
|
||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko)'
|
||
' Chrome/114.0.0.0 Safari/537.36'
|
||
}
|
||
|
||
http_session: Optional[aiohttp.ClientSession] = None
|
||
|
||
|
||
def init():
|
||
global http_session
|
||
http_session = aiohttp.ClientSession(
|
||
response_class=CustomClientResponse,
|
||
timeout=aiohttp.ClientTimeout(total=10),
|
||
)
|
||
|
||
|
||
async def shut_down():
|
||
if http_session is not None:
|
||
await http_session.close()
|
||
|
||
|
||
class CustomClientResponse(aiohttp.ClientResponse):
|
||
# 因为aiohttp的BUG，当底层连接断开时，_wait_released可能会抛出CancelledError，导致上层协程结束。这里改个错误类型
|
||
async def _wait_released(self):
|
||
try:
|
||
return await super()._wait_released()
|
||
except asyncio.CancelledError as e:
|
||
raise aiohttp.ClientConnectionError('Connection released') from e
|