basic.py 6.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189
  1. # -*- coding: utf-8 -*-
  2. from typing import Dict
  3. import numpy as np
  4. from httpx import AsyncClient
  5. from loguru import logger
  6. from app.controllers.equipment.controller import EquipmentController
  7. from app.schemas.equipment import AirValveSpeed, FCU
  8. from app.schemas.space import Space
  9. from app.services.platform import DataPlatformService
  10. from app.services.transfer import SpaceInfoService, Duoduo, Season
  11. from app.utils.math import round_half_up
  12. class FCUController(EquipmentController):
  13. def __init__(self, equipment: FCU, season: Season):
  14. super(FCUController, self).__init__()
  15. self._equipment = equipment
  16. self.season = season
  17. async def get_temperature_target(self) -> float:
  18. target = self._equipment.space.temperature_target
  19. if not np.isnan(target):
  20. setting_target = target
  21. else:
  22. setting_target = 25.0
  23. self._equipment.setting_temperature = setting_target
  24. return setting_target
  25. async def get_mode(self) -> int:
  26. temperature_target = self._equipment.space.temperature_target
  27. if self.season == Season.heating:
  28. if temperature_target >= self._equipment.space.realtime_temperature:
  29. mode = 2
  30. else:
  31. mode = 1
  32. elif self.season == Season.cooling:
  33. mode = 1
  34. else:
  35. mode = 0
  36. self._equipment.work_mode = mode
  37. return mode
  38. async def get_air_valve_speed(self):
  39. temperature_target = self._equipment.space.temperature_target
  40. if temperature_target > 0:
  41. if self.season == Season.heating:
  42. diff = abs(self._equipment.space.realtime_temperature - temperature_target)
  43. if diff >= 1.5:
  44. self._equipment.air_valve_speed = AirValveSpeed.high
  45. elif diff >= 1.0:
  46. self._equipment.air_valve_speed = AirValveSpeed.medium
  47. elif diff >= 0.5:
  48. self._equipment.air_valve_speed = AirValveSpeed.low
  49. else:
  50. self._equipment.air_valve_speed = AirValveSpeed.off
  51. else:
  52. diff = self._equipment.space.realtime_temperature - temperature_target
  53. if diff >= 1.0:
  54. self._equipment.air_valve_speed = AirValveSpeed.high
  55. elif diff >= 0.5:
  56. self._equipment.air_valve_speed = AirValveSpeed.medium
  57. elif diff >= 0.0:
  58. self._equipment.air_valve_speed = AirValveSpeed.low
  59. else:
  60. self._equipment.air_valve_speed = AirValveSpeed.off
  61. else:
  62. self._equipment.air_valve_speed = AirValveSpeed.off
  63. async def get_running_status(self):
  64. if self._equipment.air_valve_speed == AirValveSpeed.off:
  65. self._equipment.running_status = False
  66. else:
  67. self._equipment.running_status = True
  68. async def run(self):
  69. # await self.get_temperature_target()
  70. await self.get_air_valve_speed()
  71. await self.get_running_status()
  72. await self.get_mode()
  73. def get_results(self):
  74. return self._equipment
  75. class FCUControllerV2(EquipmentController):
  76. """Writen by Wenyan"""
  77. def __init__(self, equipment: FCU, season: Season):
  78. super(FCUControllerV2, self).__init__()
  79. self.equipment = equipment
  80. self.season = season
  81. def get_mode(self) -> int:
  82. if self.season == Season.heating:
  83. mode = 2
  84. elif self.season == Season.cooling:
  85. mode = 1
  86. else:
  87. mode = 0
  88. self.equipment.work_mode = mode
  89. return mode
  90. def build_air_valve_speed(self):
  91. target = self.equipment.space.temperature_target
  92. if target > 0:
  93. if self.season == Season.heating:
  94. diff = target - self.equipment.space.realtime_temperature
  95. if diff < 0:
  96. speed = AirValveSpeed.off
  97. elif diff < 0.5:
  98. speed = AirValveSpeed.low
  99. elif diff < 1.5:
  100. speed = AirValveSpeed.medium
  101. else:
  102. speed = AirValveSpeed.high
  103. self.equipment.air_valve_speed = speed
  104. else:
  105. self.equipment.air_valve_speed = AirValveSpeed.off
  106. def build_running_status(self):
  107. if self.equipment.air_valve_speed == AirValveSpeed.off:
  108. self.equipment.running_status = False
  109. else:
  110. self.equipment.running_status = True
  111. def build_temperature_set(self):
  112. if np.isnan(self.equipment.space.temperature_target):
  113. self.equipment.setting_temperature = 0.0
  114. else:
  115. self.equipment.setting_temperature = self.equipment.space.temperature_target
  116. async def run(self):
  117. self.build_air_valve_speed()
  118. self.build_running_status()
  119. self.get_mode()
  120. self.build_temperature_set()
  121. def get_results(self):
  122. return self.equipment
  123. @logger.catch()
  124. async def get_fcu_control_result(project_id: str, equipment_id: str) -> Dict:
  125. async with AsyncClient() as client:
  126. duo_duo = Duoduo(client, project_id)
  127. platform = DataPlatformService(client, project_id)
  128. spaces = await duo_duo.get_space_by_equipment(equipment_id)
  129. for sp in spaces:
  130. transfer = SpaceInfoService(client, project_id, sp.get('id'))
  131. current_target = await transfer.get_current_temperature_target()
  132. realtime_temperature = await platform.get_realtime_temperature(sp.get('id'))
  133. season = await duo_duo.get_season()
  134. temp_space_params = {
  135. 'id': sp.get('id'),
  136. 'temperature_target': current_target,
  137. 'realtime_temperature': realtime_temperature
  138. }
  139. space = Space(**temp_space_params)
  140. break
  141. fcu_temp_params = {
  142. 'id': equipment_id,
  143. 'space': space
  144. }
  145. fcu = FCU(**fcu_temp_params)
  146. if project_id == 'Pj1101050030':
  147. fcu_controller = FCUController(fcu, season)
  148. else:
  149. fcu_controller = FCUControllerV2(fcu, season)
  150. await fcu_controller.run()
  151. regulated_fcu = fcu_controller.get_results()
  152. output = {
  153. 'onOff': 1 if regulated_fcu.running_status else 0,
  154. 'mode': regulated_fcu.work_mode,
  155. 'speed': regulated_fcu.air_valve_speed.value,
  156. 'temperature': float(round_half_up(regulated_fcu.setting_temperature, 1)),
  157. 'water': 1 if regulated_fcu.running_status else 0
  158. }
  159. return output