fix: 修复了使用api查询主题时的bug

This commit is contained in:
MarkusJoe
2022-10-02 17:30:50 +08:00
parent 4a21ca730f
commit 80da8a5485
1 file changed
+6 -5
+6 -5
View File
@@ -6,9 +6,11 @@
# @File Name: view.py # @File Name: view.py
import os
from src.api import api from src.api import api
from src.db import Database from src.db import Database
from fastapi.responses import FileResponse from fastapi.responses import FileResponse
from fastapi.responses import JSONResponse
@api.get('/query/{name}') @api.get('/query/{name}')
@@ -19,6 +21,8 @@ async def query(name: str):
@api.get('/query-all/') @api.get('/query-all/')
async def query_all(limit: int = 30): async def query_all(limit: int = 30):
if os.getenv('c_not_full'):
return JSONResponse({'code': -200, 'msg': 'Redis Database do not support this action'})
data = Database().query_all()[:limit] data = Database().query_all()[:limit]
result = {} result = {}
for i in data: for i in data:
@@ -33,8 +37,5 @@ async def export():
@api.get('/query-theme/{name}') @api.get('/query-theme/{name}')
async def query_theme(name: str): async def query_theme(name: str):
themes = [] data = Database().query_image(name)
for i in range(10): return data
data = Database().query_image(name + '/' + str(i))[0]
themes.append(data)
return themes