diff options
Diffstat (limited to 'ATRI/plugins/twitter/db.py')
| -rw-r--r-- | ATRI/plugins/twitter/db.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/ATRI/plugins/twitter/db.py b/ATRI/plugins/twitter/db.py index 52bedd5..69c4611 100644 --- a/ATRI/plugins/twitter/db.py +++ b/ATRI/plugins/twitter/db.py @@ -11,8 +11,10 @@ class DB: async def add_sub(self, tid: int, group_id: int): await TwitterSubscription.create(tid=tid, group_id=group_id) - async def update_sub(self, tid: int, update_map: dict): - await TwitterSubscription.filter(tid=tid).update(**update_map) + async def update_sub(self, tid: int, group_id: int, update_map: dict): + await TwitterSubscription.filter(tid=tid, group_id=group_id).update( + **update_map + ) async def del_sub(self, query_map: dict): await TwitterSubscription.filter(**query_map).delete() |