suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2025-06-06 20:06:12 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🧑‍💻 fix type hinting

Browse Source
This commit is contained in:
felinae98 2023-04-01 19:42:22 +08:00
parent 5fbd94460b
commit f16ca6db8b
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nonebot_bison/config/subs_io/subs_io.py
View File

@ -1,6 +1,5 @@
from collections import defaultdict from collections import defaultdict
from functools import partial from typing import Any, Callable, cast
from typing import Any, Callable, TypeVar
from nonebot.log import logger from nonebot.log import logger
from nonebot_plugin_datastore.db import create_session from nonebot_plugin_datastore.db import create_session
@ -19,10 +18,8 @@ from .nbesf_model import (
) )
from .utils import subs_receipt_gen_ver_1 from .utils import subs_receipt_gen_ver_1
T = TypeVar("T", bound=Select)
async def subscribes_export(selector: Callable[[Select], Select]) -> SubGroup:
async def subscribes_export(selector: Callable[[T], T]) -> SubGroup:
""" """
将Bison订阅导出为 Nonebot Bison Exchangable Subscribes File 标准格式的 SubGroup 类型数据 将Bison订阅导出为 Nonebot Bison Exchangable Subscribes File 标准格式的 SubGroup 类型数据
@ -32,10 +29,12 @@ async def subscribes_export(selector: Callable[[T], T]) -> SubGroup:
async with create_session() as sess: async with create_session() as sess:
sub_stmt = select(Subscribe).join(User) sub_stmt = select(Subscribe).join(User)
sub_stmt = selector(sub_stmt).options(selectinload(Subscribe.target)) sub_stmt = selector(sub_stmt).options(selectinload(Subscribe.target))
sub_stmt = cast(Select[tuple[Subscribe]], sub_stmt)
sub_data = await sess.scalars(sub_stmt) sub_data = await sess.scalars(sub_stmt)
user_stmt = select(User).join(Subscribe) user_stmt = select(User).join(Subscribe)
user_stmt = selector(user_stmt).distinct() user_stmt = selector(user_stmt).distinct()
user_stmt = cast(Select[tuple[User]], user_stmt)
user_data = await sess.scalars(user_stmt) user_data = await sess.scalars(user_stmt)
groups: list[SubPack] = [] groups: list[SubPack] = []