blivechat/utils/request.py

25 lines
685 B
Python
Raw Normal View History

2022-02-15 00:18:46 +08:00
# -*- coding: utf-8 -*-
2023-07-29 01:22:21 +08:00
from typing import *
2022-02-15 00:18:46 +08:00
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'
}
2023-07-29 01:22:21 +08:00
http_session: Optional[aiohttp.ClientSession] = None
2023-07-29 12:48:57 +08:00
def init():
2023-09-06 21:34:04 +08:00
global http_session
http_session = aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(total=10))
2023-07-29 12:48:57 +08:00
2023-09-06 21:34:04 +08:00
async def shut_down():
if http_session is not None:
await http_session.close()