插件开发思路#

1
await bot.call_api(
2
            "upload_group_file",
3
            group_id=event.group_id,
4
            file=f"{path}/{jmid}.pdf",
5
            name=f"{jmid}.pdf",
6
)

1
"""
2
Author: adeptmicors279
3
Date: 2025-07-29 18:48:19
4
LastEditTime: 2024-08-08 23:15:31
5
LastEditors: 279
6
Description: jm
7
"""
8
import shutil
9
import jmcomic
10
from jmcomic import *
11
from nonebot.params import ArgPlainText
12
from nonebot.adapters import Message
13
from nonebot.params import CommandArg
14
from nonebot.rule import to_me
15
from nonebot import on_regex, on_command
16
from nonebot.typing import T_State
17
from nonebot.adapters.onebot.v11 import GroupMessageEvent, Bot, Message
18
from nonebot.matcher import Matcher
19
from nonebot.adapters.onebot.v11 import MessageSegment
20
from PIL import Image
21
import os
22
import json
23

24
def images_to_pdf(image_folder, output_path):
25

26
    # 获取文件夹中所有图片文件
27
    image_files = [f for f in os.listdir(image_folder)]
28

29
    # 存储Image对象的列表
30
    image_list = []
31

32
    # 按文件名顺序读取图片
33
    for file in image_files:
34
        file_path = os.path.join(image_folder, file)
35
        try:
36
            # 打开图片并转换为RGB模式
37
            img = Image.open(file_path)
38
            if img.mode != "RGB":
39
                img = img.convert("RGB")
40
            image_list.append(img)
41
        except Exception as e:
42
            print(f"无法处理文件 {file}: {str(e)}")
43

44
    if not image_list:
45
        print("未找到有效图片")
46
        return
47

48
    # 使用第一张图片创建PDF
49
    first_image = image_list[0]
50
    # 将其余图片附加到PDF中
51
    remaining_images = image_list[1:]
52

53
    # 保存为PDF
54
    first_image.save(output_path, save_all=True,
55
                     append_images=remaining_images)
56

57
    # 关闭所有图片
58
    for img in image_list:
59
        img.close()
60

61
    print(f"PDF已成功生成: {output_path}")
62

63
title = None
64
Test = on_regex(pattern=r"^测试$", priority=2, rule=to_me())
65
jm_get = on_command("jm", priority=1, rule=to_me())
66
test_send_file = on_command("test_send", priority=10, rule=to_me())
67
help = on_regex(pattern=r"^帮助$",priority=1)
68
jm_clean_cache = on_command("jmn",priority=3,rule=to_me())
69

70

71
@Test.handle()
72
async def Test_send(bot: Bot, event: GroupMessageEvent, state: T_State):
73
    msg = "Bot启动正常"
74
    await Test.finish(message=Message(msg))
75

76

77
@jm_get.handle()
78
async def jm_get_send(bot: Bot, event: GroupMessageEvent, state: T_State, args: Message = CommandArg()):
79
    if jmid := args.extract_plain_text():
80
        await jm_get.send(
81
            message=Message(f"正在获取JMID为{jmid}的comic\n获取时间较长,优先检查缓存内容")
82
        )
83

84
        path = os.getcwd() # 设定路径
85
        if os.path.exists(str(jmid)+'.pdf'):
86
            await jm_get.send(message=Message("检查到缓存内容"))
87
            await bot.call_api(
88
            "upload_group_file",
89
            group_id=event.group_id,
90
            file=f"{path}/{jmid}.pdf",
91
            name=f"{jmid}.pdf",
92
        )
93
            await jm_get.finish(message=Message("获取完成"))
94
            return 0
95

96

97
        if not os.path.exists(str(jmid)+'.pdf'):
98
            await jm_get.send(message=Message("未检测到缓存内容,联网下载"))
99
            client = JmOption.default().new_jm_client()
100
            # 设置client
101
            page = client.search_site(search_query=f"{jmid}")
102
            album: JmAlbumDetail = page.single_album
103

104
            jmcomic.download_album(int(jmid))
105
            images_to_pdf(image_folder=album.title + r"/",
106
                        output_path=f"./{jmid}.pdf")
107

108
            # 记录缓存内容
109
            with open (r'./src/plugins/jm/name_id.json','r')as cache:
110
                dic = json.load(cache)
111
                dic[str(album.title)] = int(jmid)
112
            with open (r'./src/plugins/jm/name_id.json','w')as new_cache:
113
                json.dump(dic,new_cache)
114

115
            # await jm_get.send(message=Message(str(event.group_id)))
116
            await bot.call_api(
117
                "upload_group_file",
118
                group_id=event.group_id,
119
                file=f"{path}/{jmid}.pdf",
120
                name=f"{jmid}.pdf",
121
            )
122
            await jm_get.finish(message=Message("获取完成"))
123
            return 0
124

125
    else:
126
        msg = "未识别到jm车牌号"
127
        await jm_get.finish(message=Message(msg))
128

129

130
@jm_clean_cache.handle()
131
async def jm_get_new(bot: Bot, event: GroupMessageEvent):
132
    await jm_clean_cache.send(
133
        message=Message(f"该操作会刷新缓存,请稍等...")
134
    )
135
    path = os.getcwd() # 设定路径
136
    with open(r'./src/plugins/jm/name_id.json','r')as cache:
137
        dic = json.load(cache)
138
        name_cache = dic.keys()
139
        jmid_cache = dic.values()
140
        for i in name_cache:
141
            shutil.rmtree(str(path+rf"/{i}"))
142
        for i in jmid_cache:
143
            os.remove(str(i)+".pdf")
144
        dic = {}
145
        json.dump(dic,cache)
146

147
    await jm_clean_cache.finish(message=Message("缓存删除成功"))
148

149

150

151
@test_send_file.handle()
152
async def test_send_file(bot: Bot, event: GroupMessageEvent, state: T_State):
153
    await bot.call_api(
154
        "upload_group_file",
155
        group_id=event.group_id,
156
        file="./1201023.pdf",
157
        name="1201023.pdf"
158
    )
159

160

161
@help.handle()
162
async def help_menu(bot: Bot, event: GroupMessageEvent, state: T_State):
163
    msg = "输入/jm xxxxx 即可获取comic\n所有功能需要@机器人才能完成"
164
    await help.finish(message=Message(msg))