mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
from nonebot.adapters.onebot.v11 import MessageSegment
|
|
from utils.image_utils import BuildImage
|
|
from utils.message_builder import image
|
|
from configs.path_config import IMAGE_PATH
|
|
from typing import Tuple, Union
|
|
from utils.http_utils import AsyncHttpx
|
|
import datetime
|
|
|
|
|
|
async def get_wbtop(url: str) -> Tuple[Union[dict, str], int]:
|
|
"""
|
|
:param url: 请求链接
|
|
"""
|
|
n = 0
|
|
while True:
|
|
try:
|
|
data = []
|
|
get_response = (await AsyncHttpx.get(url, timeout=20))
|
|
if get_response.status_code == 200:
|
|
data_json = get_response.json()['data']['realtime']
|
|
for data_item in data_json:
|
|
# 如果是广告，则不添加
|
|
if 'is_ad' in data_item:
|
|
continue
|
|
dic = {
|
|
'hot_word': data_item['note'],
|
|
'hot_word_num': str(data_item['num']),
|
|
'url': 'https://s.weibo.com/weibo?q=%23' + data_item['word'] + '%23',
|
|
}
|
|
data.append(dic)
|
|
if not data:
|
|
return "没有搜索到...", 997
|
|
return {'data': data, 'time': datetime.datetime.now()}, 200
|
|
else:
|
|
if n > 2:
|
|
return f'获取失败,请十分钟后再试', 999
|
|
else:
|
|
n += 1
|
|
continue
|
|
except TimeoutError:
|
|
return "超时了....", 998
|
|
|
|
|
|
def gen_wbtop_pic(data: dict) -> MessageSegment:
|
|
"""
|
|
生成微博热搜图片
|
|
:param data: 微博热搜数据
|
|
"""
|
|
bk = BuildImage(700, 32 * 50 + 280, 700, 32, color="#797979")
|
|
wbtop_bk = BuildImage(700, 280, background=f"{IMAGE_PATH}/other/webtop.png")
|
|
bk.paste(wbtop_bk)
|
|
text_bk = BuildImage(700, 32 * 50, 700, 32, color="#797979")
|
|
for i, data in enumerate(data):
|
|
title = f"{i + 1}. {data['hot_word']}"
|
|
hot = str(data["hot_word_num"])
|
|
img = BuildImage(700, 30, font_size=20)
|
|
w, h = img.getsize(title)
|
|
img.text((10, int((30 - h) / 2)), title)
|
|
img.text((580, int((30 - h) / 2)), hot)
|
|
text_bk.paste(img)
|
|
bk.paste(text_bk, (0, 280))
|
|
return image(b64=bk.pic2bs4())
|