From 731726371bad962ff11ec69bd57237b2f7d577cc Mon Sep 17 00:00:00 2001
From: MarkusJoe <markushammered@gmail.com>
Date: Sun, 11 Sep 2022 10:30:42 +0800
Subject: [PATCH] update: Update

---
 src/api/view.py       | 8 ++++----
 src/utils/response.py | 8 ++++----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/src/api/view.py b/src/api/view.py
index 2058f70..2b330e1 100644
--- a/src/api/view.py
+++ b/src/api/view.py
@@ -7,19 +7,19 @@
 
 
 from src.api import api
-from src.db import SQLite
+from src.db import Database
 from fastapi.responses import FileResponse
 
 
 @api.get('/query/{name}')
 async def query(name: str):
-    times = SQLite().query(name)[1]
+    times = Database().query(name)[1]
     return {name: times}
 
 
 @api.get('/query-all/')
 async def query_all(limit: int = 30):
-    data = SQLite().query_all()[:limit]
+    data = Database().query_all()[:limit]
     result = {}
     for i in data:
         result[i[0]] = i[1]
@@ -35,6 +35,6 @@ async def export():
 async def query_theme(name: str):
     themes = []
     for i in range(10):
-        data = SQLite().query_image(name + '/' + str(i))[0]
+        data = Database().query_image(name + '/' + str(i))[0]
         themes.append(data)
     return themes
diff --git a/src/utils/response.py b/src/utils/response.py
index c37c765..1d9c1df 100644
--- a/src/utils/response.py
+++ b/src/utils/response.py
@@ -6,7 +6,7 @@
 # @File Name: response.py
 
 
-from src.db import SQLite
+from src.db import Database
 
 
 async def resp(_id: str, length: int = 7, theme: str = 'lewd') -> dict:
@@ -17,16 +17,16 @@ async def resp(_id: str, length: int = 7, theme: str = 'lewd') -> dict:
     :param theme:
     :return:
     """
-    times = SQLite().query(_id)[1]
+    times = Database().query(_id)[1]
     str_number = str(times)  # 将整形转换为字符串
     len_number = len(str_number)  # 再获取字符串长度
     g_length = length * '0'  # 根据输入的位数来自动生成0的数量
     show_number = str(g_length[:-len_number] + str_number)
     context = []
-    w_h = SQLite().query_image(theme + '/0')[0]
+    w_h = Database().query_image(theme + '/0')[0]
     count = 0  # 计数器, 每次遍历一次则加一, 让图片x轴相乘
     for i in show_number:
-        data = SQLite().query_image(theme + '/' + i)[0]
+        data = Database().query_image(theme + '/' + i)[0]
         context.append({'position': data[-2] * count,
                         'width': data[-2],
                         'height': data[-1],