提交

backend/app/api/sample_sms_notification.py

@@ -6,7 +6,7 @@ from pydantic import BaseModel, ConfigDict, Field
 from sqlalchemy.orm import Session
 
 from app.core.database import get_db
-from app.core.menu_guard import require_menu_code
+from app.core.menu_guard import require_any_menu_code
 from app.core.response import error_response, success_response
 from app.crud import sample_sms_notification as svc
 from app.crud.collection_protocol import total_pages
@@ -14,6 +14,11 @@ from app.models.models import SysUser
 
 router = APIRouter(prefix="/web/admin", tags=["样本接收短信通知"])
 
+_SAMPLE_RECEPTION_MENU = require_any_menu_code(
+    "cell-storage:sample-info:list",
+    "cell-storage:sample-reception-info:list",
+)
+
 
 def _get_tenant_no(db: Session, user: dict) -> str:
     user_id = user.get("user_id")
@@ -59,7 +64,7 @@ async def list_notifications(
     receptionTimeStart: Optional[str] = Query(None, alias="receptionTimeStart"),
     receptionTimeEnd: Optional[str] = Query(None, alias="receptionTimeEnd"),
     db: Session = Depends(get_db),
-    user: dict = Depends(require_menu_code("cell-storage:sample-reception-info:list")),
+    user: dict = Depends(_SAMPLE_RECEPTION_MENU),
 ):
     try:
         tenant_no = _get_tenant_no(db, user)
@@ -87,7 +92,7 @@ async def toggle_status(
     notification_id: str,
     body: ToggleStatusBody,
     db: Session = Depends(get_db),
-    user: dict = Depends(require_menu_code("cell-storage:sample-reception-info:list")),
+    user: dict = Depends(_SAMPLE_RECEPTION_MENU),
 ):
     try:
         tenant_no = _get_tenant_no(db, user)
@@ -114,7 +119,7 @@ async def generate_sms(
     notification_id: str,
     sampleReceptionId: Optional[str] = Query(None),
     db: Session = Depends(get_db),
-    user: dict = Depends(require_menu_code("cell-storage:sample-reception-info:list")),
+    user: dict = Depends(_SAMPLE_RECEPTION_MENU),
 ):
     try:
         tenant_no = _get_tenant_no(db, user)