space.py 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. # -*- coding: utf-8 -*-
  2. from typing import List
  3. from fastapi import APIRouter, Depends, HTTPException, Query
  4. from sqlalchemy.orm import Session
  5. from app.api.dependencies.db import get_db
  6. from app.crud.space.weight import (
  7. get_weights_by_space,
  8. get_weights_by_vav,
  9. create_weight,
  10. update_weight,
  11. )
  12. from app.models.domain.space import SpaceControlResponse
  13. from app.schemas.sapce_weight import SpaceWeight, SpaceWeightCreate, SpaceWeightUpdate
  14. router = APIRouter()
  15. @router.get("/control", response_model=SpaceControlResponse)
  16. async def get_space_command(
  17. project_id: str = Query(..., max_length=50, regex="^Pj", alias="projectId"),
  18. space_id: str = Query(..., max_length=50, regex="^Sp", alias="roomId"),
  19. timestamp: str = Query(None, min_length=14, max_length=14, alias="time"),
  20. method: int = Query(3),
  21. ):
  22. response = {
  23. "projectId": project_id,
  24. "roomId": space_id,
  25. "flag": 1,
  26. "time": timestamp,
  27. "method": method,
  28. }
  29. return response
  30. @router.post("/weight", response_model=SpaceWeight)
  31. async def create_space_weight(weight: SpaceWeightCreate, db: Session = Depends(get_db)):
  32. return create_weight(db=db, weight=weight)
  33. @router.get("/weight/{space_id}", response_model=List[SpaceWeight])
  34. async def read_weight_by_space(space_id: str, db: Session = Depends(get_db)):
  35. db_weights = get_weights_by_space(db, space_id=space_id)
  36. return db_weights
  37. @router.get("/weight/vav/{vav_id}", response_model=List[SpaceWeight])
  38. async def read_weight_by_vav(vav_id: str, db: Session = Depends(get_db)):
  39. db_weights = get_weights_by_vav(db, vav_id=vav_id)
  40. return db_weights
  41. @router.put("/weight/{space_id}/{vav_id}", response_model=SpaceWeight)
  42. async def update_weight_by_space(
  43. space_id: str,
  44. vav_id: str,
  45. weight_in: SpaceWeightUpdate,
  46. db: Session = Depends(get_db),
  47. ):
  48. weights = get_weights_by_space(db, space_id=space_id)
  49. new_weight = None
  50. for weight in weights:
  51. if weight.vav_box_id == vav_id:
  52. new_weight = update_weight(db, db_weight=weight, weight_in=weight_in)
  53. break
  54. if not new_weight:
  55. raise HTTPException(status_code=404, detail="Wight not found")
  56. return new_weight