added two output

This commit is contained in:
mxr612 2024-08-23 13:21:49 +08:00
parent c421c885d4
commit 2d8b314e32
2 changed files with 18 additions and 14 deletions

31
main.py
View File

@ -11,19 +11,22 @@ app = FastAPI()
def query(query:str): def query(query:str):
res = [] res = []
list = search_paper(query) try:
for i in list: list = search_paper(query)
if mysql_connector.is_loaded(i['paperId']): for i in list:
print(i['paperId']) if mysql_connector.is_loaded(i['paperId']):
else: print(i['paperId'])
print(i['citationStyles']['bibtex']) else:
res.append({ print(i['citationStyles']['bibtex'])
'id':i['paperId'], res.append({
'q':str(i['citationStyles']['bibtex']), 'id':i['paperId'],
'a':str(i['abstract']), 'q':str(i['citationStyles']['bibtex']),
'score':[] 'a':str(i['abstract']),
}) 'score':[]
print('New load: '+str(len(res))+'/'+str(len(list))) })
print('New load: '+str(len(res))+'/'+str(len(list)))
except Exception as e:
print(str(e))
return res return res
@app.get("/fastdoi") @app.get("/fastdoi")
@ -37,7 +40,7 @@ async def get_reference(questions):
@app.get("/rss") @app.get("/rss")
async def miniflux_rss(): async def miniflux_rss():
load_rss() return load_rss()
if __name__ == '__main__': if __name__ == '__main__':
import uvicorn import uvicorn

1
rss.py
View File

@ -48,3 +48,4 @@ def load_rss():
if len(loaded) > 0: if len(loaded) > 0:
update(loaded) update(loaded)
print(f"Total entries loaded: {count}") print(f"Total entries loaded: {count}")
return f"Total entries loaded: {count}"