Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2026-10-05 20:00:04 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

补充代码

Browse Source
This commit is contained in:
DDS-DS
2022-11-05 02:17:04 +08:00
parent 5321c8d9c0
commit 91ca7b40b8
2 changed files with 43 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
plugins/bilibili_sub/model.py
+16 -7
View File
@@ -107,7 +107,7 @@ class BilibiliSub(db.Model):
return False
@classmethod
async def delete_bilibili_sub(cls, sub_id: int, sub_user: str) -> bool:
async def delete_bilibili_sub(cls, sub_id: int, sub_user: str,sub_type: Optional[str] = None) -> bool:
"""
说明:
删除订阅
@@ -117,13 +117,22 @@ class BilibiliSub(db.Model):
"""
try:
async with db.transaction():
query = (
await cls.query.where(
(cls.sub_id == sub_id) & (cls.sub_users.contains(sub_user))
if sub_type:
query = (
await cls.query.where(
(cls.sub_id == sub_id) & (cls.sub_users.contains(sub_user) & (cls.sub_type == sub_type))
)
.with_for_update()
.gino.first()
)
else:
query = (
await cls.query.where(
(cls.sub_id == sub_id) & (cls.sub_users.contains(sub_user))
)
.with_for_update()
.gino.first()
)
.with_for_update()
.gino.first()
)
if not query:
return False
await query.update(