basic.py 6.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192
  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.equipment.space.realtime_temperature > 0:
  94. if self.season == Season.heating:
  95. diff = target - self.equipment.space.realtime_temperature
  96. if diff < 0:
  97. speed = AirValveSpeed.off
  98. elif diff < 0.5:
  99. speed = AirValveSpeed.low
  100. elif diff < 1.5:
  101. speed = AirValveSpeed.medium
  102. else:
  103. speed = AirValveSpeed.high
  104. self.equipment.air_valve_speed = speed
  105. else:
  106. self.equipment.air_valve_speed = AirValveSpeed.medium
  107. else:
  108. self.equipment.air_valve_speed = AirValveSpeed.off
  109. def build_running_status(self):
  110. if self.equipment.air_valve_speed == AirValveSpeed.off:
  111. self.equipment.running_status = False
  112. else:
  113. self.equipment.running_status = True
  114. def build_temperature_set(self):
  115. if np.isnan(self.equipment.space.temperature_target):
  116. self.equipment.setting_temperature = 0.0
  117. else:
  118. self.equipment.setting_temperature = self.equipment.space.temperature_target
  119. async def run(self):
  120. self.build_air_valve_speed()
  121. self.build_running_status()
  122. self.get_mode()
  123. self.build_temperature_set()
  124. def get_results(self):
  125. return self.equipment
  126. @logger.catch()
  127. async def get_fcu_control_result(project_id: str, equipment_id: str) -> Dict:
  128. async with AsyncClient() as client:
  129. duo_duo = Duoduo(client, project_id)
  130. platform = DataPlatformService(client, project_id)
  131. spaces = await duo_duo.get_space_by_equipment(equipment_id)
  132. for sp in spaces:
  133. transfer = SpaceInfoService(client, project_id, sp.get('id'))
  134. current_target = await transfer.get_current_temperature_target()
  135. realtime_temperature = await platform.get_realtime_temperature(sp.get('id'))
  136. season = await duo_duo.get_season()
  137. temp_space_params = {
  138. 'id': sp.get('id'),
  139. 'temperature_target': current_target,
  140. 'realtime_temperature': realtime_temperature
  141. }
  142. space = Space(**temp_space_params)
  143. break
  144. fcu_temp_params = {
  145. 'id': equipment_id,
  146. 'space': space
  147. }
  148. fcu = FCU(**fcu_temp_params)
  149. if project_id == 'Pj1101050030':
  150. fcu_controller = FCUController(fcu, season)
  151. else:
  152. fcu_controller = FCUControllerV2(fcu, season)
  153. await fcu_controller.run()
  154. regulated_fcu = fcu_controller.get_results()
  155. output = {
  156. 'onOff': 1 if regulated_fcu.running_status else 0,
  157. 'mode': regulated_fcu.work_mode,
  158. 'speed': regulated_fcu.air_valve_speed.value,
  159. 'temperature': float(round_half_up(regulated_fcu.setting_temperature, 1)),
  160. 'water': 1 if regulated_fcu.running_status else 0
  161. }
  162. return output