From 34741109da2b45f24fbabc8d00c0194ab775411d Mon Sep 17 00:00:00 2001 From: Kyomotoi <kyomotoiowo@gmail.com> Date: Tue, 14 Jun 2022 21:25:04 +0800 Subject: =?UTF-8?q?=E2=9C=A8=20=E6=96=B0=E5=A2=9E=E6=8F=92=E4=BB=B6:=20Twi?= =?UTF-8?q?tter=E5=8A=A8=E6=80=81=E8=AE=A2=E9=98=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/plugins/twitter/db.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 ATRI/plugins/twitter/db.py (limited to 'ATRI/plugins/twitter/db.py') diff --git a/ATRI/plugins/twitter/db.py b/ATRI/plugins/twitter/db.py new file mode 100644 index 0000000..52bedd5 --- /dev/null +++ b/ATRI/plugins/twitter/db.py @@ -0,0 +1,24 @@ +from ATRI.database import TwitterSubscription + + +class DB: + async def __aenter__(self): + return self + + async def __aexit__(self, exc_type, exc_val, exc_tb): + pass + + 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 del_sub(self, query_map: dict): + await TwitterSubscription.filter(**query_map).delete() + + async def get_sub_list(self, query_map: dict) -> list: + return await TwitterSubscription.filter(**query_map) + + async def get_all_subs(self) -> list: + return await TwitterSubscription.all() -- cgit v1.2.3