basic.py 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293
  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, speed: Optional[str] = 'off') -> 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
  28. return self.target
  29. def get_speed_set(self) -> str:
  30. diff = self.realtime - self.target
  31. if diff < 0:
  32. speed = 'off'
  33. elif diff < 0.5:
  34. speed = 'LL'
  35. elif diff < 1.0:
  36. speed = 'L'
  37. elif diff < 1.5:
  38. speed = 'M'
  39. elif diff < 2.0:
  40. speed = 'H'
  41. elif diff >= 2.0:
  42. speed = 'HH'
  43. else:
  44. speed = 'off'
  45. self.device.speed_set = speed
  46. return speed
  47. async def run(self):
  48. if np.isnan(self.target) or np.isnan(self.realtime):
  49. self.get_switch_set()
  50. else:
  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(device_id: str, target: float, realtime: float) -> Dict:
  59. vrf = VRF(id=device_id)
  60. target = np.NAN if target == -1.0 else target
  61. realtime = np.NAN if realtime == -1.0 else realtime
  62. controller = VRFController(vrf, target, realtime)
  63. await controller.run()
  64. regulated_vrf = controller.get_results()
  65. output = {'switch_set': 'on' if regulated_vrf.equip_switch_set else 'off'}
  66. if regulated_vrf.equip_switch_set:
  67. output.update({'speed_set': regulated_vrf.speed_set})
  68. output.update({'temperature_set': regulated_vrf.temperature_set})
  69. output.update({'mode_set': regulated_vrf.mode_set})
  70. return output