equipment.py 2.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  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.basic import get_fcu_control_result
  6. from app.controllers.equipment.fcu.q_learning import get_fcu_q_learning_control_result
  7. from app.controllers.equipment.vav import get_vav_control_result
  8. from app.models.domain.equipment import EquipmentControlResponse, EquipmentControlRequest
  9. from app.utils.date import get_time_str
  10. class EquipmentName(str, Enum):
  11. FCU = 'ACATFC'
  12. VAV = 'ACATVA'
  13. router = APIRouter()
  14. @router.get('/control', response_model=EquipmentControlResponse)
  15. async def get_equipment_command_test(
  16. project_id: str = Query(..., max_length=50, regex='^Pj', alias='projectId'),
  17. equip_id: str = Query(..., max_length=50, regex='^Eq', alias='equipId'),
  18. equip_type: EquipmentName = Query(..., alias='equipType'),
  19. ):
  20. if equip_type.value == EquipmentName.FCU:
  21. if project_id == 'Pj1101050030':
  22. output = await get_fcu_control_result(project_id, equip_id)
  23. else:
  24. output = await get_fcu_q_learning_control_result(project_id, equip_id)
  25. elif equip_type.value == EquipmentName.VAV:
  26. vav = await get_vav_control_result(project_id, equip_id)
  27. output = {
  28. 'SupplyAirFlowSet': vav.supply_air_flow_set
  29. }
  30. else:
  31. output = {}
  32. response = {
  33. 'projectId': project_id,
  34. 'equipId': equip_id,
  35. 'time': get_time_str(),
  36. 'output': output
  37. }
  38. return response
  39. @router.post('/control', response_model=EquipmentControlResponse)
  40. async def get_equipment_command(equipment_control_info: EquipmentControlRequest):
  41. if equipment_control_info.equipType == EquipmentName.FCU:
  42. if equipment_control_info.projectId == 'Pj1101050030':
  43. output = await get_fcu_control_result(equipment_control_info.projectId, equipment_control_info.equipId)
  44. else:
  45. output = await get_fcu_q_learning_control_result(
  46. equipment_control_info.projectId,
  47. equipment_control_info.equipId
  48. )
  49. elif equipment_control_info.equipType == EquipmentName.VAV:
  50. vav = await get_vav_control_result(equipment_control_info.projectId, equipment_control_info.equipId)
  51. output = {
  52. 'SupplyAirFlowSet': vav.supply_air_flow_set
  53. }
  54. else:
  55. output = {}
  56. response = {
  57. 'projectId': equipment_control_info.projectId,
  58. 'equipId': equipment_control_info.equipId,
  59. 'time': get_time_str(),
  60. 'output': output
  61. }
  62. logger.info(response)
  63. return response