basic.py 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
  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: 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
  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. speed = self.get_speed_set()
  49. self.get_switch_set(speed)
  50. self.get_mode_set()
  51. self.get_temperature_set()
  52. def get_results(self):
  53. return self.device
  54. @logger.catch()
  55. async def get_vrf_instructions(device_id: str, target: float, realtime: float) -> Dict:
  56. vrf = VRF(id=device_id)
  57. target = np.NAN if target == -1.0 else target
  58. realtime = np.NAN if realtime == -1.0 else realtime
  59. controller = VRFController(vrf, target, realtime)
  60. await controller.run()
  61. regulated_vrf = controller.get_results()
  62. output = {'switch_set': 'on' if regulated_vrf.equip_switch_set else 'off'}
  63. if regulated_vrf.equip_switch_set:
  64. output.update({'speed_set': regulated_vrf.speed_set})
  65. output.update({'temperature_set': regulated_vrf.temperature_set})
  66. output.update({'mode_set': regulated_vrf.mode_set})
  67. return output