浏览代码

no message

libushang 6 月之前
父节点
当前提交
f0669ce99e
共有 1 个文件被更改,包括 8 次插入4 次删除
  1. 8 4
      routers/api/eventManagement/checkin.py

+ 8 - 4
routers/api/eventManagement/checkin.py

@@ -1,7 +1,8 @@
-from fastapi import APIRouter, Request, Depends, HTTPException, Query
+from fastapi import APIRouter, Request, Depends, HTTPException, Query, Header
 from sqlalchemy.exc import IntegrityError
 from fastapi.responses import HTMLResponse, FileResponse, StreamingResponse, RedirectResponse
 from fastapi.responses import JSONResponse
+from starlette.requests import HTTPConnection
 from database import get_db
 from sqlalchemy import text, exists, and_, or_, not_
 from sqlalchemy.orm import Session
@@ -57,12 +58,15 @@ async def get_qrcode(
 @router.get('/qrcode2')
 async def get_qrcode2(
     event_id: str,
+    user_agent: str = Header(default=''),
     db: Session = Depends(get_db)
 ):    
+    print('user_agent:', user_agent)
     redirect_url = "/signPage?event_id={}".format(event_id) # 业务页面
-    logger.info("redirect_url: {}", redirect_url)
-    # qrcode_str = f"http://19.155.220.209/api/event_management/event?event_id={event_id}"
-    detail_url = YzyApi.format_redirect_url(redirect_url)
+    if "wxworklocal" in user_agent:
+        logger.info("redirect_url: {}", redirect_url)
+        # qrcode_str = f"http://19.155.220.209/api/event_management/event?event_id={event_id}"
+        detail_url = YzyApi.format_redirect_url(redirect_url)
 
     return RedirectResponse(detail_url)