basic.py 2.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  1. from typing import Dict, Optional
  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) -> bool:
  13. if np.isnan(self.target):
  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
  28. return self.target
  29. def get_speed_set(self) -> str:
  30. diff = self.realtime - self.target
  31. if diff <= 0:
  32. speed = 'LL'
  33. elif diff < 0.8:
  34. speed = 'L'
  35. elif diff < 1.5:
  36. speed = 'M'
  37. elif diff < 2.0:
  38. speed = 'H'
  39. elif diff >= 2.0:
  40. speed = 'HH'
  41. else:
  42. speed = 'LL'
  43. self.device.speed_set = speed
  44. return speed
  45. async def run(self):
  46. if self.get_switch_set():
  47. self.get_speed_set()
  48. self.get_mode_set()
  49. self.get_temperature_set()
  50. def get_results(self):
  51. return self.device
  52. @logger.catch()
  53. async def get_vrf_instructions(device_id: str, target: float, realtime: float) -> Dict:
  54. vrf = VRF(id=device_id)
  55. target = np.NAN if target == -1.0 else target
  56. realtime = np.NAN if realtime == -1.0 else realtime
  57. controller = VRFController(vrf, target, realtime)
  58. await controller.run()
  59. regulated_vrf = controller.get_results()
  60. output = {'switch_set': 'on' if regulated_vrf.equip_switch_set else 'off'}
  61. if regulated_vrf.equip_switch_set:
  62. output.update({'speed_set': regulated_vrf.speed_set})
  63. output.update({'temperature_set': regulated_vrf.temperature_set})
  64. output.update({'mode_set': regulated_vrf.mode_set})
  65. return output