mirror of
https://github.com/ClovertaTheTrilobita/SanYeCao-Nonebot.git
synced 2026-04-01 22:04:51 +00:00
feat(src): 添加随机图功能并更新菜单
- 在 menu 和 send_menu 列表中添加 "/随机图"
This commit is contained in:
parent
3880c945c7
commit
bb58ca3854
2 changed files with 34 additions and 9 deletions
|
|
@ -8,11 +8,11 @@ from src.clover_openai import ai_chat
|
||||||
from src.clover_sqlite.models.chat import GroupChatRole
|
from src.clover_sqlite.models.chat import GroupChatRole
|
||||||
from src.clover_sqlite.models.user import UserList
|
from src.clover_sqlite.models.user import UserList
|
||||||
|
|
||||||
menu = ["/重启","/今日运势","/今日塔罗","/图","/日报","/点歌","/摸摸头","/群老婆","/今日老婆", "/开启ai","/关闭ai","/角色列表","/添加人设", "/更新人设", "/删除人设", "/切换人设", "/管理员注册",
|
menu = ["/重启","/今日运势","/今日塔罗","/图","/随机图","/日报","/点歌","/摸摸头","/群老婆","/今日老婆", "/开启ai","/关闭ai","/角色列表","/添加人设", "/更新人设", "/删除人设", "/切换人设", "/管理员注册",
|
||||||
"/待办", "/test","/天气","我喜欢你", "❤", "/待办查询", "/新建待办", "/删除待办" ,"/cf","/B站搜索", "/BV搜索", "/喜报", "/悲报", "/luxun","/鲁迅说",
|
"/待办", "/test","/天气","我喜欢你", "❤", "/待办查询", "/新建待办", "/删除待办" ,"/cf","/B站搜索", "/BV搜索", "/喜报", "/悲报", "/luxun","/鲁迅说",
|
||||||
"/奶龙", "/repo", "/info", "/menu", "/轻小说","/本季新番","/新番观察","/绝对色感"]
|
"/奶龙", "/repo", "/info", "/menu", "/轻小说","/本季新番","/新番观察","/绝对色感"]
|
||||||
|
|
||||||
send_menu = ["/menu","/今日运势","/今日塔罗","/图","/日报","/点歌","/摸摸头","/群老婆","/待办","/天气",
|
send_menu = ["/menu","/今日运势","/今日塔罗","/图","/随机图","/日报","/点歌","/摸摸头","/群老婆","/待办","/天气",
|
||||||
"/待办查询", "/新建待办", "/删除待办" ,"/cf","/B站搜索", "/BV搜索", "/喜报", "/悲报","/鲁迅说",
|
"/待办查询", "/新建待办", "/删除待办" ,"/cf","/B站搜索", "/BV搜索", "/喜报", "/悲报","/鲁迅说",
|
||||||
"/轻小说","/本季新番","/新番观察","/绝对色感"]
|
"/轻小说","/本季新番","/新番观察","/绝对色感"]
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -5,6 +5,7 @@ from nonebot.plugin import on_command
|
||||||
from nonebot.adapters.qq import MessageSegment,MessageEvent
|
from nonebot.adapters.qq import MessageSegment,MessageEvent
|
||||||
from src.clover_image.get_image import get_image_names
|
from src.clover_image.get_image import get_image_names
|
||||||
from src.clover_image.download_image import download_image
|
from src.clover_image.download_image import download_image
|
||||||
|
from src.clover_image.delete_file import delete_file
|
||||||
from src.configs.path_config import temp_path
|
from src.configs.path_config import temp_path
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -15,13 +16,37 @@ async def handle_function():
|
||||||
local_image_path = await get_image_names()
|
local_image_path = await get_image_names()
|
||||||
await image.finish(MessageSegment.file_image(Path(local_image_path)))
|
await image.finish(MessageSegment.file_image(Path(local_image_path)))
|
||||||
|
|
||||||
|
random_keyword_image = on_command("随机图", rule=to_me(), priority=10, block=False)
|
||||||
|
@random_keyword_image.handle()
|
||||||
|
async def handle_function(message: MessageEvent) -> None:
|
||||||
|
value = message.get_plaintext().split(" ")
|
||||||
|
keyword = ""
|
||||||
|
if len(value) == 2:
|
||||||
|
keyword = value[1]
|
||||||
|
if len(value) > 2:
|
||||||
|
await random_keyword_image.finish("格式不对捏,请试一下 /随机图+关键字")
|
||||||
|
|
||||||
image = on_command("/搜图", rule=to_me(), priority=10, block=True)
|
filename = f"{message.get_user_id()}{random.randint(0, 10000)}.jpg"
|
||||||
@image.handle()
|
image_path = temp_path + filename
|
||||||
async def handle_function(message: MessageEvent):
|
|
||||||
|
|
||||||
filename = str(message.get_user_id()) + str(random.randint(0, 10000)) + ".jpg"
|
await download_image(f"https://image.anosu.top/pixiv/direct?keyword={keyword}", image_path)
|
||||||
image_ptah = temp_path + filename
|
|
||||||
|
try:
|
||||||
|
await random_keyword_image.send(MessageSegment.file_image(Path(image_path)))
|
||||||
|
except Exception as e:
|
||||||
|
print(e)
|
||||||
|
await random_keyword_image.finish("图被外星人抢走啦,请重试")
|
||||||
|
|
||||||
|
await delete_file(image_path)
|
||||||
|
|
||||||
|
|
||||||
|
# search_image = on_command("/搜图", rule=to_me(), priority=10, block=True)
|
||||||
|
# @image.handle()
|
||||||
|
# async def handle_function(message: MessageEvent):
|
||||||
|
#
|
||||||
|
# filename = str(message.get_user_id()) + str(random.randint(0, 10000)) + ".jpg"
|
||||||
|
# image_ptah = temp_path + filename
|
||||||
|
#
|
||||||
|
# await download_image(message.attachments[0].url, image_ptah)
|
||||||
|
# await image.finish(MessageSegment.file_image(Path(image_ptah)))
|
||||||
|
|
||||||
await download_image(message.attachments[0].url, image_ptah)
|
|
||||||
await image.finish(MessageSegment.file_image(Path(image_ptah)))
|
|
||||||
Loading…
Reference in a new issue