from datetime import datetime from typing import Dict, Optional, Tuple import arrow import numpy as np from loguru import logger from sqlalchemy.orm import Session from sqlalchemy.exc import SQLAlchemyError from app.controllers.equipment.controller import EquipmentController from app.crud.device.device import device from app.crud.device.status_timestamp import blowy_feedback_time, high_speed_time from app.models.domain.devices import ACATVIInstructionsRequest from app.models.domain.feedback import FeedbackValue from app.schemas.device.device import DeviceCreate from app.schemas.equipment import VRF from app.schemas.season import Season from app.schemas.device.status_timestamp import ( BlowyFeedbackTimeCreate, BlowyFeedbackTimeUpdate, HighSpeedTimeCreate, HighSpeedTimeUpdate, ) from app.utils.math import round_half_up class VRFController(EquipmentController): def __init__(self, device: VRF, target: float, realtime: float): super(VRFController, self).__init__() self.device = device self.target = target self.realtime = realtime def get_switch_set(self) -> str: if np.isnan(self.target): if self.device.running_status: switch_set = "off" else: switch_set = "hold" else: if self.device.running_status: switch_set = "hold" else: switch_set = "on" self.device.equip_switch_set = switch_set return switch_set def get_mode_set(self) -> str: if self.target > self.realtime: mode_set = "cooling" else: mode_set = "cooling" self.device.mode_set = mode_set return mode_set def get_temperature_set(self) -> float: if self.realtime >= 24.0: diff = self.device.return_air_temp - self.realtime if diff > 0: new_temperature_set = self.target + diff elif diff <= 0: new_temperature_set = self.target else: if np.isnan(self.device.current_temperature_set): new_temperature_set = self.realtime else: new_temperature_set = self.device.current_temperature_set elif self.realtime <= 22.0: new_temperature_set = 28.0 else: if np.isnan(self.device.current_temperature_set): new_temperature_set = 25.0 else: new_temperature_set = self.device.current_temperature_set new_temperature_set = round_half_up(new_temperature_set) self.device.temperature_set = new_temperature_set return self.device.temperature_set def get_speed_set(self) -> str: if self.realtime >= 24.0: speed = "M" elif self.realtime <= 22.0: speed = "LL" elif self.device.return_air_temp - self.target >= 3.0: speed = "HH" elif self.device.return_air_temp - self.target >= 1.0: speed = "M" else: if self.device.speed == "missing": speed = "M" else: speed = self.device.speed self.device.speed_set = speed return speed async def run(self): if self.get_switch_set() == "on": self.get_speed_set() self.get_mode_set() self.get_temperature_set() if self.device.temperature_set == self.device.current_temperature_set: self.device.temperature_set = 0.0 if self.device.speed_set == self.device.speed: self.device.speed_set = "hold" self.device.equip_switch_set = "hold" if not self.device.running_status: self.device.equip_switch_set = "on" self.get_speed_set() else: if not self.device.running_status: self.device.equip_switch_set = "hold" def get_results(self): return self.device class VRFController1(VRFController): def __init__( self, device: VRF, target: float, realtime: float, feedback: FeedbackValue, on_time: str, off_time: str, ): super(VRFController1, self).__init__(device, target, realtime) self.feedback = feedback self.on_time = on_time self.off_time = off_time def get_switch_set(self) -> str: if self.feedback.value == FeedbackValue.switch_off: if self.device.running_status: switch_set = "off" else: switch_set = "hold" elif self.feedback.value == FeedbackValue.switch_on: if not self.device.running_status: switch_set = "on" else: switch_set = "hold" else: utc = arrow.utcnow() now = utc.to("Asia/Shanghai") ago = now.shift(seconds=-150).format("HHmmSS") later = now.shift(seconds=150).format("HHmmSS") if ago <= self.on_time <= later: if not self.device.running_status: switch_set = "on" else: switch_set = "hold" elif ago <= self.off_time <= later: if self.device.running_status: switch_set = "off" else: switch_set = "hold" else: switch_set = "hold" self.device.equip_switch_set = switch_set return switch_set def get_temperature_set(self) -> float: if self.device.work_mode == 4.0 or self.device.work_mode == 8.0: # if self.season == Season.transition: new_temperature_set = np.NAN elif self.device.work_mode == 1.0 or self.device.work_mode == 6.0: # elif self.season == Season.cooling: new_temperature_set = np.NAN if np.isnan(self.target): return new_temperature_set # Default temperature set. if not self.device.running_status: new_temperature_set = 26.0 # lower limit. if self.realtime <= 22.0: new_temperature_set = 28.0 # recovery # if self.device.temperature_set == 28.0 and self.realtime > 25.0: # new_temperature_set = self.target # feedback if self.feedback.value != "null" and self.device.current_temperature_set: if ( self.feedback == FeedbackValue.a_little_cold or self.feedback == FeedbackValue.so_cold ): if self.feedback == FeedbackValue.so_cold: if self.device.speed == "LL": new_temperature_set = ( self.device.current_temperature_set + 2.0 ) else: new_temperature_set = ( self.device.current_temperature_set + 1.0 ) else: new_temperature_set = self.device.current_temperature_set + 1.0 elif ( self.feedback == FeedbackValue.a_little_hot or self.feedback == FeedbackValue.so_hot ): if ( self.feedback == FeedbackValue.so_hot and self.device.speed == "HH" ): if self.device.speed == "HH": new_temperature_set = ( self.device.current_temperature_set - 2.0 ) else: new_temperature_set = ( self.device.current_temperature_set - 1.0 ) else: new_temperature_set = self.device.current_temperature_set - 1.0 if not np.isnan(new_temperature_set): new_temperature_set = max(24.0, min(28.0, new_temperature_set)) elif self.device.work_mode == 2.0 or self.device.work_mode == 7.0: new_temperature_set = np.NAN if np.isnan(self.target): return new_temperature_set if not self.device.running_status: new_temperature_set = 20.0 if self.realtime >= 27.0: new_temperature_set = 18.0 if self.feedback.value != "null" and self.device.current_temperature_set: if ( self.feedback == FeedbackValue.a_little_cold or self.feedback == FeedbackValue.so_cold ): if self.feedback == FeedbackValue.so_cold: if self.device.speed == "M": new_temperature_set = ( self.device.current_temperature_set + 1.0 ) else: new_temperature_set = ( self.device.current_temperature_set + 2.0 ) else: new_temperature_set = self.device.current_temperature_set + 1.0 elif ( self.feedback == FeedbackValue.a_little_hot or self.feedback == FeedbackValue.so_hot ): if self.feedback == FeedbackValue.so_hot: if self.device.speed == "M": new_temperature_set = ( self.device.current_temperature_set - 1.0 ) else: new_temperature_set = ( self.device.current_temperature_set - 2.0 ) else: new_temperature_set = self.device.current_temperature_set - 1.0 if not np.isnan(new_temperature_set): new_temperature_set = max(18.0, min(24.0, new_temperature_set)) else: new_temperature_set = np.NAN new_temperature_set = round_half_up(new_temperature_set) self.device.temperature_set = new_temperature_set return new_temperature_set def get_speed_set(self) -> str: if self.device.work_mode == 4.0 or self.device.work_mode == 8.0: # if self.season == Season.transition: if np.isnan(self.target): new_speed = "hold" else: if self.device.running_status: new_speed = "hold" if ( self.feedback == FeedbackValue.noisy_or_blowy or self.feedback == FeedbackValue.a_little_cold or self.feedback == FeedbackValue.so_cold ): if self.device.speed == "HH": new_speed = "M" elif self.device.speed == "M": new_speed = "LL" else: new_speed = "hold" elif ( self.feedback == FeedbackValue.a_little_hot or self.feedback == FeedbackValue.so_hot ): if self.device.speed == "LL": new_speed = "M" elif self.device.speed == "M": new_speed = "HH" else: new_speed = "hold" else: new_speed = "M" elif self.device.work_mode == 1.0 or self.device.work_mode == 6.0: # elif self.season == Season.cooling: new_speed = "hold" if np.isnan(self.target): return new_speed # Default speed set: if not self.device.running_status: new_speed = "M" # Lower limit. if self.realtime <= 22.0: new_speed = "LL" # Feedback. if self.feedback == FeedbackValue.so_cold: if self.device.return_air_temp and self.device.current_temperature_set: if ( self.device.return_air_temp > self.device.current_temperature_set ): if self.device.speed == "HH": new_speed = "M" elif self.device.speed == "M": new_speed = "LL" elif self.feedback == FeedbackValue.so_hot: if self.device.speed == "LL": new_speed = "M" elif self.device.speed == "M": new_speed = "HH" elif self.feedback == FeedbackValue.noisy_or_blowy: if self.device.speed == "HH": new_speed = "M" elif self.device.speed == "M": new_speed = "LL" else: new_speed = "LL" elif self.device.work_mode == 2.0 or self.device.work_mode == 7.0: new_speed = "hold" if np.isnan(self.target): return new_speed # Default speed set: if not self.device.running_status: new_speed = "M" # Lower limit. if self.realtime >= 28.0: new_speed = "LL" # Feedback. if self.feedback == FeedbackValue.so_hot: if self.device.return_air_temp and self.device.current_temperature_set: if ( self.device.return_air_temp < self.device.current_temperature_set ): if self.device.speed == "HH": new_speed = "M" elif self.device.speed == "M": new_speed = "LL" elif self.feedback == FeedbackValue.so_cold: if self.device.speed == "LL": new_speed = "M" elif self.device.speed == "M": new_speed = "HH" elif self.feedback == FeedbackValue.noisy_or_blowy: if self.device.speed == "HH": new_speed = "M" elif self.device.speed == "M": new_speed = "LL" else: new_speed = "LL" else: new_speed = "hold" # a_now = arrow.utcnow() # Restriction. # if a_now.date() == arrow.get(self.blow_feedback_time).date(): # if new_speed == "HH": # new_speed = "M" # Recovery. # logger.debug(self.high_speed_time) # if (a_now - arrow.get(self.high_speed_time)).seconds > 60 * 60: # logger.debug(self.high_speed_time) # if new_speed == "HH": # new_speed = "M" self.device.speed_set = new_speed return new_speed def ventilation_mode(self) -> str: new_speed = "hold" if np.isnan(self.target): return new_speed else: if not self.device.running_status: new_speed = "HH" else: # if self.realtime > 26.0: # new_speed = "HH" # elif self.realtime < 22.0: # new_speed = "LL" if ( self.feedback == FeedbackValue.a_little_cold or self.feedback == FeedbackValue.so_cold or self.feedback == FeedbackValue.noisy_or_blowy ): if self.device.speed == "HH": new_speed = "M" elif self.device.speed == "M": new_speed = "LL" if ( self.feedback == FeedbackValue.a_little_hot or self.feedback == FeedbackValue.so_hot ): if self.device.speed == "LL": new_speed = "M" elif self.device.speed == "M": new_speed = "HH" self.device.speed_set = new_speed return new_speed async def run(self): self.get_switch_set() self.get_speed_set() self.get_temperature_set() async def query_status_time(db: Session, device_id: str) -> Tuple[datetime, datetime]: feedback_time_in_db = blowy_feedback_time.get_time_by_device(db, device_id) if feedback_time_in_db: feedback_time = feedback_time_in_db.timestamp else: past = arrow.utcnow().shift(hours=-24) feedback_time = past.naive if not device.get(db, device_id): device.create(db=db, obj_in=DeviceCreate(id=device_id)) blowy_feedback_time.create( db=db, obj_in=BlowyFeedbackTimeCreate( timestamp=feedback_time, device_id=device_id ), ) high_speed_time_in_db = high_speed_time.get_time_by_device(db, device_id) if high_speed_time_in_db: high_time = high_speed_time_in_db.timestamp else: past = arrow.utcnow().shift(hours=-24) high_time = past.naive if not device.get(db, device_id): device.create(db=db, obj_in=DeviceCreate(id=device_id)) high_speed_time.create( db=db, obj_in=HighSpeedTimeCreate(timestamp=high_time, device_id=device_id) ) return feedback_time, high_time @logger.catch() async def build_acatvi_instructions( params: ACATVIInstructionsRequest, db: Session ) -> Dict: target = ( np.NAN if params.space_temperature_target == -1.0 else params.space_temperature_target ) realtime = ( np.NAN if params.space_realtime_temperature == -1.0 else params.space_realtime_temperature ) return_air_temp = ( np.NAN if params.return_air_temperature == -1.0 else params.return_air_temperature ) current_temperature_set = ( np.NAN if params.current_temperature_set == -1.0 else round_half_up(params.current_temperature_set) ) vrf = VRF( return_air_temp=return_air_temp, current_temperature_set=current_temperature_set, speed=params.current_speed, running_status=params.running_status, work_mode=params.work_mode, ) controller = VRFController1( vrf, target, realtime, params.feedback, params.on_time, params.off_time ) await controller.run() regulated_vrf = controller.get_results() # if params.feedback == FeedbackValue.noisy_or_blowy: # blowy_feedback_time_in_db = blowy_feedback_time.get_time_by_device( # db=db, device_id=params.device_id # ) # if blowy_feedback_time_in_db: # blowy_feedback_time.update( # db=db, # db_obj=blowy_feedback_time_in_db, # obj_in=BlowyFeedbackTimeUpdate( # timestamp=datetime.utcnow(), device_id=params.device_id # ), # ) # else: # blowy_feedback_time.create( # db=db, # obj_in=BlowyFeedbackTimeCreate( # timestamp=datetime.utcnow(), device_id=params.device_id # ), # ) # if regulated_vrf.speed_set == "HH": # high_speed_time_in_db = high_speed_time.get_time_by_device( # db=db, device_id=params.device_id # ) # if high_speed_time_in_db: # high_speed_time.update( # db=db, # db_obj=high_speed_time_in_db, # obj_in=HighSpeedTimeUpdate( # timestamp=datetime.utcnow(), device_id=params.device_id # ), # ) # else: # high_speed_time.create( # db=db, # obj_in=HighSpeedTimeCreate( # timestamp=datetime.utcnow(), device_id=params.device_id # ), # ) instructions = dict() instructions.update({"switch_set": regulated_vrf.equip_switch_set}) instructions.update({"speed_set": regulated_vrf.speed_set}) if regulated_vrf.temperature_set and not np.isnan(regulated_vrf.temperature_set): instructions.update({"temperature_set": regulated_vrf.temperature_set}) return instructions