thomas/hass-browser_mod
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Code linting

Browse Source
This commit is contained in:
Thomas Lovén
2021-04-29 10:18:23 +00:00
parent c1a000abb3
commit 862b93a175
15 changed files with 337 additions and 305 deletions
Download Patch File Download Diff File Expand all files Collapse all files

91
custom_components/browser_mod/connection.py
View File

@@ -5,7 +5,7 @@ from homeassistant.components.websocket_api import (
websocket_command,
result_message,
event_message,
async_register_command
async_register_command,
)
from .const import WS_CONNECT, WS_UPDATE
@@ -15,26 +15,28 @@ _LOGGER = logging.getLogger(__name__)
async def setup_connection(hass, config):
@websocket_command({
vol.Required("type"): WS_CONNECT,
vol.Required("deviceID"): str,
})
@websocket_command(
{
vol.Required("type"): WS_CONNECT,
vol.Required("deviceID"): str,
}
)
def handle_connect(hass, connection, msg):
deviceID = msg["deviceID"]
device = get_devices(hass).get(deviceID,
BrowserModConnection(hass, deviceID))
device = get_devices(hass).get(deviceID, BrowserModConnection(hass, deviceID))
device.connect(connection, msg["id"])
get_devices(hass)[deviceID] = device
connection.send_message(result_message(msg["id"]))
@websocket_command({
vol.Required("type"): WS_UPDATE,
vol.Required("deviceID"): str,
vol.Optional("data"): dict,
})
@websocket_command(
{
vol.Required("type"): WS_UPDATE,
vol.Required("deviceID"): str,
vol.Optional("data"): dict,
}
)
def handle_update(hass, connection, msg):
devices = get_devices(hass)
deviceID = msg["deviceID"]
@@ -69,57 +71,52 @@ class BrowserModConnection:
def send(self, command, **kwargs):
if self.connection:
connection, cid = self.connection[-1]
connection.send_message(event_message(cid, {
"command": command,
**kwargs,
}))
connection.send_message(
event_message(
cid,
{
"command": command,
**kwargs,
},
)
)
def trigger_update(self):
if is_setup_complete(self.hass):
self.send("update", **get_config(self.hass, self.deviceID))
def update(self, data):
if data.get('browser'):
if data.get("browser"):
self.sensor = self.sensor or create_entity(
self.hass,
'sensor',
self.deviceID,
self)
self.hass, "sensor", self.deviceID, self
)
if self.sensor:
self.sensor.data = data.get('browser')
self.sensor.data = data.get("browser")
if data.get('player'):
if data.get("player"):
self.media_player = self.media_player or create_entity(
self.hass,
'media_player',
self.deviceID,
self)
self.hass, "media_player", self.deviceID, self
)
if self.media_player:
self.media_player.data = data.get('player')
self.media_player.data = data.get("player")
if data.get('screen'):
if data.get("screen"):
self.screen = self.screen or create_entity(
self.hass,
'light',
self.deviceID,
self)
self.hass, "light", self.deviceID, self
)
if self.screen:
self.screen.data = data.get('screen')
self.screen.data = data.get("screen")
if data.get('fully'):
if data.get("fully"):
self.fully = self.fully or create_entity(
self.hass,
'binary_sensor',
self.deviceID,
self)
self.hass, "binary_sensor", self.deviceID, self
)
if self.fully:
self.fully.data = data.get('fully')
self.fully.data = data.get("fully")
if data.get('camera'):
if data.get("camera"):
self.camera = self.camera or create_entity(
self.hass,
'camera',
self.deviceID,
self)
self.hass, "camera", self.deviceID, self
)
if self.camera:
self.camera.data = data.get('camera')
self.camera.data = data.get("camera")