From e6a4b706ede16dd8d338bbb353e5a6ee6f35a28e Mon Sep 17 00:00:00 2001
From: MarkusJoe <markushammered@gmail.com>
Date: Tue, 1 Mar 2022 22:33:08 +0800
Subject: [PATCH] =?UTF-8?q?=E5=B0=8F=E6=9B=B4=E6=96=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/app.py b/app.py
index 87fb8ea..014860f 100644
--- a/app.py
+++ b/app.py
@@ -47,8 +47,8 @@ def main(name) -> Response or str:
     :return:
     """
     args = request.args
-    theme = args.get('theme')
-    length = args.get('length')
+    theme = args.get('theme', type=str)
+    length = args.get('length', type=int)
     if not bool(theme):  # 判断主题是否存在查询参数内如果不存在则使用配置文件内的默认主题
         theme = conf.default_style
     elif theme == 'ls':
@@ -58,8 +58,7 @@ def main(name) -> Response or str:
     if not db().exists_name(name):
         db().insert(name)
     count = db().fetch(name)
-    themes = db().fetching_table(theme)
-    if not len(themes):
+    if not db().exists_table(theme):
         return ErrorProcess(db).theme_error(theme)
     if 7 <= int(length) <= 10:  # 限定自定义长度阈值
         view_number = '0' * (int(length) - len(str(count[1]))) + str(count[1])