starting point

apps/continuous.py

@@ -0,0 +1,107 @@
+from dataclasses import dataclass
+from datetime import datetime, timedelta
+
+from appdaemon.entity import Entity
+from appdaemon.plugins.hass.hassapi import Hass
+from daylight_adjuster import DaylightAdjuster
+from pvlib.location import Location
+from rich import print
+import pandas as pd
+from astral.sun import elevation, time_at_elevation
+from astral import SunDirection
+import astral
+
+HOME_TZ = datetime.now().astimezone().tzinfo
+
+
+@dataclass(init=False)
+class Continuous(Hass):
+    entity: Entity
+    latitude: float
+    longitude: float
+
+    def initialize(self):
+        
+        # self.log(f'Brightness Range: {self.brightness_rng}, {self.current_adjuster.get_brightness()}')
+        # self.run_daily(self.refresh_daylight_times, '00:00:00')
+        # self.run_every(self.create_img, 'now', int(timedelta(minutes=5).total_seconds()))
+        self.run_every(self.adjust, 'now', int(timedelta(seconds=30).total_seconds()))
+
+    @property
+    def entity(self) -> Entity:
+        return self.get_entity(self.args['entity'])
+
+    @property
+    def light_state(self) -> bool:
+        return self.entity.is_state('on')
+
+    @light_state.setter
+    def light_state(self, new):
+        if isinstance(new, bool):
+            if new:
+                self.entity.turn_on()
+            else:
+                self.entity.turn_off()
+        elif isinstance(new, dict):
+            self.entity.turn_on(**new)
+        else:
+            raise TypeError(f'Wrong type for light state: {new}')
+
+    @property
+    def brightness(self) -> int:
+        return self.entity.get_state('brightness')
+
+    @brightness.setter
+    def brightness(self, val: int):
+        self.log(f'Setting brightness of {self.friendly_name(self.args["entity"])} to {val}')
+        self.entity.turn_on(brightness=val)
+
+    @property
+    def latitude(self) -> float:
+        return float(self.args['latitude'])
+
+    @property
+    def longitude(self) -> float:
+        return float(self.args['longitude'])
+
+    @property
+    def location(self) -> Location:
+        return Location(latitude=self.latitude,
+                        longitude=self.longitude)
+
+    @property
+    def period_df(self):
+        return pd.DataFrame(self.periods).set_index('time')
+    
+    # @property
+    # def full_df(self):
+    #     return pd.concat([df, sub_df], axis=1).sort_index().interpolate().bfill().ffill()
+
+    @property
+    def brightness_rng(self):
+        for time, brightness_rng in self.periods[::-1]:
+            if time <= self.time():
+                return brightness_rng
+        else:
+            period_start, brightness = self.periods[-1]
+            return brightness
+
+    @property
+    def adjuster(self):
+        return DaylightAdjuster(self.latitude, self.longitude, self.args['periods']) 
+
+    def adjust(self, kwargs):
+        self.log(f'Adjusting...')
+        self.log(self.adjuster.current_settings)
+        if self.light_state:
+            self.light_state = self.adjuster.current_settings
+
+    def create_img(self, kwargs):
+        self.log(f'Creating daylight curve img...')
+        try:
+            # self.current_adjuster.elevation_fig().savefig('/mnt/ha_config/www/daylight_curve.png')
+            self.adjuster.elevation_fig().savefig('/conf/daylight_curve.png')
+        except:
+            raise
+        else:
+            self.log(f'Done')