GithubBackup/gemini-balance
1
0
Fork 0
mirror of https://github.com/snailyp/gemini-balance.git synced 2026-09-05 07:26:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 增强应用日志记录并优化错误处理

Browse Source
This commit is contained in:
yinpeng
2025-02-11 21:32:21 +08:00
parent f994c5d66d
commit 17f1355099
1 changed files with 32 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/main.py
+31 -15
View File
@@ -25,8 +25,13 @@ key_manager = None
@app.on_event("startup") @app.on_event("startup")
async def startup_event(): async def startup_event():
global key_manager global key_manager
key_manager = await get_key_manager_instance(settings.API_KEYS) logger.info("Application starting up...")
try:
key_manager = await get_key_manager_instance(settings.API_KEYS)
logger.info("KeyManager initialized successfully")
except Exception as e:
logger.error(f"Failed to initialize KeyManager: {str(e)}")
raise
# 添加中间件来处理未经身份验证的请求 # 添加中间件来处理未经身份验证的请求
@app.middleware("http") @app.middleware("http")
@@ -41,7 +46,9 @@ async def auth_middleware(request: Request, call_next):
not request.url.path.startswith("/hf")): not request.url.path.startswith("/hf")):
auth_token = request.cookies.get("auth_token") auth_token = request.cookies.get("auth_token")
if not auth_token or not verify_auth_token(auth_token): if not auth_token or not verify_auth_token(auth_token):
logger.warning(f"Unauthorized access attempt to {request.url.path}")
return RedirectResponse(url="/") return RedirectResponse(url="/")
logger.debug("Request authenticated successfully")
response = await call_next(request) response = await call_next(request)
return response return response
@@ -76,31 +83,40 @@ async def authenticate(request: Request):
form = await request.form() form = await request.form()
auth_token = form.get("auth_token") auth_token = form.get("auth_token")
if not auth_token: if not auth_token:
logger.warning("Authentication attempt with empty token")
return RedirectResponse(url="/", status_code=302) return RedirectResponse(url="/", status_code=302)
if verify_auth_token(auth_token): if verify_auth_token(auth_token):
logger.info("Successful authentication")
response = RedirectResponse(url="/keys", status_code=302) response = RedirectResponse(url="/keys", status_code=302)
response.set_cookie(key="auth_token", value=auth_token, httponly=True, max_age=3600) response.set_cookie(key="auth_token", value=auth_token, httponly=True, max_age=3600)
return response return response
logger.warning("Failed authentication attempt with invalid token")
return RedirectResponse(url="/", status_code=302) return RedirectResponse(url="/", status_code=302)
except Exception as e: except Exception as e:
logger.error(f"Authentication error: {str(e)}") logger.error(f"Authentication error: {str(e)}")
return RedirectResponse(url="/", status_code=302) return RedirectResponse(url="/", status_code=302)
@app.get("/keys", response_class=HTMLResponse) @app.get("/keys", response_class=HTMLResponse)
async def keys_page(request: Request): async def keys_page(request: Request):
auth_token = request.cookies.get("auth_token") try:
if not auth_token or not verify_auth_token(auth_token): auth_token = request.cookies.get("auth_token")
return RedirectResponse(url="/", status_code=302) if not auth_token or not verify_auth_token(auth_token):
keys_status = await key_manager.get_keys_by_status() logger.warning("Unauthorized access attempt to keys page")
total = len(keys_status["valid_keys"]) + len(keys_status["invalid_keys"]) return RedirectResponse(url="/", status_code=302)
return templates.TemplateResponse("keys_status.html", {
"request": request, keys_status = await key_manager.get_keys_by_status()
"valid_keys": keys_status["valid_keys"], total = len(keys_status["valid_keys"]) + len(keys_status["invalid_keys"])
"invalid_keys": keys_status["invalid_keys"], logger.info(f"Keys status retrieved successfully. Total keys: {total}")
"total": total return templates.TemplateResponse("keys_status.html", {
}) "request": request,
"valid_keys": keys_status["valid_keys"],
"invalid_keys": keys_status["invalid_keys"],
"total": total
})
except Exception as e:
logger.error(f"Error retrieving keys status: {str(e)}")
raise
@app.get("/health") @app.get("/health")
@@ -110,5 +126,5 @@ async def health_check(request: Request):
if __name__ == "__main__": if __name__ == "__main__":
logger.info("Starting application server...")
uvicorn.run(app, host="0.0.0.0", port=8001) uvicorn.run(app, host="0.0.0.0", port=8001)