support vercel kv db and remove mysql

src/api/view.py

@@ -68,9 +68,9 @@ async def query_theme(name: str):
     }
     for i in data:
         response["data"].append({
-            "index": i[0],
-            "image": i[1],
-            "width": i[2],
-            "height": i[3]
+            "index": i["id"],
+            "image": i["base64"],
+            "width": i["width"],
+            "height": i["height"]
         })
     return JSONResponse(response)

src/config.py

@@ -14,6 +14,7 @@ class Config:
     mysql -> mysql://user:pwd@host:port/db
     deta -> deta
     """
-    database = os.getenv("COUNTER_DB") or "sqlite3"  # 数据库类型
-    if os.getenv("PJ_DETA"):
-        database = "deta"  # 自动设置为deta
+    database = os.getenv("COUNTER_DB") or "vercelkv"  # 数据库类型
+    api_key = os.getenv("VERCEL_KV_KEY") or "<KEY>"
+    vercel_kv_url = os.getenv("VERCEL_KV_URL") or "http://127.0.0.1:8000"
+    themes = {}

src/db/__init__.py

@@ -8,6 +8,9 @@
 
 import os
 from urllib.request import urlretrieve
+
+import requests
+
 from src.config import Config
 
 database = Config.database  # operator
@@ -26,8 +29,22 @@ if database == "sqlite3":
 
 if database == "sqlite3":
     from src.db.db import SQLite as Database
-elif database == "deta":
-    from src.db.db import DetaBase as Database
 else:
-    from src.db.db import MySQL as Database
+    from src.db.db import VercelKV as Database
+
+    base_url = "https://static.rtast.cn/static/counter"
+    themes_file = [
+        "lisu.json",
+        "moebooru.json",
+        "asoul.json",
+        "blacked.json",
+        "hgelbooru.json",
+        "lewd.json",
+        "rule34.json",
+        "hmoebooru.json"
+    ]
+    for i in themes_file:
+        result = requests.get(base_url + f"/{i}").json()
+        print(f"{i}, Done!")
+        Config.themes[i.replace(".json", "")] = result
 __all__ = [Database]

src/db/db.py

@@ -4,17 +4,12 @@
 # @Development Tool: PyCharm
 # @Create Time: 2022/9/11
 # @File Name: db.py
+import requests
 
-
-import os
 from src.config import Config
 
 if Config.database in ["sqlite3", "sqlite"]:
     import sqlite3 as operator
-elif Config.database == "deta":
-    from deta import Deta
-else:
-    import pymysql as operator
 
 
 class SQL:
@@ -63,102 +58,35 @@ class SQLite(SQL):
         self.cursor = self.conn.cursor()
 
 
-class MySQL(SQL):
-    def __init__(self):
-        super().__init__()
-        _CONFIG = Config.database.replace("mysql://", "").split("@")
-        user = _CONFIG[0].split(":")[0]
-        pwd = _CONFIG[0].split(":")[1]
-        host = _CONFIG[1].split(":")[0]
-        port = int(_CONFIG[1].split(":")[1].split("/")[0])
-        db = _CONFIG[1].split("/")[1]
-
-        self.conn = operator.connect(user=user,
-                                     passwd=pwd,
-                                     host=host,
-                                     port=port,
-                                     database=db)
-        self.cursor = self.conn.cursor()
-
-
-class DetaBase:
-    def __init__(self):
-        self.__deta = Deta(os.getenv("PJ_DETA"))
-        self.__data = self.__deta.AsyncBase("times")
-        self.__image = self.__deta.AsyncBase("images")
-
-    async def __get(self, _id: str) -> tuple:
-        result = await self.__data.get(_id)
-        if result is None:
-            await self.__put_data(_id)
-            return tuple([_id, 0])
-        await self.update(_id, result["times"])
-        await self.__image.close()
-        await self.__data.close()
-        return tuple([_id, result["times"]])
-
-    async def __put_data(self, _id: str) -> bool:
-        await self.__data.put({"times": 0}, _id)
-        await self.__image.close()
-        await self.__data.close()
-        return True
-
-    async def __update_data(self, _id: str, times: int) -> bool:
-        new = {"times": times + 1}
-        await self.__data.update(new, _id)
-        await self.__image.close()
-        await self.__data.close()
-        return True
-
-    async def __insert_data(self, _id: str) -> bool:
-        await self.__put_data(_id)
-        await self.__image.close()
-        await self.__data.close()
-        return True
-
-    async def __get_images(self, theme: str) -> list:
-        response = []
-        result = await self.__image.get(theme)
-        for i in result:
-            if i != "key":
-                response.append(tuple([
-                    i,
-                    result[i]["base64"],
-                    result[i]["width"],
-                    result[i]["height"]
-                ]))
-        await self.__image.close()
-        await self.__data.close()
-        return response
-
+class VercelKV:
     async def query(self, _id: str) -> tuple:
-        result = await self.__get(_id)
-        return result
+        times = requests.get(
+            f"{Config.vercel_kv_url}/get/{_id}",
+            headers={"Authorization": f"Bearer {Config.api_key}"}
+        ).json()["result"]
+        if times is None:
+            await self.insert(_id)
+            return tuple([_id, 0])
+        await self.update(_id, int(times))
+        return tuple([_id, times])
 
     async def insert(self, _id: str) -> bool:
-        await self.__insert_data(_id)
+        requests.get(
+            f"{Config.vercel_kv_url}/set/{_id}/1",
+            headers={"Authorization": f"Bearer {Config.api_key}"}
+        )
         return True
 
     async def update(self, _id: str, times: int) -> bool:
-        await self.__update_data(_id, times)
+        times += 1
+        requests.get(
+            f"{Config.vercel_kv_url}/set/{_id}/{times}",
+            headers={"Authorization": f"Bearer {Config.api_key}"}
+        )
         return True
 
     async def query_all(self) -> list:
-        res = await self.__data.fetch()
-        all_items = res.items
-        while res.last:
-            res = await self.__data.fetch(last=res.last)
-            all_items += res.items
-        await self.__image.close()
-        await self.__data.close()
-        result = []
-        for i in all_items:
-            result.append((
-                i["key"],
-                i["times"]
-            ))
-        return result
+        return list("")
 
     async def query_image(self, theme: str) -> list:
-        result = await self.__get_images(theme)
-        return result
+        return Config.themes[theme]

src/utils/response.py

@@ -28,15 +28,15 @@ async def resp(_id: str, length: int = 7, theme: str = "lewd") -> dict:
     headers = {"cache-control": "max-age=0, no-cache, no-store, must-revalidate",
                "Content-Type": "image/svg+xml; charset=utf-8"}
     data = await Database().query_image(theme)
-    height = data[0][-1]
-    width = data[0][-2]
+    height = data[0]["height"]
+    width = data[0]["width"]
     counter = 0
     for i, n in zip(data, show_number):
         context.append({
-            "position": i[-2] * counter,
-            "width": i[-2],
-            "height": i[-1],
-            "base64": data[int(n)][1]
+            "position": i["width"] * counter,
+            "width": i["width"],
+            "height": i["height"],
+            "base64": data[int(n)]["base64"]
         })
         counter += 1