blob: e6bb8bcb972e1396470dfd70630de4c5d1620673 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
from ATRI.database import BilibiliSubscription
class DB:
async def __aenter__(self):
return self
async def __aexit__(self, exc_type, exc_val, exc_tb):
pass
async def add_sub(self, uid: int, group_id: int):
await BilibiliSubscription.create(uid=uid, group_id=group_id)
async def update_sub(self, uid: int, group_id: int, update_map: dict):
await BilibiliSubscription.filter(uid=uid, group_id=group_id).update(
**update_map
)
async def del_sub(self, query_map: dict):
await BilibiliSubscription.filter(**query_map).delete()
async def get_sub_list(self, query_map: dict) -> list:
return await BilibiliSubscription.filter(**query_map)
async def get_all_subs(self) -> list:
return await BilibiliSubscription.all()
|
