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

:recycles: DBConfig中 替换platform_name为site_name

Browse Source
This commit is contained in:
suyiiyii 2024-09-08 18:38:38 +08:00
parent 275bc0cb53
commit 4db7e7b911
1 changed files with 124 additions and 124 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

248
nonebot_bison/config/db_config.py
View File

@ -259,130 +259,130 @@ class DBConfig:
) )
return res return res
# async def get_cookie(self, platform_name: str = None, target: T_Target = None) -> Sequence[Cookie]: async def get_cookie(self, site_name: str = None, target: T_Target = None) -> Sequence[Cookie]:
# """根据平台名和订阅名获取 cookie，不会返回匿名cookie""" """根据 site_name 和 target 获取 cookie，不会返回匿名cookie"""
# async with create_session() as sess: async with create_session() as sess:
# query = select(Cookie).distinct().where(Cookie.is_universal == False) # noqa: E712 query = select(Cookie).distinct().where(Cookie.is_universal == False) # noqa: E712
# if platform_name: if site_name:
# query = query.where(Cookie.platform_name == platform_name) query = query.where(Cookie.site_name == site_name)
# query = query.outerjoin(CookieTarget).options(selectinload(Cookie.targets)) query = query.outerjoin(CookieTarget).options(selectinload(Cookie.targets))
# res = (await sess.scalars(query)).all() res = (await sess.scalars(query)).all()
# if target: if target:
# query = select(CookieTarget.cookie_id).join(Target).where(Target.target == target) query = select(CookieTarget.cookie_id).join(Target).where(Target.target == target)
# ids = set((await sess.scalars(query)).all()) ids = set((await sess.scalars(query)).all())
# res = [cookie for cookie in res if cookie.id in ids] res = [cookie for cookie in res if cookie.id in ids]
# return res return res
#
# async def get_unviersal_cookie(self, platform_name: str = None) -> Sequence[Cookie]: async def get_unviersal_cookie(self, site_name: str = None) -> Sequence[Cookie]:
# async with create_session() as sess: async with create_session() as sess:
# query = select(Cookie).distinct().where(Cookie.is_universal == True) # noqa: E712 query = select(Cookie).distinct().where(Cookie.is_universal == True) # noqa: E712
# if platform_name: if site_name:
# query = query.where(Cookie.platform_name == platform_name) query = query.where(Cookie.site_name == site_name)
# res = (await sess.scalars(query)).all() res = (await sess.scalars(query)).all()
# return res return res
#
# async def add_cookie_with_content(self, platform_name: str, content: str) -> int: async def add_cookie_with_content(self, site_name: str, content: str) -> int:
# async with create_session() as sess: async with create_session() as sess:
# cookie = Cookie(platform_name=platform_name, content=content) cookie = Cookie(site_name=site_name, content=content)
# sess.add(cookie) sess.add(cookie)
# await sess.commit() await sess.commit()
# await sess.refresh(cookie) await sess.refresh(cookie)
# return cookie.id return cookie.id
#
# async def add_cookie(self, cookie: Cookie) -> int: async def add_cookie(self, cookie: Cookie) -> int:
# async with create_session() as sess: async with create_session() as sess:
# sess.add(cookie) sess.add(cookie)
# await sess.commit() await sess.commit()
# await sess.refresh(cookie) await sess.refresh(cookie)
# return cookie.id return cookie.id
#
# async def update_cookie(self, cookie: Cookie): async def update_cookie(self, cookie: Cookie):
# async with create_session() as sess: async with create_session() as sess:
# cookie_in_db: Cookie | None = await sess.scalar(select(Cookie).where(Cookie.id == cookie.id)) cookie_in_db: Cookie | None = await sess.scalar(select(Cookie).where(Cookie.id == cookie.id))
# if not cookie_in_db: if not cookie_in_db:
# return return
# cookie_in_db.content = cookie.content cookie_in_db.content = cookie.content
# cookie_in_db.last_usage = cookie.last_usage cookie_in_db.last_usage = cookie.last_usage
# cookie_in_db.status = cookie.status cookie_in_db.status = cookie.status
# cookie_in_db.tags = cookie.tags cookie_in_db.tags = cookie.tags
# await sess.commit() await sess.commit()
#
# async def delete_cookie_by_id(self, cookie_id: int): async def delete_cookie_by_id(self, cookie_id: int):
# async with create_session() as sess: async with create_session() as sess:
# cookie = await sess.scalar( cookie = await sess.scalar(
# select(Cookie) select(Cookie)
# .where(Cookie.id == cookie_id) .where(Cookie.id == cookie_id)
# .outerjoin(CookieTarget) .outerjoin(CookieTarget)
# .options(selectinload(Cookie.targets)) .options(selectinload(Cookie.targets))
# ) )
# if len(cookie.targets) > 0: if len(cookie.targets) > 0:
# raise Exception(f"cookie {cookie.id} in use") raise Exception(f"cookie {cookie.id} in use")
# await sess.execute(delete(Cookie).where(Cookie.id == cookie_id)) await sess.execute(delete(Cookie).where(Cookie.id == cookie_id))
# await sess.commit() await sess.commit()
#
# async def get_cookie_by_target(self, target: T_Target, platform_name: str) -> Sequence[Cookie]: async def get_cookie_by_target(self, target: T_Target, site_name: str) -> Sequence[Cookie]:
# async with create_session() as sess: async with create_session() as sess:
# query = ( query = (
# select(Cookie) select(Cookie)
# .join(CookieTarget) .join(CookieTarget)
# .join(Target) .join(Target)
# .where(Target.platform_name == platform_name, Target.target == target) .where(Target.site_name == site_name, Target.target == target)
# ) )
# return (await sess.scalars(query)).all() return (await sess.scalars(query)).all()
#
# async def get_universal_cookie(self, platform_name: str) -> Sequence[Cookie]: async def get_universal_cookie(self, site_name: str) -> Sequence[Cookie]:
# async with create_session() as sess: async with create_session() as sess:
# query = ( query = (
# select(Cookie) select(Cookie)
# .where(Cookie.platform_name == platform_name) .where(Cookie.site_name == site_name)
# .where(Cookie.is_universal == True) # noqa: E712 .where(Cookie.is_universal == True) # noqa: E712
# ) )
# return (await sess.scalars(query)).all() return (await sess.scalars(query)).all()
#
# async def add_cookie_target(self, target: T_Target, platform_name: str, cookie_id: int): async def add_cookie_target(self, target: T_Target, site_name: str, cookie_id: int):
# async with create_session() as sess: async with create_session() as sess:
# target_obj = await sess.scalar( target_obj = await sess.scalar(
# select(Target).where(Target.platform_name == platform_name, Target.target == target) select(Target).where(Target.site_name == site_name, Target.target == target)
# ) )
# # check if relation exists # check if relation exists
# cookie_target = await sess.scalar( cookie_target = await sess.scalar(
# select(CookieTarget).where(CookieTarget.target == target_obj, CookieTarget.cookie_id == cookie_id) select(CookieTarget).where(CookieTarget.target == target_obj, CookieTarget.cookie_id == cookie_id)
# ) )
# if cookie_target: if cookie_target:
# raise DuplicateCookieTargetException() raise DuplicateCookieTargetException()
# cookie_obj = await sess.scalar(select(Cookie).where(Cookie.id == cookie_id)) cookie_obj = await sess.scalar(select(Cookie).where(Cookie.id == cookie_id))
# cookie_target = CookieTarget(target=target_obj, cookie=cookie_obj) cookie_target = CookieTarget(target=target_obj, cookie=cookie_obj)
# sess.add(cookie_target) sess.add(cookie_target)
# await sess.commit() await sess.commit()
#
# async def delete_cookie_target(self, target: T_Target, platform_name: str, cookie_id: int): async def delete_cookie_target(self, target: T_Target, site_name: str, cookie_id: int):
# async with create_session() as sess: async with create_session() as sess:
# target_obj = await sess.scalar( target_obj = await sess.scalar(
# select(Target).where(Target.platform_name == platform_name, Target.target == target) select(Target).where(Target.site_name == site_name, Target.target == target)
# ) )
# cookie_obj = await sess.scalar(select(Cookie).where(Cookie.id == cookie_id)) cookie_obj = await sess.scalar(select(Cookie).where(Cookie.id == cookie_id))
# await sess.execute( await sess.execute(
# delete(CookieTarget).where(CookieTarget.target == target_obj, CookieTarget.cookie == cookie_obj) delete(CookieTarget).where(CookieTarget.target == target_obj, CookieTarget.cookie == cookie_obj)
# ) )
# await sess.commit() await sess.commit()
#
# async def delete_cookie_target_by_id(self, cookie_target_id: int): async def delete_cookie_target_by_id(self, cookie_target_id: int):
# async with create_session() as sess: async with create_session() as sess:
# await sess.execute(delete(CookieTarget).where(CookieTarget.id == cookie_target_id)) await sess.execute(delete(CookieTarget).where(CookieTarget.id == cookie_target_id))
# await sess.commit() await sess.commit()
#
# async def get_cookie_target(self) -> list[CookieTarget]: async def get_cookie_target(self) -> list[CookieTarget]:
# async with create_session() as sess: async with create_session() as sess:
# query = ( query = (
# select(CookieTarget) select(CookieTarget)
# .outerjoin(Target) .outerjoin(Target)
# .options(selectinload(CookieTarget.target)) .options(selectinload(CookieTarget.target))
# .outerjoin(Cookie) .outerjoin(Cookie)
# .options(selectinload(CookieTarget.cookie)) .options(selectinload(CookieTarget.cookie))
# ) )
# res = list((await sess.scalars(query)).all()) res = list((await sess.scalars(query)).all())
# res.sort(key=lambda x: (x.target.platform_name, x.cookie_id, x.target_id)) res.sort(key=lambda x: (x.target.platform_name, x.cookie_id, x.target_id))
# return res return res
config = DBConfig() config = DBConfig()