65 lines
2.5 KiB
Python
65 lines
2.5 KiB
Python
from dataclasses import dataclass, field, fields
|
|
from datetime import timedelta
|
|
|
|
from appdaemon.entity import Entity
|
|
from appdaemon.plugins.hass.hassapi import Hass
|
|
|
|
|
|
@dataclass(init=False)
|
|
class LivingRoomTV(Hass):
|
|
remote_entity: str = None
|
|
playing_entity: str = None
|
|
# media_entity: str = None
|
|
# hardware_entity: str = None
|
|
|
|
def initialize(self):
|
|
self.log(self.args)
|
|
for f in fields(self):
|
|
if f.name in self.args:
|
|
entity_id = self.args[f.name]
|
|
if not self.entity_exists(entity_id):
|
|
self.log(f'{entity_id} does not exist', level="WARNING")
|
|
else:
|
|
setattr(self, f.name, entity_id)
|
|
else:
|
|
self.log(f'{f.name} field is unset', level="WARNING")
|
|
self.log(repr(self))
|
|
|
|
self.listen_state(self.turn_on_soundbar, self.playing_entity, new='playing')
|
|
self.log(f'Waiting for {self.playing_entity} to go to playing')
|
|
|
|
def turn_on_soundbar(self, entity=None, attribute=None, old=None, new=None, kwargs=None):
|
|
if self.get_state(self.remote_entity) != 'on':
|
|
self.log('Turning on remote')
|
|
self.turn_on(self.remote_entity)
|
|
# if self.remote_entity.state != 'on':
|
|
# self.log('Turning on remote')
|
|
# self.remote_entity.turn_on()
|
|
|
|
self.log('Turning on soundbar')
|
|
self.call_service(service='remote/send_command',
|
|
entity_id=self.remote_entity,
|
|
device='BoseTV', command='TV')
|
|
|
|
def state_change(self, entity=None, attribute=None, old=None, new=None, kwargs=None):
|
|
try:
|
|
if entity.endswith('2') or entity.endswith('3'):
|
|
name = new['attributes']['entity_id']
|
|
app_name = new['attributes']['app_name']
|
|
state = new['state']
|
|
self.log(f'{name}: {app_name} {state}')
|
|
else:
|
|
self.log(f'{entity}: {new}')
|
|
except KeyError as e:
|
|
self.log(f'Failed getting attribute {e} from {entity}')
|
|
|
|
def cancel_callbacks(self, entity=None, attribute=None, old=None, new=None, kwargs=None):
|
|
if (callbacks := self.get_scheduler_entries().get(self.name)):
|
|
for handle, info in callbacks.items():
|
|
self.cancel_timer(handle)
|
|
self.log(f'Cancelled {handle}')
|
|
|
|
def hardware_off_callback(self, entity=None, attribute=None, old=None, new=None, kwargs=None):
|
|
self.log('Hardware turned off')
|
|
self.media_entity.turn_off()
|