fcu.py 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. # -*- coding: utf-8 -*-
  2. import numpy as np
  3. from httpx import AsyncClient
  4. from loguru import logger
  5. from app.controllers.equipment.controller import EquipmentController
  6. from app.schemas.equipment import AirValveSpeed, FCU, Space
  7. from app.services.platform import DataPlatformService
  8. from app.services.transfer import SpaceInfoService, EquipmentInfoService
  9. class FCUController(EquipmentController):
  10. def __init__(self, equipment: FCU):
  11. super(FCUController, self).__init__()
  12. self._equipment = equipment
  13. async def get_temperature_target(self) -> float:
  14. target = self._equipment.space.temperature_target
  15. if not np.isnan(target) and target > 0:
  16. setting_target = target
  17. else:
  18. setting_target = 25.0
  19. self._equipment.setting_temperature = setting_target
  20. return setting_target
  21. async def get_air_valve_speed(self):
  22. temperature_target = await self.get_temperature_target()
  23. if temperature_target and temperature_target > 0:
  24. self._equipment.air_valve_speed = AirValveSpeed.auto
  25. else:
  26. self._equipment.air_valve_speed = AirValveSpeed.off
  27. async def run(self):
  28. await self.get_temperature_target()
  29. await self.get_air_valve_speed()
  30. self._equipment.running_status = True
  31. def get_results(self):
  32. return self._equipment
  33. @logger.catch()
  34. async def get_fcu_control_result(project_id: str, equipment_id: str) -> FCU:
  35. async with AsyncClient() as client:
  36. duo_duo = EquipmentInfoService(client, project_id)
  37. platform = DataPlatformService(client, project_id)
  38. spaces = await duo_duo.get_space_by_equipment(equipment_id)
  39. for sp in spaces:
  40. transfer = SpaceInfoService(client, project_id, sp.get('id'))
  41. current_target = await transfer.get_current_temperature_target()
  42. realtime_temperature = await platform.get_realtime_temperature(sp.get('id'))
  43. temp_space_params = {
  44. 'id': sp.get('id'),
  45. 'temperature_target': current_target,
  46. 'realtime_temperature': realtime_temperature
  47. }
  48. space = Space(**temp_space_params)
  49. break
  50. fcu_temp_params = {
  51. 'id': equipment_id,
  52. 'space': space
  53. }
  54. fcu = FCU(**fcu_temp_params)
  55. fcu_controller = FCUController(fcu)
  56. await fcu_controller.run()
  57. regulated_fcu = fcu_controller.get_results()
  58. return regulated_fcu