equipment.py 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778
  1. # -*- coding: utf-8 -*-
  2. from enum import Enum
  3. from fastapi import APIRouter, Query
  4. from loguru import logger
  5. from app.controllers.equipment.fcu import get_fcu_control_result
  6. from app.controllers.equipment.vav import get_vav_control_result
  7. from app.models.equipment import EquipmentControlInResponse, EquipmentControlInRequest
  8. from app.utils.date import get_time_str
  9. class EquipmentName(str, Enum):
  10. FCU = 'ACATFC'
  11. VAV = 'ACATVA'
  12. router = APIRouter()
  13. @router.get('/control', response_model=EquipmentControlInResponse, tags=['equipment'])
  14. async def get_equipment_command(
  15. projectId: str = Query(..., max_length=50, regex='^Pj'),
  16. equipId: str = Query(..., max_length=50, regex='^Eq'),
  17. equipType: EquipmentName = Query(...),
  18. method: int = Query(3),
  19. ):
  20. if equipType.value == EquipmentName.FCU:
  21. fcu = await get_fcu_control_result(projectId, equipId)
  22. output = {
  23. 'RunStatus': 1 if fcu.running_status else 0,
  24. 'WorkModeSet': 1,
  25. 'FanGearSet': fcu.air_valve_speed.value,
  26. 'IndoorAirTempSet': fcu.setting_temperature
  27. }
  28. elif equipType.value == EquipmentName.VAV:
  29. vav = await get_vav_control_result(projectId, equipId)
  30. output = {
  31. 'SupplyAirFlowSet': vav.supply_air_flow_set
  32. }
  33. else:
  34. output = {}
  35. response = {
  36. 'projectId': projectId,
  37. 'equipId': equipId,
  38. 'time': get_time_str(),
  39. 'output': output
  40. }
  41. return response
  42. @router.post('/control', response_model=EquipmentControlInResponse, tags=['equipment'])
  43. async def get_equipment_command_v2(equipment_control_info: EquipmentControlInRequest):
  44. if equipment_control_info.equipType == EquipmentName.FCU:
  45. fcu = await get_fcu_control_result(equipment_control_info.projectId, equipment_control_info.equipId)
  46. output = {
  47. 'RunStatus': 1 if fcu.running_status else 0,
  48. 'WorkModeSet': 1,
  49. 'FanGearSet': fcu.air_valve_speed.value,
  50. 'IndoorAirTempSet': fcu.setting_temperature
  51. }
  52. elif equipment_control_info.equipType == EquipmentName.VAV:
  53. vav = await get_vav_control_result(equipment_control_info.projectId, equipment_control_info.equipId)
  54. output = {
  55. 'SupplyAirFlowSet': vav.supply_air_flow_set
  56. }
  57. else:
  58. output = {}
  59. response = {
  60. 'projectId': equipment_control_info.projectId,
  61. 'equipId': equipment_control_info.equipId,
  62. 'time': get_time_str(),
  63. 'output': output
  64. }
  65. return response