Add state detection for Plex on Fire TV (#121)

JeffLIrion · Nov 13, 2019 · e83bcba · e83bcba
1 parent cd58cd8
commit e83bcba
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/androidtv/firetv.py b/androidtv/firetv.py
@@ -181,6 +181,16 @@ def update(self, get_running_apps=True):
                 else:
                     state = constants.STATE_STANDBY
 
+            # Plex
+            elif current_app == constants.APP_PLEX:
+                if media_session_state == 3:
+                    if wake_lock_size == 2:
+                        state = constants.STATE_PAUSED
+                    else:
+                        state = constants.STATE_PLAYING
+                else:
+                    state = constants.STATE_STANDBY
+
             # Sport 1
             elif current_app == constants.APP_SPORT1:
                 if wake_lock_size == 2:

diff --git a/tests/test_firetv.py b/tests/test_firetv.py
@@ -348,6 +348,16 @@ def test_state_detection(self):
         self.assertUpdate([True, True, 1, constants.APP_NETFLIX, 1, [constants.APP_NETFLIX]],
                           (constants.STATE_STANDBY, constants.APP_NETFLIX, [constants.APP_NETFLIX]))
 
+        # Plex
+        self.assertUpdate([True, True, 1, constants.APP_PLEX, 3, [constants.APP_PLEX]],
+                          (constants.STATE_PLAYING, constants.APP_PLEX, [constants.APP_PLEX]))
+
+        self.assertUpdate([True, True, 2, constants.APP_PLEX, 3, [constants.APP_PLEX]],
+                          (constants.STATE_PAUSED, constants.APP_PLEX, [constants.APP_PLEX]))
+
+        self.assertUpdate([True, True, 1, constants.APP_PLEX, 1, [constants.APP_PLEX]],
+                          (constants.STATE_STANDBY, constants.APP_PLEX, [constants.APP_PLEX]))
+
         # Sport 1
         self.assertUpdate([True, True, 3, constants.APP_SPORT1, 3, [constants.APP_SPORT1]],
                           (constants.STATE_PLAYING, constants.APP_SPORT1, [constants.APP_SPORT1]))