event.py 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. from fastapi import APIRouter, Request, Depends, HTTPException, Query
  2. from sqlalchemy.exc import IntegrityError
  3. from fastapi.responses import HTMLResponse, FileResponse
  4. from fastapi.responses import JSONResponse
  5. from database import get_db
  6. from sqlalchemy import text, exists, and_, or_, not_
  7. from sqlalchemy.orm import Session
  8. from models import *
  9. import json
  10. import random
  11. from sqlalchemy import create_engine, select
  12. from typing import Optional
  13. from utils.StripTagsHTMLParser import *
  14. from common.db import db_event_management, db_user, db_area, db_emergency_plan
  15. from common.security import valid_access_token
  16. import traceback
  17. from utils import *
  18. from datetime import datetime, timedelta
  19. router = APIRouter()
  20. # 列出预案单位职责
  21. @router.post('/unit_list')
  22. async def unit_list(
  23. request: Request,
  24. db: Session = Depends(get_db),
  25. body = Depends(remove_xss_json),
  26. user_id = Depends(valid_access_token)
  27. ):
  28. plan_id = body['plan_id']
  29. rows = db.query(EmergencyUnit).filter(EmergencyUnit.plan_id == plan_id).order_by(EmergencyUnit.dept_order.asc()).all()
  30. data = [
  31. {
  32. "id": row.id,
  33. "dept_id": row.dept_id,
  34. "dept_name": row.dept_name,
  35. "content": row.content
  36. }
  37. for row in rows
  38. ]
  39. # 返回结果
  40. return {
  41. "code": 200,
  42. "msg": "查询成功",
  43. "data": data
  44. }