Browse Source

no message

libushang 6 months ago
parent
commit
f0669ce99e
1 changed files with 8 additions and 4 deletions
  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 sqlalchemy.exc import IntegrityError
 from fastapi.responses import HTMLResponse, FileResponse, StreamingResponse, RedirectResponse
 from fastapi.responses import HTMLResponse, FileResponse, StreamingResponse, RedirectResponse
 from fastapi.responses import JSONResponse
 from fastapi.responses import JSONResponse
+from starlette.requests import HTTPConnection
 from database import get_db
 from database import get_db
 from sqlalchemy import text, exists, and_, or_, not_
 from sqlalchemy import text, exists, and_, or_, not_
 from sqlalchemy.orm import Session
 from sqlalchemy.orm import Session
@@ -57,12 +58,15 @@ async def get_qrcode(
 @router.get('/qrcode2')
 @router.get('/qrcode2')
 async def get_qrcode2(
 async def get_qrcode2(
     event_id: str,
     event_id: str,
+    user_agent: str = Header(default=''),
     db: Session = Depends(get_db)
     db: Session = Depends(get_db)
 ):    
 ):    
+    print('user_agent:', user_agent)
     redirect_url = "/signPage?event_id={}".format(event_id) # 业务页面
     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)
     return RedirectResponse(detail_url)