From ba131c6096009ce4025470ea913be0efbb17ec41 Mon Sep 17 00:00:00 2001
From: Changhua <lichanghua0821@gmail.com>
Date: Sat, 6 May 2023 22:20:11 +0800
Subject: [PATCH] Impl get tables

---
 http/wcfhttp/core.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/http/wcfhttp/core.py b/http/wcfhttp/core.py
index 1ca552f..3e1178e 100644
--- a/http/wcfhttp/core.py
+++ b/http/wcfhttp/core.py
@@ -45,6 +45,7 @@ class Http(FastAPI):
         self.add_api_route("/msg-types", self.get_msg_types, methods=["GET"], summary="获取消息类型")
         self.add_api_route("/contacts", self.get_contacts, methods=["GET"], summary="获取完整通讯录")
         self.add_api_route("/dbs", self.get_dbs, methods=["GET"], summary="获取所有数据库")
+        self.add_api_route("/{db}/tables", self.get_tables, methods=["GET"], summary="获取 db 中所有表")
 
     def _set_cb(self, cb):
         def callback(msg: WxMsg):
@@ -149,3 +150,10 @@ class Http(FastAPI):
         if ret:
             return {"status": 0, "message": "成功", "data": {"dbs": ret}}
         return {"status": -1, "message": "失败"}
+
+    def get_tables(self, db: str) -> dict:
+        """获取 db 中所有表"""
+        ret = self.wcf.get_tables(db)
+        if ret:
+            return {"status": 0, "message": "成功", "data": {"tables": ret}}
+        return {"status": -1, "message": "失败"}