mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-05 19:36:43 +08:00
4e304a43b1
* feat: 实现导出存储的订阅信息的功能 * test: 编写导出功能测试 * test: 使用tmp_path * feat: 实现导入订阅文件功能 * refactor: 将订阅导入导出部分独立出来 * fix: 修复一些拼写错误 test: 完成import的第一个测试 * feat: 将订阅导入导出函数加入nb script test: 添加cli测试 * test: 完善subs import测试 * 🐛 fix nb cli entrypoint name error * fix: 修改错误的entry_point, 关闭yaml导出时对键名的排序 * fix: 使用更简短的命令名 * 🚚 将subs_io迁移到config下 * ♻️ 不再使用抛出异常的方式创建目录 * refactor: 将subscribe_export类转为函数 * refactor: 将subscribe_import类转为函数 * refactor: 根据重写的subs_io重新调整cli * test: 调整重写subs_io后的test * chore: 清理未使用的import内容 * feat(cli): 将--yaml更改为--format * test: 调整测试 * fix(cli): 为import添加不支持格式的报错 * ⚡ improve export performace * feat: subscribes_import函数不再需要传入参数函数，而是指定为add_subscribes fix: nbesf_parser在传入str时将调用parse_raw, 否则调用parse_obj * feat: subscribes_import现在会根据nbesf_data的版本选择合适的导入方式 * fix(test): 调整测试 * feat: nb bison export命令不再将文件导出到data目录，而是当前工作目录 * docs: 增添相关文档 * fix(test): 修复错误的变量名 --------- Co-authored-by: felinae98 <731499577@qq.com>
29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
from functools import partial
|
|
|
|
from nonebot.log import logger
|
|
|
|
from ..db_config import SubscribeDupException, config
|
|
from .nbesf_model import NBESF_VERSION, SubGroup, SubReceipt
|
|
|
|
|
|
async def subs_receipt_gen_ver_1(nbesf_data: SubGroup):
|
|
for item in nbesf_data.groups:
|
|
sub_receipt = partial(SubReceipt, user=item.user.uid, user_type=item.user.type)
|
|
|
|
for sub in item.subs:
|
|
receipt = sub_receipt(
|
|
target=sub.target.target,
|
|
target_name=sub.target.target_name,
|
|
platform_name=sub.target.platform_name,
|
|
cats=sub.categories,
|
|
tags=sub.tags,
|
|
)
|
|
try:
|
|
await config.add_subscribe(**receipt.dict())
|
|
except SubscribeDupException:
|
|
logger.warning(f"！添加订阅条目 {repr(receipt)} 失败: 相同的订阅已存在")
|
|
except Exception as e:
|
|
logger.error(f"！添加订阅条目 {repr(receipt)} 失败: {repr(e)}")
|
|
else:
|
|
logger.success(f"添加订阅条目 {repr(receipt)} 成功！")
|