basic.py 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495
  1. from typing import Dict
  2. import numpy as np
  3. from loguru import logger
  4. from app.controllers.equipment.controller import EquipmentController
  5. from app.schemas.equipment import VRF
  6. class VRFController(EquipmentController):
  7. def __init__(self, device: VRF, target: float, realtime: float):
  8. super(VRFController, self).__init__()
  9. self.device = device
  10. self.target = target
  11. self.realtime = realtime
  12. def get_switch_set(self, speed: str) -> bool:
  13. if speed == 'off':
  14. switch_set = False
  15. else:
  16. switch_set = True
  17. self.device.equip_switch_set = switch_set
  18. return switch_set
  19. def get_mode_set(self) -> str:
  20. if self.target > self.realtime:
  21. mode_set = 'cooling'
  22. else:
  23. mode_set = 'cooling'
  24. self.device.mode_set = mode_set
  25. return mode_set
  26. def get_temperature_set(self) -> float:
  27. self.device.temperature_set = self.target + 1
  28. return self.target
  29. def get_speed_set(self) -> str:
  30. if self.realtime > self.device.return_air_temp:
  31. diff = self.realtime - self.target
  32. else:
  33. diff = self.device.return_air_temp - (self.target + 1)
  34. if diff < 0:
  35. speed = 'off'
  36. elif diff < 0.5:
  37. speed = 'LL'
  38. elif diff < 1.0:
  39. speed = 'L'
  40. elif diff < 1.5:
  41. speed = 'M'
  42. elif diff < 2.0:
  43. speed = 'H'
  44. elif diff > 2.0:
  45. speed = 'HH'
  46. else:
  47. speed = 'off'
  48. self.device.speed_set = speed
  49. return speed
  50. async def run(self):
  51. speed = self.get_speed_set()
  52. self.get_switch_set(speed)
  53. self.get_mode_set()
  54. self.get_temperature_set()
  55. def get_results(self):
  56. return self.device
  57. @logger.catch()
  58. async def get_vrf_instructions(target: float, return_air_temp: float, realtime: float) -> Dict:
  59. target = np.NAN if target == -1.0 else target
  60. realtime = np.NAN if realtime == -1.0 else realtime
  61. return_air_temp = np.NAN if realtime == -1.0 else return_air_temp
  62. vrf = VRF(return_air_temp=return_air_temp)
  63. controller = VRFController(vrf, target, realtime)
  64. await controller.run()
  65. regulated_vrf = controller.get_results()
  66. output = {'switch_set': 'on' if regulated_vrf.equip_switch_set else 'off'}
  67. if regulated_vrf.equip_switch_set:
  68. output.update({'speed_set': regulated_vrf.speed_set})
  69. output.update({'temperature_set': regulated_vrf.temperature_set})
  70. output.update({'mode_set': regulated_vrf.mode_set})
  71. return output