Soulter
115 lines
4.0 KiB
Python
115 lines
4.0 KiB
Python
import requests
|
|
import asyncio
|
|
import websockets
|
|
from websockets import WebSocketClientProtocol
|
|
import threading
|
|
import json
|
|
|
|
class UnofficialQQBotSDK:
|
|
|
|
GET_APP_ACCESS_TOKEN_URL = "https://bots.qq.com/app/getAppAccessToken"
|
|
OPENAPI_BASE_URL = "https://api.sgroup.qq.com"
|
|
|
|
def __init__(self, appid: str, client_secret: str) -> None:
|
|
self.appid = appid
|
|
self.client_secret = client_secret
|
|
self.get_access_token()
|
|
self.get_wss_endpoint()
|
|
# self.ws_connect()
|
|
# print(self.ws_recv())
|
|
asyncio.get_event_loop().run_until_complete(self.ws_client())
|
|
|
|
|
|
def get_access_token(self) -> None:
|
|
# self.access_token = requests.post(self.GET_APP_ACCESS_TOKEN_URL, data={
|
|
# "appid": self.appid,
|
|
# "clientSecret": self.client_secret
|
|
# }).json()['access_token']
|
|
res = requests.post(self.GET_APP_ACCESS_TOKEN_URL, json={
|
|
"appId": self.appid,
|
|
"clientSecret": self.client_secret
|
|
}, headers={
|
|
"Content-Type": "application/json"
|
|
})
|
|
print(res.text)
|
|
self.access_token = 'QQBot ' + res.json()['access_token']
|
|
print("access_token: " + self.access_token)
|
|
|
|
def auth_header(self) -> str:
|
|
return {
|
|
'Authorization': self.access_token,
|
|
'X-Union-Appid': self.appid,
|
|
}
|
|
|
|
def get_wss_endpoint(self):
|
|
# self.wss_endpoint = requests.get(self.OPENAPI_BASE_URL + "/gateway", headers=self.auth_header()).json()['url']
|
|
res = requests.get(self.OPENAPI_BASE_URL + "/gateway", headers=self.auth_header())
|
|
print(res.text)
|
|
self.wss_endpoint = res.json()['url']
|
|
print("wss_endpoint: " + self.wss_endpoint)
|
|
|
|
async def behav_heartbeat(self, ws: WebSocketClientProtocol, t: int):
|
|
while True:
|
|
await asyncio.sleep(t - 1)
|
|
try:
|
|
print("heartbeat., s: " + str(self.s))
|
|
await ws.send(json.dumps({
|
|
"op": 1,
|
|
"d": self.s
|
|
}))
|
|
except:
|
|
print("heartbeat error.")
|
|
|
|
async def handle_msg(self, ws: WebSocketClientProtocol, msg: dict):
|
|
if msg['op'] == 10:
|
|
# hello
|
|
# 创建心跳任务
|
|
print("hello.")
|
|
asyncio.get_event_loop().create_task(self.behav_heartbeat(ws, msg['d']['heartbeat_interval'] / 1000))
|
|
# 鉴权，获得session
|
|
await ws.send(json.dumps({
|
|
"op": 2,
|
|
"d": {
|
|
"token": self.access_token,
|
|
"intents": 33554432,
|
|
"shard": [0, 1],
|
|
"properties": {
|
|
"$os": "linux",
|
|
"$browser": "my_library",
|
|
"$device": "my_library"
|
|
}
|
|
}
|
|
}))
|
|
|
|
if msg['op'] == 0:
|
|
# ready
|
|
print("ready.")
|
|
data = msg['d']
|
|
print(data)
|
|
|
|
if 'group_openid' in data:
|
|
group_openid = data['group_openid']
|
|
message_str = data['content'].strip()
|
|
message_id = data['id']
|
|
# 发送消息
|
|
requests.post(self.OPENAPI_BASE_URL + f"/v2/groups/{group_openid}/messages", headers=self.auth_header(), json={
|
|
"content": message_str,
|
|
"message_type": 0,
|
|
"msg_id": message_id
|
|
})
|
|
|
|
async def ws_client(self):
|
|
self.s = 0
|
|
async with websockets.connect(self.wss_endpoint) as websocket:
|
|
print("ws connected.")
|
|
while True:
|
|
msg = await websocket.recv()
|
|
msg = json.loads(msg)
|
|
if 's' in msg:
|
|
self.s = msg['s']
|
|
print("recv: " + str(msg))
|
|
await self.handle_msg(websocket, msg)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
UnofficialQQBotSDK("102041113", "b9VebKp7B2g7KK7f") |