Use Plex websocket payloads to reduce overhead (#42332)

Co-authored-by: Paulus Schoutsen <balloob@gmail.com>

.coveragerc

@@ -690,6 +690,7 @@ omit =
     homeassistant/components/pjlink/media_player.py
     homeassistant/components/plaato/*
     homeassistant/components/plex/media_player.py
+    homeassistant/components/plex/models.py
     homeassistant/components/plex/sensor.py
     homeassistant/components/plum_lightpad/light.py
     homeassistant/components/pocketcasts/sensor.py

homeassistant/components/plex/__init__.py

@@ -1,6 +1,5 @@
 """Support to embed Plex."""
 import asyncio
-import functools
 from functools import partial
 import logging
 
@@ -35,10 +34,7 @@
 from homeassistant.helpers import config_validation as cv
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
 from homeassistant.helpers.debounce import Debouncer
-from homeassistant.helpers.dispatcher import (
-    async_dispatcher_connect,
-    async_dispatcher_send,
-)
+from homeassistant.helpers.dispatcher import async_dispatcher_connect
 
 from .const import (
     CONF_SERVER,
@@ -176,6 +172,7 @@ def plex_websocket_callback(signal, data, error):
 
             if data == STATE_CONNECTED:
                 _LOGGER.debug("Websocket to %s successful", entry.data[CONF_SERVER])
+                hass.async_create_task(async_update_plex())
             elif data == STATE_DISCONNECTED:
                 _LOGGER.debug(
                     "Websocket to %s disconnected, retrying", entry.data[CONF_SERVER]
@@ -190,7 +187,7 @@ def plex_websocket_callback(signal, data, error):
                 hass.async_create_task(hass.config_entries.async_reload(entry.entry_id))
 
         elif signal == SIGNAL_DATA:
-            async_dispatcher_send(hass, PLEX_UPDATE_PLATFORMS_SIGNAL.format(server_id))
+            hass.async_create_task(plex_server.async_update_session(data))
 
     session = async_get_clientsession(hass)
     verify_ssl = server_config.get(CONF_VERIFY_SSL)
@@ -219,7 +216,7 @@ def close_websocket_session(_):
         task = hass.async_create_task(
             hass.config_entries.async_forward_entry_setup(entry, platform)
         )
-        task.add_done_callback(functools.partial(start_websocket_session, platform))
+        task.add_done_callback(partial(start_websocket_session, platform))
 
     async def async_play_on_sonos_service(service_call):
         await hass.async_add_executor_job(play_on_sonos, hass, service_call)

homeassistant/components/plex/const.py

@@ -24,6 +24,7 @@
 PLEX_SERVER_CONFIG = "server_config"
 
 PLEX_NEW_MP_SIGNAL = "plex_new_mp_signal.{}"
+PLEX_UPDATE_MEDIA_PLAYER_SESSION_SIGNAL = "plex_update_session_signal.{}"
 PLEX_UPDATE_MEDIA_PLAYER_SIGNAL = "plex_update_mp_signal.{}"
 PLEX_UPDATE_PLATFORMS_SIGNAL = "plex_update_platforms_signal.{}"
 PLEX_UPDATE_SENSOR_SIGNAL = "plex_update_sensor_signal.{}"